mirror of https://github.com/FFmpeg/FFmpeg.git
This code is independent from the decoder, so it makes more sense for it to to have its own header.pull/62/merge
Anton Khirnov
9 years ago
parent
150c896a9e
commit
f6e2f8a9ff
7 changed files with 339 additions and 302 deletions
@ -0,0 +1,320 @@ |
||||
/*
|
||||
* HEVC parameter set parsing |
||||
* |
||||
* This file is part of Libav. |
||||
* |
||||
* Libav 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. |
||||
* |
||||
* Libav 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 Libav; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#ifndef AVCODEC_HEVC_PS_H |
||||
#define AVCODEC_HEVC_PS_H |
||||
|
||||
#include <stdint.h> |
||||
|
||||
#include "libavutil/buffer.h" |
||||
#include "libavutil/pixfmt.h" |
||||
#include "libavutil/rational.h" |
||||
|
||||
#include "avcodec.h" |
||||
#include "get_bits.h" |
||||
#include "hevc.h" |
||||
|
||||
typedef struct ShortTermRPS { |
||||
unsigned int num_negative_pics; |
||||
int num_delta_pocs; |
||||
int rps_idx_num_delta_pocs; |
||||
int32_t delta_poc[32]; |
||||
uint8_t used[32]; |
||||
} ShortTermRPS; |
||||
|
||||
typedef struct HEVCWindow { |
||||
unsigned int left_offset; |
||||
unsigned int right_offset; |
||||
unsigned int top_offset; |
||||
unsigned int bottom_offset; |
||||
} HEVCWindow; |
||||
|
||||
typedef struct VUI { |
||||
AVRational sar; |
||||
|
||||
int overscan_info_present_flag; |
||||
int overscan_appropriate_flag; |
||||
|
||||
int video_signal_type_present_flag; |
||||
int video_format; |
||||
int video_full_range_flag; |
||||
int colour_description_present_flag; |
||||
uint8_t colour_primaries; |
||||
uint8_t transfer_characteristic; |
||||
uint8_t matrix_coeffs; |
||||
|
||||
int chroma_loc_info_present_flag; |
||||
int chroma_sample_loc_type_top_field; |
||||
int chroma_sample_loc_type_bottom_field; |
||||
int neutra_chroma_indication_flag; |
||||
|
||||
int field_seq_flag; |
||||
int frame_field_info_present_flag; |
||||
|
||||
int default_display_window_flag; |
||||
HEVCWindow def_disp_win; |
||||
|
||||
int vui_timing_info_present_flag; |
||||
uint32_t vui_num_units_in_tick; |
||||
uint32_t vui_time_scale; |
||||
int vui_poc_proportional_to_timing_flag; |
||||
int vui_num_ticks_poc_diff_one_minus1; |
||||
int vui_hrd_parameters_present_flag; |
||||
|
||||
int bitstream_restriction_flag; |
||||
int tiles_fixed_structure_flag; |
||||
int motion_vectors_over_pic_boundaries_flag; |
||||
int restricted_ref_pic_lists_flag; |
||||
int min_spatial_segmentation_idc; |
||||
int max_bytes_per_pic_denom; |
||||
int max_bits_per_min_cu_denom; |
||||
int log2_max_mv_length_horizontal; |
||||
int log2_max_mv_length_vertical; |
||||
} VUI; |
||||
|
||||
typedef struct PTLCommon { |
||||
uint8_t profile_space; |
||||
uint8_t tier_flag; |
||||
uint8_t profile_idc; |
||||
uint8_t profile_compatibility_flag[32]; |
||||
uint8_t level_idc; |
||||
uint8_t progressive_source_flag; |
||||
uint8_t interlaced_source_flag; |
||||
uint8_t non_packed_constraint_flag; |
||||
uint8_t frame_only_constraint_flag; |
||||
} PTLCommon; |
||||
|
||||
typedef struct PTL { |
||||
PTLCommon general_ptl; |
||||
PTLCommon sub_layer_ptl[HEVC_MAX_SUB_LAYERS]; |
||||
|
||||
uint8_t sub_layer_profile_present_flag[HEVC_MAX_SUB_LAYERS]; |
||||
uint8_t sub_layer_level_present_flag[HEVC_MAX_SUB_LAYERS]; |
||||
} PTL; |
||||
|
||||
typedef struct HEVCVPS { |
||||
uint8_t vps_temporal_id_nesting_flag; |
||||
int vps_max_layers; |
||||
int vps_max_sub_layers; ///< vps_max_temporal_layers_minus1 + 1
|
||||
|
||||
PTL ptl; |
||||
int vps_sub_layer_ordering_info_present_flag; |
||||
unsigned int vps_max_dec_pic_buffering[HEVC_MAX_SUB_LAYERS]; |
||||
unsigned int vps_num_reorder_pics[HEVC_MAX_SUB_LAYERS]; |
||||
unsigned int vps_max_latency_increase[HEVC_MAX_SUB_LAYERS]; |
||||
int vps_max_layer_id; |
||||
int vps_num_layer_sets; ///< vps_num_layer_sets_minus1 + 1
|
||||
uint8_t vps_timing_info_present_flag; |
||||
uint32_t vps_num_units_in_tick; |
||||
uint32_t vps_time_scale; |
||||
uint8_t vps_poc_proportional_to_timing_flag; |
||||
int vps_num_ticks_poc_diff_one; ///< vps_num_ticks_poc_diff_one_minus1 + 1
|
||||
int vps_num_hrd_parameters; |
||||
} HEVCVPS; |
||||
|
||||
typedef struct ScalingList { |
||||
/* This is a little wasteful, since sizeID 0 only needs 8 coeffs,
|
||||
* and size ID 3 only has 2 arrays, not 6. */ |
||||
uint8_t sl[4][6][64]; |
||||
uint8_t sl_dc[2][6]; |
||||
} ScalingList; |
||||
|
||||
typedef struct HEVCSPS { |
||||
int vps_id; |
||||
int chroma_format_idc; |
||||
uint8_t separate_colour_plane_flag; |
||||
|
||||
///< output (i.e. cropped) values
|
||||
int output_width, output_height; |
||||
HEVCWindow output_window; |
||||
|
||||
HEVCWindow pic_conf_win; |
||||
|
||||
int bit_depth; |
||||
int pixel_shift; |
||||
enum AVPixelFormat pix_fmt; |
||||
|
||||
unsigned int log2_max_poc_lsb; |
||||
int pcm_enabled_flag; |
||||
|
||||
int max_sub_layers; |
||||
struct { |
||||
int max_dec_pic_buffering; |
||||
int num_reorder_pics; |
||||
int max_latency_increase; |
||||
} temporal_layer[HEVC_MAX_SUB_LAYERS]; |
||||
|
||||
VUI vui; |
||||
PTL ptl; |
||||
|
||||
uint8_t scaling_list_enable_flag; |
||||
ScalingList scaling_list; |
||||
|
||||
unsigned int nb_st_rps; |
||||
ShortTermRPS st_rps[HEVC_MAX_SHORT_TERM_RPS_COUNT]; |
||||
|
||||
uint8_t amp_enabled_flag; |
||||
uint8_t sao_enabled; |
||||
|
||||
uint8_t long_term_ref_pics_present_flag; |
||||
uint16_t lt_ref_pic_poc_lsb_sps[32]; |
||||
uint8_t used_by_curr_pic_lt_sps_flag[32]; |
||||
uint8_t num_long_term_ref_pics_sps; |
||||
|
||||
struct { |
||||
uint8_t bit_depth; |
||||
uint8_t bit_depth_chroma; |
||||
unsigned int log2_min_pcm_cb_size; |
||||
unsigned int log2_max_pcm_cb_size; |
||||
uint8_t loop_filter_disable_flag; |
||||
} pcm; |
||||
uint8_t sps_temporal_mvp_enabled_flag; |
||||
uint8_t sps_strong_intra_smoothing_enable_flag; |
||||
|
||||
unsigned int log2_min_cb_size; |
||||
unsigned int log2_diff_max_min_coding_block_size; |
||||
unsigned int log2_min_tb_size; |
||||
unsigned int log2_max_trafo_size; |
||||
unsigned int log2_ctb_size; |
||||
unsigned int log2_min_pu_size; |
||||
|
||||
int max_transform_hierarchy_depth_inter; |
||||
int max_transform_hierarchy_depth_intra; |
||||
|
||||
///< coded frame dimension in various units
|
||||
int width; |
||||
int height; |
||||
int ctb_width; |
||||
int ctb_height; |
||||
int ctb_size; |
||||
int min_cb_width; |
||||
int min_cb_height; |
||||
int min_tb_width; |
||||
int min_tb_height; |
||||
int min_pu_width; |
||||
int min_pu_height; |
||||
|
||||
int hshift[3]; |
||||
int vshift[3]; |
||||
|
||||
int qp_bd_offset; |
||||
} HEVCSPS; |
||||
|
||||
typedef struct HEVCPPS { |
||||
unsigned int sps_id; ///< seq_parameter_set_id
|
||||
|
||||
uint8_t sign_data_hiding_flag; |
||||
|
||||
uint8_t cabac_init_present_flag; |
||||
|
||||
int num_ref_idx_l0_default_active; ///< num_ref_idx_l0_default_active_minus1 + 1
|
||||
int num_ref_idx_l1_default_active; ///< num_ref_idx_l1_default_active_minus1 + 1
|
||||
int pic_init_qp_minus26; |
||||
|
||||
uint8_t constrained_intra_pred_flag; |
||||
uint8_t transform_skip_enabled_flag; |
||||
|
||||
uint8_t cu_qp_delta_enabled_flag; |
||||
int diff_cu_qp_delta_depth; |
||||
|
||||
int cb_qp_offset; |
||||
int cr_qp_offset; |
||||
uint8_t pic_slice_level_chroma_qp_offsets_present_flag; |
||||
uint8_t weighted_pred_flag; |
||||
uint8_t weighted_bipred_flag; |
||||
uint8_t output_flag_present_flag; |
||||
uint8_t transquant_bypass_enable_flag; |
||||
|
||||
uint8_t dependent_slice_segments_enabled_flag; |
||||
uint8_t tiles_enabled_flag; |
||||
uint8_t entropy_coding_sync_enabled_flag; |
||||
|
||||
int num_tile_columns; ///< num_tile_columns_minus1 + 1
|
||||
int num_tile_rows; ///< num_tile_rows_minus1 + 1
|
||||
uint8_t uniform_spacing_flag; |
||||
uint8_t loop_filter_across_tiles_enabled_flag; |
||||
|
||||
uint8_t seq_loop_filter_across_slices_enabled_flag; |
||||
|
||||
uint8_t deblocking_filter_control_present_flag; |
||||
uint8_t deblocking_filter_override_enabled_flag; |
||||
uint8_t disable_dbf; |
||||
int beta_offset; ///< beta_offset_div2 * 2
|
||||
int tc_offset; ///< tc_offset_div2 * 2
|
||||
|
||||
uint8_t scaling_list_data_present_flag; |
||||
ScalingList scaling_list; |
||||
|
||||
uint8_t lists_modification_present_flag; |
||||
int log2_parallel_merge_level; ///< log2_parallel_merge_level_minus2 + 2
|
||||
int num_extra_slice_header_bits; |
||||
uint8_t slice_header_extension_present_flag; |
||||
|
||||
// Inferred parameters
|
||||
unsigned int *column_width; ///< ColumnWidth
|
||||
unsigned int *row_height; ///< RowHeight
|
||||
unsigned int *col_bd; ///< ColBd
|
||||
unsigned int *row_bd; ///< RowBd
|
||||
int *col_idxX; |
||||
|
||||
int *ctb_addr_rs_to_ts; ///< CtbAddrRSToTS
|
||||
int *ctb_addr_ts_to_rs; ///< CtbAddrTSToRS
|
||||
int *tile_id; ///< TileId
|
||||
int *tile_pos_rs; ///< TilePosRS
|
||||
int *min_tb_addr_zs; ///< MinTbAddrZS
|
||||
} HEVCPPS; |
||||
|
||||
typedef struct HEVCParamSets { |
||||
AVBufferRef *vps_list[HEVC_MAX_VPS_COUNT]; |
||||
AVBufferRef *sps_list[HEVC_MAX_SPS_COUNT]; |
||||
AVBufferRef *pps_list[HEVC_MAX_PPS_COUNT]; |
||||
|
||||
/* currently active parameter sets */ |
||||
const HEVCVPS *vps; |
||||
const HEVCSPS *sps; |
||||
const HEVCPPS *pps; |
||||
} HEVCParamSets; |
||||
|
||||
/**
|
||||
* Parse the SPS from the bitstream into the provided HEVCSPS struct. |
||||
* |
||||
* @param sps_id the SPS id will be written here |
||||
* @param apply_defdispwin if set 1, the default display window from the VUI |
||||
* will be applied to the video dimensions |
||||
* @param vps_list if non-NULL, this function will validate that the SPS refers |
||||
* to an existing VPS |
||||
*/ |
||||
int ff_hevc_parse_sps(HEVCSPS *sps, GetBitContext *gb, unsigned int *sps_id, |
||||
int apply_defdispwin, AVBufferRef **vps_list, AVCodecContext *avctx); |
||||
|
||||
int ff_hevc_decode_nal_vps(GetBitContext *gb, AVCodecContext *avctx, |
||||
HEVCParamSets *ps); |
||||
int ff_hevc_decode_nal_sps(GetBitContext *gb, AVCodecContext *avctx, |
||||
HEVCParamSets *ps, int apply_defdispwin); |
||||
int ff_hevc_decode_nal_pps(GetBitContext *gb, AVCodecContext *avctx, |
||||
HEVCParamSets *ps); |
||||
|
||||
int ff_hevc_decode_short_term_rps(GetBitContext *gb, AVCodecContext *avctx, |
||||
ShortTermRPS *rps, const HEVCSPS *sps, int is_slice_header); |
||||
|
||||
int ff_hevc_encode_nal_vps(HEVCVPS *vps, unsigned int id, |
||||
uint8_t *buf, int buf_size); |
||||
|
||||
#endif /* AVCODEC_HEVC_PS_H */ |
||||
Loading…
Reference in new issue