libavcodec/cbs_sei_syntax_template.c - third_party/ffmpeg - Git at Google

 /*
  * This file is part of FFmpeg.
  *
  * FFmpeg is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
  * License as published by the Free Software Foundation; either
  * version 2.1 of the License, or (at your option) any later version.
  *
  * FFmpeg is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public
  * License along with FFmpeg; if not, write to the Free Software
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */

 SEI_FUNC(filler_payload, (CodedBitstreamContext *ctx, RWContext *rw,
                           SEIRawFillerPayload *current,
                           SEIMessageState *state))
 {
     int err, i;

     HEADER("Filler Payload");

 #ifdef READ
     current->payload_size = state->payload_size;
 #endif

     for (i = 0; i < current->payload_size; i++)
         fixed(8, ff_byte, 0xff);

     return 0;
 }

 SEI_FUNC(user_data_registered, (CodedBitstreamContext *ctx, RWContext *rw,
                                 SEIRawUserDataRegistered *current,
                                 SEIMessageState *state))
 {
     int err, i, j;

     HEADER("User Data Registered ITU-T T.35");

     u(8, itu_t_t35_country_code, 0x00, 0xff);
     if (current->itu_t_t35_country_code != 0xff)
         i = 1;
     else {
         u(8, itu_t_t35_country_code_extension_byte, 0x00, 0xff);
         i = 2;
     }

 #ifdef READ
     if (state->payload_size < i) {
         av_log(ctx->log_ctx, AV_LOG_ERROR,
                "Invalid SEI user data registered payload.\n");
         return AVERROR_INVALIDDATA;
     }
     current->data_length = state->payload_size - i;
 #endif

     allocate(current->data, current->data_length);
     for (j = 0; j < current->data_length; j++)
         xu(8, itu_t_t35_payload_byte[], current->data[j], 0x00, 0xff, 1, i + j);

     return 0;
 }

 SEI_FUNC(user_data_unregistered, (CodedBitstreamContext *ctx, RWContext *rw,
                                   SEIRawUserDataUnregistered *current,
                                   SEIMessageState *state))
 {
     int err, i;

     HEADER("User Data Unregistered");

 #ifdef READ
     if (state->payload_size < 16) {
         av_log(ctx->log_ctx, AV_LOG_ERROR,
                "Invalid SEI user data unregistered payload.\n");
         return AVERROR_INVALIDDATA;
     }
     current->data_length = state->payload_size - 16;
 #endif

     for (i = 0; i < 16; i++)
         us(8, uuid_iso_iec_11578[i], 0x00, 0xff, 1, i);

     allocate(current->data, current->data_length);

     for (i = 0; i < current->data_length; i++)
         xu(8, user_data_payload_byte[i], current->data[i], 0x00, 0xff, 1, i);

     return 0;
 }

 SEI_FUNC(frame_packing_arrangement, (CodedBitstreamContext *ctx, RWContext *rw,
                                      SEIRawFramePackingArrangement *current,
                                      SEIMessageState *unused))
 {
     int err;

     HEADER("Frame Packing Arrangement");

     ue(fp_arrangement_id, 0, MAX_UINT_BITS(31));
     flag(fp_arrangement_cancel_flag);
     if (!current->fp_arrangement_cancel_flag) {
         u(7, fp_arrangement_type, 3, 5);
         flag(fp_quincunx_sampling_flag);
         u(6, fp_content_interpretation_type, 0, 2);
         flag(fp_spatial_flipping_flag);
         flag(fp_frame0_flipped_flag);
         flag(fp_field_views_flag);
         flag(fp_current_frame_is_frame0_flag);
         flag(fp_frame0_self_contained_flag);
         flag(fp_frame1_self_contained_flag);
         if (!current->fp_quincunx_sampling_flag && current->fp_arrangement_type != 5) {
             ub(4, fp_frame0_grid_position_x);
             ub(4, fp_frame0_grid_position_y);
             ub(4, fp_frame1_grid_position_x);
             ub(4, fp_frame1_grid_position_y);
         }
         fixed(8, fp_arrangement_reserved_byte, 0);
         flag(fp_arrangement_persistence_flag);
     }
     flag(fp_upsampled_aspect_ratio_flag);

     return 0;
 }

 SEI_FUNC(decoded_picture_hash, (CodedBitstreamContext *ctx,
                                 RWContext *rw,
                                 SEIRawDecodedPictureHash *current,
                                 SEIMessageState *unused))
 {
     int err, c_idx, i;

     HEADER("Decoded Picture Hash");

     u(8, dph_sei_hash_type, 0, 2);
     flag(dph_sei_single_component_flag);
     ub(7, dph_sei_reserved_zero_7bits);

     for (c_idx = 0; c_idx < (current->dph_sei_single_component_flag ? 1 : 3);
          c_idx++) {
         if (current->dph_sei_hash_type == 0) {
             for (i = 0; i < 16; i++)
                 us(8, dph_sei_picture_md5[c_idx][i], 0x00, 0xff, 2, c_idx, i);
         } else if (current->dph_sei_hash_type == 1) {
             us(16, dph_sei_picture_crc[c_idx], 0x0000, 0xffff, 1, c_idx);
         } else if (current->dph_sei_hash_type == 2) {
             us(32, dph_sei_picture_checksum[c_idx], 0x00000000, 0xffffffff, 1,
                c_idx);
         }
     }
     return 0;
 }

 SEI_FUNC(mastering_display_colour_volume,
          (CodedBitstreamContext *ctx, RWContext *rw,
           SEIRawMasteringDisplayColourVolume *current,
           SEIMessageState *state))
 {
     int err, c;

     HEADER("Mastering Display Colour Volume");

     for (c = 0; c < 3; c++) {
         ubs(16, display_primaries_x[c], 1, c);
         ubs(16, display_primaries_y[c], 1, c);
     }

     ub(16, white_point_x);
     ub(16, white_point_y);

     ub(32, max_display_mastering_luminance);
     ub(32, min_display_mastering_luminance);

     return 0;
 }

 SEI_FUNC(content_light_level_info, (CodedBitstreamContext *ctx, RWContext *rw,
                                     SEIRawContentLightLevelInfo *current,
                                     SEIMessageState *state))
 {
     int err;

     HEADER("Content Light Level Information");

     ub(16, max_content_light_level);
     ub(16, max_pic_average_light_level);

     return 0;
 }

 SEI_FUNC(alternative_transfer_characteristics,
          (CodedBitstreamContext *ctx, RWContext *rw,
           SEIRawAlternativeTransferCharacteristics *current,
           SEIMessageState *state))
 {
     int err;

     HEADER("Alternative Transfer Characteristics");

     ub(8, preferred_transfer_characteristics);

     return 0;
 }

 SEI_FUNC(ambient_viewing_environment,
          (CodedBitstreamContext *ctx, RWContext *rw,
           SEIRawAmbientViewingEnvironment *current,
           SEIMessageState *state))
 {
     static const uint16_t max_ambient_light_value = 50000;
     int err;

     HEADER("Ambient Viewing Environment");

     u(32, ambient_illuminance, 1, MAX_UINT_BITS(32));
     u(16, ambient_light_x, 0, max_ambient_light_value);
     u(16, ambient_light_y, 0, max_ambient_light_value);

     return 0;
 }

 SEI_FUNC(film_grain_characteristics,
         (CodedBitstreamContext *ctx, RWContext *rw,
          SEIRawFilmGrainCharacteristics *current,
          SEIMessageState *state))
 {
     int err, c, i, j;

     HEADER("Film Grain Characteristics");

     flag(fg_characteristics_cancel_flag);
     if (!current->fg_characteristics_cancel_flag) {
         int filmGrainBitDepth[3];

         u(2, fg_model_id, 0, 1);
         flag(fg_separate_colour_description_present_flag);
         if (current->fg_separate_colour_description_present_flag) {
             ub(3, fg_bit_depth_luma_minus8);
             ub(3, fg_bit_depth_chroma_minus8);
             flag(fg_full_range_flag);
             ub(8, fg_colour_primaries);
             ub(8, fg_transfer_characteristics);
             ub(8, fg_matrix_coeffs);
         }

         filmGrainBitDepth[0] = current->fg_bit_depth_luma_minus8 + 8;
         filmGrainBitDepth[1] =
         filmGrainBitDepth[2] = current->fg_bit_depth_chroma_minus8 + 8;

         u(2, fg_blending_mode_id, 0, 1);
         ub(4, fg_log2_scale_factor);
         for (c = 0; c < 3; c++)
             flags(fg_comp_model_present_flag[c], 1, c);

         for (c = 0; c < 3; c++) {
             if (current->fg_comp_model_present_flag[c]) {
                 ubs(8, fg_num_intensity_intervals_minus1[c], 1, c);
                 us(3, fg_num_model_values_minus1[c], 0, 5, 1, c);
                 for (i = 0; i <= current->fg_num_intensity_intervals_minus1[c]; i++) {
                     ubs(8, fg_intensity_interval_lower_bound[c][i], 2, c, i);
                     ubs(8, fg_intensity_interval_upper_bound[c][i], 2, c, i);
                     for (j = 0; j <= current->fg_num_model_values_minus1[c]; j++)
                         ses(fg_comp_model_value[c][i][j],  0  - current->fg_model_id * (1 << (filmGrainBitDepth[c] - 1)),
                             ((1 << filmGrainBitDepth[c]) - 1) - current->fg_model_id * (1 << (filmGrainBitDepth[c] - 1)),
                             3, c, i, j);
                 }
             }
         }
         flag(fg_characteristics_persistence_flag);
     }

     return 0;
 }

 SEI_FUNC(display_orientation, (CodedBitstreamContext *ctx, RWContext *rw,
                                SEIRawDisplayOrientation *current,
                                SEIMessageState *state))
 {
     int err;

     HEADER("Display Orientation");

     flag(display_orientation_cancel_flag);
     if (!current->display_orientation_cancel_flag) {
         flag(display_orientation_persistence_flag);
         u(3, display_orientation_transform_type, 0, 7);
         ub(3, display_orientation_reserved_zero_3bits);
     }

     return 0;
 }

 SEI_FUNC(frame_field_information, (CodedBitstreamContext *ctx, RWContext *rw,
                                        SEIRawFrameFieldInformation *current,
                                        SEIMessageState *state))
 {
     int err;

     HEADER("Frame-field information");

     flag(ffi_field_pic_flag);
     if (current->ffi_field_pic_flag) {
         flag(ffi_bottom_field_flag);
         flag(ffi_pairing_indicated_flag);
         if (current->ffi_pairing_indicated_flag)
             flag(ffi_paired_with_next_field_flag);
     } else {
         flag(ffi_display_fields_from_frame_flag);
         if (current->ffi_display_fields_from_frame_flag)
             flag(ffi_top_field_first_flag);
         u(8, ffi_display_elemental_periods_minus1, 0, 0xff);
     }
     u(2, ffi_source_scan_type, 0, 3);
     flag(ffi_duplicate_flag);

     return 0;
 }

 static int FUNC(message)(CodedBitstreamContext *ctx, RWContext *rw,
                          SEIRawMessage *current)
 {
     const SEIMessageTypeDescriptor *desc;
     int err, i;

     desc = ff_cbs_sei_find_type(ctx, current->payload_type);
     if (desc) {
         SEIMessageState state = {
             .payload_type      = current->payload_type,
             .payload_size      = current->payload_size,
             .extension_present = current->extension_bit_length > 0,
         };
         int start_position, current_position, bits_written;

 #ifdef READ
         CHECK(ff_cbs_sei_alloc_message_payload(current, desc));
 #endif

         start_position = bit_position(rw);

         CHECK(desc->READWRITE(ctx, rw, current->payload, &state));

         current_position = bit_position(rw);
         bits_written = current_position - start_position;

         if (byte_alignment(rw) || state.extension_present ||
             bits_written < 8 * current->payload_size) {
             size_t bits_left;

 #ifdef READ
             GetBitContext tmp = *rw;
             int trailing_bits, trailing_zero_bits;

             bits_left = 8 * current->payload_size - bits_written;
             if (bits_left > 8)
                 skip_bits_long(&tmp, bits_left - 8);
             trailing_bits = get_bits(&tmp, FFMIN(bits_left, 8));
             if (trailing_bits == 0) {
                 // The trailing bits must contain a bit_equal_to_one, so
                 // they can't all be zero.
                 return AVERROR_INVALIDDATA;
             }
             trailing_zero_bits = ff_ctz(trailing_bits);
             current->extension_bit_length =
                 bits_left - 1 - trailing_zero_bits;
 #endif

             if (current->extension_bit_length > 0) {
                 allocate(current->extension_data,
                          (current->extension_bit_length + 7) / 8);

                 bits_left = current->extension_bit_length;
                 for (i = 0; bits_left > 0; i++) {
                     int length = FFMIN(bits_left, 8);
                     xu(length, reserved_payload_extension_data,
                        current->extension_data[i],
                        0, MAX_UINT_BITS(length), 0);
                     bits_left -= length;
                 }
             }

             fixed(1, bit_equal_to_one, 1);
             while (byte_alignment(rw))
                 fixed(1, bit_equal_to_zero, 0);
         }

 #ifdef WRITE
         current->payload_size = (put_bits_count(rw) - start_position) / 8;
 #endif
     } else {
         uint8_t *data;

 #ifdef READ
         allocate(current->payload_ref, current->payload_size);
         current->payload = current->payload_ref;
 #else
         allocate(current->payload, current->payload_size);
 #endif
         data = current->payload;

         for (i = 0; i < current->payload_size; i++)
             xu(8, payload_byte[i], data[i], 0, 255, 1, i);
     }

     return 0;
 }

 static int FUNC(message_list)(CodedBitstreamContext *ctx, RWContext *rw,
                               SEIRawMessageList *current, int prefix)
 {
     SEIRawMessage *message;
     int err, k;

 #ifdef READ
     for (k = 0;; k++) {
         uint32_t payload_type = 0;
         uint32_t payload_size = 0;
         uint32_t tmp;
         GetBitContext payload_gbc;

         while (show_bits(rw, 8) == 0xff) {
             fixed(8, ff_byte, 0xff);
             payload_type += 255;
         }
         xu(8, last_payload_type_byte, tmp, 0, 254, 0);
         payload_type += tmp;

         while (show_bits(rw, 8) == 0xff) {
             fixed(8, ff_byte, 0xff);
             payload_size += 255;
         }
         xu(8, last_payload_size_byte, tmp, 0, 254, 0);
         payload_size += tmp;

         // There must be space remaining for both the payload and
         // the trailing bits on the SEI NAL unit.
         if (payload_size + 1 > get_bits_left(rw) / 8) {
             av_log(ctx->log_ctx, AV_LOG_ERROR,
                    "Invalid SEI message: payload_size too large "
                    "(%"PRIu32" bytes).\n", payload_size);
             return AVERROR_INVALIDDATA;
         }
         CHECK(init_get_bits(&payload_gbc, rw->buffer,
                             get_bits_count(rw) + 8 * payload_size));
         skip_bits_long(&payload_gbc, get_bits_count(rw));

         CHECK(ff_cbs_sei_list_add(current));
         message = &current->messages[k];

         message->payload_type = payload_type;
         message->payload_size = payload_size;

         CHECK(FUNC(message)(ctx, &payload_gbc, message));

         skip_bits_long(rw, 8 * payload_size);

         if (!cbs_h2645_read_more_rbsp_data(rw))
             break;
     }
 #else
     for (k = 0; k < current->nb_messages; k++) {
         PutBitContext start_state;
         uint32_t tmp;
         int trace, i;

         message = &current->messages[k];

         // We write the payload twice in order to find the size.  Trace
         // output is switched off for the first write.
         trace = ctx->trace_enable;
         ctx->trace_enable = 0;

         start_state = *rw;
         for (i = 0; i < 2; i++) {
             *rw = start_state;

             tmp = message->payload_type;
             while (tmp >= 255) {
                 fixed(8, ff_byte, 0xff);
                 tmp -= 255;
             }
             xu(8, last_payload_type_byte, tmp, 0, 254, 0);

             tmp = message->payload_size;
             while (tmp >= 255) {
                 fixed(8, ff_byte, 0xff);
                 tmp -= 255;
             }
             xu(8, last_payload_size_byte, tmp, 0, 254, 0);

             err = FUNC(message)(ctx, rw, message);
             ctx->trace_enable = trace;
             if (err < 0)
                 return err;
         }
     }
 #endif

     return 0;
 }
	/*
	* This file is part of FFmpeg.
	*
	* FFmpeg is free software; you can redistribute it and/or
	* modify it under the terms of the GNU Lesser General Public
	* License as published by the Free Software Foundation; either
	* version 2.1 of the License, or (at your option) any later version.
	*
	* FFmpeg is distributed in the hope that it will be useful,
	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	* Lesser General Public License for more details.
	*
	* You should have received a copy of the GNU Lesser General Public
	* License along with FFmpeg; if not, write to the Free Software
	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
	*/

	SEI_FUNC(filler_payload, (CodedBitstreamContext ctx, RWContext rw,
	SEIRawFillerPayload *current,
	SEIMessageState *state))
	{
	int err, i;

	HEADER("Filler Payload");

	#ifdef READ
	current->payload_size = state->payload_size;
	#endif

	for (i = 0; i < current->payload_size; i++)
	fixed(8, ff_byte, 0xff);

	return 0;
	}

	SEI_FUNC(user_data_registered, (CodedBitstreamContext ctx, RWContext rw,
	SEIRawUserDataRegistered *current,
	SEIMessageState *state))
	{
	int err, i, j;

	HEADER("User Data Registered ITU-T T.35");

	u(8, itu_t_t35_country_code, 0x00, 0xff);
	if (current->itu_t_t35_country_code != 0xff)
	i = 1;
	else {
	u(8, itu_t_t35_country_code_extension_byte, 0x00, 0xff);
	i = 2;
	}

	#ifdef READ
	if (state->payload_size < i) {
	av_log(ctx->log_ctx, AV_LOG_ERROR,
	"Invalid SEI user data registered payload.\n");
	return AVERROR_INVALIDDATA;
	}
	current->data_length = state->payload_size - i;
	#endif

	allocate(current->data, current->data_length);
	for (j = 0; j < current->data_length; j++)
	xu(8, itu_t_t35_payload_byte[], current->data[j], 0x00, 0xff, 1, i + j);

	return 0;
	}

	SEI_FUNC(user_data_unregistered, (CodedBitstreamContext ctx, RWContext rw,
	SEIRawUserDataUnregistered *current,
	SEIMessageState *state))
	{
	int err, i;

	HEADER("User Data Unregistered");

	#ifdef READ
	if (state->payload_size < 16) {
	av_log(ctx->log_ctx, AV_LOG_ERROR,
	"Invalid SEI user data unregistered payload.\n");
	return AVERROR_INVALIDDATA;
	}
	current->data_length = state->payload_size - 16;
	#endif

	for (i = 0; i < 16; i++)
	us(8, uuid_iso_iec_11578[i], 0x00, 0xff, 1, i);

	allocate(current->data, current->data_length);

	for (i = 0; i < current->data_length; i++)
	xu(8, user_data_payload_byte[i], current->data[i], 0x00, 0xff, 1, i);

	return 0;
	}

	SEI_FUNC(frame_packing_arrangement, (CodedBitstreamContext ctx, RWContext rw,
	SEIRawFramePackingArrangement *current,
	SEIMessageState *unused))
	{
	int err;

	HEADER("Frame Packing Arrangement");

	ue(fp_arrangement_id, 0, MAX_UINT_BITS(31));
	flag(fp_arrangement_cancel_flag);
	if (!current->fp_arrangement_cancel_flag) {
	u(7, fp_arrangement_type, 3, 5);
	flag(fp_quincunx_sampling_flag);
	u(6, fp_content_interpretation_type, 0, 2);
	flag(fp_spatial_flipping_flag);
	flag(fp_frame0_flipped_flag);
	flag(fp_field_views_flag);
	flag(fp_current_frame_is_frame0_flag);
	flag(fp_frame0_self_contained_flag);
	flag(fp_frame1_self_contained_flag);
	if (!current->fp_quincunx_sampling_flag && current->fp_arrangement_type != 5) {
	ub(4, fp_frame0_grid_position_x);
	ub(4, fp_frame0_grid_position_y);
	ub(4, fp_frame1_grid_position_x);
	ub(4, fp_frame1_grid_position_y);
	}
	fixed(8, fp_arrangement_reserved_byte, 0);
	flag(fp_arrangement_persistence_flag);
	}
	flag(fp_upsampled_aspect_ratio_flag);

	return 0;
	}

	SEI_FUNC(decoded_picture_hash, (CodedBitstreamContext *ctx,
	RWContext *rw,
	SEIRawDecodedPictureHash *current,
	SEIMessageState *unused))
	{
	int err, c_idx, i;

	HEADER("Decoded Picture Hash");

	u(8, dph_sei_hash_type, 0, 2);
	flag(dph_sei_single_component_flag);
	ub(7, dph_sei_reserved_zero_7bits);

	for (c_idx = 0; c_idx < (current->dph_sei_single_component_flag ? 1 : 3);
	c_idx++) {
	if (current->dph_sei_hash_type == 0) {
	for (i = 0; i < 16; i++)
	us(8, dph_sei_picture_md5[c_idx][i], 0x00, 0xff, 2, c_idx, i);
	} else if (current->dph_sei_hash_type == 1) {
	us(16, dph_sei_picture_crc[c_idx], 0x0000, 0xffff, 1, c_idx);
	} else if (current->dph_sei_hash_type == 2) {
	us(32, dph_sei_picture_checksum[c_idx], 0x00000000, 0xffffffff, 1,
	c_idx);
	}
	}
	return 0;
	}

	SEI_FUNC(mastering_display_colour_volume,
	(CodedBitstreamContext ctx, RWContext rw,
	SEIRawMasteringDisplayColourVolume *current,
	SEIMessageState *state))
	{
	int err, c;

	HEADER("Mastering Display Colour Volume");

	for (c = 0; c < 3; c++) {
	ubs(16, display_primaries_x[c], 1, c);
	ubs(16, display_primaries_y[c], 1, c);
	}

	ub(16, white_point_x);
	ub(16, white_point_y);

	ub(32, max_display_mastering_luminance);
	ub(32, min_display_mastering_luminance);

	return 0;
	}

	SEI_FUNC(content_light_level_info, (CodedBitstreamContext ctx, RWContext rw,
	SEIRawContentLightLevelInfo *current,
	SEIMessageState *state))
	{
	int err;

	HEADER("Content Light Level Information");

	ub(16, max_content_light_level);
	ub(16, max_pic_average_light_level);

	return 0;
	}

	SEI_FUNC(alternative_transfer_characteristics,
	(CodedBitstreamContext ctx, RWContext rw,
	SEIRawAlternativeTransferCharacteristics *current,
	SEIMessageState *state))
	{
	int err;

	HEADER("Alternative Transfer Characteristics");

	ub(8, preferred_transfer_characteristics);

	return 0;
	}

	SEI_FUNC(ambient_viewing_environment,
	(CodedBitstreamContext ctx, RWContext rw,
	SEIRawAmbientViewingEnvironment *current,
	SEIMessageState *state))
	{
	static const uint16_t max_ambient_light_value = 50000;
	int err;

	HEADER("Ambient Viewing Environment");

	u(32, ambient_illuminance, 1, MAX_UINT_BITS(32));
	u(16, ambient_light_x, 0, max_ambient_light_value);
	u(16, ambient_light_y, 0, max_ambient_light_value);

	return 0;
	}

	SEI_FUNC(film_grain_characteristics,
	(CodedBitstreamContext ctx, RWContext rw,
	SEIRawFilmGrainCharacteristics *current,
	SEIMessageState *state))
	{
	int err, c, i, j;

	HEADER("Film Grain Characteristics");

	flag(fg_characteristics_cancel_flag);
	if (!current->fg_characteristics_cancel_flag) {
	int filmGrainBitDepth[3];

	u(2, fg_model_id, 0, 1);
	flag(fg_separate_colour_description_present_flag);
	if (current->fg_separate_colour_description_present_flag) {
	ub(3, fg_bit_depth_luma_minus8);
	ub(3, fg_bit_depth_chroma_minus8);
	flag(fg_full_range_flag);
	ub(8, fg_colour_primaries);
	ub(8, fg_transfer_characteristics);
	ub(8, fg_matrix_coeffs);
	}

	filmGrainBitDepth[0] = current->fg_bit_depth_luma_minus8 + 8;
	filmGrainBitDepth[1] =
	filmGrainBitDepth[2] = current->fg_bit_depth_chroma_minus8 + 8;

	u(2, fg_blending_mode_id, 0, 1);
	ub(4, fg_log2_scale_factor);
	for (c = 0; c < 3; c++)
	flags(fg_comp_model_present_flag[c], 1, c);

	for (c = 0; c < 3; c++) {
	if (current->fg_comp_model_present_flag[c]) {
	ubs(8, fg_num_intensity_intervals_minus1[c], 1, c);
	us(3, fg_num_model_values_minus1[c], 0, 5, 1, c);
	for (i = 0; i <= current->fg_num_intensity_intervals_minus1[c]; i++) {
	ubs(8, fg_intensity_interval_lower_bound[c][i], 2, c, i);
	ubs(8, fg_intensity_interval_upper_bound[c][i], 2, c, i);
	for (j = 0; j <= current->fg_num_model_values_minus1[c]; j++)
	ses(fg_comp_model_value[c][i][j], 0 - current->fg_model_id * (1 << (filmGrainBitDepth[c] - 1)),
	((1 << filmGrainBitDepth[c]) - 1) - current->fg_model_id * (1 << (filmGrainBitDepth[c] - 1)),
	3, c, i, j);
	}
	}
	}
	flag(fg_characteristics_persistence_flag);
	}

	return 0;
	}

	SEI_FUNC(display_orientation, (CodedBitstreamContext ctx, RWContext rw,
	SEIRawDisplayOrientation *current,
	SEIMessageState *state))
	{
	int err;

	HEADER("Display Orientation");

	flag(display_orientation_cancel_flag);
	if (!current->display_orientation_cancel_flag) {
	flag(display_orientation_persistence_flag);
	u(3, display_orientation_transform_type, 0, 7);
	ub(3, display_orientation_reserved_zero_3bits);
	}

	return 0;
	}

	SEI_FUNC(frame_field_information, (CodedBitstreamContext ctx, RWContext rw,
	SEIRawFrameFieldInformation *current,
	SEIMessageState *state))
	{
	int err;

	HEADER("Frame-field information");

	flag(ffi_field_pic_flag);
	if (current->ffi_field_pic_flag) {
	flag(ffi_bottom_field_flag);
	flag(ffi_pairing_indicated_flag);
	if (current->ffi_pairing_indicated_flag)
	flag(ffi_paired_with_next_field_flag);
	} else {
	flag(ffi_display_fields_from_frame_flag);
	if (current->ffi_display_fields_from_frame_flag)
	flag(ffi_top_field_first_flag);
	u(8, ffi_display_elemental_periods_minus1, 0, 0xff);
	}
	u(2, ffi_source_scan_type, 0, 3);
	flag(ffi_duplicate_flag);

	return 0;
	}

	static int FUNC(message)(CodedBitstreamContext ctx, RWContext rw,
	SEIRawMessage *current)
	{
	const SEIMessageTypeDescriptor *desc;
	int err, i;

	desc = ff_cbs_sei_find_type(ctx, current->payload_type);
	if (desc) {
	SEIMessageState state = {
	.payload_type = current->payload_type,
	.payload_size = current->payload_size,
	.extension_present = current->extension_bit_length > 0,
	};
	int start_position, current_position, bits_written;

	#ifdef READ
	CHECK(ff_cbs_sei_alloc_message_payload(current, desc));
	#endif

	start_position = bit_position(rw);

	CHECK(desc->READWRITE(ctx, rw, current->payload, &state));

	current_position = bit_position(rw);
	bits_written = current_position - start_position;

	if (byte_alignment(rw) \|\| state.extension_present \|\|
	bits_written < 8 * current->payload_size) {
	size_t bits_left;

	#ifdef READ
	GetBitContext tmp = *rw;
	int trailing_bits, trailing_zero_bits;

	bits_left = 8 * current->payload_size - bits_written;
	if (bits_left > 8)
	skip_bits_long(&tmp, bits_left - 8);
	trailing_bits = get_bits(&tmp, FFMIN(bits_left, 8));
	if (trailing_bits == 0) {
	// The trailing bits must contain a bit_equal_to_one, so
	// they can't all be zero.
	return AVERROR_INVALIDDATA;
	}
	trailing_zero_bits = ff_ctz(trailing_bits);
	current->extension_bit_length =
	bits_left - 1 - trailing_zero_bits;
	#endif

	if (current->extension_bit_length > 0) {
	allocate(current->extension_data,
	(current->extension_bit_length + 7) / 8);

	bits_left = current->extension_bit_length;
	for (i = 0; bits_left > 0; i++) {
	int length = FFMIN(bits_left, 8);
	xu(length, reserved_payload_extension_data,
	current->extension_data[i],
	0, MAX_UINT_BITS(length), 0);
	bits_left -= length;
	}
	}

	fixed(1, bit_equal_to_one, 1);
	while (byte_alignment(rw))
	fixed(1, bit_equal_to_zero, 0);
	}

	#ifdef WRITE
	current->payload_size = (put_bits_count(rw) - start_position) / 8;
	#endif
	} else {
	uint8_t *data;

	#ifdef READ
	allocate(current->payload_ref, current->payload_size);
	current->payload = current->payload_ref;
	#else
	allocate(current->payload, current->payload_size);
	#endif
	data = current->payload;

	for (i = 0; i < current->payload_size; i++)
	xu(8, payload_byte[i], data[i], 0, 255, 1, i);
	}

	return 0;
	}

	static int FUNC(message_list)(CodedBitstreamContext ctx, RWContext rw,
	SEIRawMessageList *current, int prefix)
	{
	SEIRawMessage *message;
	int err, k;

	#ifdef READ
	for (k = 0;; k++) {
	uint32_t payload_type = 0;
	uint32_t payload_size = 0;
	uint32_t tmp;
	GetBitContext payload_gbc;

	while (show_bits(rw, 8) == 0xff) {
	fixed(8, ff_byte, 0xff);
	payload_type += 255;
	}
	xu(8, last_payload_type_byte, tmp, 0, 254, 0);
	payload_type += tmp;

	while (show_bits(rw, 8) == 0xff) {
	fixed(8, ff_byte, 0xff);
	payload_size += 255;
	}
	xu(8, last_payload_size_byte, tmp, 0, 254, 0);
	payload_size += tmp;

	// There must be space remaining for both the payload and
	// the trailing bits on the SEI NAL unit.
	if (payload_size + 1 > get_bits_left(rw) / 8) {
	av_log(ctx->log_ctx, AV_LOG_ERROR,
	"Invalid SEI message: payload_size too large "
	"(%"PRIu32" bytes).\n", payload_size);
	return AVERROR_INVALIDDATA;
	}
	CHECK(init_get_bits(&payload_gbc, rw->buffer,
	get_bits_count(rw) + 8 * payload_size));
	skip_bits_long(&payload_gbc, get_bits_count(rw));

	CHECK(ff_cbs_sei_list_add(current));
	message = &current->messages[k];

	message->payload_type = payload_type;
	message->payload_size = payload_size;

	CHECK(FUNC(message)(ctx, &payload_gbc, message));

	skip_bits_long(rw, 8 * payload_size);

	if (!cbs_h2645_read_more_rbsp_data(rw))
	break;
	}
	#else
	for (k = 0; k < current->nb_messages; k++) {
	PutBitContext start_state;
	uint32_t tmp;
	int trace, i;

	message = &current->messages[k];

	// We write the payload twice in order to find the size. Trace
	// output is switched off for the first write.
	trace = ctx->trace_enable;
	ctx->trace_enable = 0;

	start_state = *rw;
	for (i = 0; i < 2; i++) {
	*rw = start_state;

	tmp = message->payload_type;
	while (tmp >= 255) {
	fixed(8, ff_byte, 0xff);
	tmp -= 255;
	}
	xu(8, last_payload_type_byte, tmp, 0, 254, 0);

	tmp = message->payload_size;
	while (tmp >= 255) {
	fixed(8, ff_byte, 0xff);
	tmp -= 255;
	}
	xu(8, last_payload_size_byte, tmp, 0, 254, 0);

	err = FUNC(message)(ctx, rw, message);
	ctx->trace_enable = trace;
	if (err < 0)
	return err;
	}
	}
	#endif

	return 0;
	}