265 lines
9.2 KiB
C
265 lines
9.2 KiB
C
/*
|
|
* 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
|
|
*/
|
|
|
|
#ifndef AVCODEC_CBS_SEI_H
|
|
#define AVCODEC_CBS_SEI_H
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
|
|
#include "cbs.h"
|
|
#include "sei.h"
|
|
|
|
|
|
typedef struct SEIRawFillerPayload {
|
|
uint32_t payload_size;
|
|
} SEIRawFillerPayload;
|
|
|
|
typedef struct SEIRawUserDataRegistered {
|
|
uint8_t itu_t_t35_country_code;
|
|
uint8_t itu_t_t35_country_code_extension_byte;
|
|
uint8_t *data; ///< RefStruct reference
|
|
size_t data_length;
|
|
} SEIRawUserDataRegistered;
|
|
|
|
typedef struct SEIRawUserDataUnregistered {
|
|
uint8_t uuid_iso_iec_11578[16];
|
|
uint8_t *data; ///< RefStruct reference
|
|
size_t data_length;
|
|
} SEIRawUserDataUnregistered;
|
|
|
|
typedef struct SEIRawFramePackingArrangement {
|
|
uint32_t fp_arrangement_id;
|
|
uint8_t fp_arrangement_cancel_flag;
|
|
uint8_t fp_arrangement_type;
|
|
uint8_t fp_quincunx_sampling_flag;
|
|
uint8_t fp_content_interpretation_type;
|
|
uint8_t fp_spatial_flipping_flag;
|
|
uint8_t fp_frame0_flipped_flag;
|
|
uint8_t fp_field_views_flag;
|
|
uint8_t fp_current_frame_is_frame0_flag;
|
|
uint8_t fp_frame0_self_contained_flag;
|
|
uint8_t fp_frame1_self_contained_flag;
|
|
uint8_t fp_frame0_grid_position_x;
|
|
uint8_t fp_frame0_grid_position_y;
|
|
uint8_t fp_frame1_grid_position_x;
|
|
uint8_t fp_frame1_grid_position_y;
|
|
uint8_t fp_arrangement_persistence_flag;
|
|
uint8_t fp_upsampled_aspect_ratio_flag;
|
|
} SEIRawFramePackingArrangement;
|
|
|
|
typedef struct SEIRawDecodedPictureHash {
|
|
uint8_t dph_sei_hash_type;
|
|
uint8_t dph_sei_single_component_flag;
|
|
uint8_t dph_sei_picture_md5[3][16];
|
|
uint16_t dph_sei_picture_crc[3];
|
|
uint32_t dph_sei_picture_checksum[3];
|
|
|
|
uint8_t dph_sei_reserved_zero_7bits;
|
|
} SEIRawDecodedPictureHash;
|
|
|
|
typedef struct SEIRawMasteringDisplayColourVolume {
|
|
uint16_t display_primaries_x[3];
|
|
uint16_t display_primaries_y[3];
|
|
uint16_t white_point_x;
|
|
uint16_t white_point_y;
|
|
uint32_t max_display_mastering_luminance;
|
|
uint32_t min_display_mastering_luminance;
|
|
} SEIRawMasteringDisplayColourVolume;
|
|
|
|
typedef struct SEIRawContentLightLevelInfo {
|
|
uint16_t max_content_light_level;
|
|
uint16_t max_pic_average_light_level;
|
|
} SEIRawContentLightLevelInfo;
|
|
|
|
typedef struct SEIRawAlternativeTransferCharacteristics {
|
|
uint8_t preferred_transfer_characteristics;
|
|
} SEIRawAlternativeTransferCharacteristics;
|
|
|
|
typedef struct SEIRawAmbientViewingEnvironment {
|
|
uint32_t ambient_illuminance;
|
|
uint16_t ambient_light_x;
|
|
uint16_t ambient_light_y;
|
|
} SEIRawAmbientViewingEnvironment;
|
|
|
|
typedef struct SEIRawFilmGrainCharacteristics {
|
|
uint8_t fg_characteristics_cancel_flag;
|
|
uint8_t fg_model_id;
|
|
uint8_t fg_separate_colour_description_present_flag;
|
|
uint8_t fg_bit_depth_luma_minus8;
|
|
uint8_t fg_bit_depth_chroma_minus8;
|
|
uint8_t fg_full_range_flag;
|
|
uint8_t fg_colour_primaries;
|
|
uint8_t fg_transfer_characteristics;
|
|
uint8_t fg_matrix_coeffs;
|
|
uint8_t fg_blending_mode_id;
|
|
uint8_t fg_log2_scale_factor;
|
|
uint8_t fg_comp_model_present_flag[3];
|
|
uint8_t fg_num_intensity_intervals_minus1[3];
|
|
uint8_t fg_num_model_values_minus1[3];
|
|
uint8_t fg_intensity_interval_lower_bound[3][256];
|
|
uint8_t fg_intensity_interval_upper_bound[3][256];
|
|
int16_t fg_comp_model_value[3][256][6];
|
|
uint8_t fg_characteristics_persistence_flag;
|
|
} SEIRawFilmGrainCharacteristics;
|
|
|
|
typedef struct SEIRawDisplayOrientation {
|
|
uint8_t display_orientation_cancel_flag;
|
|
uint8_t display_orientation_persistence_flag;
|
|
uint8_t display_orientation_transform_type;
|
|
uint8_t display_orientation_reserved_zero_3bits;
|
|
} SEIRawDisplayOrientation;
|
|
|
|
typedef struct SEIRawFrameFieldInformation {
|
|
uint8_t ffi_field_pic_flag;
|
|
uint8_t ffi_bottom_field_flag;
|
|
uint8_t ffi_pairing_indicated_flag;
|
|
uint8_t ffi_paired_with_next_field_flag;
|
|
uint8_t ffi_display_fields_from_frame_flag;
|
|
uint8_t ffi_top_field_first_flag;
|
|
uint8_t ffi_display_elemental_periods_minus1;
|
|
uint8_t ffi_source_scan_type;
|
|
uint8_t ffi_duplicate_flag;
|
|
} SEIRawFrameFieldInformation;
|
|
|
|
typedef struct SEIRawMessage {
|
|
uint32_t payload_type;
|
|
uint32_t payload_size;
|
|
void *payload;
|
|
void *payload_ref; ///< RefStruct reference
|
|
uint8_t *extension_data; ///< RefStruct reference
|
|
size_t extension_bit_length;
|
|
} SEIRawMessage;
|
|
|
|
typedef struct SEIRawMessageList {
|
|
SEIRawMessage *messages;
|
|
int nb_messages;
|
|
int nb_messages_allocated;
|
|
} SEIRawMessageList;
|
|
|
|
|
|
typedef struct SEIMessageState {
|
|
// The type of the payload being written.
|
|
uint32_t payload_type;
|
|
// When reading, contains the size of the payload to allow finding the
|
|
// end of variable-length fields (such as user_data_payload_byte[]).
|
|
// (When writing, the size will be derived from the total number of
|
|
// bytes actually written.)
|
|
uint32_t payload_size;
|
|
// When writing, indicates that payload extension data is present so
|
|
// all extended fields must be written. May be updated by the writer
|
|
// to indicate that extended fields have been written, so the extension
|
|
// end bits must be written too.
|
|
uint8_t extension_present;
|
|
} SEIMessageState;
|
|
|
|
struct GetBitContext;
|
|
struct PutBitContext;
|
|
|
|
typedef int (*SEIMessageReadFunction)(CodedBitstreamContext *ctx,
|
|
struct GetBitContext *rw,
|
|
void *current,
|
|
SEIMessageState *sei);
|
|
|
|
typedef int (*SEIMessageWriteFunction)(CodedBitstreamContext *ctx,
|
|
struct PutBitContext *rw,
|
|
void *current,
|
|
SEIMessageState *sei);
|
|
|
|
typedef struct SEIMessageTypeDescriptor {
|
|
// Payload type for the message. (-1 in this field ends a list.)
|
|
int type;
|
|
// Valid in a prefix SEI NAL unit (always for H.264).
|
|
uint8_t prefix;
|
|
// Valid in a suffix SEI NAL unit (never for H.264).
|
|
uint8_t suffix;
|
|
// Size of the decomposed structure.
|
|
size_t size;
|
|
// Read bitstream into SEI message.
|
|
SEIMessageReadFunction read;
|
|
// Write bitstream from SEI message.
|
|
SEIMessageWriteFunction write;
|
|
} SEIMessageTypeDescriptor;
|
|
|
|
// End-of-list sentinel element.
|
|
#define SEI_MESSAGE_TYPE_END { .type = -1 }
|
|
|
|
|
|
/**
|
|
* Find the type descriptor for the given payload type.
|
|
*
|
|
* Returns NULL if the payload type is not known.
|
|
*/
|
|
const SEIMessageTypeDescriptor *ff_cbs_sei_find_type(CodedBitstreamContext *ctx,
|
|
int payload_type);
|
|
|
|
/**
|
|
* Allocate a new payload for the given SEI message.
|
|
*/
|
|
int ff_cbs_sei_alloc_message_payload(SEIRawMessage *message,
|
|
const SEIMessageTypeDescriptor *desc);
|
|
|
|
/**
|
|
* Allocate a new empty SEI message in a message list.
|
|
*
|
|
* The new message is in place nb_messages - 1.
|
|
*/
|
|
int ff_cbs_sei_list_add(SEIRawMessageList *list);
|
|
|
|
/**
|
|
* Free all SEI messages in a message list.
|
|
*/
|
|
void ff_cbs_sei_free_message_list(SEIRawMessageList *list);
|
|
|
|
/**
|
|
* Add an SEI message to an access unit.
|
|
*
|
|
* Will add to an existing SEI NAL unit, or create a new one for the
|
|
* message if there is no suitable existing one.
|
|
*
|
|
* If set, payload_ref must be a RefStruct reference backing payload_data.
|
|
* This function creates a new reference to payload_ref in this case.
|
|
* If payload_ref is NULL, the new message will not be reference counted.
|
|
*/
|
|
int ff_cbs_sei_add_message(CodedBitstreamContext *ctx,
|
|
CodedBitstreamFragment *au,
|
|
int prefix,
|
|
uint32_t payload_type,
|
|
void *payload_data,
|
|
void *payload_ref);
|
|
|
|
/**
|
|
* Iterate over messages with the given payload type in an access unit.
|
|
*
|
|
* Set message to NULL in the first call. Returns 0 while more messages
|
|
* are available, AVERROR(ENOENT) when all messages have been found.
|
|
*/
|
|
int ff_cbs_sei_find_message(CodedBitstreamContext *ctx,
|
|
CodedBitstreamFragment *au,
|
|
uint32_t payload_type,
|
|
SEIRawMessage **message);
|
|
|
|
/**
|
|
* Delete all messages with the given payload type from an access unit.
|
|
*/
|
|
void ff_cbs_sei_delete_message_type(CodedBitstreamContext *ctx,
|
|
CodedBitstreamFragment *au,
|
|
uint32_t payload_type);
|
|
|
|
#endif /* AVCODEC_CBS_SEI_H */
|