* rocDecode/HEVC: Code clean up. (#163)
- Moved HEVC standard specific defines out of HevcVideoParser class to a new file.
- Changed some namings to be more specification compliant.
- No functional changes.
[ROCm/rocdecode commit: 2a4026470f]
This commit is contained in:
jeffqjiangNew
committad av
GitHub
GitHub
förälder
347c25c6cc
incheckning
f90684cc40
@@ -0,0 +1,573 @@
|
||||
/*
|
||||
Copyright (c) 2023 - 2024 Advanced Micro Devices, Inc. All rights reserved.
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in
|
||||
all copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
THE SOFTWARE.
|
||||
*/
|
||||
#pragma once
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
#define MAX_VPS_COUNT 16 // 7.3.2.1
|
||||
#define MAX_SPS_COUNT 16 // 7.3.2.2.1
|
||||
#define MAX_PPS_COUNT 64 // 7.4.3.3.1
|
||||
|
||||
#define HEVC_SCALING_LIST_NUM 6 ///< list number for quantization matrix
|
||||
#define HEVC_SCALING_LIST_MAX_INDEX 64
|
||||
|
||||
#define RBSP_BUF_SIZE 1024 // enough to parse any parameter sets or slice headers
|
||||
#define HEVC_MAX_DPB_FRAMES 16 // (A-2)
|
||||
#define HEVC_MAX_NUM_REF_PICS 16
|
||||
// 7.4.7.1. (num_tile_columns_minus1 + 1) * PicHeightInCtbsY − 1. Max tile columns = 20 (A.4.2). Pic height in 16x16 CTB of 8K = 270.
|
||||
#define MAX_ENTRY_POINT_OFFSETS 20 * 270
|
||||
#define INIT_SEI_MESSAGE_COUNT 16 // initial SEI message count
|
||||
#define INIT_SEI_PAYLOAD_BUF_SIZE 1024 * 1024 // initial SEI payload buffer size, 1 MB
|
||||
|
||||
/*! \brief Enumerator for the NAL Unit types - ISO-IEC 14496-15-2004.pdf, page 14, table 1 " NAL unit types in elementary streams
|
||||
*/
|
||||
enum HevcNalUnitType {
|
||||
NAL_UNIT_CODED_SLICE_TRAIL_N = 0, // 0
|
||||
NAL_UNIT_CODED_SLICE_TRAIL_R, // 1
|
||||
|
||||
NAL_UNIT_CODED_SLICE_TSA_N, // 2
|
||||
NAL_UNIT_CODED_SLICE_TLA_R, // 3
|
||||
|
||||
NAL_UNIT_CODED_SLICE_STSA_N, // 4
|
||||
NAL_UNIT_CODED_SLICE_STSA_R, // 5
|
||||
|
||||
NAL_UNIT_CODED_SLICE_RADL_N, // 6
|
||||
NAL_UNIT_CODED_SLICE_RADL_R, // 7
|
||||
|
||||
NAL_UNIT_CODED_SLICE_RASL_N, // 8
|
||||
NAL_UNIT_CODED_SLICE_RASL_R, // 9
|
||||
|
||||
NAL_UNIT_RESERVED_VCL_N10,
|
||||
NAL_UNIT_RESERVED_VCL_R11,
|
||||
NAL_UNIT_RESERVED_VCL_N12,
|
||||
NAL_UNIT_RESERVED_VCL_R13,
|
||||
NAL_UNIT_RESERVED_VCL_N14,
|
||||
NAL_UNIT_RESERVED_VCL_R15,
|
||||
|
||||
NAL_UNIT_CODED_SLICE_BLA_W_LP, // 16
|
||||
NAL_UNIT_CODED_SLICE_BLA_W_RADL, // 17
|
||||
NAL_UNIT_CODED_SLICE_BLA_N_LP, // 18
|
||||
NAL_UNIT_CODED_SLICE_IDR_W_RADL, // 19
|
||||
NAL_UNIT_CODED_SLICE_IDR_N_LP, // 20
|
||||
NAL_UNIT_CODED_SLICE_CRA_NUT, // 21
|
||||
NAL_UNIT_RESERVED_IRAP_VCL22,
|
||||
NAL_UNIT_RESERVED_IRAP_VCL23,
|
||||
|
||||
NAL_UNIT_RESERVED_VCL24,
|
||||
NAL_UNIT_RESERVED_VCL25,
|
||||
NAL_UNIT_RESERVED_VCL26,
|
||||
NAL_UNIT_RESERVED_VCL27,
|
||||
NAL_UNIT_RESERVED_VCL28,
|
||||
NAL_UNIT_RESERVED_VCL29,
|
||||
NAL_UNIT_RESERVED_VCL30,
|
||||
NAL_UNIT_RESERVED_VCL31,
|
||||
|
||||
NAL_UNIT_VPS, // 32
|
||||
NAL_UNIT_SPS, // 33
|
||||
NAL_UNIT_PPS, // 34
|
||||
NAL_UNIT_ACCESS_UNIT_DELIMITER, // 35
|
||||
NAL_UNIT_EOS, // 36
|
||||
NAL_UNIT_EOB, // 37
|
||||
NAL_UNIT_FILLER_DATA, // 38
|
||||
NAL_UNIT_PREFIX_SEI, // 39
|
||||
NAL_UNIT_SUFFIX_SEI, // 40
|
||||
NAL_UNIT_RESERVED_NVCL41,
|
||||
NAL_UNIT_RESERVED_NVCL42,
|
||||
NAL_UNIT_RESERVED_NVCL43,
|
||||
NAL_UNIT_RESERVED_NVCL44,
|
||||
NAL_UNIT_RESERVED_NVCL45,
|
||||
NAL_UNIT_RESERVED_NVCL46,
|
||||
NAL_UNIT_RESERVED_NVCL47,
|
||||
NAL_UNIT_UNSPECIFIED_48,
|
||||
NAL_UNIT_UNSPECIFIED_49,
|
||||
NAL_UNIT_UNSPECIFIED_50,
|
||||
NAL_UNIT_UNSPECIFIED_51,
|
||||
NAL_UNIT_UNSPECIFIED_52,
|
||||
NAL_UNIT_UNSPECIFIED_53,
|
||||
NAL_UNIT_UNSPECIFIED_54,
|
||||
NAL_UNIT_UNSPECIFIED_55,
|
||||
NAL_UNIT_UNSPECIFIED_56,
|
||||
NAL_UNIT_UNSPECIFIED_57,
|
||||
NAL_UNIT_UNSPECIFIED_58,
|
||||
NAL_UNIT_UNSPECIFIED_59,
|
||||
NAL_UNIT_UNSPECIFIED_60,
|
||||
NAL_UNIT_UNSPECIFIED_61,
|
||||
NAL_UNIT_UNSPECIFIED_62,
|
||||
NAL_UNIT_UNSPECIFIED_63,
|
||||
NAL_UNIT_INVALID,
|
||||
};
|
||||
|
||||
/*! \brief Structure to hold the NAL Unit Header
|
||||
*/
|
||||
struct HevcNalUnitHeader {
|
||||
uint32_t forbidden_zero_bit;
|
||||
uint32_t nal_unit_type;
|
||||
uint32_t nuh_layer_id;
|
||||
uint32_t nuh_temporal_id_plus1;
|
||||
uint32_t num_emu_byte_removed;
|
||||
};
|
||||
|
||||
/*! \brief Enumerator for the scaling list sizes
|
||||
*/
|
||||
enum HevcScalingListSize {
|
||||
HEVC_SCALING_LIST_4x4 = 0,
|
||||
HEVC_SCALING_LIST_8x8,
|
||||
HEVC_SCALING_LIST_16x16,
|
||||
HEVC_SCALING_LIST_32x32,
|
||||
HEVC_SCALING_LIST_SIZE_NUM
|
||||
};
|
||||
|
||||
/*! \brief Slice type
|
||||
*/
|
||||
enum HevcSliceType {
|
||||
HEVC_SLICE_TYPE_B = 0,
|
||||
HEVC_SLICE_TYPE_P,
|
||||
HEVC_SLICE_TYPE_I
|
||||
};
|
||||
|
||||
/*! \brief Reference picture type
|
||||
*/
|
||||
enum HevcRefMarking {
|
||||
kUnusedForReference = 0,
|
||||
kUsedForShortTerm = 1,
|
||||
kUsedForLongTerm = 2
|
||||
};
|
||||
|
||||
/*! \brief Structure for Profile Tier Levels
|
||||
*/
|
||||
typedef struct {
|
||||
uint32_t general_profile_space; //u(2)
|
||||
bool general_tier_flag; //u(1)
|
||||
uint32_t general_profile_idc; //u(5)
|
||||
bool general_profile_compatibility_flag[32]; //u(1)
|
||||
bool general_progressive_source_flag; //u(1)
|
||||
bool general_interlaced_source_flag; //u(1)
|
||||
bool general_non_packed_constraint_flag; //u(1)
|
||||
bool general_frame_only_constraint_flag; //u(1)
|
||||
uint64_t general_reserved_zero_44bits; //u(44)
|
||||
uint32_t general_level_idc; //u(8)
|
||||
//max_num_sub_layers_minus1 max is 7 - 1 = 6
|
||||
bool sub_layer_profile_present_flag[6]; //u(1)
|
||||
bool sub_layer_level_present_flag[6]; //u(1)
|
||||
|
||||
uint32_t reserved_zero_2bits[8]; //u(2)
|
||||
|
||||
uint32_t sub_layer_profile_space[6]; //u(2)
|
||||
bool sub_layer_tier_flag[6]; //u(1)
|
||||
uint32_t sub_layer_profile_idc[6]; //u(5)
|
||||
bool sub_layer_profile_compatibility_flag[6][32]; //u(1)
|
||||
bool sub_layer_progressive_source_flag[6]; //u(1)
|
||||
bool sub_layer_interlaced_source_flag[6]; //u(1)
|
||||
bool sub_layer_non_packed_constraint_flag[6]; //u(1)
|
||||
bool sub_layer_frame_only_constraint_flag[6]; //u(1)
|
||||
uint64_t sub_layer_reserved_zero_44bits[6]; //u(44)
|
||||
uint32_t sub_layer_level_idc[6]; //u(8)
|
||||
} HevcProfileTierLevel;
|
||||
|
||||
/*! \brief Structure for Scaling List Data
|
||||
*/
|
||||
typedef struct {
|
||||
bool scaling_list_pred_mode_flag[4][6]; //u(1)
|
||||
uint32_t scaling_list_pred_matrix_id_delta[4][6]; //ue(v)
|
||||
int32_t scaling_list_dc_coef_minus8[4][6]; //se(v)
|
||||
int32_t scaling_list_delta_coef; //se(v)
|
||||
int32_t scaling_list[HEVC_SCALING_LIST_SIZE_NUM][HEVC_SCALING_LIST_NUM][HEVC_SCALING_LIST_MAX_INDEX];
|
||||
int32_t scaling_list_dc_coef[2][6]; // DC coefficient for 16x16 and 32x32
|
||||
} HevcScalingListData;
|
||||
|
||||
/*! \brief Structure for Short Term Reference Picture Set
|
||||
*/
|
||||
typedef struct {
|
||||
uint8_t inter_ref_pic_set_prediction_flag;
|
||||
uint32_t delta_idx_minus1;
|
||||
uint8_t delta_rps_sign;
|
||||
uint32_t abs_delta_rps_minus1;
|
||||
uint8_t used_by_curr_pic_flag[HEVC_MAX_DPB_FRAMES];
|
||||
uint8_t use_delta_flag[HEVC_MAX_DPB_FRAMES];
|
||||
|
||||
uint32_t delta_poc_s0_minus1[HEVC_MAX_DPB_FRAMES];
|
||||
uint8_t used_by_curr_pic_s0_flag[HEVC_MAX_DPB_FRAMES];
|
||||
uint32_t delta_poc_s1_minus1[HEVC_MAX_DPB_FRAMES];
|
||||
uint8_t used_by_curr_pic_s1_flag[HEVC_MAX_DPB_FRAMES];
|
||||
|
||||
uint32_t num_negative_pics; // NumNegativePics
|
||||
uint32_t num_positive_pics; // NumPositivePics
|
||||
uint32_t num_of_delta_pocs; // NumDeltaPocs
|
||||
uint8_t used_by_curr_pic_s0[HEVC_MAX_DPB_FRAMES]; // UsedByCurrPicS0
|
||||
uint8_t used_by_curr_pic_s1[HEVC_MAX_DPB_FRAMES]; // UsedByCurrPicS1
|
||||
int32_t delta_poc_s0[HEVC_MAX_DPB_FRAMES]; // DeltaPocS0
|
||||
int32_t delta_poc_s1[HEVC_MAX_DPB_FRAMES]; // DeltaPocS1
|
||||
} HevcShortTermRps;
|
||||
|
||||
/*! \brief Structure for Long Term Reference Picture Set
|
||||
*/
|
||||
typedef struct {
|
||||
int32_t num_of_pics;
|
||||
int32_t pocs[32]; // PocLsbLt
|
||||
bool used_by_curr_pic[32]; // UsedByCurrPicLt
|
||||
} HevcLongTermRps;
|
||||
|
||||
/*! \brief Structure for Sub Layer Hypothetical Reference Decoder Parameters
|
||||
*/
|
||||
typedef struct {
|
||||
//CpbCnt = cpb_cnt_minus1
|
||||
uint32_t bit_rate_value_minus1[32]; //ue(v)
|
||||
uint32_t cpb_size_value_minus1[32]; //ue(v)
|
||||
uint32_t cpb_size_du_value_minus1[32]; //ue(v)
|
||||
uint32_t bit_rate_du_value_minus1[32]; //ue(v)
|
||||
bool cbr_flag[32]; //u(1)
|
||||
} HevcSubLayerHrdParameters;
|
||||
|
||||
/*! \brief Structure for Hypothetical Reference Decoder Parameters
|
||||
*/
|
||||
typedef struct {
|
||||
bool nal_hrd_parameters_present_flag; //u(1)
|
||||
bool vcl_hrd_parameters_present_flag; //u(1)
|
||||
bool sub_pic_hrd_params_present_flag; //u(1)
|
||||
uint32_t tick_divisor_minus2; //u(8)
|
||||
uint32_t du_cpb_removal_delay_increment_length_minus1; //u(5)
|
||||
bool sub_pic_cpb_params_in_pic_timing_sei_flag; //u(1)
|
||||
uint32_t dpb_output_delay_du_length_minus1; //u(5)
|
||||
uint32_t bit_rate_scale; //u(4)
|
||||
uint32_t cpb_size_scale; //u(4)
|
||||
uint32_t cpb_size_du_scale; //u(4)
|
||||
uint32_t initial_cpb_removal_delay_length_minus1; //u(5)
|
||||
uint32_t au_cpb_removal_delay_length_minus1; //u(5)
|
||||
uint32_t dpb_output_delay_length_minus1; //u(5)
|
||||
bool fixed_pic_rate_general_flag[7]; //u(1)
|
||||
bool fixed_pic_rate_within_cvs_flag[7]; //u(1)
|
||||
uint32_t elemental_duration_in_tc_minus1[7]; //ue(v)
|
||||
bool low_delay_hrd_flag[7]; //u(1)
|
||||
uint32_t cpb_cnt_minus1[7]; //ue(v)
|
||||
//sub_layer_hrd_parameters()
|
||||
HevcSubLayerHrdParameters sub_layer_hrd_parameters_0[7];
|
||||
//sub_layer_hrd_parameters()
|
||||
HevcSubLayerHrdParameters sub_layer_hrd_parameters_1[7];
|
||||
} HevcHrdParameters;
|
||||
|
||||
/*! \brief Structure for Video Usability Information Parameters
|
||||
*/
|
||||
typedef struct {
|
||||
bool aspect_ratio_info_present_flag; //u(1)
|
||||
uint32_t aspect_ratio_idc; //u(8)
|
||||
uint32_t sar_width; //u(16)
|
||||
uint32_t sar_height; //u(16)
|
||||
bool overscan_info_present_flag; //u(1)
|
||||
bool overscan_appropriate_flag; //u(1)
|
||||
bool video_signal_type_present_flag; //u(1)
|
||||
uint32_t video_format; //u(3)
|
||||
bool video_full_range_flag; //u(1)
|
||||
bool colour_description_present_flag; //u(1)
|
||||
uint32_t colour_primaries; //u(8)
|
||||
uint32_t transfer_characteristics; //u(8)
|
||||
uint32_t matrix_coeffs; //u(8)
|
||||
bool chroma_loc_info_present_flag; //u(1)
|
||||
uint32_t chroma_sample_loc_type_top_field; //ue(v)
|
||||
uint32_t chroma_sample_loc_type_bottom_field; //ue(v)
|
||||
bool neutral_chroma_indication_flag; //u(1)
|
||||
bool field_seq_flag; //u(1)
|
||||
bool frame_field_info_present_flag; //u(1)
|
||||
bool default_display_window_flag; //u(1)
|
||||
uint32_t def_disp_win_left_offset; //ue(v)
|
||||
uint32_t def_disp_win_right_offset; //ue(v)
|
||||
uint32_t def_disp_win_top_offset; //ue(v)
|
||||
uint32_t def_disp_win_bottom_offset; //ue(v)
|
||||
bool vui_timing_info_present_flag; //u(1)
|
||||
uint32_t vui_num_units_in_tick; //u(32)
|
||||
uint32_t vui_time_scale; //u(32)
|
||||
bool vui_poc_proportional_to_timing_flag; //u(1)
|
||||
uint32_t vui_num_ticks_poc_diff_one_minus1; //ue(v)
|
||||
bool vui_hrd_parameters_present_flag; //u(1)
|
||||
//hrd_parameters()
|
||||
HevcHrdParameters hrd_parameters;
|
||||
bool bitstream_restriction_flag; //u(1)
|
||||
bool tiles_fixed_structure_flag; //u(1)
|
||||
bool motion_vectors_over_pic_boundaries_flag; //u(1)
|
||||
bool restricted_ref_pic_lists_flag; //u(1)
|
||||
uint32_t min_spatial_segmentation_idc; //ue(v)
|
||||
uint32_t max_bytes_per_pic_denom; //ue(v)
|
||||
uint32_t max_bits_per_min_cu_denom; //ue(v)
|
||||
uint32_t log2_max_mv_length_horizontal; //ue(v)
|
||||
uint32_t log2_max_mv_length_vertical; //ue(v)
|
||||
} HevcVuiParameters;
|
||||
|
||||
typedef struct {
|
||||
uint32_t luma_log2_weight_denom; //ue(v)
|
||||
int32_t delta_chroma_log2_weight_denom; //se(v)
|
||||
uint8_t luma_weight_l0_flag[16]; //u(1)
|
||||
uint8_t chroma_weight_l0_flag[16]; //u(1)
|
||||
int32_t delta_luma_weight_l0[16]; //se(v)
|
||||
int32_t luma_offset_l0[16]; //se(v)
|
||||
int32_t delta_chroma_weight_l0[16][2]; //se(v)
|
||||
int32_t delta_chroma_offset_l0[16][2]; //se(v)
|
||||
int32_t chroma_weight_l0[16][2]; //ChromaWeightL0[]
|
||||
int32_t chroma_offset_l0[16][2]; //ChromaOffsetL0[]
|
||||
uint8_t luma_weight_l1_flag[16]; //u(1)
|
||||
uint8_t chroma_weight_l1_flag[16]; //u(1)
|
||||
int32_t delta_luma_weight_l1[16]; //se(v)
|
||||
int32_t luma_offset_l1[16]; //se(v)
|
||||
int32_t delta_chroma_weight_l1[16][2]; //se(v)
|
||||
int32_t delta_chroma_offset_l1[16][2]; //se(v)
|
||||
int32_t chroma_weight_l1[16][2]; //ChromaWeightL1[]
|
||||
int32_t chroma_offset_l1[16][2]; //ChromaOffsetL1[]
|
||||
} HevcPredWeightTable;
|
||||
|
||||
/*! \brief Structure for Raw Byte Sequence Payload Trialing Bits
|
||||
*/
|
||||
typedef struct {
|
||||
uint32_t rbsp_stop_one_bit; /* equal to 1 */
|
||||
uint32_t rbsp_alignment_zero_bit; /* equal to 0 */
|
||||
} HevcRbspTrailingBits;
|
||||
|
||||
/*! \brief Structure for Video Parameter Set
|
||||
*/
|
||||
typedef struct{
|
||||
uint32_t vps_video_parameter_set_id; //u(4)
|
||||
uint32_t vps_base_layer_internal_flag; //u(1)
|
||||
uint32_t vps_base_layer_available_flag; //u(1)
|
||||
uint32_t vps_max_layers_minus1; //u(6)
|
||||
uint32_t vps_max_sub_layers_minus1; //u(3)
|
||||
bool vps_temporal_id_nesting_flag; //u(1)
|
||||
uint32_t vps_reserved_0xffff_16bits; //u(16)
|
||||
//profile_tier_level( vps_max_sub_layers_minus1 )
|
||||
HevcProfileTierLevel profile_tier_level;
|
||||
bool vps_sub_layer_ordering_info_present_flag; //u(1)
|
||||
//vps_max_sub_layers_minus1 max is 6, need to +1
|
||||
uint32_t vps_max_dec_pic_buffering_minus1[7]; //ue(v)
|
||||
uint32_t vps_max_num_reorder_pics[7]; //ue(v)
|
||||
uint32_t vps_max_latency_increase_plus1[7]; //ue(v)
|
||||
uint32_t vps_max_layer_id; //u(6)
|
||||
uint32_t vps_num_layer_sets_minus1; //ue(v)
|
||||
//vps_num_layer_sets_minus1 max is 1023 (dont +1 since starts from 1)
|
||||
//vps_max_layer_id max is 62 (+1 since starts from 0 and <= condition)
|
||||
bool layer_id_included_flag[1023][63]; //u(1)
|
||||
bool vps_timing_info_present_flag; //u(1)
|
||||
uint32_t vps_num_units_in_tick; //u(32)
|
||||
uint32_t vps_time_scale; //u(32)
|
||||
bool vps_poc_proportional_to_timing_flag; //u(1)
|
||||
uint32_t vps_num_ticks_poc_diff_one_minus1; //ue(v)
|
||||
uint32_t vps_num_hrd_parameters; //ue(v)
|
||||
//vps_num_hrd_parameters max is 1024
|
||||
uint32_t hrd_layer_set_idx[1024]; //ue(v)
|
||||
bool cprms_present_flag[1024]; //u(1)
|
||||
//hrd_parameters()
|
||||
HevcHrdParameters hrd_parameters[1024];
|
||||
bool vps_extension_flag; //u(1)
|
||||
bool vps_extension_data_flag; //u(1)
|
||||
//rbsp_trailing_bits()
|
||||
HevcRbspTrailingBits rbsp_trailing_bits;
|
||||
} HevcVideoParamSet;
|
||||
|
||||
/*! \brief Structure for Sequence Parameter Set
|
||||
*/
|
||||
typedef struct {
|
||||
uint32_t sps_video_parameter_set_id; //u(4)
|
||||
uint32_t sps_max_sub_layers_minus1; //u(3)
|
||||
bool sps_temporal_id_nesting_flag; //u(1)
|
||||
//profile_tier_level( sps_max_sub_layers_minus1 )
|
||||
HevcProfileTierLevel profile_tier_level;
|
||||
uint32_t sps_seq_parameter_set_id; //ue(v)
|
||||
uint32_t chroma_format_idc; //ue(v)
|
||||
bool separate_colour_plane_flag; //u(1)
|
||||
uint32_t pic_width_in_luma_samples; //ue(v)
|
||||
uint32_t pic_height_in_luma_samples; //ue(v)
|
||||
uint32_t max_cu_width;
|
||||
uint32_t max_cu_height;
|
||||
uint32_t max_cu_depth;
|
||||
bool conformance_window_flag; //u(1)
|
||||
uint32_t conf_win_left_offset; //ue(v)
|
||||
uint32_t conf_win_right_offset; //ue(v)
|
||||
uint32_t conf_win_top_offset; //ue(v)
|
||||
uint32_t conf_win_bottom_offset; //ue(v)
|
||||
uint32_t bit_depth_luma_minus8; //ue(v)
|
||||
uint32_t bit_depth_chroma_minus8; //ue(v)
|
||||
uint32_t log2_max_pic_order_cnt_lsb_minus4; //ue(v)
|
||||
bool sps_sub_layer_ordering_info_present_flag; //u(1)
|
||||
uint32_t sps_max_dec_pic_buffering_minus1[7]; //ue(v)
|
||||
uint32_t sps_max_num_reorder_pics[7]; //ue(v)
|
||||
uint32_t sps_max_latency_increase_plus1[7]; //ue(v)
|
||||
uint32_t log2_min_luma_coding_block_size_minus3; //ue(v)
|
||||
uint32_t log2_diff_max_min_luma_coding_block_size; //ue(v)
|
||||
uint32_t log2_min_transform_block_size_minus2; //ue(v)
|
||||
uint32_t log2_diff_max_min_transform_block_size; //ue(v)
|
||||
uint32_t max_transform_hierarchy_depth_inter; //ue(v)
|
||||
uint32_t max_transform_hierarchy_depth_intra; //ue(v)
|
||||
bool scaling_list_enabled_flag; //u(1)
|
||||
bool sps_scaling_list_data_present_flag; //u(1)
|
||||
//scaling_list_data()
|
||||
HevcScalingListData scaling_list_data;
|
||||
bool amp_enabled_flag; //u(1)
|
||||
bool sample_adaptive_offset_enabled_flag; //u(1)
|
||||
bool pcm_enabled_flag; //u(1)
|
||||
uint32_t pcm_sample_bit_depth_luma_minus1; //u(4)
|
||||
uint32_t pcm_sample_bit_depth_chroma_minus1; //u(4)
|
||||
uint32_t log2_min_pcm_luma_coding_block_size_minus3; //ue(v)
|
||||
uint32_t log2_diff_max_min_pcm_luma_coding_block_size; //ue(v)
|
||||
bool pcm_loop_filter_disabled_flag; //u(1)
|
||||
uint32_t num_short_term_ref_pic_sets; //ue(v)
|
||||
//short_term_ref_pic_set(i) max is 64
|
||||
HevcShortTermRps st_rps[64];
|
||||
HevcLongTermRps lt_rps;
|
||||
bool long_term_ref_pics_present_flag; //u(1)
|
||||
uint32_t num_long_term_ref_pics_sps; //ue(v)
|
||||
//max is 32
|
||||
uint32_t lt_ref_pic_poc_lsb_sps[32]; //u(v)
|
||||
bool used_by_curr_pic_lt_sps_flag[32]; //u(1)
|
||||
bool sps_temporal_mvp_enabled_flag; //u(1)
|
||||
bool strong_intra_smoothing_enabled_flag; //u(1)
|
||||
bool vui_parameters_present_flag; //u(1)
|
||||
//vui_parameters()
|
||||
HevcVuiParameters vui_parameters;
|
||||
bool sps_extension_flag; //u(1)
|
||||
bool sps_extension_data_flag; //u(1)
|
||||
//rbsp_trailing_bits( )
|
||||
HevcRbspTrailingBits rbsp_trailing_bits;
|
||||
} HevcSeqParamSet;
|
||||
|
||||
/*! \brief Structure for Picture Parameter Set
|
||||
*/
|
||||
typedef struct {
|
||||
uint32_t pps_pic_parameter_set_id; //ue(v)
|
||||
uint32_t pps_seq_parameter_set_id; //ue(v)
|
||||
bool dependent_slice_segments_enabled_flag; //u(1)
|
||||
bool output_flag_present_flag; //u(1)
|
||||
uint32_t num_extra_slice_header_bits; //u(3)
|
||||
bool sign_data_hiding_enabled_flag; //u(1)
|
||||
bool cabac_init_present_flag; //u(1)
|
||||
uint32_t num_ref_idx_l0_default_active_minus1; //ue(v)
|
||||
uint32_t num_ref_idx_l1_default_active_minus1; //ue(v)
|
||||
int32_t init_qp_minus26; //se(v)
|
||||
bool constrained_intra_pred_flag; //u(1)
|
||||
bool transform_skip_enabled_flag; //u(1)
|
||||
bool cu_qp_delta_enabled_flag; //u(1)
|
||||
uint32_t diff_cu_qp_delta_depth; //ue(v)
|
||||
int32_t pps_cb_qp_offset; //se(v)
|
||||
int32_t pps_cr_qp_offset; //se(v)
|
||||
bool pps_slice_chroma_qp_offsets_present_flag; //u(1)
|
||||
bool weighted_pred_flag; //u(1)
|
||||
bool weighted_bipred_flag; //u(1)
|
||||
bool transquant_bypass_enabled_flag; //u(1)
|
||||
bool tiles_enabled_flag; //u(1)
|
||||
bool entropy_coding_sync_enabled_flag; //u(1)
|
||||
uint32_t num_tile_columns_minus1; //ue(v)
|
||||
uint32_t num_tile_rows_minus1; //ue(v)
|
||||
bool uniform_spacing_flag; //u(1)
|
||||
//PicWidthInCtbsY = Ceil( pic_width_in_luma_samples / CtbSizeY ) = 256 assume max width is 4096
|
||||
//CtbSizeY = 1<<CtbLog2SizeY so min is 16
|
||||
// 4 <= CtbLog2SizeY <= 6
|
||||
uint32_t column_width_minus1[265]; //ue(v)
|
||||
//2304/16=144 assume max height is 2304
|
||||
uint32_t row_height_minus1[144]; //ue(v)
|
||||
bool loop_filter_across_tiles_enabled_flag; //u(1)
|
||||
bool pps_loop_filter_across_slices_enabled_flag; //u(1)
|
||||
bool deblocking_filter_control_present_flag; //u(1)
|
||||
bool deblocking_filter_override_enabled_flag; //u(1)
|
||||
bool pps_deblocking_filter_disabled_flag; //u(1)
|
||||
int32_t pps_beta_offset_div2; //se(v)
|
||||
int32_t pps_tc_offset_div2; //se(v)
|
||||
bool pps_scaling_list_data_present_flag; //u(1)
|
||||
//scaling_list_data( )
|
||||
HevcScalingListData scaling_list_data;
|
||||
bool lists_modification_present_flag; //u(1)
|
||||
uint32_t log2_parallel_merge_level_minus2; //ue(v)
|
||||
bool slice_segment_header_extension_present_flag; //u(1)
|
||||
bool pps_extension_present_flag; //u(1)
|
||||
bool pps_range_extension_flag; //u(1)
|
||||
bool pps_multilayer_extension_flag; //u(1)
|
||||
uint32_t pps_extension_6bits; //u(6)
|
||||
// pps_range_extension()
|
||||
uint32_t log2_max_transform_skip_block_size_minus2; //ue(v)
|
||||
uint8_t cross_component_prediction_enabled_flag; //u(1)
|
||||
uint8_t chroma_qp_offset_list_enabled_flag; //u(1)
|
||||
uint32_t diff_cu_chroma_qp_offset_depth; //ue(v)
|
||||
uint32_t chroma_qp_offset_list_len_minus1; //ue(v)
|
||||
int32_t cb_qp_offset_list[6]; //se(v)
|
||||
int32_t cr_qp_offset_list[6]; //se(v)
|
||||
uint32_t log2_sao_offset_scale_luma; //ue(v)
|
||||
uint32_t log2_sao_offset_scale_chroma; //ue(v)
|
||||
bool pps_extension_data_flag; //u(1)
|
||||
//rbsp_trailing_bits( )
|
||||
HevcRbspTrailingBits rbsp_trailing_bits;
|
||||
} HevcPicParamSet;
|
||||
|
||||
/*! \brief Structure for Slice Segment Header
|
||||
*/
|
||||
typedef struct {
|
||||
bool first_slice_segment_in_pic_flag; //u(1)
|
||||
bool no_output_of_prior_pics_flag; //u(1)
|
||||
uint32_t slice_pic_parameter_set_id; //ue(v)
|
||||
bool dependent_slice_segment_flag; //u(1)
|
||||
uint32_t slice_segment_address; //u(v)
|
||||
//num_extra_slice_header_bits is u(3), so max is 7
|
||||
bool slice_reserved_flag[7]; //u(1)
|
||||
uint32_t slice_type; //ue(v)
|
||||
bool pic_output_flag; //u(1)
|
||||
uint32_t colour_plane_id; //u(2)
|
||||
uint32_t slice_pic_order_cnt_lsb; //u(v)
|
||||
bool short_term_ref_pic_set_sps_flag; //u(1)
|
||||
//short_term_ref_pic_set( num_short_term_ref_pic_sets )
|
||||
uint32_t short_term_ref_pic_set_size; //MM
|
||||
HevcShortTermRps st_rps;
|
||||
uint32_t short_term_ref_pic_set_idx; //u(v)
|
||||
uint32_t num_long_term_sps; //ue(v)
|
||||
uint32_t num_long_term_pics; //ue(v)
|
||||
//num_long_term_sps + num_long_term_pics max is 32
|
||||
HevcLongTermRps lt_rps;
|
||||
uint32_t lt_idx_sps[32]; //u(v)
|
||||
uint32_t poc_lsb_lt[32]; //u(v)
|
||||
bool used_by_curr_pic_lt_flag[32]; //u(1)
|
||||
bool delta_poc_msb_present_flag[32]; //u(1)
|
||||
uint32_t delta_poc_msb_cycle_lt[32]; //ue(v)
|
||||
bool slice_temporal_mvp_enabled_flag; //u(1)
|
||||
bool slice_sao_luma_flag; //u(1)
|
||||
bool slice_sao_chroma_flag; //u(1)
|
||||
bool num_ref_idx_active_override_flag; //u(1)
|
||||
uint32_t num_ref_idx_l0_active_minus1; //ue(v)
|
||||
uint32_t num_ref_idx_l1_active_minus1; //ue(v)
|
||||
// Reference picture list modification
|
||||
uint32_t ref_pic_list_modification_flag_l0; //u(1)
|
||||
uint32_t list_entry_l0[16]; //u(v)
|
||||
uint32_t ref_pic_list_modification_flag_l1; //u(1)
|
||||
uint32_t list_entry_l1[16]; //u(v)
|
||||
bool mvd_l1_zero_flag; //u(1)
|
||||
bool cabac_init_flag; //u(1)
|
||||
bool collocated_from_l0_flag; //u(1)
|
||||
HevcPredWeightTable pred_weight_table;
|
||||
uint32_t collocated_ref_idx; //ue(v)
|
||||
uint32_t five_minus_max_num_merge_cand; //ue(v)
|
||||
int32_t slice_qp_delta; //se(v)
|
||||
int32_t slice_cb_qp_offset; //se(v)
|
||||
int32_t slice_cr_qp_offset; //se(v)
|
||||
uint8_t cu_chroma_qp_offset_enabled_flag; //u(1)
|
||||
bool deblocking_filter_override_flag; //u(1)
|
||||
bool slice_deblocking_filter_disabled_flag; //u(1)
|
||||
int32_t slice_beta_offset_div2; //se(v)
|
||||
int32_t slice_tc_offset_div2; //se(v)
|
||||
bool slice_loop_filter_across_slices_enabled_flag; //u(1)
|
||||
uint32_t num_entry_point_offsets; //ue(v)
|
||||
uint32_t offset_len_minus1; //ue(v)
|
||||
uint32_t entry_point_offset_minus1[MAX_ENTRY_POINT_OFFSETS]; //u(v)
|
||||
uint32_t slice_segment_header_extension_length; //ue(v)
|
||||
//slice_segment_header_extension_length max is 256
|
||||
uint8_t slice_segment_header_extension_data_byte[256]; //u(8)
|
||||
} HevcSliceSegHeader;
|
||||
|
||||
@@ -35,7 +35,7 @@ inline T *AllocStruct(const int max_cnt) {
|
||||
return p;
|
||||
}
|
||||
|
||||
HEVCVideoParser::HEVCVideoParser() {
|
||||
HevcVideoParser::HevcVideoParser() {
|
||||
pic_count_ = 0;
|
||||
first_pic_after_eos_nal_unit_ = 0;
|
||||
m_active_vps_id_ = -1;
|
||||
@@ -43,11 +43,11 @@ HEVCVideoParser::HEVCVideoParser() {
|
||||
m_active_pps_id_ = -1;
|
||||
b_new_picture_ = false;
|
||||
// allocate all fixed size structors here
|
||||
m_vps_ = AllocStruct<VpsData>(MAX_VPS_COUNT);
|
||||
m_sps_ = AllocStruct<SpsData>(MAX_SPS_COUNT);
|
||||
m_pps_ = AllocStruct<PpsData>(MAX_PPS_COUNT);
|
||||
m_sh_ = AllocStruct<SliceHeaderData>(1);
|
||||
m_sh_copy_ = AllocStruct<SliceHeaderData>(1);
|
||||
m_vps_ = AllocStruct<HevcVideoParamSet>(MAX_VPS_COUNT);
|
||||
m_sps_ = AllocStruct<HevcSeqParamSet>(MAX_SPS_COUNT);
|
||||
m_pps_ = AllocStruct<HevcPicParamSet>(MAX_PPS_COUNT);
|
||||
m_sh_ = AllocStruct<HevcSliceSegHeader>(1);
|
||||
m_sh_copy_ = AllocStruct<HevcSliceSegHeader>(1);
|
||||
|
||||
sei_rbsp_buf_ = nullptr;
|
||||
sei_rbsp_buf_size_ = 0;
|
||||
@@ -59,7 +59,7 @@ HEVCVideoParser::HEVCVideoParser() {
|
||||
InitDpb();
|
||||
}
|
||||
|
||||
rocDecStatus HEVCVideoParser::Initialize(RocdecParserParams *p_params) {
|
||||
rocDecStatus HevcVideoParser::Initialize(RocdecParserParams *p_params) {
|
||||
ParserResult status = Init();
|
||||
if (status)
|
||||
return ROCDEC_RUNTIME_ERROR;
|
||||
@@ -72,13 +72,13 @@ rocDecStatus HEVCVideoParser::Initialize(RocdecParserParams *p_params) {
|
||||
*
|
||||
* @return rocDecStatus
|
||||
*/
|
||||
rocDecStatus HEVCVideoParser::UnInitialize() {
|
||||
rocDecStatus HevcVideoParser::UnInitialize() {
|
||||
//todo:: do any uninitialization here
|
||||
return ROCDEC_SUCCESS;
|
||||
}
|
||||
|
||||
|
||||
rocDecStatus HEVCVideoParser::ParseVideoData(RocdecSourceDataPacket *p_data) {
|
||||
rocDecStatus HevcVideoParser::ParseVideoData(RocdecSourceDataPacket *p_data) {
|
||||
if (p_data->payload && p_data->payload_size) {
|
||||
// Clear DPB output/display buffer number
|
||||
dpb_buffer_.num_output_pics = 0;
|
||||
@@ -130,7 +130,7 @@ rocDecStatus HEVCVideoParser::ParseVideoData(RocdecSourceDataPacket *p_data) {
|
||||
return ROCDEC_SUCCESS;
|
||||
}
|
||||
|
||||
HEVCVideoParser::~HEVCVideoParser() {
|
||||
HevcVideoParser::~HevcVideoParser() {
|
||||
if (m_vps_) {
|
||||
delete [] m_vps_;
|
||||
}
|
||||
@@ -154,12 +154,12 @@ HEVCVideoParser::~HEVCVideoParser() {
|
||||
}
|
||||
}
|
||||
|
||||
ParserResult HEVCVideoParser::Init() {
|
||||
ParserResult HevcVideoParser::Init() {
|
||||
b_new_picture_ = false;
|
||||
return PARSER_OK;
|
||||
}
|
||||
|
||||
int HEVCVideoParser::FillSeqCallbackFn(SpsData* sps_data) {
|
||||
int HevcVideoParser::FillSeqCallbackFn(HevcSeqParamSet* sps_data) {
|
||||
video_format_params_.codec = rocDecVideoCodec_HEVC;
|
||||
video_format_params_.frame_rate.numerator = frame_rate_.numerator;
|
||||
video_format_params_.frame_rate.denominator = frame_rate_.denominator;
|
||||
@@ -242,7 +242,7 @@ int HEVCVideoParser::FillSeqCallbackFn(SpsData* sps_data) {
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::FillSeiMessageCallbackFn() {
|
||||
void HevcVideoParser::FillSeiMessageCallbackFn() {
|
||||
sei_message_info_params_.sei_message_count = sei_message_count_;
|
||||
sei_message_info_params_.pSEIMessage = sei_message_list_.data();
|
||||
sei_message_info_params_.pSEIData = (void*)sei_payload_buf_;
|
||||
@@ -252,10 +252,10 @@ void HEVCVideoParser::FillSeiMessageCallbackFn() {
|
||||
if (pfn_get_sei_message_cb_) pfn_get_sei_message_cb_(parser_params_.pUserData, &sei_message_info_params_);
|
||||
}
|
||||
|
||||
int HEVCVideoParser::SendPicForDecode() {
|
||||
int HevcVideoParser::SendPicForDecode() {
|
||||
int i, j, ref_idx, buf_idx;
|
||||
SpsData *sps_ptr = &m_sps_[m_active_sps_id_];
|
||||
PpsData *pps_ptr = &m_pps_[m_active_pps_id_];
|
||||
HevcSeqParamSet *sps_ptr = &m_sps_[m_active_sps_id_];
|
||||
HevcPicParamSet *pps_ptr = &m_pps_[m_active_pps_id_];
|
||||
dec_pic_params_ = {0};
|
||||
|
||||
dec_pic_params_.PicWidth = sps_ptr->pic_width_in_luma_samples;
|
||||
@@ -512,7 +512,7 @@ int HEVCVideoParser::SendPicForDecode() {
|
||||
/// Fill scaling lists
|
||||
if (sps_ptr->scaling_list_enabled_flag) {
|
||||
RocdecHevcIQMatrix *iq_matrix_ptr = &dec_pic_params_.iq_matrix.hevc;
|
||||
H265ScalingListData *scaling_list_data_ptr = &pps_ptr->scaling_list_data;
|
||||
HevcScalingListData *scaling_list_data_ptr = &pps_ptr->scaling_list_data;
|
||||
for (i = 0; i < 6; i++) {
|
||||
for (j = 0; j < 16; j++) {
|
||||
iq_matrix_ptr->ScalingList4x4[i][j] = scaling_list_data_ptr->scaling_list[0][i][j];
|
||||
@@ -541,7 +541,7 @@ int HEVCVideoParser::SendPicForDecode() {
|
||||
}
|
||||
}
|
||||
|
||||
int HEVCVideoParser::OutputDecodedPictures() {
|
||||
int HevcVideoParser::OutputDecodedPictures() {
|
||||
RocdecParserDispInfo disp_info = {0};
|
||||
disp_info.progressive_frame = m_sps_[m_active_sps_id_].profile_tier_level.general_progressive_source_flag;
|
||||
disp_info.top_field_first = 1;
|
||||
@@ -555,7 +555,7 @@ int HEVCVideoParser::OutputDecodedPictures() {
|
||||
return PARSER_OK;
|
||||
}
|
||||
|
||||
bool HEVCVideoParser::ParseFrameData(const uint8_t* p_stream, uint32_t frame_data_size) {
|
||||
bool HevcVideoParser::ParseFrameData(const uint8_t* p_stream, uint32_t frame_data_size) {
|
||||
int ret = PARSER_OK;
|
||||
|
||||
frame_data_buffer_ptr_ = (uint8_t*)p_stream;
|
||||
@@ -725,7 +725,7 @@ bool HEVCVideoParser::ParseFrameData(const uint8_t* p_stream, uint32_t frame_dat
|
||||
return true;
|
||||
}
|
||||
|
||||
int HEVCVideoParser::GetNalUnit() {
|
||||
int HevcVideoParser::GetNalUnit() {
|
||||
bool start_code_found = false;
|
||||
|
||||
nal_unit_size_ = 0;
|
||||
@@ -766,7 +766,7 @@ int HEVCVideoParser::GetNalUnit() {
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ParsePtl(H265ProfileTierLevel *ptl, bool profile_present_flag, uint32_t max_num_sub_layers_minus1, uint8_t *nalu, size_t size, size_t& offset) {
|
||||
void HevcVideoParser::ParsePtl(HevcProfileTierLevel *ptl, bool profile_present_flag, uint32_t max_num_sub_layers_minus1, uint8_t *nalu, size_t size, size_t& offset) {
|
||||
if (profile_present_flag) {
|
||||
ptl->general_profile_space = Parser::ReadBits(nalu, offset, 2);
|
||||
ptl->general_tier_flag = Parser::GetBit(nalu, offset);
|
||||
@@ -815,7 +815,7 @@ void HEVCVideoParser::ParsePtl(H265ProfileTierLevel *ptl, bool profile_present_f
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ParseSubLayerHrdParameters(H265SubLayerHrdParameters *sub_hrd, uint32_t cpb_cnt, bool sub_pic_hrd_params_present_flag, uint8_t *nalu, size_t /*size*/, size_t& offset) {
|
||||
void HevcVideoParser::ParseSubLayerHrdParameters(HevcSubLayerHrdParameters *sub_hrd, uint32_t cpb_cnt, bool sub_pic_hrd_params_present_flag, uint8_t *nalu, size_t /*size*/, size_t& offset) {
|
||||
for (uint32_t i = 0; i <= cpb_cnt; i++) {
|
||||
sub_hrd->bit_rate_value_minus1[i] = Parser::ExpGolomb::ReadUe(nalu, offset);
|
||||
sub_hrd->cpb_size_value_minus1[i] = Parser::ExpGolomb::ReadUe(nalu, offset);
|
||||
@@ -827,7 +827,7 @@ void HEVCVideoParser::ParseSubLayerHrdParameters(H265SubLayerHrdParameters *sub_
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ParseHrdParameters(H265HrdParameters *hrd, bool common_inf_present_flag, uint32_t max_num_sub_layers_minus1, uint8_t *nalu, size_t size,size_t &offset) {
|
||||
void HevcVideoParser::ParseHrdParameters(HevcHrdParameters *hrd, bool common_inf_present_flag, uint32_t max_num_sub_layers_minus1, uint8_t *nalu, size_t size,size_t &offset) {
|
||||
if (common_inf_present_flag) {
|
||||
hrd->nal_hrd_parameters_present_flag = Parser::GetBit(nalu, offset);
|
||||
hrd->vcl_hrd_parameters_present_flag = Parser::GetBit(nalu, offset);
|
||||
@@ -923,7 +923,7 @@ static const int diag_scan_8x8[64] = {
|
||||
46,39,61,54,47,62,55,63
|
||||
};
|
||||
|
||||
void HEVCVideoParser::SetDefaultScalingList(H265ScalingListData *sl_ptr) {
|
||||
void HevcVideoParser::SetDefaultScalingList(HevcScalingListData *sl_ptr) {
|
||||
int size_id, matrix_id, i;
|
||||
|
||||
// DC coefficient for 16x16 and 32x32
|
||||
@@ -958,7 +958,7 @@ void HEVCVideoParser::SetDefaultScalingList(H265ScalingListData *sl_ptr) {
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ParseScalingList(H265ScalingListData * sl_ptr, uint8_t *nalu, size_t size, size_t& offset, SpsData *sps_ptr) {
|
||||
void HevcVideoParser::ParseScalingList(HevcScalingListData * sl_ptr, uint8_t *nalu, size_t size, size_t& offset, HevcSeqParamSet *sps_ptr) {
|
||||
for (int size_id = 0; size_id < 4; size_id++) {
|
||||
for (int matrix_id = 0; matrix_id < 6; matrix_id += (size_id == 3) ? 3 : 1) {
|
||||
sl_ptr->scaling_list_pred_mode_flag[size_id][matrix_id] = Parser::GetBit(nalu, offset);
|
||||
@@ -1015,10 +1015,10 @@ void HEVCVideoParser::ParseScalingList(H265ScalingListData * sl_ptr, uint8_t *na
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ParseShortTermRefPicSet(H265ShortTermRPS *rps, uint32_t st_rps_idx, uint32_t number_short_term_ref_pic_sets, H265ShortTermRPS rps_ref[], uint8_t *nalu, size_t /*size*/, size_t& offset) {
|
||||
void HevcVideoParser::ParseShortTermRefPicSet(HevcShortTermRps *rps, uint32_t st_rps_idx, uint32_t number_short_term_ref_pic_sets, HevcShortTermRps rps_ref[], uint8_t *nalu, size_t /*size*/, size_t& offset) {
|
||||
int i, j;
|
||||
|
||||
memset(rps, 0, sizeof(H265ShortTermRPS));
|
||||
memset(rps, 0, sizeof(HevcShortTermRps));
|
||||
if (st_rps_idx != 0) {
|
||||
rps->inter_ref_pic_set_prediction_flag = Parser::GetBit(nalu, offset);
|
||||
} else {
|
||||
@@ -1035,7 +1035,7 @@ void HEVCVideoParser::ParseShortTermRefPicSet(H265ShortTermRPS *rps, uint32_t st
|
||||
int ref_rps_idx = st_rps_idx - (rps->delta_idx_minus1 + 1); // (7-59)
|
||||
int delta_rps = (1 - 2 * rps->delta_rps_sign) * (rps->abs_delta_rps_minus1 + 1); // (7-60)
|
||||
|
||||
H265ShortTermRPS *ref_rps = &rps_ref[ref_rps_idx];
|
||||
HevcShortTermRps *ref_rps = &rps_ref[ref_rps_idx];
|
||||
for (j = 0; j <= ref_rps->num_of_delta_pocs; j++) {
|
||||
rps->used_by_curr_pic_flag[j] = Parser::GetBit(nalu, offset);
|
||||
if (!rps->used_by_curr_pic_flag[j]) {
|
||||
@@ -1116,7 +1116,7 @@ void HEVCVideoParser::ParseShortTermRefPicSet(H265ShortTermRPS *rps, uint32_t st
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ParsePredWeightTable(HEVCVideoParser::SliceHeaderData *slice_header_ptr, int chroma_array_type, uint8_t *stream_ptr, size_t &offset) {
|
||||
void HevcVideoParser::ParsePredWeightTable(HevcSliceSegHeader *slice_header_ptr, int chroma_array_type, uint8_t *stream_ptr, size_t &offset) {
|
||||
HevcPredWeightTable *pred_weight_table_ptr = &slice_header_ptr->pred_weight_table;
|
||||
int chroma_log2_weight_denom; // ChromaLog2WeightDenom
|
||||
int i, j;
|
||||
@@ -1186,7 +1186,7 @@ void HEVCVideoParser::ParsePredWeightTable(HEVCVideoParser::SliceHeaderData *sli
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ParseVui(H265VuiParameters *vui, uint32_t max_num_sub_layers_minus1, uint8_t *nalu, size_t size, size_t &offset) {
|
||||
void HevcVideoParser::ParseVui(HevcVuiParameters *vui, uint32_t max_num_sub_layers_minus1, uint8_t *nalu, size_t size, size_t &offset) {
|
||||
vui->aspect_ratio_info_present_flag = Parser::GetBit(nalu, offset);
|
||||
if (vui->aspect_ratio_info_present_flag) {
|
||||
vui->aspect_ratio_idc = Parser::ReadBits(nalu, offset, 8);
|
||||
@@ -1251,11 +1251,11 @@ void HEVCVideoParser::ParseVui(H265VuiParameters *vui, uint32_t max_num_sub_laye
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ParseVps(uint8_t *nalu, size_t size) {
|
||||
void HevcVideoParser::ParseVps(uint8_t *nalu, size_t size) {
|
||||
size_t offset = 0; // current bit offset
|
||||
uint32_t vps_id = Parser::ReadBits(nalu, offset, 4);
|
||||
VpsData *p_vps = &m_vps_[vps_id];
|
||||
memset(p_vps, 0, sizeof(VpsData));
|
||||
HevcVideoParamSet *p_vps = &m_vps_[vps_id];
|
||||
memset(p_vps, 0, sizeof(HevcVideoParamSet));
|
||||
|
||||
p_vps->vps_video_parameter_set_id = vps_id;
|
||||
p_vps->vps_base_layer_internal_flag = Parser::GetBit(nalu, offset);
|
||||
@@ -1310,21 +1310,21 @@ void HEVCVideoParser::ParseVps(uint8_t *nalu, size_t size) {
|
||||
#endif // DBGINFO
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ParseSps(uint8_t *nalu, size_t size) {
|
||||
SpsData *sps_ptr = nullptr;
|
||||
void HevcVideoParser::ParseSps(uint8_t *nalu, size_t size) {
|
||||
HevcSeqParamSet *sps_ptr = nullptr;
|
||||
size_t offset = 0;
|
||||
|
||||
uint32_t vps_id = Parser::ReadBits(nalu, offset, 4);
|
||||
uint32_t max_sub_layer_minus1 = Parser::ReadBits(nalu, offset, 3);
|
||||
uint32_t sps_temporal_id_nesting_flag = Parser::GetBit(nalu, offset);
|
||||
H265ProfileTierLevel ptl;
|
||||
HevcProfileTierLevel ptl;
|
||||
memset (&ptl, 0, sizeof(ptl));
|
||||
ParsePtl(&ptl, true, max_sub_layer_minus1, nalu, size, offset);
|
||||
|
||||
uint32_t sps_id = Parser::ExpGolomb::ReadUe(nalu, offset);
|
||||
sps_ptr = &m_sps_[sps_id];
|
||||
|
||||
memset(sps_ptr, 0, sizeof(SpsData));
|
||||
memset(sps_ptr, 0, sizeof(HevcSeqParamSet));
|
||||
sps_ptr->sps_video_parameter_set_id = vps_id;
|
||||
sps_ptr->sps_max_sub_layers_minus1 = max_sub_layer_minus1;
|
||||
sps_ptr->sps_temporal_id_nesting_flag = sps_temporal_id_nesting_flag;
|
||||
@@ -1437,12 +1437,12 @@ void HEVCVideoParser::ParseSps(uint8_t *nalu, size_t size) {
|
||||
#endif // DBGINFO
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ParsePps(uint8_t *nalu, size_t size) {
|
||||
void HevcVideoParser::ParsePps(uint8_t *nalu, size_t size) {
|
||||
int i;
|
||||
size_t offset = 0;
|
||||
uint32_t pps_id = Parser::ExpGolomb::ReadUe(nalu, offset);
|
||||
PpsData *pps_ptr = &m_pps_[pps_id];
|
||||
memset(pps_ptr, 0, sizeof(PpsData));
|
||||
HevcPicParamSet *pps_ptr = &m_pps_[pps_id];
|
||||
memset(pps_ptr, 0, sizeof(HevcPicParamSet));
|
||||
|
||||
pps_ptr->pps_pic_parameter_set_id = pps_id;
|
||||
pps_ptr->pps_seq_parameter_set_id = Parser::ExpGolomb::ReadUe(nalu, offset);
|
||||
@@ -1554,12 +1554,12 @@ void HEVCVideoParser::ParsePps(uint8_t *nalu, size_t size) {
|
||||
#endif // DBGINFO
|
||||
}
|
||||
|
||||
bool HEVCVideoParser::ParseSliceHeader(uint8_t *nalu, size_t size) {
|
||||
PpsData *pps_ptr = nullptr;
|
||||
SpsData *sps_ptr = nullptr;
|
||||
bool HevcVideoParser::ParseSliceHeader(uint8_t *nalu, size_t size) {
|
||||
HevcPicParamSet *pps_ptr = nullptr;
|
||||
HevcSeqParamSet *sps_ptr = nullptr;
|
||||
size_t offset = 0;
|
||||
SliceHeaderData temp_sh;
|
||||
memset(m_sh_, 0, sizeof(SliceHeaderData));
|
||||
HevcSliceSegHeader temp_sh;
|
||||
memset(m_sh_, 0, sizeof(HevcSliceSegHeader));
|
||||
memset(&temp_sh, 0, sizeof(temp_sh));
|
||||
|
||||
temp_sh.first_slice_segment_in_pic_flag = m_sh_->first_slice_segment_in_pic_flag = Parser::GetBit(nalu, offset);
|
||||
@@ -1727,7 +1727,7 @@ bool HEVCVideoParser::ParseSliceHeader(uint8_t *nalu, size_t size) {
|
||||
// 7.3.6.2 Reference picture list modification
|
||||
// Calculate NumPicTotalCurr
|
||||
num_pic_total_curr_ = 0;
|
||||
H265ShortTermRPS *st_rps_ptr = &m_sh_->st_rps;
|
||||
HevcShortTermRps *st_rps_ptr = &m_sh_->st_rps;
|
||||
for (int i = 0; i < st_rps_ptr->num_negative_pics; i++) {
|
||||
if (st_rps_ptr->used_by_curr_pic_s0[i]) {
|
||||
num_pic_total_curr_++;
|
||||
@@ -1739,7 +1739,7 @@ bool HEVCVideoParser::ParseSliceHeader(uint8_t *nalu, size_t size) {
|
||||
}
|
||||
}
|
||||
|
||||
H265LongTermRPS *lt_rps_ptr = &m_sh_->lt_rps;
|
||||
HevcLongTermRps *lt_rps_ptr = &m_sh_->lt_rps;
|
||||
// Check the combined list
|
||||
for (int i = 0; i < lt_rps_ptr->num_of_pics; i++) {
|
||||
if (lt_rps_ptr->used_by_curr_pic[i]) {
|
||||
@@ -1817,10 +1817,10 @@ bool HEVCVideoParser::ParseSliceHeader(uint8_t *nalu, size_t size) {
|
||||
m_sh_->slice_loop_filter_across_slices_enabled_flag = Parser::GetBit(nalu, offset);
|
||||
}
|
||||
|
||||
memcpy(m_sh_copy_, m_sh_, sizeof(SliceHeaderData));
|
||||
memcpy(m_sh_copy_, m_sh_, sizeof(HevcSliceSegHeader));
|
||||
} else {
|
||||
//dependant slice
|
||||
memcpy(m_sh_, m_sh_copy_, sizeof(SliceHeaderData));
|
||||
memcpy(m_sh_, m_sh_copy_, sizeof(HevcSliceSegHeader));
|
||||
m_sh_->first_slice_segment_in_pic_flag = temp_sh.first_slice_segment_in_pic_flag;
|
||||
m_sh_->no_output_of_prior_pics_flag = temp_sh.no_output_of_prior_pics_flag;
|
||||
m_sh_->slice_pic_parameter_set_id = temp_sh.slice_pic_parameter_set_id;
|
||||
@@ -1867,7 +1867,7 @@ bool HEVCVideoParser::ParseSliceHeader(uint8_t *nalu, size_t size) {
|
||||
return false;
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ParseSeiMessage(uint8_t *nalu, size_t size) {
|
||||
void HevcVideoParser::ParseSeiMessage(uint8_t *nalu, size_t size) {
|
||||
int offset = 0; // byte offset
|
||||
int payload_type;
|
||||
int payload_size;
|
||||
@@ -1918,31 +1918,31 @@ void HEVCVideoParser::ParseSeiMessage(uint8_t *nalu, size_t size) {
|
||||
} while (offset < size && nalu[offset] != 0x80);
|
||||
}
|
||||
|
||||
bool HEVCVideoParser::IsIdrPic(NalUnitHeader *nal_header_ptr) {
|
||||
bool HevcVideoParser::IsIdrPic(HevcNalUnitHeader *nal_header_ptr) {
|
||||
return (nal_header_ptr->nal_unit_type == NAL_UNIT_CODED_SLICE_IDR_W_RADL || nal_header_ptr->nal_unit_type == NAL_UNIT_CODED_SLICE_IDR_N_LP);
|
||||
}
|
||||
|
||||
bool HEVCVideoParser::IsBlaPic(NalUnitHeader *nal_header_ptr) {
|
||||
bool HevcVideoParser::IsBlaPic(HevcNalUnitHeader *nal_header_ptr) {
|
||||
return (nal_header_ptr->nal_unit_type == NAL_UNIT_CODED_SLICE_BLA_W_LP || nal_header_ptr->nal_unit_type == NAL_UNIT_CODED_SLICE_BLA_W_RADL || nal_header_ptr->nal_unit_type == NAL_UNIT_CODED_SLICE_BLA_N_LP);
|
||||
}
|
||||
|
||||
bool HEVCVideoParser::IsCraPic(NalUnitHeader *nal_header_ptr) {
|
||||
bool HevcVideoParser::IsCraPic(HevcNalUnitHeader *nal_header_ptr) {
|
||||
return (nal_header_ptr->nal_unit_type == NAL_UNIT_CODED_SLICE_CRA_NUT);
|
||||
}
|
||||
|
||||
bool HEVCVideoParser::IsRaslPic(NalUnitHeader *nal_header_ptr) {
|
||||
bool HevcVideoParser::IsRaslPic(HevcNalUnitHeader *nal_header_ptr) {
|
||||
return (nal_header_ptr->nal_unit_type == NAL_UNIT_CODED_SLICE_RASL_N || nal_header_ptr->nal_unit_type == NAL_UNIT_CODED_SLICE_RASL_R);
|
||||
}
|
||||
|
||||
bool HEVCVideoParser::IsRadlPic(NalUnitHeader *nal_header_ptr) {
|
||||
bool HevcVideoParser::IsRadlPic(HevcNalUnitHeader *nal_header_ptr) {
|
||||
return (nal_header_ptr->nal_unit_type == NAL_UNIT_CODED_SLICE_RADL_N || nal_header_ptr->nal_unit_type == NAL_UNIT_CODED_SLICE_RADL_R);
|
||||
}
|
||||
|
||||
bool HEVCVideoParser::IsIrapPic(NalUnitHeader *nal_header_ptr) {
|
||||
bool HevcVideoParser::IsIrapPic(HevcNalUnitHeader *nal_header_ptr) {
|
||||
return (nal_header_ptr->nal_unit_type >= NAL_UNIT_CODED_SLICE_BLA_W_LP && nal_header_ptr->nal_unit_type <= NAL_UNIT_RESERVED_IRAP_VCL23);
|
||||
}
|
||||
|
||||
bool HEVCVideoParser::IsRefPic(NalUnitHeader *nal_header_ptr) {
|
||||
bool HevcVideoParser::IsRefPic(HevcNalUnitHeader *nal_header_ptr) {
|
||||
if (((nal_header_ptr->nal_unit_type <= NAL_UNIT_RESERVED_VCL_R15) && ((nal_header_ptr->nal_unit_type % 2) != 0)) ||
|
||||
((nal_header_ptr->nal_unit_type >= NAL_UNIT_CODED_SLICE_BLA_W_LP) && (nal_header_ptr->nal_unit_type <= NAL_UNIT_RESERVED_IRAP_VCL23))) {
|
||||
return true;
|
||||
@@ -1951,7 +1951,7 @@ bool HEVCVideoParser::IsRefPic(NalUnitHeader *nal_header_ptr) {
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::CalculateCurrPOC() {
|
||||
void HevcVideoParser::CalculateCurrPOC() {
|
||||
// Recode decode order count
|
||||
curr_pic_info_.decode_order_count = pic_count_;
|
||||
if (IsIdrPic(&slice_nal_unit_header_)) {
|
||||
@@ -1984,7 +1984,7 @@ void HEVCVideoParser::CalculateCurrPOC() {
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::DecodeRps() {
|
||||
void HevcVideoParser::DecodeRps() {
|
||||
int i, j, k;
|
||||
int curr_delta_poc_msb_present_flag[HEVC_MAX_NUM_REF_PICS] = {0}; // CurrDeltaPocMsbPresentFlag
|
||||
int foll_delta_poc_msb_present_flag[HEVC_MAX_NUM_REF_PICS] = {0}; // FollDeltaPocMsbPresentFlag
|
||||
@@ -2010,7 +2010,7 @@ void HEVCVideoParser::DecodeRps() {
|
||||
memset(poc_lt_curr_, 0, sizeof(int32_t) * HEVC_MAX_NUM_REF_PICS);
|
||||
memset(poc_lt_foll_, 0, sizeof(int32_t) * HEVC_MAX_NUM_REF_PICS);
|
||||
} else {
|
||||
H265ShortTermRPS *rps_ptr = &m_sh_->st_rps;
|
||||
HevcShortTermRps *rps_ptr = &m_sh_->st_rps;
|
||||
for (i = 0, j = 0, k = 0; i < rps_ptr->num_negative_pics; i++) {
|
||||
if (rps_ptr->used_by_curr_pic_s0[i]) {
|
||||
poc_st_curr_before_[j++] = curr_pic_info_.pic_order_cnt + rps_ptr->delta_poc_s0[i];
|
||||
@@ -2030,7 +2030,7 @@ void HEVCVideoParser::DecodeRps() {
|
||||
num_poc_st_curr_after_ = j;
|
||||
num_poc_st_foll_ = k;
|
||||
|
||||
H265LongTermRPS *lt_rps_ptr = &m_sh_->lt_rps;
|
||||
HevcLongTermRps *lt_rps_ptr = &m_sh_->lt_rps;
|
||||
for (i = 0, j = 0, k = 0; i < lt_rps_ptr->num_of_pics; i++) {
|
||||
uint32_t poc_lt = lt_rps_ptr->pocs[i]; // oocLt
|
||||
if (m_sh_->delta_poc_msb_present_flag[i]) {
|
||||
@@ -2135,7 +2135,7 @@ void HEVCVideoParser::DecodeRps() {
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::ConstructRefPicLists() {
|
||||
void HevcVideoParser::ConstructRefPicLists() {
|
||||
|
||||
uint32_t num_rps_curr_temp_list; // NumRpsCurrTempList0 or NumRpsCurrTempList1;
|
||||
int i, j;
|
||||
@@ -2189,7 +2189,7 @@ void HEVCVideoParser::ConstructRefPicLists() {
|
||||
}
|
||||
}
|
||||
|
||||
void HEVCVideoParser::InitDpb() {
|
||||
void HevcVideoParser::InitDpb() {
|
||||
memset(&dpb_buffer_, 0, sizeof(DecodedPictureBuffer));
|
||||
for (int i = 0; i < HEVC_MAX_DPB_FRAMES; i++) {
|
||||
dpb_buffer_.frame_buffer_list[i].pic_idx = i;
|
||||
@@ -2204,7 +2204,7 @@ void HEVCVideoParser::InitDpb() {
|
||||
dpb_buffer_.num_output_pics = 0;
|
||||
}
|
||||
|
||||
void HEVCVideoParser::EmptyDpb() {
|
||||
void HevcVideoParser::EmptyDpb() {
|
||||
for (int i = 0; i < HEVC_MAX_DPB_FRAMES; i++) {
|
||||
dpb_buffer_.frame_buffer_list[i].is_reference = kUnusedForReference;
|
||||
dpb_buffer_.frame_buffer_list[i].pic_output_flag = 0;
|
||||
@@ -2216,7 +2216,7 @@ void HEVCVideoParser::EmptyDpb() {
|
||||
dpb_buffer_.num_output_pics = 0;
|
||||
}
|
||||
|
||||
int HEVCVideoParser::FlushDpb() {
|
||||
int HevcVideoParser::FlushDpb() {
|
||||
if (dpb_buffer_.num_needed_for_output) {
|
||||
// Bump the remaining pictures
|
||||
while (dpb_buffer_.num_needed_for_output) {
|
||||
@@ -2233,7 +2233,7 @@ int HEVCVideoParser::FlushDpb() {
|
||||
return PARSER_OK;
|
||||
}
|
||||
|
||||
int HEVCVideoParser::MarkOutputPictures() {
|
||||
int HevcVideoParser::MarkOutputPictures() {
|
||||
int i;
|
||||
|
||||
if (IsIrapPic(&slice_nal_unit_header_) && no_rasl_output_flag_ == 1 && pic_count_ != 0) {
|
||||
@@ -2264,7 +2264,7 @@ int HEVCVideoParser::MarkOutputPictures() {
|
||||
}
|
||||
}
|
||||
|
||||
SpsData *sps_ptr = &m_sps_[m_active_sps_id_];
|
||||
HevcSeqParamSet *sps_ptr = &m_sps_[m_active_sps_id_];
|
||||
uint32_t highest_tid = sps_ptr->sps_max_sub_layers_minus1; // HighestTid
|
||||
uint32_t max_num_reorder_pics = sps_ptr->sps_max_num_reorder_pics[highest_tid];
|
||||
uint32_t max_dec_pic_buffering = sps_ptr->sps_max_dec_pic_buffering_minus1[highest_tid] + 1;
|
||||
@@ -2287,7 +2287,7 @@ int HEVCVideoParser::MarkOutputPictures() {
|
||||
return PARSER_OK;
|
||||
}
|
||||
|
||||
int HEVCVideoParser::FindFreeBufAndMark() {
|
||||
int HevcVideoParser::FindFreeBufAndMark() {
|
||||
int i, j;
|
||||
|
||||
// Look for an empty buffer with longest decode history (lowest decode count)
|
||||
@@ -2333,7 +2333,7 @@ int HEVCVideoParser::FindFreeBufAndMark() {
|
||||
}
|
||||
dpb_buffer_.dpb_fullness++;
|
||||
|
||||
SpsData *sps_ptr = &m_sps_[m_active_sps_id_];
|
||||
HevcSeqParamSet *sps_ptr = &m_sps_[m_active_sps_id_];
|
||||
uint32_t highest_tid = sps_ptr->sps_max_sub_layers_minus1; // HighestTid
|
||||
uint32_t max_num_reorder_pics = sps_ptr->sps_max_num_reorder_pics[highest_tid];
|
||||
|
||||
@@ -2349,7 +2349,7 @@ int HEVCVideoParser::FindFreeBufAndMark() {
|
||||
return PARSER_OK;
|
||||
}
|
||||
|
||||
int HEVCVideoParser::BumpPicFromDpb() {
|
||||
int HevcVideoParser::BumpPicFromDpb() {
|
||||
int32_t min_poc = 0x7FFFFFFF; // largest possible POC value 2^31 - 1
|
||||
int min_poc_pic_idx = HEVC_MAX_DPB_FRAMES;
|
||||
int i;
|
||||
@@ -2393,7 +2393,7 @@ int HEVCVideoParser::BumpPicFromDpb() {
|
||||
return PARSER_OK;
|
||||
}
|
||||
|
||||
size_t HEVCVideoParser::EBSPtoRBSP(uint8_t *streamBuffer,size_t begin_bytepos, size_t end_bytepos) {
|
||||
size_t HevcVideoParser::EBSPtoRBSP(uint8_t *streamBuffer,size_t begin_bytepos, size_t end_bytepos) {
|
||||
int count = 0;
|
||||
if (end_bytepos < begin_bytepos) {
|
||||
return end_bytepos;
|
||||
@@ -2434,7 +2434,7 @@ size_t HEVCVideoParser::EBSPtoRBSP(uint8_t *streamBuffer,size_t begin_bytepos, s
|
||||
}
|
||||
|
||||
#if DBGINFO
|
||||
void HEVCVideoParser::PrintVps(HEVCVideoParser::VpsData *vps_ptr) {
|
||||
void HevcVideoParser::PrintVps(HevcVideoParser::HevcVideoParamSet *vps_ptr) {
|
||||
MSG("=== hevc_video_parameter_set_t ===");
|
||||
MSG("vps_video_parameter_set_id = " << vps_ptr->vps_video_parameter_set_id);
|
||||
MSG("vps_base_layer_internal_flag = " << vps_ptr->vps_base_layer_internal_flag);
|
||||
@@ -2485,7 +2485,7 @@ void HEVCVideoParser::PrintVps(HEVCVideoParser::VpsData *vps_ptr) {
|
||||
MSG("");
|
||||
}
|
||||
|
||||
void HEVCVideoParser::PrintSps(HEVCVideoParser::SpsData *sps_ptr) {
|
||||
void HevcVideoParser::PrintSps(HevcVideoParser::HevcSeqParamSet *sps_ptr) {
|
||||
MSG("=== hevc_sequence_parameter_set_t ===");
|
||||
MSG("sps_video_parameter_set_id = " << sps_ptr->sps_video_parameter_set_id);
|
||||
MSG("sps_max_sub_layers_minus1 = " << sps_ptr->sps_max_sub_layers_minus1);
|
||||
@@ -2545,10 +2545,10 @@ void HEVCVideoParser::PrintSps(HEVCVideoParser::SpsData *sps_ptr) {
|
||||
MSG("scaling_list_enabled_flag = " << sps_ptr->scaling_list_enabled_flag);
|
||||
MSG("sps_scaling_list_data_present_flag = " << sps_ptr->sps_scaling_list_data_present_flag);
|
||||
MSG("Scaling list:");
|
||||
for (int i = 0; i < H265_SCALING_LIST_SIZE_NUM; i++) {
|
||||
for (int j = 0; j < H265_SCALING_LIST_NUM; j++) {
|
||||
for (int i = 0; i < HEVC_SCALING_LIST_SIZE_NUM; i++) {
|
||||
for (int j = 0; j < HEVC_SCALING_LIST_NUM; j++) {
|
||||
MSG_NO_NEWLINE("scaling_list[" << i <<"][" << j << "][]:")
|
||||
for (int k = 0; k < H265_SCALING_LIST_MAX_I; k++) {
|
||||
for (int k = 0; k < HEVC_SCALING_LIST_MAX_INDEX; k++) {
|
||||
MSG_NO_NEWLINE(" " << sps_ptr->scaling_list_data.scaling_list[i][j][k]);
|
||||
}
|
||||
MSG("");
|
||||
@@ -2597,7 +2597,7 @@ void HEVCVideoParser::PrintSps(HEVCVideoParser::SpsData *sps_ptr) {
|
||||
MSG("");
|
||||
}
|
||||
|
||||
void HEVCVideoParser::PrintPps(HEVCVideoParser::PpsData *pps_ptr) {
|
||||
void HevcVideoParser::PrintPps(HevcVideoParser::HevcPicParamSet *pps_ptr) {
|
||||
MSG("=== hevc_picture_parameter_set_t ===");
|
||||
MSG("pps_pic_parameter_set_id = " << pps_ptr->pps_pic_parameter_set_id);
|
||||
MSG("pps_seq_parameter_set_id = " << pps_ptr->pps_seq_parameter_set_id);
|
||||
@@ -2645,10 +2645,10 @@ void HEVCVideoParser::PrintPps(HEVCVideoParser::PpsData *pps_ptr) {
|
||||
MSG("pps_tc_offset_div2 = " << pps_ptr->pps_tc_offset_div2);
|
||||
MSG("pps_scaling_list_data_present_flag = " << pps_ptr->pps_scaling_list_data_present_flag);
|
||||
MSG("Scaling list:");
|
||||
for (int i = 0; i < H265_SCALING_LIST_SIZE_NUM; i++) {
|
||||
for (int j = 0; j < H265_SCALING_LIST_NUM; j++) {
|
||||
for (int i = 0; i < HEVC_SCALING_LIST_SIZE_NUM; i++) {
|
||||
for (int j = 0; j < HEVC_SCALING_LIST_NUM; j++) {
|
||||
MSG_NO_NEWLINE("scaling_list[" << i <<"][" << j << "][]:")
|
||||
for (int k = 0; k < H265_SCALING_LIST_MAX_I; k++) {
|
||||
for (int k = 0; k < HEVC_SCALING_LIST_MAX_INDEX; k++) {
|
||||
MSG_NO_NEWLINE(" " << pps_ptr->scaling_list_data.scaling_list[i][j][k]);
|
||||
}
|
||||
MSG("");
|
||||
@@ -2662,7 +2662,7 @@ void HEVCVideoParser::PrintPps(HEVCVideoParser::PpsData *pps_ptr) {
|
||||
MSG("");
|
||||
}
|
||||
|
||||
void HEVCVideoParser::PrintSliceSegHeader(HEVCVideoParser::SliceHeaderData *slice_header_ptr) {
|
||||
void HevcVideoParser::PrintSliceSegHeader(HevcSliceSegHeader *slice_header_ptr) {
|
||||
MSG("=== hevc_slice_segment_header_t ===");
|
||||
MSG("first_slice_segment_in_pic_flag = " << slice_header_ptr->first_slice_segment_in_pic_flag);
|
||||
MSG("no_output_of_prior_pics_flag = " << slice_header_ptr->no_output_of_prior_pics_flag);
|
||||
@@ -2747,7 +2747,7 @@ void HEVCVideoParser::PrintSliceSegHeader(HEVCVideoParser::SliceHeaderData *slic
|
||||
MSG("");
|
||||
}
|
||||
|
||||
void HEVCVideoParser::PrintStRps(HEVCVideoParser::H265ShortTermRPS *rps_ptr) {
|
||||
void HevcVideoParser::PrintStRps(HevcVideoParser::HevcShortTermRps *rps_ptr) {
|
||||
MSG("==== Short-term reference picture set =====")
|
||||
MSG("inter_ref_pic_set_prediction_flag = " << rps_ptr->inter_ref_pic_set_prediction_flag);
|
||||
MSG("delta_idx_minus1 = " << rps_ptr->delta_idx_minus1);
|
||||
@@ -2810,7 +2810,7 @@ void HEVCVideoParser::PrintStRps(HEVCVideoParser::H265ShortTermRPS *rps_ptr) {
|
||||
MSG("");
|
||||
}
|
||||
|
||||
void HEVCVideoParser::PrintLtRefInfo(HEVCVideoParser::H265LongTermRPS *lt_info_ptr) {
|
||||
void HevcVideoParser::PrintLtRefInfo(HevcVideoParser::HevcLongTermRps *lt_info_ptr) {
|
||||
MSG("==== Long-term reference picture info =====");
|
||||
MSG("num_of_pics = " << lt_info_ptr->num_of_pics);
|
||||
MSG_NO_NEWLINE("pocs[]:");
|
||||
|
||||
@@ -23,6 +23,7 @@ THE SOFTWARE.
|
||||
|
||||
#include "../commons.h"
|
||||
#include "roc_video_parser.h"
|
||||
#include "hevc_defines.h"
|
||||
|
||||
#include <map>
|
||||
#include <vector>
|
||||
@@ -37,28 +38,19 @@ extern int scaling_list_default_1_2[2][6][64];
|
||||
//size_id = 3
|
||||
extern int scaling_list_default_3[1][2][64];
|
||||
|
||||
#define MAX_VPS_COUNT 16 // 7.3.2.1
|
||||
#define MAX_SPS_COUNT 16 // 7.3.2.2.1
|
||||
#define MAX_PPS_COUNT 64 // 7.4.3.3.1
|
||||
#define RBSP_BUF_SIZE 1024 // enough to parse any parameter sets or slice headers
|
||||
#define HEVC_MAX_DPB_FRAMES 16 // (A-2)
|
||||
#define HEVC_MAX_NUM_REF_PICS 16
|
||||
// 7.4.7.1. (num_tile_columns_minus1 + 1) * PicHeightInCtbsY − 1. Max tile columns = 20 (A.4.2). Pic height in 16x16 CTB of 8K = 270.
|
||||
#define MAX_ENTRY_POINT_OFFSETS 20 * 270
|
||||
#define INIT_SEI_MESSAGE_COUNT 16 // initial SEI message count
|
||||
#define INIT_SEI_PAYLOAD_BUF_SIZE 1024 * 1024 // initial SEI payload buffer size, 1 MB
|
||||
|
||||
class HEVCVideoParser : public RocVideoParser {
|
||||
class HevcVideoParser : public RocVideoParser {
|
||||
|
||||
public:
|
||||
/*! \brief Construct a new HEVCParser object
|
||||
*/
|
||||
HEVCVideoParser();
|
||||
HevcVideoParser();
|
||||
|
||||
/*! \brief Function to Initialize the parser
|
||||
* \param [in] p_params Input of <tt>RocdecParserParams</tt> with codec type to initialize parser.
|
||||
* \return <tt>rocDecStatus</tt> Returns success on completion, else error code for failure
|
||||
*/
|
||||
virtual rocDecStatus Initialize(RocdecParserParams *p_params);
|
||||
|
||||
/*! \brief Function to Parse video data: Typically called from application when a demuxed picture is ready to be parsed
|
||||
* \param [in] p_data Pointer to picture data of type <tt>RocdecSourceDataPacket</tt>
|
||||
* @return <tt>rocDecStatus</tt> Returns success on completion, else error_code for failure
|
||||
@@ -75,545 +67,16 @@ public:
|
||||
/**
|
||||
* @brief HEVCParser object destructor
|
||||
*/
|
||||
virtual ~HEVCVideoParser();
|
||||
virtual ~HevcVideoParser();
|
||||
|
||||
protected:
|
||||
/*! \brief Enumerator for the NAL Unit types - ISO-IEC 14496-15-2004.pdf, page 14, table 1 " NAL unit types in elementary streams
|
||||
*/
|
||||
enum NalUnitType {
|
||||
NAL_UNIT_CODED_SLICE_TRAIL_N = 0, // 0
|
||||
NAL_UNIT_CODED_SLICE_TRAIL_R, // 1
|
||||
|
||||
NAL_UNIT_CODED_SLICE_TSA_N, // 2
|
||||
NAL_UNIT_CODED_SLICE_TLA_R, // 3
|
||||
|
||||
NAL_UNIT_CODED_SLICE_STSA_N, // 4
|
||||
NAL_UNIT_CODED_SLICE_STSA_R, // 5
|
||||
|
||||
NAL_UNIT_CODED_SLICE_RADL_N, // 6
|
||||
NAL_UNIT_CODED_SLICE_RADL_R, // 7
|
||||
|
||||
NAL_UNIT_CODED_SLICE_RASL_N, // 8
|
||||
NAL_UNIT_CODED_SLICE_RASL_R, // 9
|
||||
|
||||
NAL_UNIT_RESERVED_VCL_N10,
|
||||
NAL_UNIT_RESERVED_VCL_R11,
|
||||
NAL_UNIT_RESERVED_VCL_N12,
|
||||
NAL_UNIT_RESERVED_VCL_R13,
|
||||
NAL_UNIT_RESERVED_VCL_N14,
|
||||
NAL_UNIT_RESERVED_VCL_R15,
|
||||
|
||||
NAL_UNIT_CODED_SLICE_BLA_W_LP, // 16
|
||||
NAL_UNIT_CODED_SLICE_BLA_W_RADL, // 17
|
||||
NAL_UNIT_CODED_SLICE_BLA_N_LP, // 18
|
||||
NAL_UNIT_CODED_SLICE_IDR_W_RADL, // 19
|
||||
NAL_UNIT_CODED_SLICE_IDR_N_LP, // 20
|
||||
NAL_UNIT_CODED_SLICE_CRA_NUT, // 21
|
||||
NAL_UNIT_RESERVED_IRAP_VCL22,
|
||||
NAL_UNIT_RESERVED_IRAP_VCL23,
|
||||
|
||||
NAL_UNIT_RESERVED_VCL24,
|
||||
NAL_UNIT_RESERVED_VCL25,
|
||||
NAL_UNIT_RESERVED_VCL26,
|
||||
NAL_UNIT_RESERVED_VCL27,
|
||||
NAL_UNIT_RESERVED_VCL28,
|
||||
NAL_UNIT_RESERVED_VCL29,
|
||||
NAL_UNIT_RESERVED_VCL30,
|
||||
NAL_UNIT_RESERVED_VCL31,
|
||||
|
||||
NAL_UNIT_VPS, // 32
|
||||
NAL_UNIT_SPS, // 33
|
||||
NAL_UNIT_PPS, // 34
|
||||
NAL_UNIT_ACCESS_UNIT_DELIMITER, // 35
|
||||
NAL_UNIT_EOS, // 36
|
||||
NAL_UNIT_EOB, // 37
|
||||
NAL_UNIT_FILLER_DATA, // 38
|
||||
NAL_UNIT_PREFIX_SEI, // 39
|
||||
NAL_UNIT_SUFFIX_SEI, // 40
|
||||
NAL_UNIT_RESERVED_NVCL41,
|
||||
NAL_UNIT_RESERVED_NVCL42,
|
||||
NAL_UNIT_RESERVED_NVCL43,
|
||||
NAL_UNIT_RESERVED_NVCL44,
|
||||
NAL_UNIT_RESERVED_NVCL45,
|
||||
NAL_UNIT_RESERVED_NVCL46,
|
||||
NAL_UNIT_RESERVED_NVCL47,
|
||||
NAL_UNIT_UNSPECIFIED_48,
|
||||
NAL_UNIT_UNSPECIFIED_49,
|
||||
NAL_UNIT_UNSPECIFIED_50,
|
||||
NAL_UNIT_UNSPECIFIED_51,
|
||||
NAL_UNIT_UNSPECIFIED_52,
|
||||
NAL_UNIT_UNSPECIFIED_53,
|
||||
NAL_UNIT_UNSPECIFIED_54,
|
||||
NAL_UNIT_UNSPECIFIED_55,
|
||||
NAL_UNIT_UNSPECIFIED_56,
|
||||
NAL_UNIT_UNSPECIFIED_57,
|
||||
NAL_UNIT_UNSPECIFIED_58,
|
||||
NAL_UNIT_UNSPECIFIED_59,
|
||||
NAL_UNIT_UNSPECIFIED_60,
|
||||
NAL_UNIT_UNSPECIFIED_61,
|
||||
NAL_UNIT_UNSPECIFIED_62,
|
||||
NAL_UNIT_UNSPECIFIED_63,
|
||||
NAL_UNIT_INVALID,
|
||||
};
|
||||
|
||||
/*! \brief Structure to hold the NAL Unit Header
|
||||
*/
|
||||
struct NalUnitHeader {
|
||||
uint32_t forbidden_zero_bit;
|
||||
uint32_t nal_unit_type;
|
||||
uint32_t nuh_layer_id;
|
||||
uint32_t nuh_temporal_id_plus1;
|
||||
uint32_t num_emu_byte_removed;
|
||||
};
|
||||
|
||||
/*! \brief Enumerator for the scaling list sizes
|
||||
*/
|
||||
enum H265ScalingListSize {
|
||||
H265_SCALING_LIST_4x4 = 0,
|
||||
H265_SCALING_LIST_8x8,
|
||||
H265_SCALING_LIST_16x16,
|
||||
H265_SCALING_LIST_32x32,
|
||||
H265_SCALING_LIST_SIZE_NUM
|
||||
};
|
||||
|
||||
/*! \brief Slice type
|
||||
*/
|
||||
enum HevcSliceType {
|
||||
HEVC_SLICE_TYPE_B = 0,
|
||||
HEVC_SLICE_TYPE_P,
|
||||
HEVC_SLICE_TYPE_I
|
||||
};
|
||||
|
||||
/*! \brief Structure for Profile Tier Levels
|
||||
*/
|
||||
typedef struct {
|
||||
uint32_t general_profile_space; //u(2)
|
||||
bool general_tier_flag; //u(1)
|
||||
uint32_t general_profile_idc; //u(5)
|
||||
bool general_profile_compatibility_flag[32]; //u(1)
|
||||
bool general_progressive_source_flag; //u(1)
|
||||
bool general_interlaced_source_flag; //u(1)
|
||||
bool general_non_packed_constraint_flag; //u(1)
|
||||
bool general_frame_only_constraint_flag; //u(1)
|
||||
uint64_t general_reserved_zero_44bits; //u(44)
|
||||
uint32_t general_level_idc; //u(8)
|
||||
//max_num_sub_layers_minus1 max is 7 - 1 = 6
|
||||
bool sub_layer_profile_present_flag[6]; //u(1)
|
||||
bool sub_layer_level_present_flag[6]; //u(1)
|
||||
|
||||
uint32_t reserved_zero_2bits[8]; //u(2)
|
||||
|
||||
uint32_t sub_layer_profile_space[6]; //u(2)
|
||||
bool sub_layer_tier_flag[6]; //u(1)
|
||||
uint32_t sub_layer_profile_idc[6]; //u(5)
|
||||
bool sub_layer_profile_compatibility_flag[6][32]; //u(1)
|
||||
bool sub_layer_progressive_source_flag[6]; //u(1)
|
||||
bool sub_layer_interlaced_source_flag[6]; //u(1)
|
||||
bool sub_layer_non_packed_constraint_flag[6]; //u(1)
|
||||
bool sub_layer_frame_only_constraint_flag[6]; //u(1)
|
||||
uint64_t sub_layer_reserved_zero_44bits[6]; //u(44)
|
||||
uint32_t sub_layer_level_idc[6]; //u(8)
|
||||
} H265ProfileTierLevel;
|
||||
|
||||
#define H265_SCALING_LIST_NUM 6 ///< list number for quantization matrix
|
||||
#define H265_SCALING_LIST_MAX_I 64
|
||||
|
||||
/*! \brief Structure for Scaling List Data
|
||||
*/
|
||||
typedef struct {
|
||||
bool scaling_list_pred_mode_flag[4][6]; //u(1)
|
||||
uint32_t scaling_list_pred_matrix_id_delta[4][6]; //ue(v)
|
||||
int32_t scaling_list_dc_coef_minus8[4][6]; //se(v)
|
||||
int32_t scaling_list_delta_coef; //se(v)
|
||||
int32_t scaling_list[H265_SCALING_LIST_SIZE_NUM][H265_SCALING_LIST_NUM][H265_SCALING_LIST_MAX_I];
|
||||
int32_t scaling_list_dc_coef[2][6]; // DC coefficient for 16x16 and 32x32
|
||||
} H265ScalingListData;
|
||||
|
||||
/*! \brief Structure for Short Term Reference Picture Set
|
||||
*/
|
||||
typedef struct {
|
||||
uint8_t inter_ref_pic_set_prediction_flag;
|
||||
uint32_t delta_idx_minus1;
|
||||
uint8_t delta_rps_sign;
|
||||
uint32_t abs_delta_rps_minus1;
|
||||
uint8_t used_by_curr_pic_flag[HEVC_MAX_DPB_FRAMES];
|
||||
uint8_t use_delta_flag[HEVC_MAX_DPB_FRAMES];
|
||||
|
||||
uint32_t delta_poc_s0_minus1[HEVC_MAX_DPB_FRAMES];
|
||||
uint8_t used_by_curr_pic_s0_flag[HEVC_MAX_DPB_FRAMES];
|
||||
uint32_t delta_poc_s1_minus1[HEVC_MAX_DPB_FRAMES];
|
||||
uint8_t used_by_curr_pic_s1_flag[HEVC_MAX_DPB_FRAMES];
|
||||
|
||||
uint32_t num_negative_pics; // NumNegativePics
|
||||
uint32_t num_positive_pics; // NumPositivePics
|
||||
uint32_t num_of_delta_pocs; // NumDeltaPocs
|
||||
uint8_t used_by_curr_pic_s0[HEVC_MAX_DPB_FRAMES]; // UsedByCurrPicS0
|
||||
uint8_t used_by_curr_pic_s1[HEVC_MAX_DPB_FRAMES]; // UsedByCurrPicS1
|
||||
int32_t delta_poc_s0[HEVC_MAX_DPB_FRAMES]; // DeltaPocS0
|
||||
int32_t delta_poc_s1[HEVC_MAX_DPB_FRAMES]; // DeltaPocS1
|
||||
} H265ShortTermRPS;
|
||||
|
||||
/*! \brief Structure for Long Term Reference Picture Set
|
||||
*/
|
||||
typedef struct {
|
||||
int32_t num_of_pics;
|
||||
int32_t pocs[32]; // PocLsbLt
|
||||
bool used_by_curr_pic[32]; // UsedByCurrPicLt
|
||||
} H265LongTermRPS;
|
||||
|
||||
/*! \brief Structure for Sub Layer Hypothetical Reference Decoder Parameters
|
||||
*/
|
||||
typedef struct {
|
||||
//CpbCnt = cpb_cnt_minus1
|
||||
uint32_t bit_rate_value_minus1[32]; //ue(v)
|
||||
uint32_t cpb_size_value_minus1[32]; //ue(v)
|
||||
uint32_t cpb_size_du_value_minus1[32]; //ue(v)
|
||||
uint32_t bit_rate_du_value_minus1[32]; //ue(v)
|
||||
bool cbr_flag[32]; //u(1)
|
||||
} H265SubLayerHrdParameters;
|
||||
|
||||
/*! \brief Structure for Hypothetical Reference Decoder Parameters
|
||||
*/
|
||||
typedef struct {
|
||||
bool nal_hrd_parameters_present_flag; //u(1)
|
||||
bool vcl_hrd_parameters_present_flag; //u(1)
|
||||
bool sub_pic_hrd_params_present_flag; //u(1)
|
||||
uint32_t tick_divisor_minus2; //u(8)
|
||||
uint32_t du_cpb_removal_delay_increment_length_minus1; //u(5)
|
||||
bool sub_pic_cpb_params_in_pic_timing_sei_flag; //u(1)
|
||||
uint32_t dpb_output_delay_du_length_minus1; //u(5)
|
||||
uint32_t bit_rate_scale; //u(4)
|
||||
uint32_t cpb_size_scale; //u(4)
|
||||
uint32_t cpb_size_du_scale; //u(4)
|
||||
uint32_t initial_cpb_removal_delay_length_minus1; //u(5)
|
||||
uint32_t au_cpb_removal_delay_length_minus1; //u(5)
|
||||
uint32_t dpb_output_delay_length_minus1; //u(5)
|
||||
bool fixed_pic_rate_general_flag[7]; //u(1)
|
||||
bool fixed_pic_rate_within_cvs_flag[7]; //u(1)
|
||||
uint32_t elemental_duration_in_tc_minus1[7]; //ue(v)
|
||||
bool low_delay_hrd_flag[7]; //u(1)
|
||||
uint32_t cpb_cnt_minus1[7]; //ue(v)
|
||||
//sub_layer_hrd_parameters()
|
||||
H265SubLayerHrdParameters sub_layer_hrd_parameters_0[7];
|
||||
//sub_layer_hrd_parameters()
|
||||
H265SubLayerHrdParameters sub_layer_hrd_parameters_1[7];
|
||||
} H265HrdParameters;
|
||||
|
||||
/*! \brief Structure for Video Usability Information Parameters
|
||||
*/
|
||||
typedef struct {
|
||||
bool aspect_ratio_info_present_flag; //u(1)
|
||||
uint32_t aspect_ratio_idc; //u(8)
|
||||
uint32_t sar_width; //u(16)
|
||||
uint32_t sar_height; //u(16)
|
||||
bool overscan_info_present_flag; //u(1)
|
||||
bool overscan_appropriate_flag; //u(1)
|
||||
bool video_signal_type_present_flag; //u(1)
|
||||
uint32_t video_format; //u(3)
|
||||
bool video_full_range_flag; //u(1)
|
||||
bool colour_description_present_flag; //u(1)
|
||||
uint32_t colour_primaries; //u(8)
|
||||
uint32_t transfer_characteristics; //u(8)
|
||||
uint32_t matrix_coeffs; //u(8)
|
||||
bool chroma_loc_info_present_flag; //u(1)
|
||||
uint32_t chroma_sample_loc_type_top_field; //ue(v)
|
||||
uint32_t chroma_sample_loc_type_bottom_field; //ue(v)
|
||||
bool neutral_chroma_indication_flag; //u(1)
|
||||
bool field_seq_flag; //u(1)
|
||||
bool frame_field_info_present_flag; //u(1)
|
||||
bool default_display_window_flag; //u(1)
|
||||
uint32_t def_disp_win_left_offset; //ue(v)
|
||||
uint32_t def_disp_win_right_offset; //ue(v)
|
||||
uint32_t def_disp_win_top_offset; //ue(v)
|
||||
uint32_t def_disp_win_bottom_offset; //ue(v)
|
||||
bool vui_timing_info_present_flag; //u(1)
|
||||
uint32_t vui_num_units_in_tick; //u(32)
|
||||
uint32_t vui_time_scale; //u(32)
|
||||
bool vui_poc_proportional_to_timing_flag; //u(1)
|
||||
uint32_t vui_num_ticks_poc_diff_one_minus1; //ue(v)
|
||||
bool vui_hrd_parameters_present_flag; //u(1)
|
||||
//hrd_parameters()
|
||||
H265HrdParameters hrd_parameters;
|
||||
bool bitstream_restriction_flag; //u(1)
|
||||
bool tiles_fixed_structure_flag; //u(1)
|
||||
bool motion_vectors_over_pic_boundaries_flag; //u(1)
|
||||
bool restricted_ref_pic_lists_flag; //u(1)
|
||||
uint32_t min_spatial_segmentation_idc; //ue(v)
|
||||
uint32_t max_bytes_per_pic_denom; //ue(v)
|
||||
uint32_t max_bits_per_min_cu_denom; //ue(v)
|
||||
uint32_t log2_max_mv_length_horizontal; //ue(v)
|
||||
uint32_t log2_max_mv_length_vertical; //ue(v)
|
||||
} H265VuiParameters;
|
||||
|
||||
typedef struct {
|
||||
uint32_t luma_log2_weight_denom; //ue(v)
|
||||
int32_t delta_chroma_log2_weight_denom; //se(v)
|
||||
uint8_t luma_weight_l0_flag[16]; //u(1)
|
||||
uint8_t chroma_weight_l0_flag[16]; //u(1)
|
||||
int32_t delta_luma_weight_l0[16]; //se(v)
|
||||
int32_t luma_offset_l0[16]; //se(v)
|
||||
int32_t delta_chroma_weight_l0[16][2]; //se(v)
|
||||
int32_t delta_chroma_offset_l0[16][2]; //se(v)
|
||||
int32_t chroma_weight_l0[16][2]; //ChromaWeightL0[]
|
||||
int32_t chroma_offset_l0[16][2]; //ChromaOffsetL0[]
|
||||
uint8_t luma_weight_l1_flag[16]; //u(1)
|
||||
uint8_t chroma_weight_l1_flag[16]; //u(1)
|
||||
int32_t delta_luma_weight_l1[16]; //se(v)
|
||||
int32_t luma_offset_l1[16]; //se(v)
|
||||
int32_t delta_chroma_weight_l1[16][2]; //se(v)
|
||||
int32_t delta_chroma_offset_l1[16][2]; //se(v)
|
||||
int32_t chroma_weight_l1[16][2]; //ChromaWeightL1[]
|
||||
int32_t chroma_offset_l1[16][2]; //ChromaOffsetL1[]
|
||||
} HevcPredWeightTable;
|
||||
|
||||
/*! \brief Structure for Raw Byte Sequence Payload Trialing Bits
|
||||
*/
|
||||
typedef struct {
|
||||
uint32_t rbsp_stop_one_bit; /* equal to 1 */
|
||||
uint32_t rbsp_alignment_zero_bit; /* equal to 0 */
|
||||
} H265RbspTrailingBits;
|
||||
|
||||
/*! \brief Structure for Video Parameter Set
|
||||
*/
|
||||
typedef struct{
|
||||
uint32_t vps_video_parameter_set_id; //u(4)
|
||||
uint32_t vps_base_layer_internal_flag; //u(1)
|
||||
uint32_t vps_base_layer_available_flag; //u(1)
|
||||
uint32_t vps_max_layers_minus1; //u(6)
|
||||
uint32_t vps_max_sub_layers_minus1; //u(3)
|
||||
bool vps_temporal_id_nesting_flag; //u(1)
|
||||
uint32_t vps_reserved_0xffff_16bits; //u(16)
|
||||
//profile_tier_level( vps_max_sub_layers_minus1 )
|
||||
H265ProfileTierLevel profile_tier_level;
|
||||
bool vps_sub_layer_ordering_info_present_flag; //u(1)
|
||||
//vps_max_sub_layers_minus1 max is 6, need to +1
|
||||
uint32_t vps_max_dec_pic_buffering_minus1[7]; //ue(v)
|
||||
uint32_t vps_max_num_reorder_pics[7]; //ue(v)
|
||||
uint32_t vps_max_latency_increase_plus1[7]; //ue(v)
|
||||
uint32_t vps_max_layer_id; //u(6)
|
||||
uint32_t vps_num_layer_sets_minus1; //ue(v)
|
||||
//vps_num_layer_sets_minus1 max is 1023 (dont +1 since starts from 1)
|
||||
//vps_max_layer_id max is 62 (+1 since starts from 0 and <= condition)
|
||||
bool layer_id_included_flag[1023][63]; //u(1)
|
||||
bool vps_timing_info_present_flag; //u(1)
|
||||
uint32_t vps_num_units_in_tick; //u(32)
|
||||
uint32_t vps_time_scale; //u(32)
|
||||
bool vps_poc_proportional_to_timing_flag; //u(1)
|
||||
uint32_t vps_num_ticks_poc_diff_one_minus1; //ue(v)
|
||||
uint32_t vps_num_hrd_parameters; //ue(v)
|
||||
//vps_num_hrd_parameters max is 1024
|
||||
uint32_t hrd_layer_set_idx[1024]; //ue(v)
|
||||
bool cprms_present_flag[1024]; //u(1)
|
||||
//hrd_parameters()
|
||||
H265HrdParameters hrd_parameters[1024];
|
||||
bool vps_extension_flag; //u(1)
|
||||
bool vps_extension_data_flag; //u(1)
|
||||
//rbsp_trailing_bits()
|
||||
H265RbspTrailingBits rbsp_trailing_bits;
|
||||
} VpsData;
|
||||
|
||||
/*! \brief Structure for Sequence Parameter Set
|
||||
*/
|
||||
typedef struct {
|
||||
uint32_t sps_video_parameter_set_id; //u(4)
|
||||
uint32_t sps_max_sub_layers_minus1; //u(3)
|
||||
bool sps_temporal_id_nesting_flag; //u(1)
|
||||
//profile_tier_level( sps_max_sub_layers_minus1 )
|
||||
H265ProfileTierLevel profile_tier_level;
|
||||
uint32_t sps_seq_parameter_set_id; //ue(v)
|
||||
uint32_t chroma_format_idc; //ue(v)
|
||||
bool separate_colour_plane_flag; //u(1)
|
||||
uint32_t pic_width_in_luma_samples; //ue(v)
|
||||
uint32_t pic_height_in_luma_samples; //ue(v)
|
||||
uint32_t max_cu_width;
|
||||
uint32_t max_cu_height;
|
||||
uint32_t max_cu_depth;
|
||||
bool conformance_window_flag; //u(1)
|
||||
uint32_t conf_win_left_offset; //ue(v)
|
||||
uint32_t conf_win_right_offset; //ue(v)
|
||||
uint32_t conf_win_top_offset; //ue(v)
|
||||
uint32_t conf_win_bottom_offset; //ue(v)
|
||||
uint32_t bit_depth_luma_minus8; //ue(v)
|
||||
uint32_t bit_depth_chroma_minus8; //ue(v)
|
||||
uint32_t log2_max_pic_order_cnt_lsb_minus4; //ue(v)
|
||||
bool sps_sub_layer_ordering_info_present_flag; //u(1)
|
||||
uint32_t sps_max_dec_pic_buffering_minus1[7]; //ue(v)
|
||||
uint32_t sps_max_num_reorder_pics[7]; //ue(v)
|
||||
uint32_t sps_max_latency_increase_plus1[7]; //ue(v)
|
||||
uint32_t log2_min_luma_coding_block_size_minus3; //ue(v)
|
||||
uint32_t log2_diff_max_min_luma_coding_block_size; //ue(v)
|
||||
uint32_t log2_min_transform_block_size_minus2; //ue(v)
|
||||
uint32_t log2_diff_max_min_transform_block_size; //ue(v)
|
||||
uint32_t max_transform_hierarchy_depth_inter; //ue(v)
|
||||
uint32_t max_transform_hierarchy_depth_intra; //ue(v)
|
||||
bool scaling_list_enabled_flag; //u(1)
|
||||
bool sps_scaling_list_data_present_flag; //u(1)
|
||||
//scaling_list_data()
|
||||
H265ScalingListData scaling_list_data;
|
||||
bool amp_enabled_flag; //u(1)
|
||||
bool sample_adaptive_offset_enabled_flag; //u(1)
|
||||
bool pcm_enabled_flag; //u(1)
|
||||
uint32_t pcm_sample_bit_depth_luma_minus1; //u(4)
|
||||
uint32_t pcm_sample_bit_depth_chroma_minus1; //u(4)
|
||||
uint32_t log2_min_pcm_luma_coding_block_size_minus3; //ue(v)
|
||||
uint32_t log2_diff_max_min_pcm_luma_coding_block_size; //ue(v)
|
||||
bool pcm_loop_filter_disabled_flag; //u(1)
|
||||
uint32_t num_short_term_ref_pic_sets; //ue(v)
|
||||
//short_term_ref_pic_set(i) max is 64
|
||||
H265ShortTermRPS st_rps[64];
|
||||
H265LongTermRPS lt_rps;
|
||||
//H265_short_term_ref_pic_set_t short_term_ref_pic_set[64];
|
||||
bool long_term_ref_pics_present_flag; //u(1)
|
||||
uint32_t num_long_term_ref_pics_sps; //ue(v)
|
||||
//max is 32
|
||||
uint32_t lt_ref_pic_poc_lsb_sps[32]; //u(v)
|
||||
bool used_by_curr_pic_lt_sps_flag[32]; //u(1)
|
||||
bool sps_temporal_mvp_enabled_flag; //u(1)
|
||||
bool strong_intra_smoothing_enabled_flag; //u(1)
|
||||
bool vui_parameters_present_flag; //u(1)
|
||||
//vui_parameters()
|
||||
H265VuiParameters vui_parameters;
|
||||
bool sps_extension_flag; //u(1)
|
||||
bool sps_extension_data_flag; //u(1)
|
||||
//rbsp_trailing_bits( )
|
||||
H265RbspTrailingBits rbsp_trailing_bits;
|
||||
} SpsData;
|
||||
|
||||
/*! \brief Structure for Picture Parameter Set
|
||||
*/
|
||||
typedef struct {
|
||||
uint32_t pps_pic_parameter_set_id; //ue(v)
|
||||
uint32_t pps_seq_parameter_set_id; //ue(v)
|
||||
bool dependent_slice_segments_enabled_flag; //u(1)
|
||||
bool output_flag_present_flag; //u(1)
|
||||
uint32_t num_extra_slice_header_bits; //u(3)
|
||||
bool sign_data_hiding_enabled_flag; //u(1)
|
||||
bool cabac_init_present_flag; //u(1)
|
||||
uint32_t num_ref_idx_l0_default_active_minus1; //ue(v)
|
||||
uint32_t num_ref_idx_l1_default_active_minus1; //ue(v)
|
||||
int32_t init_qp_minus26; //se(v)
|
||||
bool constrained_intra_pred_flag; //u(1)
|
||||
bool transform_skip_enabled_flag; //u(1)
|
||||
bool cu_qp_delta_enabled_flag; //u(1)
|
||||
uint32_t diff_cu_qp_delta_depth; //ue(v)
|
||||
int32_t pps_cb_qp_offset; //se(v)
|
||||
int32_t pps_cr_qp_offset; //se(v)
|
||||
bool pps_slice_chroma_qp_offsets_present_flag; //u(1)
|
||||
bool weighted_pred_flag; //u(1)
|
||||
bool weighted_bipred_flag; //u(1)
|
||||
bool transquant_bypass_enabled_flag; //u(1)
|
||||
bool tiles_enabled_flag; //u(1)
|
||||
bool entropy_coding_sync_enabled_flag; //u(1)
|
||||
uint32_t num_tile_columns_minus1; //ue(v)
|
||||
uint32_t num_tile_rows_minus1; //ue(v)
|
||||
bool uniform_spacing_flag; //u(1)
|
||||
//PicWidthInCtbsY = Ceil( pic_width_in_luma_samples / CtbSizeY ) = 256 assume max width is 4096
|
||||
//CtbSizeY = 1<<CtbLog2SizeY so min is 16
|
||||
// 4 <= CtbLog2SizeY <= 6
|
||||
uint32_t column_width_minus1[265]; //ue(v)
|
||||
//2304/16=144 assume max height is 2304
|
||||
uint32_t row_height_minus1[144]; //ue(v)
|
||||
bool loop_filter_across_tiles_enabled_flag; //u(1)
|
||||
bool pps_loop_filter_across_slices_enabled_flag; //u(1)
|
||||
bool deblocking_filter_control_present_flag; //u(1)
|
||||
bool deblocking_filter_override_enabled_flag; //u(1)
|
||||
bool pps_deblocking_filter_disabled_flag; //u(1)
|
||||
int32_t pps_beta_offset_div2; //se(v)
|
||||
int32_t pps_tc_offset_div2; //se(v)
|
||||
bool pps_scaling_list_data_present_flag; //u(1)
|
||||
//scaling_list_data( )
|
||||
H265ScalingListData scaling_list_data;
|
||||
bool lists_modification_present_flag; //u(1)
|
||||
uint32_t log2_parallel_merge_level_minus2; //ue(v)
|
||||
bool slice_segment_header_extension_present_flag; //u(1)
|
||||
bool pps_extension_present_flag; //u(1)
|
||||
bool pps_range_extension_flag; //u(1)
|
||||
bool pps_multilayer_extension_flag; //u(1)
|
||||
uint32_t pps_extension_6bits; //u(6)
|
||||
// pps_range_extension()
|
||||
uint32_t log2_max_transform_skip_block_size_minus2; //ue(v)
|
||||
uint8_t cross_component_prediction_enabled_flag; //u(1)
|
||||
uint8_t chroma_qp_offset_list_enabled_flag; //u(1)
|
||||
uint32_t diff_cu_chroma_qp_offset_depth; //ue(v)
|
||||
uint32_t chroma_qp_offset_list_len_minus1; //ue(v)
|
||||
int32_t cb_qp_offset_list[6]; //se(v)
|
||||
int32_t cr_qp_offset_list[6]; //se(v)
|
||||
uint32_t log2_sao_offset_scale_luma; //ue(v)
|
||||
uint32_t log2_sao_offset_scale_chroma; //ue(v)
|
||||
bool pps_extension_data_flag; //u(1)
|
||||
//rbsp_trailing_bits( )
|
||||
H265RbspTrailingBits rbsp_trailing_bits;
|
||||
} PpsData;
|
||||
|
||||
/*! \brief Structure for Slice Header Data
|
||||
*/
|
||||
typedef struct {
|
||||
bool first_slice_segment_in_pic_flag; //u(1)
|
||||
bool no_output_of_prior_pics_flag; //u(1)
|
||||
uint32_t slice_pic_parameter_set_id; //ue(v)
|
||||
bool dependent_slice_segment_flag; //u(1)
|
||||
uint32_t slice_segment_address; //u(v)
|
||||
//num_extra_slice_header_bits is u(3), so max is 7
|
||||
bool slice_reserved_flag[7]; //u(1)
|
||||
uint32_t slice_type; //ue(v)
|
||||
bool pic_output_flag; //u(1)
|
||||
uint32_t colour_plane_id; //u(2)
|
||||
uint32_t slice_pic_order_cnt_lsb; //u(v)
|
||||
bool short_term_ref_pic_set_sps_flag; //u(1)
|
||||
//short_term_ref_pic_set( num_short_term_ref_pic_sets )
|
||||
uint32_t short_term_ref_pic_set_size; //MM
|
||||
H265ShortTermRPS st_rps;
|
||||
uint32_t short_term_ref_pic_set_idx; //u(v)
|
||||
uint32_t num_long_term_sps; //ue(v)
|
||||
uint32_t num_long_term_pics; //ue(v)
|
||||
//num_long_term_sps + num_long_term_pics max is 32
|
||||
H265LongTermRPS lt_rps;
|
||||
uint32_t lt_idx_sps[32]; //u(v)
|
||||
uint32_t poc_lsb_lt[32]; //u(v)
|
||||
bool used_by_curr_pic_lt_flag[32]; //u(1)
|
||||
bool delta_poc_msb_present_flag[32]; //u(1)
|
||||
uint32_t delta_poc_msb_cycle_lt[32]; //ue(v)
|
||||
bool slice_temporal_mvp_enabled_flag; //u(1)
|
||||
bool slice_sao_luma_flag; //u(1)
|
||||
bool slice_sao_chroma_flag; //u(1)
|
||||
bool num_ref_idx_active_override_flag; //u(1)
|
||||
uint32_t num_ref_idx_l0_active_minus1; //ue(v)
|
||||
uint32_t num_ref_idx_l1_active_minus1; //ue(v)
|
||||
// Reference picture list modification
|
||||
uint32_t ref_pic_list_modification_flag_l0; //u(1)
|
||||
uint32_t list_entry_l0[16]; //u(v)
|
||||
uint32_t ref_pic_list_modification_flag_l1; //u(1)
|
||||
uint32_t list_entry_l1[16]; //u(v)
|
||||
bool mvd_l1_zero_flag; //u(1)
|
||||
bool cabac_init_flag; //u(1)
|
||||
bool collocated_from_l0_flag; //u(1)
|
||||
HevcPredWeightTable pred_weight_table;
|
||||
uint32_t collocated_ref_idx; //ue(v)
|
||||
uint32_t five_minus_max_num_merge_cand; //ue(v)
|
||||
int32_t slice_qp_delta; //se(v)
|
||||
int32_t slice_cb_qp_offset; //se(v)
|
||||
int32_t slice_cr_qp_offset; //se(v)
|
||||
uint8_t cu_chroma_qp_offset_enabled_flag; //u(1)
|
||||
bool deblocking_filter_override_flag; //u(1)
|
||||
bool slice_deblocking_filter_disabled_flag; //u(1)
|
||||
int32_t slice_beta_offset_div2; //se(v)
|
||||
int32_t slice_tc_offset_div2; //se(v)
|
||||
bool slice_loop_filter_across_slices_enabled_flag; //u(1)
|
||||
uint32_t num_entry_point_offsets; //ue(v)
|
||||
uint32_t offset_len_minus1; //ue(v)
|
||||
uint32_t entry_point_offset_minus1[MAX_ENTRY_POINT_OFFSETS]; //u(v)
|
||||
uint32_t slice_segment_header_extension_length; //ue(v)
|
||||
//slice_segment_header_extension_length max is 256
|
||||
uint8_t slice_segment_header_extension_data_byte[256]; //u(8)
|
||||
} SliceHeaderData;
|
||||
|
||||
/*! \brief Inline function to Parse the NAL Unit Header
|
||||
*
|
||||
* \param [in] nal_unit A pointer of <tt>uint8_t</tt> containing the Demuxed output stream
|
||||
* \return Returns an object of NALUnitHeader
|
||||
*/
|
||||
static inline NalUnitHeader ParseNalUnitHeader(uint8_t *nal_unit) {
|
||||
NalUnitHeader nalu_header;
|
||||
static inline HevcNalUnitHeader ParseNalUnitHeader(uint8_t *nal_unit) {
|
||||
HevcNalUnitHeader nalu_header;
|
||||
nalu_header.num_emu_byte_removed = 0;
|
||||
//read nalu header
|
||||
nalu_header.forbidden_zero_bit = (uint32_t) ((nal_unit[0] >> 7) & 1);
|
||||
@@ -624,12 +87,6 @@ protected:
|
||||
return nalu_header;
|
||||
}
|
||||
|
||||
enum HevcRefMarking {
|
||||
kUnusedForReference = 0,
|
||||
kUsedForShortTerm = 1,
|
||||
kUsedForLongTerm = 2
|
||||
};
|
||||
|
||||
/*! \brief Picture info for decoding process
|
||||
*/
|
||||
typedef struct {
|
||||
@@ -671,17 +128,17 @@ protected:
|
||||
// Data members of HEVC class
|
||||
uint32_t pic_count_; // decoded picture count for the current bitstream
|
||||
|
||||
NalUnitHeader nal_unit_header_;
|
||||
HevcNalUnitHeader nal_unit_header_;
|
||||
int32_t m_active_vps_id_;
|
||||
int32_t m_active_sps_id_;
|
||||
int32_t m_active_pps_id_;
|
||||
VpsData* m_vps_ = nullptr;
|
||||
SpsData* m_sps_ = nullptr;
|
||||
PpsData* m_pps_ = nullptr;
|
||||
SliceHeaderData* m_sh_ = nullptr;
|
||||
SliceHeaderData* m_sh_copy_ = nullptr;
|
||||
HevcVideoParamSet* m_vps_ = nullptr;
|
||||
HevcSeqParamSet* m_sps_ = nullptr;
|
||||
HevcPicParamSet* m_pps_ = nullptr;
|
||||
HevcSliceSegHeader* m_sh_ = nullptr;
|
||||
HevcSliceSegHeader* m_sh_copy_ = nullptr;
|
||||
|
||||
NalUnitHeader slice_nal_unit_header_;
|
||||
HevcNalUnitHeader slice_nal_unit_header_;
|
||||
HevcPicInfo curr_pic_info_;
|
||||
bool b_new_picture_;
|
||||
int m_packet_count_;
|
||||
@@ -767,7 +224,7 @@ protected:
|
||||
void ParsePps(uint8_t *nalu, size_t size);
|
||||
|
||||
/*! \brief Function to parse Profiles, Tiers and Levels
|
||||
* \param [out] ptl A pointer of <tt>H265ProfileTierLevel</tt> for the output from teh parsed stream
|
||||
* \param [out] ptl A pointer of <tt>HevcProfileTierLevel</tt> for the output from teh parsed stream
|
||||
* \param [in] profile_present_flag Input of <tt>bool</tt> - 1 specifies profile information is present, else 0
|
||||
* \param [in] max_num_sub_layers_minus1 Input of <tt>uint32_t</tt> - plus 1 specifies the maximum number of temporal sub-layers that may be present
|
||||
* \param [in] nalu A pointer of <tt>uint8_t</tt> for the input stream to be parsed
|
||||
@@ -775,10 +232,10 @@ protected:
|
||||
* \param [in] offset Reference to the offset in the input buffer
|
||||
* \return No return value
|
||||
*/
|
||||
void ParsePtl(H265ProfileTierLevel *ptl, bool profile_present_flag, uint32_t max_num_sub_layers_minus1, uint8_t *nalu, size_t size, size_t &offset);
|
||||
void ParsePtl(HevcProfileTierLevel *ptl, bool profile_present_flag, uint32_t max_num_sub_layers_minus1, uint8_t *nalu, size_t size, size_t &offset);
|
||||
|
||||
/*! \brief Function to parse Sub Layer Hypothetical Reference Decoder Parameters
|
||||
* \param [out] sub_hrd A pointer of <tt>H265SubLayerHrdParameters</tt> for the output from teh parsed stream
|
||||
* \param [out] sub_hrd A pointer of <tt>HevcSubLayerHrdParameters</tt> for the output from teh parsed stream
|
||||
* \param [in] cpb_cnt Input of <tt>uint32_t</tt> - specifies the coded picture buffer count in a HRD buffer
|
||||
* \param [in] sub_pic_hrd_params_present_flag Input of <tt>bool</tt> - 1 specifies sub layer HRD information is present, else 0
|
||||
* \param [in] nalu A pointer of <tt>uint8_t</tt> for the input stream to be parsed
|
||||
@@ -786,10 +243,10 @@ protected:
|
||||
* \param [in] offset Reference to the offset in the input buffer
|
||||
* \return No return value
|
||||
*/
|
||||
void ParseSubLayerHrdParameters(H265SubLayerHrdParameters *sub_hrd, uint32_t cpb_cnt, bool sub_pic_hrd_params_present_flag, uint8_t *nalu, size_t size, size_t &offset);
|
||||
void ParseSubLayerHrdParameters(HevcSubLayerHrdParameters *sub_hrd, uint32_t cpb_cnt, bool sub_pic_hrd_params_present_flag, uint8_t *nalu, size_t size, size_t &offset);
|
||||
|
||||
/*! \brief Function to parse Hypothetical Reference Decoder Parameters
|
||||
* \param [out] hrd A pointer of <tt>H265HrdParameters</tt> for the output from the parsed stream
|
||||
* \param [out] hrd A pointer of <tt>HevcHrdParameters</tt> for the output from the parsed stream
|
||||
* \param [in] common_inf_present_flag Input of <tt>bool</tt> - 1 specifies HRD information is present, else 0
|
||||
* \param [in] max_num_sub_layers_minus1 Input of <tt>uint32_t</tt> - plus 1 specifies the maximum number of temporal sub-layers that may be present
|
||||
* \param [in] nalu A pointer of <tt>uint8_t</tt> for the input stream to be parsed
|
||||
@@ -797,45 +254,45 @@ protected:
|
||||
* \param [in] offset Reference to the offset in the input buffer
|
||||
* \return No return value
|
||||
*/
|
||||
void ParseHrdParameters(H265HrdParameters *hrd, bool common_inf_present_flag, uint32_t max_num_sub_layers_minus1, uint8_t *nalu, size_t size, size_t &offset);
|
||||
void ParseHrdParameters(HevcHrdParameters *hrd, bool common_inf_present_flag, uint32_t max_num_sub_layers_minus1, uint8_t *nalu, size_t size, size_t &offset);
|
||||
|
||||
/*! \brief Function to set the default values to the scaling list
|
||||
* \param [out] sl_ptr A pointer to the scaling list <tt>H265ScalingListData</tt>
|
||||
* \param [out] sl_ptr A pointer to the scaling list <tt>HevcScalingListData</tt>
|
||||
* \return No return value
|
||||
*/
|
||||
void SetDefaultScalingList(H265ScalingListData *sl_ptr);
|
||||
void SetDefaultScalingList(HevcScalingListData *sl_ptr);
|
||||
|
||||
/*! \brief Function to parse Scaling List
|
||||
* \param [out] sl_ptr A pointer of <tt>H265ScalingListData</tt> for the output from the parsed stream
|
||||
* \param [out] sl_ptr A pointer of <tt>HevcScalingListData</tt> for the output from the parsed stream
|
||||
* \param [in] data A pointer of <tt>uint8_t</tt> for the input stream to be parsed
|
||||
* \param [in] size Size of the input stream
|
||||
* \param [in] offset Reference to the offset in the input buffer
|
||||
* \param [in] sps_ptr Pointer to the current SPS
|
||||
* \return No return value
|
||||
*/
|
||||
void ParseScalingList(H265ScalingListData * sl_ptr, uint8_t *data, size_t size, size_t &offset, SpsData *sps_ptr);
|
||||
void ParseScalingList(HevcScalingListData * sl_ptr, uint8_t *data, size_t size, size_t &offset, HevcSeqParamSet *sps_ptr);
|
||||
|
||||
/*! \brief Function to parse Video Usability Information
|
||||
* \param [out] vui A pointer of <tt>H265VuiParameters</tt> for the output from the parsed stream
|
||||
* \param [out] vui A pointer of <tt>HevcVuiParameters</tt> for the output from the parsed stream
|
||||
* \param [in] max_num_sub_layers_minus1 Input of <tt>uint32_t</tt> - plus 1 specifies the maximum number of temporal sub-layers that may be present
|
||||
* \param [in] data A pointer of <tt>uint8_t</tt> for the input stream to be parsed
|
||||
* \param [in] size Size of the input stream
|
||||
* \param [in] offset Reference to the offset in the input buffer
|
||||
* \return No return value
|
||||
*/
|
||||
void ParseVui(H265VuiParameters *vui, uint32_t max_num_sub_layers_minus1, uint8_t *data, size_t size, size_t &offset);
|
||||
void ParseVui(HevcVuiParameters *vui, uint32_t max_num_sub_layers_minus1, uint8_t *data, size_t size, size_t &offset);
|
||||
|
||||
/*! \brief Function to parse Short Term Reference Picture Set
|
||||
* \param [out] rps A pointer of <tt>H265ShortTermRPS</tt> for the output from the parsed stream
|
||||
* \param [out] rps A pointer of <tt>HevcShortTermRps</tt> for the output from the parsed stream
|
||||
* \param [in] st_rps_idx specifies the index in the RPS buffer
|
||||
* \param [in] num_short_term_ref_pic_sets Specifies the count of Short Term RPS in <tt>uint32_t</tt>
|
||||
* \param [in] rps_ref A reference of <tt>H265ShortTermRPS</tt> to the RPS buffer
|
||||
* \param [in] rps_ref A reference of <tt>HevcShortTermRps</tt> to the RPS buffer
|
||||
* \param [in] data A pointer of <tt>uint8_t</tt> for the input stream to be parsed
|
||||
* \param [in] size Size of the input stream
|
||||
* \param [in] offset Reference to the offset in the input buffer
|
||||
* \return No return value
|
||||
*/
|
||||
void ParseShortTermRefPicSet(H265ShortTermRPS *rps, uint32_t st_rps_idx, uint32_t num_short_term_ref_pic_sets, H265ShortTermRPS rps_ref[], uint8_t *data, size_t size, size_t &offset);
|
||||
void ParseShortTermRefPicSet(HevcShortTermRps *rps, uint32_t st_rps_idx, uint32_t num_short_term_ref_pic_sets, HevcShortTermRps rps_ref[], uint8_t *data, size_t size, size_t &offset);
|
||||
|
||||
/*! \brief Function to parse weighted prediction table
|
||||
* \param [in/out] Slice_header_ptr Pointer to the slice segment header
|
||||
@@ -843,7 +300,7 @@ protected:
|
||||
* \param [in] stream_ptr Bit stream pointer
|
||||
* \param [in/out] offset Bit offset of the current parsing action
|
||||
*/
|
||||
void ParsePredWeightTable(HEVCVideoParser::SliceHeaderData *slice_header_ptr, int chroma_array_type, uint8_t *stream_ptr, size_t &offset);
|
||||
void ParsePredWeightTable(HevcSliceSegHeader *slice_header_ptr, int chroma_array_type, uint8_t *stream_ptr, size_t &offset);
|
||||
|
||||
/*! \brief Function to parse Slice Header
|
||||
* \param [in] nal_unit_type Input of <tt>uint32_t</tt> containing the enumerator value to the NAL Unit Type
|
||||
@@ -914,12 +371,12 @@ protected:
|
||||
int GetNalUnit();
|
||||
|
||||
#if DBGINFO
|
||||
void PrintVps(HEVCVideoParser::VpsData *vps_ptr);
|
||||
void PrintSps(HEVCVideoParser::SpsData *sps_ptr);
|
||||
void PrintPps(HEVCVideoParser::PpsData *pps_ptr);
|
||||
void PrintSliceSegHeader(HEVCVideoParser::SliceHeaderData *slice_header_ptr);
|
||||
void PrintStRps(HEVCVideoParser::H265ShortTermRPS *rps_ptr);
|
||||
void PrintLtRefInfo(HEVCVideoParser::H265LongTermRPS *lt_info_ptr);
|
||||
void PrintVps(HevcVideoParamSet *vps_ptr);
|
||||
void PrintSps(HevcSeqParamSet *sps_ptr);
|
||||
void PrintPps(HevcPicParamSet *pps_ptr);
|
||||
void PrintSliceSegHeader(HevcSliceSegHeader *slice_header_ptr);
|
||||
void PrintStRps(HevcShortTermRps *rps_ptr);
|
||||
void PrintLtRefInfo(HevcLongTermRps *lt_info_ptr);
|
||||
#endif // DBGINFO
|
||||
|
||||
private:
|
||||
@@ -929,7 +386,7 @@ private:
|
||||
ParserResult Init();
|
||||
|
||||
// functions to fill structures for callback functions
|
||||
int FillSeqCallbackFn(SpsData* sps_data);
|
||||
int FillSeqCallbackFn(HevcSeqParamSet* sps_data);
|
||||
void FillSeiMessageCallbackFn();
|
||||
|
||||
/*! \brief Function to fill the decode parameters and call back decoder to decode a picture
|
||||
@@ -942,11 +399,11 @@ private:
|
||||
*/
|
||||
int OutputDecodedPictures();
|
||||
|
||||
bool IsIdrPic(NalUnitHeader *nal_header_ptr);
|
||||
bool IsCraPic(NalUnitHeader *nal_header_ptr);
|
||||
bool IsBlaPic(NalUnitHeader *nal_header_ptr);
|
||||
bool IsIrapPic(NalUnitHeader *nal_header_ptr);
|
||||
bool IsRaslPic(NalUnitHeader *nal_header_ptr);
|
||||
bool IsRadlPic(NalUnitHeader *nal_header_ptr);
|
||||
bool IsRefPic(NalUnitHeader *nal_header_ptr);
|
||||
bool IsIdrPic(HevcNalUnitHeader *nal_header_ptr);
|
||||
bool IsCraPic(HevcNalUnitHeader *nal_header_ptr);
|
||||
bool IsBlaPic(HevcNalUnitHeader *nal_header_ptr);
|
||||
bool IsIrapPic(HevcNalUnitHeader *nal_header_ptr);
|
||||
bool IsRaslPic(HevcNalUnitHeader *nal_header_ptr);
|
||||
bool IsRadlPic(HevcNalUnitHeader *nal_header_ptr);
|
||||
bool IsRefPic(HevcNalUnitHeader *nal_header_ptr);
|
||||
};
|
||||
@@ -47,7 +47,7 @@ private:
|
||||
roc_parser_ = std::make_shared<H264VideoParser>();
|
||||
break;
|
||||
case rocDecVideoCodec_HEVC:
|
||||
roc_parser_ = std::make_shared<HEVCVideoParser>();
|
||||
roc_parser_ = std::make_shared<HevcVideoParser>();
|
||||
break;
|
||||
default:
|
||||
THROW("Unsupported parser type "+ TOSTR(params->CodecType));
|
||||
|
||||
Referens i nytt ärende
Block a user