diff --git a/ext-codec/H264.cpp b/ext-codec/H264.cpp index 28a6d9d0..9b255d03 100644 --- a/ext-codec/H264.cpp +++ b/ext-codec/H264.cpp @@ -11,7 +11,6 @@ #include "H264.h" #include "H264Rtmp.h" #include "H264Rtp.h" -#include "SPSParser.h" #include "Util/logger.h" #include "Util/base64.h" #include "Common/Parser.h" @@ -22,28 +21,194 @@ #include "mpeg4-avc.h" #endif +#include +#include + using namespace std; using namespace toolkit; namespace mediakit { -static bool getAVCInfo(const char *sps, size_t sps_len, int &iVideoWidth, int &iVideoHeight, float &iVideoFps) { - if (sps_len < 4) { - return false; +// ---- 内部比特流工具 ---- +namespace { + +// 去除 RBSP 防竞争字节 (0x00 0x00 0x03 -> 0x00 0x00) +static std::vector rbsp_from_nalu(const uint8_t *data, size_t size) { + std::vector out; + out.reserve(size); + for (size_t i = 0; i < size; ) { + if (i + 2 < size && data[i] == 0x00 && data[i+1] == 0x00 && data[i+2] == 0x03) { + out.push_back(0x00); + out.push_back(0x00); + i += 3; + } else { + out.push_back(data[i++]); + } } - T_GetBitContext tGetBitBuf; - T_SPS tH264SpsInfo; - memset(&tGetBitBuf, 0, sizeof(tGetBitBuf)); - memset(&tH264SpsInfo, 0, sizeof(tH264SpsInfo)); - tGetBitBuf.pu8Buf = (uint8_t *)sps + 1; - tGetBitBuf.iBufSize = (int)(sps_len - 1); - if (0 != h264DecSeqParameterSet((void *)&tGetBitBuf, &tH264SpsInfo)) { - return false; + return out; +} + +struct BitStream { + const uint8_t *buf; + size_t size; // bytes + size_t pos; // bit position + + BitStream(const uint8_t *b, size_t s) : buf(b), size(s), pos(0) {} + + bool eof() const { return pos >= size * 8; } + + size_t bits_left() const { return size * 8 - pos; } + + uint32_t read_bits(int n) { + if (n < 0 || n > 32 || (size_t)n > bits_left()) { + throw std::runtime_error("eof"); + } + uint32_t val = 0; + for (int i = 0; i < n; i++) { + val = (val << 1) | ((buf[pos / 8] >> (7 - pos % 8)) & 1); + pos++; + } + return val; + } + + void skip_bits(int n) { + if (n < 0 || (size_t)n > bits_left()) { + throw std::runtime_error("eof"); + } + pos += n; + } + + uint32_t read_ue() { // Exp-Golomb unsigned + int zeros = 0; + while (!eof() && read_bits(1) == 0) zeros++; + if (zeros == 0) return 0; + if (zeros >= 32) throw std::runtime_error("exp-golomb overflow"); + return (1u << zeros) - 1 + read_bits(zeros); + } + + int32_t read_se() { // Exp-Golomb signed + uint32_t v = read_ue(); + return (v & 1) ? (int32_t)((v + 1) >> 1) : -(int32_t)(v >> 1); + } +}; + +} // anonymous namespace + +// ---- H264 SPS 解析 ---- +static bool getAVCInfo(const char *sps_raw, size_t sps_len, int &iVideoWidth, int &iVideoHeight, float &iVideoFps) { + if (sps_len < 4) return false; + // sps_raw[0] 是 NAL header，从第 1 字节开始是 RBSP + auto rbsp = rbsp_from_nalu((const uint8_t *)sps_raw + 1, sps_len - 1); + if (rbsp.size() < 3) return false; + + try { + BitStream bs(rbsp.data(), rbsp.size()); + + uint8_t profile_idc = (uint8_t)bs.read_bits(8); // profile_idc + bs.skip_bits(8); // constraint flags + reserved + bs.skip_bits(8); // level_idc + bs.read_ue(); // seq_parameter_set_id + + uint32_t chroma_format_idc = 1; + if (profile_idc == 100 || profile_idc == 110 || profile_idc == 122 || + profile_idc == 244 || profile_idc == 44 || profile_idc == 83 || + profile_idc == 86 || profile_idc == 118 || profile_idc == 128) { + chroma_format_idc = bs.read_ue(); + if (chroma_format_idc > 3) { + return false; + } + if (chroma_format_idc == 3) bs.skip_bits(1); // separate_colour_plane_flag + bs.read_ue(); // bit_depth_luma_minus8 + bs.read_ue(); // bit_depth_chroma_minus8 + bs.skip_bits(1); // qpprime_y_zero_transform_bypass_flag + if (bs.read_bits(1)) { // seq_scaling_matrix_present_flag + int cnt = (chroma_format_idc != 3) ? 8 : 12; + for (int i = 0; i < cnt; i++) { + if (bs.read_bits(1)) { // seq_scaling_list_present_flag + int sz = (i < 6) ? 16 : 64; + int last = 8, next = 8; + for (int j = 0; j < sz; j++) { + if (next != 0) next = (last + bs.read_se() + 256) % 256; + last = (next == 0) ? last : next; + } + } + } + } + } + + bs.read_ue(); // log2_max_frame_num_minus4 + uint32_t pic_order_cnt_type = bs.read_ue(); + if (pic_order_cnt_type == 0) { + bs.read_ue(); // log2_max_pic_order_cnt_lsb_minus4 + } else if (pic_order_cnt_type == 1) { + bs.skip_bits(1); // delta_pic_order_always_zero_flag + bs.read_se(); // offset_for_non_ref_pic + bs.read_se(); // offset_for_top_to_bottom_field + uint32_t n = bs.read_ue(); + for (uint32_t i = 0; i < n; i++) bs.read_se(); + } + bs.read_ue(); // max_num_ref_frames + bs.skip_bits(1); // gaps_in_frame_num_value_allowed_flag + + uint32_t pic_width_in_mbs_minus1 = bs.read_ue(); + uint32_t pic_height_in_map_units_minus1 = bs.read_ue(); + uint32_t frame_mbs_only_flag = bs.read_bits(1); + if (!frame_mbs_only_flag) bs.skip_bits(1); // mb_adaptive_frame_field_flag + bs.skip_bits(1); // direct_8x8_inference_flag + + uint32_t crop_left = 0, crop_right = 0, crop_top = 0, crop_bottom = 0; + if (bs.read_bits(1)) { // frame_cropping_flag + crop_left = bs.read_ue(); + crop_right = bs.read_ue(); + crop_top = bs.read_ue(); + crop_bottom = bs.read_ue(); + } + + uint32_t crop_unit_x = 1; + uint32_t crop_unit_y = 2 - frame_mbs_only_flag; + if (chroma_format_idc == 1) { + crop_unit_x = 2; + crop_unit_y = 2 * (2 - frame_mbs_only_flag); + } else if (chroma_format_idc == 2) { + crop_unit_x = 2; + crop_unit_y = 2 - frame_mbs_only_flag; + } + + uint64_t raw_width = (uint64_t)(pic_width_in_mbs_minus1 + 1) * 16; + uint64_t raw_height = (uint64_t)(pic_height_in_map_units_minus1 + 1) * 16 * (2 - frame_mbs_only_flag); + uint64_t crop_w = ((uint64_t)crop_left + crop_right) * crop_unit_x; + uint64_t crop_h = ((uint64_t)crop_top + crop_bottom) * crop_unit_y; + if (crop_w >= raw_width || crop_h >= raw_height) return false; + iVideoWidth = (int)(raw_width - crop_w); + iVideoHeight = (int)(raw_height - crop_h); + + iVideoFps = 0.0f; + if (bs.read_bits(1)) { // vui_parameters_present_flag + if (bs.read_bits(1)) { // aspect_ratio_info_present_flag + uint32_t ar = bs.read_bits(8); + if (ar == 255) bs.skip_bits(32); // sar width+height + } + if (bs.read_bits(1)) bs.skip_bits(1); // overscan + if (bs.read_bits(1)) { // video_signal_type_present_flag + bs.skip_bits(3 + 1); // video_format + video_full_range_flag + if (bs.read_bits(1)) bs.skip_bits(24); // colour_description_present_flag + } + if (bs.read_bits(1)) { // chroma_loc_info_present_flag + bs.read_ue(); bs.read_ue(); + } + if (bs.read_bits(1)) { // timing_info_present_flag + uint32_t num_units_in_tick = bs.read_bits(32); + uint32_t time_scale = bs.read_bits(32); + bs.skip_bits(1); // fixed_frame_rate_flag + if (num_units_in_tick > 0) { + iVideoFps = (float)time_scale / (2.0f * (float)num_units_in_tick); + } + } + } + return iVideoWidth > 0 && iVideoHeight > 0; + } catch (...) { + return iVideoWidth > 0 && iVideoHeight > 0; } - h264GetWidthHeight(&tH264SpsInfo, &iVideoWidth, &iVideoHeight); - h264GeFramerate(&tH264SpsInfo, &iVideoFps); - // ErrorL << iVideoWidth << " " << iVideoHeight << " " << iVideoFps; - return true; } bool getAVCInfo(const string &strSps, int &iVideoWidth, int &iVideoHeight, float &iVideoFps) { diff --git a/ext-codec/H265.cpp b/ext-codec/H265.cpp index 906973a0..0a21ddad 100644 --- a/ext-codec/H265.cpp +++ b/ext-codec/H265.cpp @@ -11,11 +11,13 @@ #include "H265.h" #include "H265Rtp.h" #include "H265Rtmp.h" -#include "SPSParser.h" #include "Util/base64.h" #include "Common/Parser.h" #include "Extension/Factory.h" +#include +#include + #ifdef ENABLE_MP4 #include "mpeg4-hevc.h" #endif @@ -25,35 +27,283 @@ using namespace toolkit; namespace mediakit { -bool getHEVCInfo(const char * vps, size_t vps_len,const char * sps,size_t sps_len,int &iVideoWidth, int &iVideoHeight, float &iVideoFps){ - T_GetBitContext tGetBitBuf; - T_HEVCSPS tH265SpsInfo; - T_HEVCVPS tH265VpsInfo; - if ( vps_len > 2 ){ - memset(&tGetBitBuf,0,sizeof(tGetBitBuf)); - memset(&tH265VpsInfo,0,sizeof(tH265VpsInfo)); - tGetBitBuf.pu8Buf = (uint8_t*)vps+2; - tGetBitBuf.iBufSize = (int)(vps_len-2); - if(0 != h265DecVideoParameterSet((void *) &tGetBitBuf, &tH265VpsInfo)){ +// ---- 内部比特流工具（H265） ---- +namespace { + +static std::vector h265_rbsp_from_nalu(const uint8_t *data, size_t size) { + std::vector out; + out.reserve(size); + for (size_t i = 0; i < size; ) { + if (i + 2 < size && data[i] == 0x00 && data[i+1] == 0x00 && data[i+2] == 0x03) { + out.push_back(0x00); + out.push_back(0x00); + i += 3; + } else { + out.push_back(data[i++]); + } + } + return out; +} + +struct H265BS { + const uint8_t *buf; + size_t size; + size_t pos; + H265BS(const uint8_t *b, size_t s) : buf(b), size(s), pos(0) {} + bool eof() const { return pos >= size * 8; } + size_t bits_left() const { return size * 8 - pos; } + uint32_t read_bits(int n) { + if (n < 0 || n > 32 || (size_t)n > bits_left()) { + throw std::runtime_error("eof"); + } + uint32_t val = 0; + for (int i = 0; i < n; i++) { + val = (val << 1) | ((buf[pos / 8] >> (7 - pos % 8)) & 1); + pos++; + } + return val; + } + void skip_bits(int n) { + if (n < 0 || (size_t)n > bits_left()) { + throw std::runtime_error("eof"); + } + pos += n; + } + uint32_t read_ue() { + int z = 0; + while (!eof() && read_bits(1) == 0) z++; + if (z == 0) return 0; + if (z >= 32) throw std::runtime_error("exp-golomb overflow"); + return (1u << z) - 1 + read_bits(z); + } + int32_t read_se() { + uint32_t v = read_ue(); + return (v & 1) ? (int32_t)((v + 1) >> 1) : -(int32_t)(v >> 1); + } + // profile_tier_level(profilePresentFlag, maxNumSubLayersMinus1) + void skip_profile_tier_level(bool profilePresentFlag, uint32_t maxNumSubLayersMinus1) { + if (profilePresentFlag) { + skip_bits(2 + 1 + 5); // profile_space + tier_flag + profile_idc + skip_bits(32); // profile_compatibility_flag[32] + skip_bits(4); // progressive/interlaced/non_packed/frame_only + skip_bits(44); // reserved_zero_44bits + } + skip_bits(8); // general_level_idc + // sub_layer flags + std::vector profile_present(maxNumSubLayersMinus1), level_present(maxNumSubLayersMinus1); + for (uint32_t i = 0; i < maxNumSubLayersMinus1; i++) { + profile_present[i] = read_bits(1) != 0; + level_present[i] = read_bits(1) != 0; + } + if (maxNumSubLayersMinus1 > 0) { + for (uint32_t i = maxNumSubLayersMinus1; i < 8; i++) skip_bits(2); + } + for (uint32_t i = 0; i < maxNumSubLayersMinus1; i++) { + if (profile_present[i]) { + skip_bits(2 + 1 + 5 + 32 + 4 + 44); + } + if (level_present[i]) skip_bits(8); + } + } +}; + +} // anonymous namespace + +// ---- H265 VPS 解析（只提取帧率用的 timing info） ---- +static bool parse_hevc_vps_fps(const uint8_t *data, size_t size, float &fps) { + // data 为 NALU 原始数据（含 NAL header） + if (size < 3) return false; + auto rbsp = h265_rbsp_from_nalu(data, size); + try { + H265BS bs(rbsp.data(), rbsp.size()); + // NALU header: forbidden_zero_bit(1) + nal_unit_type(6) + nuh_layer_id(6) + nuh_temporal_id_plus1(3) + bs.skip_bits(16); + // vps_video_parameter_set_id(4) + vps_reserved_three_2bits(2) + vps_max_layers_minus1(6) + bs.skip_bits(4 + 2 + 6); + uint32_t vps_max_sub_layers_minus1 = bs.read_bits(3); + bs.skip_bits(1); // vps_temporal_id_nesting_flag + bs.skip_bits(16); // vps_reserved_0xffff_16bits + bs.skip_profile_tier_level(true, vps_max_sub_layers_minus1); + bool vps_sub_layer_ordering_info_present_flag = bs.read_bits(1) != 0; + uint32_t start = vps_sub_layer_ordering_info_present_flag ? 0 : vps_max_sub_layers_minus1; + for (uint32_t i = start; i <= vps_max_sub_layers_minus1; i++) { + bs.read_ue(); // vps_max_dec_pic_buffering_minus1 + bs.read_ue(); // vps_max_num_reorder_pics + bs.read_ue(); // vps_max_latency_increase_plus1 + } + uint32_t vps_max_layer_id = bs.read_bits(6); + uint32_t vps_num_layer_sets_minus1 = bs.read_ue(); + for (uint32_t i = 1; i <= vps_num_layer_sets_minus1; i++) { + for (uint32_t j = 0; j <= vps_max_layer_id; j++) bs.skip_bits(1); + } + if (bs.read_bits(1)) { // vps_timing_info_present_flag + uint32_t vps_num_units_in_tick = bs.read_bits(32); + uint32_t vps_time_scale = bs.read_bits(32); + if (vps_num_units_in_tick > 0) { + fps = (float)vps_time_scale / (float)vps_num_units_in_tick; + } + } + return true; + } catch (...) { + return false; + } +} + +// ---- H265 SPS 解析（宽高 + 备用帧率） ---- +static bool parse_hevc_sps(const uint8_t *data, size_t size, + int &width, int &height, float &fps) { + if (size < 3) return false; + auto rbsp = h265_rbsp_from_nalu(data, size); + try { + H265BS bs(rbsp.data(), rbsp.size()); + bs.skip_bits(16); // NALU header + bs.skip_bits(4); // sps_video_parameter_set_id + uint32_t sps_max_sub_layers_minus1 = bs.read_bits(3); + bs.skip_bits(1); // sps_temporal_id_nesting_flag + bs.skip_profile_tier_level(true, sps_max_sub_layers_minus1); + bs.read_ue(); // sps_seq_parameter_set_id + uint32_t chroma_format_idc = bs.read_ue(); + if (chroma_format_idc > 3) { return false; } + if (chroma_format_idc == 3) bs.skip_bits(1); // separate_colour_plane_flag + uint32_t pic_width = bs.read_ue(); + uint32_t pic_height = bs.read_ue(); + + if (bs.read_bits(1)) { // conformance_window_flag + uint32_t sub_width_c = (chroma_format_idc == 1 || chroma_format_idc == 2) ? 2 : 1; + uint32_t sub_height_c = (chroma_format_idc == 1) ? 2 : 1; + uint32_t crop_left = bs.read_ue() * sub_width_c; + uint32_t crop_right = bs.read_ue() * sub_width_c; + uint32_t crop_top = bs.read_ue() * sub_height_c; + uint32_t crop_bottom = bs.read_ue() * sub_height_c; + if (crop_left + crop_right > pic_width || crop_top + crop_bottom > pic_height) { + return false; + } + pic_width -= crop_left + crop_right; + pic_height -= crop_top + crop_bottom; + } + + width = (int)pic_width; + height = (int)pic_height; + + bs.read_ue(); // bit_depth_luma_minus8 + bs.read_ue(); // bit_depth_chroma_minus8 + uint32_t log2_max_pic_order_cnt_lsb_minus4 = bs.read_ue(); + + bool sps_sub_layer_ordering_info_present_flag = bs.read_bits(1) != 0; + uint32_t start = sps_sub_layer_ordering_info_present_flag ? 0 : sps_max_sub_layers_minus1; + for (uint32_t i = start; i <= sps_max_sub_layers_minus1; i++) { + bs.read_ue(); bs.read_ue(); bs.read_ue(); + } + bs.read_ue(); // log2_min_luma_coding_block_size_minus3 + bs.read_ue(); // log2_diff_max_min_luma_coding_block_size + bs.read_ue(); // log2_min_luma_transform_block_size_minus2 + bs.read_ue(); // log2_diff_max_min_luma_transform_block_size + bs.read_ue(); // max_transform_hierarchy_depth_inter + bs.read_ue(); // max_transform_hierarchy_depth_intra + + if (bs.read_bits(1)) { // scaling_list_enabled_flag + if (bs.read_bits(1)) { // sps_scaling_list_data_present_flag + for (int sizeId = 0; sizeId < 4; sizeId++) { + for (int matrixId = 0; matrixId < (sizeId == 3 ? 2 : 6); matrixId++) { + if (!bs.read_bits(1)) { // scaling_list_pred_mode_flag + bs.read_ue(); // scaling_list_pred_matrix_id_delta + } else { + int coefNum = (std::min)(64, 1 << (4 + (sizeId << 1))); + if (sizeId > 1) bs.read_se(); // scaling_list_dc_coef_minus8 + for (int i = 0; i < coefNum; i++) bs.read_se(); + } + } + } + } + } + + bs.skip_bits(2); // amp_enabled_flag + sample_adaptive_offset_enabled_flag + if (bs.read_bits(1)) { // pcm_enabled_flag + bs.skip_bits(4 + 4); // pcm_sample_bit_depth_luma/chroma_minus1 + bs.read_ue(); bs.read_ue(); // log2_min/max pcm_luma_coding_block_size + bs.skip_bits(1); // pcm_loop_filter_disabled_flag + } + + uint32_t num_short_term_ref_pic_sets = bs.read_ue(); + uint32_t prev_num_delta_pocs = 0; + for (uint32_t i = 0; i < num_short_term_ref_pic_sets; i++) { + bool inter_ref = (i != 0) && bs.read_bits(1) != 0; + if (inter_ref) { + bs.skip_bits(1); // delta_rps_sign + bs.read_ue(); // abs_delta_rps_minus1 + uint32_t n = prev_num_delta_pocs + 1; + uint32_t cnt = 0; + for (uint32_t j = 0; j < n; j++) { + bool used = bs.read_bits(1) != 0; + bool use = !used && bs.read_bits(1) != 0; + if (used || use) cnt++; + } + prev_num_delta_pocs = cnt; + } else { + uint32_t num_neg = bs.read_ue(); + uint32_t num_pos = bs.read_ue(); + prev_num_delta_pocs = num_neg + num_pos; + for (uint32_t j = 0; j < num_neg; j++) { bs.read_ue(); bs.skip_bits(1); } + for (uint32_t j = 0; j < num_pos; j++) { bs.read_ue(); bs.skip_bits(1); } + } + } + + if (bs.read_bits(1)) { // long_term_ref_pics_present_flag + uint32_t n = bs.read_ue(); + uint32_t log2_max = log2_max_pic_order_cnt_lsb_minus4 + 4; + for (uint32_t i = 0; i < n; i++) { + bs.skip_bits(log2_max); // lt_ref_pic_poc_lsb_sps + bs.skip_bits(1); // used_by_curr_pic_lt_sps_flag + } + } + + bs.skip_bits(2); // sps_temporal_mvp_enabled_flag + strong_intra_smoothing_enabled_flag + + if (bs.read_bits(1)) { // vui_parameters_present_flag + if (bs.read_bits(1)) { // aspect_ratio_info_present_flag + if (bs.read_bits(8) == 255) bs.skip_bits(32); + } + if (bs.read_bits(1)) bs.skip_bits(1); // overscan + if (bs.read_bits(1)) { // video_signal_type_present_flag + bs.skip_bits(3 + 1); + if (bs.read_bits(1)) bs.skip_bits(24); + } + if (bs.read_bits(1)) { bs.read_ue(); bs.read_ue(); } // chroma_loc_info + bs.skip_bits(3); // neutral_chroma/field_seq/frame_field_info + if (bs.read_bits(1)) { // default_display_window_flag + bs.read_ue(); // def_disp_win_left_offset + bs.read_ue(); // def_disp_win_right_offset + bs.read_ue(); // def_disp_win_top_offset + bs.read_ue(); // def_disp_win_bottom_offset + } + if (bs.read_bits(1)) { // vui_timing_info_present_flag + uint32_t num_units = bs.read_bits(32); + uint32_t time_scale = bs.read_bits(32); + if (num_units > 0 && fps <= 0.0f) { + fps = (float)time_scale / (float)num_units; + } + } + } + return width > 0 && height > 0; + } catch (...) { + return width > 0 && height > 0; + } +} + +bool getHEVCInfo(const char *vps, size_t vps_len, const char *sps, size_t sps_len, + int &iVideoWidth, int &iVideoHeight, float &iVideoFps) { + iVideoWidth = 0; iVideoHeight = 0; iVideoFps = 0.0f; + + // 先从 VPS 提取帧率 + if (vps_len > 2) { + parse_hevc_vps_fps((const uint8_t *)vps, vps_len, iVideoFps); } - if ( sps_len > 2 ){ - memset(&tGetBitBuf,0,sizeof(tGetBitBuf)); - memset(&tH265SpsInfo,0,sizeof(tH265SpsInfo)); - tGetBitBuf.pu8Buf = (uint8_t*)sps+2; - tGetBitBuf.iBufSize = (int)(sps_len-2); - if(0 != h265DecSeqParameterSet((void *) &tGetBitBuf, &tH265SpsInfo)){ - return false; - } - } - else - return false; - h265GetWidthHeight(&tH265SpsInfo, &iVideoWidth, &iVideoHeight); - iVideoFps = 0; - h265GeFramerate(&tH265VpsInfo, &tH265SpsInfo, &iVideoFps); - return true; + // 再从 SPS 提取宽高（如果 VPS 没有帧率，SPS VUI 里也可能有） + if (sps_len <= 2) return false; + return parse_hevc_sps((const uint8_t *)sps, sps_len, iVideoWidth, iVideoHeight, iVideoFps); } bool getHEVCInfo(const string &strVps, const string &strSps, int &iVideoWidth, int &iVideoHeight, float &iVideoFps) { diff --git a/ext-codec/SPSParser.c b/ext-codec/SPSParser.c deleted file mode 100644 index 0a33ae80..00000000 --- a/ext-codec/SPSParser.c +++ /dev/null @@ -1,2321 +0,0 @@ -#include -#include -#include -#include /* for uint32_t, etc */ -#include "SPSParser.h" - -/******************************************** -*define here -********************************************/ -#define SPS_PPS_DEBUG -//#undef SPS_PPS_DEBUG - - -#define MAX_LEN 32 -#define EXTENDED_SAR 255 -#define FF_ARRAY_ELEMS(a) (sizeof(a) / sizeof((a)[0])) -#define MAX_SPS_COUNT 32 -#define MAX_LOG2_MAX_FRAME_NUM (12 + 4) -#define MIN_LOG2_MAX_FRAME_NUM 4 -#define H264_MAX_PICTURE_COUNT 36 -#define CODEC_FLAG2_IGNORE_CROP 0x00010000 ///< Discard cropping information from SPS. -#ifndef INT_MAX -#define INT_MAX 65535 -#endif //INT_MAX - -#ifndef FFMIN -#define FFMIN(a,b) ((a) > (b) ? (b) : (a)) -#endif -#ifndef FFMAX -#define FFMAX(a,b) ((a) > (b) ? (a) : (b)) -#endif - -/* report level */ -#define RPT_ERR (1) // error, system error -#define RPT_WRN (2) // warning, maybe wrong, maybe OK -#define RPT_INF (3) // important information -#define RPT_DBG (4) // debug information - -static int rpt_lvl = RPT_WRN; /* report level: ERR, WRN, INF, DBG */ - -/* report micro */ -#define RPT(lvl, ...) \ - do { \ - if(lvl <= rpt_lvl) { \ - switch(lvl) { \ - case RPT_ERR: \ - fprintf(stderr, "\"%s\" line %d [err]: ", __FILE__, __LINE__); \ - break; \ - case RPT_WRN: \ - fprintf(stderr, "\"%s\" line %d [wrn]: ", __FILE__, __LINE__); \ - break; \ - case RPT_INF: \ - fprintf(stderr, "\"%s\" line %d [inf]: ", __FILE__, __LINE__); \ - break; \ - case RPT_DBG: \ - fprintf(stderr, "\"%s\" line %d [dbg]: ", __FILE__, __LINE__); \ - break; \ - default: \ - fprintf(stderr, "\"%s\" line %d [???]: ", __FILE__, __LINE__); \ - break; \ - } \ - fprintf(stderr, __VA_ARGS__); \ - fprintf(stderr, "\n"); \ - } \ - } while(0) - -static const uint8_t sg_aau8DefaultScaling4[2][16] = { - { 6, 13, 20, 28, 13, 20, 28, 32, - 20, 28, 32, 37, 28, 32, 37, 42 }, - { 10, 14, 20, 24, 14, 20, 24, 27, - 20, 24, 27, 30, 24, 27, 30, 34 } -}; - -static const uint8_t sg_aau8DefaultScaling8[2][64] = { - { 6, 10, 13, 16, 18, 23, 25, 27, - 10, 11, 16, 18, 23, 25, 27, 29, - 13, 16, 18, 23, 25, 27, 29, 31, - 16, 18, 23, 25, 27, 29, 31, 33, - 18, 23, 25, 27, 29, 31, 33, 36, - 23, 25, 27, 29, 31, 33, 36, 38, - 25, 27, 29, 31, 33, 36, 38, 40, - 27, 29, 31, 33, 36, 38, 40, 42 }, - { 9, 13, 15, 17, 19, 21, 22, 24, - 13, 13, 17, 19, 21, 22, 24, 25, - 15, 17, 19, 21, 22, 24, 25, 27, - 17, 19, 21, 22, 24, 25, 27, 28, - 19, 21, 22, 24, 25, 27, 28, 30, - 21, 22, 24, 25, 27, 28, 30, 32, - 22, 24, 25, 27, 28, 30, 32, 33, - 24, 25, 27, 28, 30, 32, 33, 35 } -}; - -static const T_AVRational sg_atFfH264PixelSspect[17] = { - { 0, 1 }, - { 1, 1 }, - { 12, 11 }, - { 10, 11 }, - { 16, 11 }, - { 40, 33 }, - { 24, 11 }, - { 20, 11 }, - { 32, 11 }, - { 80, 33 }, - { 18, 11 }, - { 15, 11 }, - { 64, 33 }, - { 160, 99 }, - { 4, 3 }, - { 3, 2 }, - { 2, 1 }, -}; - -static const uint8_t sg_au8ZigzagScan[16+1] = { - 0 + 0 * 4, 1 + 0 * 4, 0 + 1 * 4, 0 + 2 * 4, - 1 + 1 * 4, 2 + 0 * 4, 3 + 0 * 4, 2 + 1 * 4, - 1 + 2 * 4, 0 + 3 * 4, 1 + 3 * 4, 2 + 2 * 4, - 3 + 1 * 4, 3 + 2 * 4, 2 + 3 * 4, 3 + 3 * 4, -}; - -const uint8_t g_au8FfZigzagDirect[64] = { - 0, 1, 8, 16, 9, 2, 3, 10, - 17, 24, 32, 25, 18, 11, 4, 5, - 12, 19, 26, 33, 40, 48, 41, 34, - 27, 20, 13, 6, 7, 14, 21, 28, - 35, 42, 49, 56, 57, 50, 43, 36, - 29, 22, 15, 23, 30, 37, 44, 51, - 58, 59, 52, 45, 38, 31, 39, 46, - 53, 60, 61, 54, 47, 55, 62, 63 -}; - - -static const uint8_t sg_au8HevcSubWidthC[] = { - 1, 2, 2, 1 -}; - -static const uint8_t sg_au8HevcSubHeightC[] = { - 1, 2, 1, 1 -}; - -static const uint8_t sg_au8DefaultScalingListIntra[] = { - 16, 16, 16, 16, 17, 18, 21, 24, - 16, 16, 16, 16, 17, 19, 22, 25, - 16, 16, 17, 18, 20, 22, 25, 29, - 16, 16, 18, 21, 24, 27, 31, 36, - 17, 17, 20, 24, 30, 35, 41, 47, - 18, 19, 22, 27, 35, 44, 54, 65, - 21, 22, 25, 31, 41, 54, 70, 88, - 24, 25, 29, 36, 47, 65, 88, 115 -}; - -static const uint8_t sg_au8DefaultScalingListInter[] = { - 16, 16, 16, 16, 17, 18, 20, 24, - 16, 16, 16, 17, 18, 20, 24, 25, - 16, 16, 17, 18, 20, 24, 25, 28, - 16, 17, 18, 20, 24, 25, 28, 33, - 17, 18, 20, 24, 25, 28, 33, 41, - 18, 20, 24, 25, 28, 33, 41, 54, - 20, 24, 25, 28, 33, 41, 54, 71, - 24, 25, 28, 33, 41, 54, 71, 91 -}; - - -const uint8_t g_au8HevcDiagScan4x4X[16] = { - 0, 0, 1, 0, - 1, 2, 0, 1, - 2, 3, 1, 2, - 3, 2, 3, 3, -}; - -const uint8_t g_au8HevcDiagScan4x4Y[16] = { - 0, 1, 0, 2, - 1, 0, 3, 2, - 1, 0, 3, 2, - 1, 3, 2, 3, -}; - -const uint8_t g_au8HevcDiagScan8x8X[64] = { - 0, 0, 1, 0, - 1, 2, 0, 1, - 2, 3, 0, 1, - 2, 3, 4, 0, - 1, 2, 3, 4, - 5, 0, 1, 2, - 3, 4, 5, 6, - 0, 1, 2, 3, - 4, 5, 6, 7, - 1, 2, 3, 4, - 5, 6, 7, 2, - 3, 4, 5, 6, - 7, 3, 4, 5, - 6, 7, 4, 5, - 6, 7, 5, 6, - 7, 6, 7, 7, -}; - -const uint8_t g_au8HevcDiagScan8x8Y[64] = { - 0, 1, 0, 2, - 1, 0, 3, 2, - 1, 0, 4, 3, - 2, 1, 0, 5, - 4, 3, 2, 1, - 0, 6, 5, 4, - 3, 2, 1, 0, - 7, 6, 5, 4, - 3, 2, 1, 0, - 7, 6, 5, 4, - 3, 2, 1, 7, - 6, 5, 4, 3, - 2, 7, 6, 5, - 4, 3, 7, 6, - 5, 4, 7, 6, - 5, 7, 6, 7, -}; - -static const T_AVRational sg_atVuiSar[] = { - { 0, 1 }, - { 1, 1 }, - { 12, 11 }, - { 10, 11 }, - { 16, 11 }, - { 40, 33 }, - { 24, 11 }, - { 20, 11 }, - { 32, 11 }, - { 80, 33 }, - { 18, 11 }, - { 15, 11 }, - { 64, 33 }, - { 160, 99 }, - { 4, 3 }, - { 3, 2 }, - { 2, 1 }, -}; - -static inline int getBitsLeft(void *pvHandle) -{ - int iResLen = 0; - T_GetBitContext *ptPtr = (T_GetBitContext *)pvHandle; - if(ptPtr->iBufSize <= 0 || ptPtr->iTotalBit <= 0) - { - RPT(RPT_WRN, "buffer size is zero"); - return 0; - } - - - iResLen = ptPtr->iTotalBit - ptPtr->iBitPos; - return iResLen; -} - - -/******************************************** -*functions -********************************************/ -/** - * @brief Function getOneBit() get next bit - * @param[in] h T_GetBitContext structrue - * @retval other : success, -1 : failure - * @pre - * @post - */ -static int getOneBit(void *pvHandle) -{ - T_GetBitContext *ptPtr = (T_GetBitContext *)pvHandle; - int iRet = 0; - uint8_t *pu8CurChar = NULL; - uint8_t u8Shift; - int iResoLen = 0; - - if(NULL == ptPtr) - { - RPT(RPT_ERR, "NULL pointer"); - iRet = -1; - goto exit; - } - iResoLen = getBitsLeft(ptPtr); - if(iResoLen < 1) - { - iRet = -1; - goto exit; - } - - pu8CurChar = ptPtr->pu8Buf + (ptPtr->iBitPos >> 3); - u8Shift = 7 - (ptPtr->iCurBitPos); - ptPtr->iBitPos++; - ptPtr->iCurBitPos = ptPtr->iBitPos & 0x7; - iRet = ((*pu8CurChar) >> u8Shift) & 0x01; - -exit: - return iRet; -} - - -/** - * @brief Function getBits() get next bits - * @param[in] h T_GetBitContext structrue - * @param[in] n how many bits you want? - * @retval other : success, -1 : failure - * @pre - * @post - */ -static int getBits(void *pvHandle, int iN) -{ - T_GetBitContext *ptPtr = (T_GetBitContext *)pvHandle; - uint8_t au8Temp[5] = {0}; - uint8_t *pu8CurChar = NULL; - uint8_t u8Nbyte; - uint8_t u8Shift; - uint32_t u32Result = 0; - uint32_t iRet = 0; - int iResoLen = 0; - - if(NULL == ptPtr) - { - RPT(RPT_ERR, "NULL pointer"); - goto exit; - } - - if(iN > MAX_LEN) - { - iN = MAX_LEN; - } - - iResoLen = getBitsLeft(ptPtr); - if(iResoLen < iN) - { - goto exit; - } - - - if((ptPtr->iBitPos + iN) > ptPtr->iTotalBit) - { - iN = ptPtr->iTotalBit- ptPtr->iBitPos; - } - - pu8CurChar = ptPtr->pu8Buf+ (ptPtr->iBitPos>>3); - u8Nbyte = (ptPtr->iCurBitPos + iN + 7) >> 3; - u8Shift = (8 - (ptPtr->iCurBitPos + iN))& 0x07; - - if(iN == MAX_LEN) - { - RPT(RPT_DBG, "12(ptPtr->iBitPos(:%d) + iN(:%d)) > ptPtr->iTotalBit(:%d)!!! ",\ - ptPtr->iBitPos, iN, ptPtr->iTotalBit); - RPT(RPT_DBG, "0x%x 0x%x 0x%x 0x%x", (*pu8CurChar), *(pu8CurChar+1),*(pu8CurChar+2),*(pu8CurChar+3)); - } - - memcpy(&au8Temp[5-u8Nbyte], pu8CurChar, u8Nbyte); - iRet = (uint32_t)au8Temp[0] << 24; - iRet = iRet << 8; - iRet = ((uint32_t)au8Temp[1]<<24)|((uint32_t)au8Temp[2] << 16)\ - |((uint32_t)au8Temp[3] << 8)|au8Temp[4]; - - iRet = (iRet >> u8Shift) & (((uint64_t)1<iBitPos += iN; - ptPtr->iCurBitPos = ptPtr->iBitPos & 0x7; - -exit: - return u32Result; -} - - -/** - * Show 1-25 bits. - */ -static inline unsigned int showBits(void *pvHandle, int iN) -{ - T_GetBitContext *ptPtr = (T_GetBitContext *)pvHandle; - uint8_t au8Temp[5] = {0}; - uint8_t *pu8CurChar = NULL; - uint8_t u8Nbyte; - uint8_t u8Shift; - uint32_t u32Result = 0; - int iRet = 0; - int iResoLen = 0; - - if(NULL == ptPtr) - { - RPT(RPT_ERR, "NULL pointer"); - iRet = -1; - goto exit; - } - - if(iN > MAX_LEN) - { - iN = MAX_LEN; - } - - iResoLen = getBitsLeft(ptPtr); - if(iResoLen < iN) - { - iRet = -1; - goto exit; - } - - - if((ptPtr->iBitPos + iN) > ptPtr->iTotalBit) - { - iN = ptPtr->iTotalBit- ptPtr->iBitPos; - } - - pu8CurChar = ptPtr->pu8Buf+ (ptPtr->iBitPos>>3); - u8Nbyte = (ptPtr->iCurBitPos + iN + 7) >> 3; - u8Shift = (8 - (ptPtr->iCurBitPos + iN))& 0x07; - - if(iN == MAX_LEN) - { - RPT(RPT_DBG, "12(ptPtr->iBitPos(:%d) + iN(:%d)) > ptPtr->iTotalBit(:%d)!!! ",\ - ptPtr->iBitPos, iN, ptPtr->iTotalBit); - RPT(RPT_DBG, "0x%x 0x%x 0x%x 0x%x", (*pu8CurChar), *(pu8CurChar+1),*(pu8CurChar+2),*(pu8CurChar+3)); - } - - memcpy(&au8Temp[5-u8Nbyte], pu8CurChar, u8Nbyte); - iRet = (uint32_t)au8Temp[0] << 24; - iRet = iRet << 8; - iRet = ((uint32_t)au8Temp[1]<<24)|((uint32_t)au8Temp[2] << 16)\ - |((uint32_t)au8Temp[3] << 8)|au8Temp[4]; - - iRet = (iRet >> u8Shift) & (((uint64_t)1<iBitPos += iN; -// ptPtr->iCurBitPos = ptPtr->iBitPos & 0x7; - -exit: - return u32Result; -} - - - -/** - * Show 0-32 bits. - */ -static inline unsigned int showBitsLong(void *pvHandle, int iN) -{ - T_GetBitContext *ptPtr = (T_GetBitContext *)pvHandle; - - if (iN <= 32) { - return showBits(ptPtr, iN); - } - return 0; -} - - - -/** - * @brief Function parseCodenum() - * @param[in] buf - * @retval u32CodeNum - * @pre - * @post - */ -static int parseCodenum(void *pvBuf) -{ - uint8_t u8LeadingZeroBits = -1; - uint8_t u8B; - uint32_t u32CodeNum = 0; - - for(u8B=0; !u8B; u8LeadingZeroBits++) - { - u8B = getOneBit(pvBuf); - } - - u32CodeNum = ((uint32_t)1 << u8LeadingZeroBits) - 1 + getBits(pvBuf, u8LeadingZeroBits); - - return u32CodeNum; -} - -/** - * @brief Function parseUe() - * @param[in] buf sps_pps parse buf - * @retval u32CodeNum - * @pre - * @post - */ -static int parseUe(void *pvBuf) -{ - return parseCodenum(pvBuf); -} - - -/** - * @brief Function parseSe() - * @param[in] buf sps_pps parse buf - * @retval u32CodeNum - * @pre - * @post - */ -static int parseSe(void *pvBuf) -{ - int iRet = 0; - int u32CodeNum; - - u32CodeNum = parseCodenum(pvBuf); - iRet = (u32CodeNum + 1) >> 1; - iRet = (u32CodeNum & 0x01)? iRet : -iRet; - - return iRet; -} - - -/** - * @brief Function getBitContextFree() - * @param[in] buf T_GetBitContext buf - * @retval none - * @pre - * @post - */ -static void getBitContextFree(void *pvBuf) -{ - T_GetBitContext *ptPtr = (T_GetBitContext *)pvBuf; - - if(ptPtr) - { - if(ptPtr->pu8Buf) - { - free(ptPtr->pu8Buf); - } - - free(ptPtr); - } -} - - - - -/** - * @brief Function deEmulationPrevention() - * @param[in] buf T_GetBitContext buf - * @retval none - * @pre - * @post - * @note: - * http://www.cnblogs.com/eustoma/archive/2012/02/13/2415764.html - * 0x000000 >>>>>> 0x00000300 - * 0x000001 >>>>>> 0x00000301 - * 0x000002 >>>>>> 0x00000302 - * 0x000003 >>>>>> 0x00000303 - */ -static void *deEmulationPrevention(void *pvBuf) -{ - T_GetBitContext *ptPtr = NULL; - T_GetBitContext *ptBufPtr = (T_GetBitContext *)pvBuf; - int i = 0, j = 0; - uint8_t *pu8TmpPtr = NULL; - int tmp_buf_size = 0; - int iVal = 0; - - if(NULL == ptBufPtr) - { - RPT(RPT_ERR, "NULL ptPtr"); - goto exit; - } - - ptPtr = (T_GetBitContext *)malloc(sizeof(T_GetBitContext)); - if(NULL == ptPtr) - { - RPT(RPT_ERR, "NULL ptPtr"); - goto exit; - } - - memcpy(ptPtr, ptBufPtr, sizeof(T_GetBitContext)); - - ptPtr->pu8Buf = (uint8_t *)malloc(ptPtr->iBufSize); - if(NULL == ptPtr->pu8Buf) - { - RPT(RPT_ERR, "NULL ptPtr"); - goto exit; - } - - memcpy(ptPtr->pu8Buf, ptBufPtr->pu8Buf, ptBufPtr->iBufSize); - - pu8TmpPtr = ptPtr->pu8Buf; - tmp_buf_size = ptPtr->iBufSize; - for(i=0; i<(tmp_buf_size-2); i++) - { - - iVal = (pu8TmpPtr[i]^0x00) + (pu8TmpPtr[i+1]^0x00) + (pu8TmpPtr[i+2]^0x03); - if(iVal == 0) - { - - for(j=i+2; jiBufSize--; - } - } - ptPtr->iTotalBit = ptPtr->iBufSize << 3; - - return (void *)ptPtr; - -exit: - getBitContextFree(ptPtr); - return NULL; -} - -static void decodeScalingList(void *pvBuf, uint8_t *pu8Factors, int iSize, - const uint8_t *pu8JvtList, - const uint8_t *pu8FallbackList) -{ - int i; - int iLast = 8; - int iNext = 8; - const uint8_t *pu8Scan = iSize == 16 ? sg_au8ZigzagScan : g_au8FfZigzagDirect; - - if (!getOneBit(pvBuf)) /* matrix not written, we use the predicted one */ - memcpy(pu8Factors, pu8FallbackList, iSize * sizeof(uint8_t)); - else - for (i = 0; i < iSize; i++) - { - if (iNext) - { - - iNext = (iLast + parseSe(pvBuf)) & 0xff; - } - if (!i && !iNext) - { - /* matrix not written, we use the preset one */ - memcpy(pu8Factors, pu8JvtList, iSize * sizeof(uint8_t)); - break; - } - iLast = pu8Factors[pu8Scan[i]] = iNext ? iNext : iLast; - } -} - - - int decodeScalingMatrices(void *pvBuf, T_SPS *ptSps, - T_PPS *ptPps, int iIsSps, - uint8_t(*pau8ScalingMatrix4)[16], - uint8_t(*pau8ScalingMatrix8)[64]) -{ - int iFallbackSps = !iIsSps && ptSps->iScalingMatrixPresent; - const uint8_t *pau8Fallback[4] = { - iFallbackSps ? ptSps->aau8ScalingMatrix4[0] : sg_aau8DefaultScaling4[0], - iFallbackSps ? ptSps->aau8ScalingMatrix4[3] : sg_aau8DefaultScaling4[1], - iFallbackSps ? ptSps->aau8ScalingMatrix8[0] : sg_aau8DefaultScaling8[0], - iFallbackSps ? ptSps->aau8ScalingMatrix8[3] : sg_aau8DefaultScaling8[1] - }; - if (getOneBit(pvBuf)) { - ptSps->iScalingMatrixPresent |= iIsSps; - decodeScalingList(pvBuf, pau8ScalingMatrix4[0], 16, sg_aau8DefaultScaling4[0], pau8Fallback[0]); // Intra, Y - decodeScalingList(pvBuf, pau8ScalingMatrix4[1], 16, sg_aau8DefaultScaling4[0], pau8ScalingMatrix4[0]); // Intra, Cr - decodeScalingList(pvBuf, pau8ScalingMatrix4[2], 16, sg_aau8DefaultScaling4[0], pau8ScalingMatrix4[1]); // Intra, Cb - - decodeScalingList(pvBuf, pau8ScalingMatrix4[3], 16, sg_aau8DefaultScaling4[1], pau8Fallback[1]); // Inter, Y - - decodeScalingList(pvBuf, pau8ScalingMatrix4[4], 16, sg_aau8DefaultScaling4[1], pau8ScalingMatrix4[3]); // Inter, Cr - - decodeScalingList(pvBuf, pau8ScalingMatrix4[5], 16, sg_aau8DefaultScaling4[1], pau8ScalingMatrix4[4]); // Inter, Cb - - - if (iIsSps || ptPps->iTransform8x8Mode) - { - decodeScalingList(pvBuf, pau8ScalingMatrix8[0], 64, sg_aau8DefaultScaling8[0], pau8Fallback[2]); // Intra, Y - decodeScalingList(pvBuf, pau8ScalingMatrix8[3], 64, sg_aau8DefaultScaling8[1], pau8Fallback[3]); // Inter, Y - if (ptSps->iChromaFormatIdc == 3) { - decodeScalingList(pvBuf, pau8ScalingMatrix8[1], 64, sg_aau8DefaultScaling8[0], pau8ScalingMatrix8[0]); // Intra, Cr - decodeScalingList(pvBuf, pau8ScalingMatrix8[4], 64, sg_aau8DefaultScaling8[1], pau8ScalingMatrix8[3]); // Inter, Cr - decodeScalingList(pvBuf, pau8ScalingMatrix8[2], 64, sg_aau8DefaultScaling8[0], pau8ScalingMatrix8[1]); // Intra, Cb - decodeScalingList(pvBuf, pau8ScalingMatrix8[5], 64, sg_aau8DefaultScaling8[1], pau8ScalingMatrix8[4]); // Inter, Cb - } - } - } - return 0; -} - -static int decodeHrdPAarameters(void *pvBuf, T_SPS *ptSps) -{ - int iCpbCount = 0; - int i; - iCpbCount = parseUe(pvBuf); - - if (iCpbCount > 32U) - { - RPT(RPT_ERR,"iCpbCount %d invalid\n", iCpbCount); - return -1; - - } - - getBits(pvBuf, 4); /* bit_rate_scale */ - getBits(pvBuf, 4); /* cpb_size_scale */ - for (i = 0; i < iCpbCount; i++) - { - parseUe(pvBuf); - parseUe(pvBuf); - //get_ue_golomb_long(&h->gb); /* bit_rate_value_minus1 */ - //get_ue_golomb_long(&h->gb); /* cpb_size_value_minus1 */ - getOneBit(pvBuf); /* cbr_flag */ - } - ptSps->iInitialCpbRemovalDelayLength = getBits(pvBuf, 5) + 1; - ptSps->iCpbRemovalDelayLength = getBits(pvBuf, 5) + 1; - ptSps->iDpbOutputDelayLength = getBits(pvBuf, 5) + 1; - ptSps->iTimeOffsetLength = getBits(pvBuf, 5); - ptSps->iCpbCnt = iCpbCount; - return 0; -} - - -static inline int decodeVuiParameters(void *pvBuf, T_SPS *ptSps) -{ - int iAspectRatioInfoPresentFlag; - unsigned int uiAspectRatioIdc; - - iAspectRatioInfoPresentFlag = getOneBit(pvBuf); - - if (iAspectRatioInfoPresentFlag) { - uiAspectRatioIdc = getBits(pvBuf, 8); - if (uiAspectRatioIdc == EXTENDED_SAR) { - ptSps->tSar.num = getBits(pvBuf, 16); - ptSps->tSar.den = getBits(pvBuf, 16); - } else if (uiAspectRatioIdc < FF_ARRAY_ELEMS(sg_atFfH264PixelSspect)) { - ptSps->tSar = sg_atFfH264PixelSspect[uiAspectRatioIdc]; - } else - { - RPT(RPT_ERR,"illegal aspect ratio\n"); - return -1; - } - } else - { - ptSps->tSar.num = - ptSps->tSar.den = 0; - } - - if (getOneBit(pvBuf)) /* iOverscanInfoPresentFlag */ - getOneBit(pvBuf); /* iOverscanAppropriateFlag */ - - ptSps->iVideoSignalTypePresentFlag = getOneBit(pvBuf); - if (ptSps->iVideoSignalTypePresentFlag) { - getBits(pvBuf, 3); /* video_format */ - ptSps->iFullRange = getOneBit(pvBuf); /* iVideoFullRangeFlag */ - - ptSps->iColourDescriptionPresentFlag = getOneBit(pvBuf); - if (ptSps->iColourDescriptionPresentFlag) { - ptSps->tColorPrimaries = getBits(pvBuf, 8); /* u8ColourPrimaries */ - ptSps->tColorTrc = getBits(pvBuf, 8); /* transfer_characteristics */ - ptSps->tColorspace = getBits(pvBuf, 8); /* matrix_coefficients */ - if (ptSps->tColorPrimaries >= AVCOL_PRI_NB) - ptSps->tColorPrimaries = AVCOL_PRI_UNSPECIFIED; - if (ptSps->tColorTrc >= AVCOL_TRC_NB) - ptSps->tColorTrc = AVCOL_TRC_UNSPECIFIED; - if (ptSps->tColorspace >= AVCOL_SPC_NB) - ptSps->tColorspace = AVCOL_SPC_UNSPECIFIED; - } - } - - /* chroma_location_info_present_flag */ - if (getOneBit(pvBuf)) - { - parseUe(pvBuf); /* chroma_sample_location_type_top_field */ - parseUe(pvBuf); /* chroma_sample_location_type_bottom_field */ - } - if(getBitsLeft(pvBuf) < 10) - { - return 0; - } - - - ptSps->iTimingInfoPresentFlag = getOneBit(pvBuf); - if (ptSps->iTimingInfoPresentFlag) { - unsigned u32NumUnitsInTick = getBits(pvBuf, 32); - unsigned u32TimeScale = getBits(pvBuf, 32); - if (!u32NumUnitsInTick || !u32TimeScale) { - - RPT(RPT_ERR,"u32TimeScale/u32NumUnitsInTick invalid or unsupported (%u/%u)\n",u32TimeScale, u32NumUnitsInTick); - ptSps->iTimingInfoPresentFlag = 0; - } else { - ptSps->u32NumUnitsInTick = u32NumUnitsInTick; - ptSps->u32TimeScale = u32TimeScale; - } - ptSps->iFixedFrameRateFlag = getOneBit(pvBuf); - } - - ptSps->iNalHrdParametersPresentFlag = getOneBit(pvBuf); - if (ptSps->iNalHrdParametersPresentFlag) - if (decodeHrdPAarameters(pvBuf, ptSps) < 0) - return -1; - ptSps->iVclHrdParametersPresentFlag = getOneBit(pvBuf); - if (ptSps->iVclHrdParametersPresentFlag) - if (decodeHrdPAarameters(pvBuf, ptSps) < 0) - return -1; - if (ptSps->iNalHrdParametersPresentFlag || - ptSps->iVclHrdParametersPresentFlag) - getOneBit(pvBuf); /* low_delay_hrd_flag */ - ptSps->iPicStructPresentFlag = getOneBit(pvBuf); - - if(getBitsLeft(pvBuf) == 0) - return 0; - ptSps->iBitstreamRestrictionFlag = getOneBit(pvBuf); - if (ptSps->iBitstreamRestrictionFlag) { - getOneBit(pvBuf); /* motion_vectors_over_pic_boundaries_flag */ - parseUe(pvBuf); - //get_ue_golomb(&h->gb); /* max_bytes_per_pic_denom */ - parseUe(pvBuf); - //get_ue_golomb(&h->gb); /* max_bits_per_mb_denom */ - parseUe(pvBuf); - //get_ue_golomb(&h->gb); /* log2_max_mv_length_horizontal */ - parseUe(pvBuf); - //get_ue_golomb(&h->gb); /* log2_max_mv_length_vertical */ - ptSps->iNumReorderFrames = parseUe(pvBuf); - - parseUe(pvBuf); - //get_ue_golomb(&h->gb); /*max_dec_frame_buffering*/ - - if (getBitsLeft(pvBuf) < 0) - { - ptSps->iNumReorderFrames = 0; - ptSps->iBitstreamRestrictionFlag = 0; - } - - if (ptSps->iNumReorderFrames > 16U - /* max_dec_frame_buffering || max_dec_frame_buffering > 16 */) - { - RPT(RPT_DBG, "Clipping illegal iNumReorderFrames %d\n", - ptSps->iNumReorderFrames); - ptSps->iNumReorderFrames = 16; - return -1; - } - } - - return 0; -} - - -int h264DecSeqParameterSet(void *pvBufSrc, T_SPS *ptSps) -{ - int iLevelIdc; - int iConstraintSetFlags = 0; - unsigned int uiSpsId; - int i; - int iLog2MaxFrameNumMinus4; - - int iRet = 0; - int iProfileIdc = 0; - void *pvBuf = NULL; - - - - if(NULL == pvBufSrc || NULL == ptSps) - { - RPT(RPT_ERR,"ERR null pointer\n"); - iRet = -1; - goto exit; - } - - memset((void *)ptSps, 0, sizeof(T_SPS)); - - pvBuf = deEmulationPrevention(pvBufSrc); - if(NULL == pvBuf) - { - RPT(RPT_ERR,"ERR null pointer\n"); - iRet = -1; - goto exit; - } - - iProfileIdc = getBits(pvBuf, 8); - iConstraintSetFlags |= getOneBit(pvBuf) << 0; // constraint_set0_flag - iConstraintSetFlags |= getOneBit(pvBuf) << 1; // constraint_set1_flag - iConstraintSetFlags |= getOneBit(pvBuf) << 2; // constraint_set2_flag - iConstraintSetFlags |= getOneBit(pvBuf) << 3; // constraint_set3_flag - iConstraintSetFlags |= getOneBit(pvBuf) << 4; // constraint_set4_flag - iConstraintSetFlags |= getOneBit(pvBuf) << 5; // constraint_set5_flag - getBits(pvBuf, 2); // reserved_zero_2bits - iLevelIdc = getBits(pvBuf, 8); - uiSpsId = parseUe(pvBuf); - - if (uiSpsId >= MAX_SPS_COUNT) { - RPT(RPT_ERR, "uiSpsId %u out of range\n", uiSpsId); - iRet = -1; - goto exit; - - } - - - ptSps->uiSpsId = uiSpsId; - ptSps->iTimeOffsetLength = 24; - ptSps->iProfileIdc = iProfileIdc; - ptSps->iConstraintSetFlags = iConstraintSetFlags; - ptSps->iLevelIdc = iLevelIdc; - ptSps->iFullRange = -1; - - memset(ptSps->aau8ScalingMatrix4, 16, sizeof(ptSps->aau8ScalingMatrix4)); - memset(ptSps->aau8ScalingMatrix8, 16, sizeof(ptSps->aau8ScalingMatrix8)); - ptSps->iScalingMatrixPresent = 0; - ptSps->tColorspace = 2; //AVCOL_SPC_UNSPECIFIED - - if (ptSps->iProfileIdc == 100 || // High profile - ptSps->iProfileIdc == 110 || // High10 profile - ptSps->iProfileIdc == 122 || // High422 profile - ptSps->iProfileIdc == 244 || // High444 Predictive profile - ptSps->iProfileIdc == 44 || // Cavlc444 profile - ptSps->iProfileIdc == 83 || // Scalable Constrained High profile (SVC) - ptSps->iProfileIdc == 86 || // Scalable High Intra profile (SVC) - ptSps->iProfileIdc == 118 || // Stereo High profile (MVC) - ptSps->iProfileIdc == 128 || // Multiview High profile (MVC) - ptSps->iProfileIdc == 138 || // Multiview Depth High profile (MVCD) - ptSps->iProfileIdc == 144) { // old High444 profile - ptSps->iChromaFormatIdc = parseUe(pvBuf); - if (ptSps->iChromaFormatIdc > 3U) - { - RPT(RPT_ERR, "iChromaFormatIdc %u",ptSps->iChromaFormatIdc); - iRet = -1; - goto exit; - } - else if (ptSps->iChromaFormatIdc == 3) - { - ptSps->iResidualColorTransformFlag = getOneBit(pvBuf); - if (ptSps->iResidualColorTransformFlag) - { - RPT(RPT_ERR, "separate color planes are not supported\n"); - iRet = -1; - goto exit; - - } - } - ptSps->iBitDepthLuma = parseUe(pvBuf) + 8; - ptSps->iBitDepthChroma = parseUe(pvBuf) + 8; - if (ptSps->iBitDepthChroma != ptSps->iBitDepthLuma) - { - RPT(RPT_ERR, "Different chroma and luma bit depth"); - iRet = -1; - goto exit; - - } - if (ptSps->iBitDepthLuma < 8 || ptSps->iBitDepthLuma > 14 || - ptSps->iBitDepthChroma < 8 || ptSps->iBitDepthChroma > 14) - { - RPT(RPT_ERR, "illegal bit depth value (%d, %d)\n",ptSps->iBitDepthLuma, ptSps->iBitDepthChroma); - iRet = -1; - goto exit; - } - ptSps->iTransformBypass = getOneBit(pvBuf); - decodeScalingMatrices(pvBuf, ptSps, NULL, 1, - ptSps->aau8ScalingMatrix4, ptSps->aau8ScalingMatrix8); - } - else - { - - ptSps->iChromaFormatIdc = 1; - ptSps->iBitDepthLuma = 8; - ptSps->iBitDepthChroma = 8; - } - - iLog2MaxFrameNumMinus4 = parseUe(pvBuf); - - if (iLog2MaxFrameNumMinus4 < MIN_LOG2_MAX_FRAME_NUM - 4 || - iLog2MaxFrameNumMinus4 > MAX_LOG2_MAX_FRAME_NUM - 4) - { - RPT(RPT_ERR, "iLog2MaxFrameNumMinus4 out of range (0-12): %d\n", iLog2MaxFrameNumMinus4); - iRet = -1; - goto exit; - - } - ptSps->iLog2MaxFrameNum = iLog2MaxFrameNumMinus4 + 4; - - ptSps->iPocType = parseUe(pvBuf); - - if (ptSps->iPocType == 0) - { - // FIXME #define - unsigned t = parseUe(pvBuf); - if (t>12) - { - RPT(RPT_ERR, "iLog2MaxPocLsb (%d) is out of range\n", t); - iRet = -1; - goto exit; - - } - ptSps->iLog2MaxPocLsb = t + 4; - } - else if (ptSps->iPocType == 1) - { - // FIXME #define - ptSps->iDeltaPicOrderAlwaysZeroFlag = getOneBit(pvBuf); - ptSps->iOffsetForNonRefPic = parseSe(pvBuf); - ptSps->iOffsetForTopToBottomField = parseSe(pvBuf); - ptSps->iPocCycleLength = parseUe(pvBuf); - - if ((unsigned)ptSps->iPocCycleLength >= FF_ARRAY_ELEMS(ptSps->asOffsetForRefFrame)) - { - RPT(RPT_ERR, "iPocCycleLength overflow %d\n", ptSps->iPocCycleLength); - iRet = -1; - goto exit; - - } - - for (i = 0; i < ptSps->iPocCycleLength; i++) - ptSps->asOffsetForRefFrame[i] = parseSe(pvBuf); - } - else if (ptSps->iPocType != 2) - { - RPT(RPT_ERR, "illegal POC type %d\n", ptSps->iPocType); - iRet = -1; - goto exit; - - } - - ptSps->iRefFrameCount = parseUe(pvBuf); - if (ptSps->iRefFrameCount > H264_MAX_PICTURE_COUNT - 2 || - ptSps->iRefFrameCount > 16U) - { - RPT(RPT_ERR, "too many reference frames %d\n", ptSps->iRefFrameCount); - iRet = -1; - goto exit; - - } - ptSps->iGapsInFrameNumAllowedFlag = getOneBit(pvBuf); - ptSps->iMbWidth = parseUe(pvBuf) + 1; - ptSps->iMbHeight = parseUe(pvBuf) + 1; - - - ptSps->iFrameMbsOnlyFlag = getOneBit(pvBuf); - if (!ptSps->iFrameMbsOnlyFlag) - ptSps->iMbAff = getOneBit(pvBuf); - else - ptSps->iMbAff = 0; - - ptSps->iDirect8x8InferenceFlag = getOneBit(pvBuf); - - ptSps->iCrop = getOneBit(pvBuf); - if (ptSps->iCrop) { - unsigned int uiCropLeft = parseUe(pvBuf); - unsigned int uiCropRight = parseUe(pvBuf); - unsigned int uiCropTop = parseUe(pvBuf); - unsigned int uiCropBottom = parseUe(pvBuf); - int iWidth = 16 * ptSps->iMbWidth; - int iHeight = 16 * ptSps->iMbHeight * (2 - ptSps->iFrameMbsOnlyFlag); - - if(1) - { - int vsub = (ptSps->iChromaFormatIdc == 1) ? 1 : 0; - int hsub = (ptSps->iChromaFormatIdc == 1 || - ptSps->iChromaFormatIdc == 2) ? 1 : 0; - int step_x = 1 << hsub; - int step_y = (2 - ptSps->iFrameMbsOnlyFlag) << vsub; - - if (uiCropLeft & (0x1F >> (ptSps->iBitDepthLuma > 8))) - { - uiCropLeft &= ~(0x1F >> (ptSps->iBitDepthLuma > 8)); - } - - if (uiCropLeft > (unsigned)INT_MAX / 4 / step_x || - uiCropRight > (unsigned)INT_MAX / 4 / step_x || - uiCropTop > (unsigned)INT_MAX / 4 / step_y || - uiCropBottom> (unsigned)INT_MAX / 4 / step_y || - (uiCropLeft + uiCropRight ) * step_x >= iWidth || - (uiCropTop + uiCropBottom) * step_y >= iHeight - ) - { - RPT(RPT_ERR, "crop values invalid %d %d %d %d / %d %d\n", uiCropLeft, uiCropRight, uiCropTop, uiCropBottom, iWidth, iHeight); - iRet = -1; - goto exit; - } - - ptSps->uiCropLeft = uiCropLeft * step_x; - ptSps->uiCropRight = uiCropRight * step_x; - ptSps->uiCropTop = uiCropTop * step_y; - ptSps->uiCropBottom = uiCropBottom * step_y; - } - } - else - { - ptSps->uiCropLeft = - ptSps->uiCropRight = - ptSps->uiCropTop = - ptSps->uiCropBottom = - ptSps->iCrop = 0; - } - - ptSps->iVuiParametersPresentFlag = getOneBit(pvBuf); - if (ptSps->iVuiParametersPresentFlag) { - int ret = decodeVuiParameters(pvBuf, ptSps); - if (ret < 0) - goto exit; - } - - if (getBitsLeft(pvBuf) < 0) - { - RPT(RPT_ERR, "Overread %s by %d bits\n", ptSps->iVuiParametersPresentFlag ? "VUI" : "SPS", -getBitsLeft(pvBuf)); - iRet = -1; - } - - if (!ptSps->tSar.den) - ptSps->tSar.den = 1; - - ptSps->iNew = 1; - -exit: -#ifdef SPS_PPS_DEBUG - - if (1) - { - static const char csp[4][5] = { "Gray", "420", "422", "444" }; - RPT(RPT_DBG, - "ptSps:%u profile:%d/%d poc:%d ref:%d %dx%d %s %s crop:%u/%u/%u/%u %s %s %d/%d b%d reo:%d\n", - uiSpsId, ptSps->iProfileIdc, ptSps->iLevelIdc, - ptSps->iPocType, - ptSps->iRefFrameCount, - ptSps->iMbWidth, ptSps->iMbHeight, - ptSps->iFrameMbsOnlyFlag ? "FRM" : (ptSps->iMbAff ? "MB-AFF" : "PIC-AFF"), - ptSps->iDirect8x8InferenceFlag ? "8B8" : "", - ptSps->uiCropLeft, ptSps->uiCropRight, - ptSps->uiCropTop, ptSps->uiCropBottom, - ptSps->iVuiParametersPresentFlag ? "VUI" : "", - csp[ptSps->iChromaFormatIdc], - ptSps->iTimingInfoPresentFlag ? ptSps->u32NumUnitsInTick : 0, - ptSps->iTimingInfoPresentFlag ? ptSps->u32TimeScale : 0, - ptSps->iBitDepthLuma, - ptSps->iBitstreamRestrictionFlag ? ptSps->iNumReorderFrames : -1 - ); - } - -#endif - getBitContextFree(pvBuf); - - return iRet; -} - - -static int decodeProfileTierLevel(T_GetBitContext *pvBuf, T_PTLCommon *tPtl) -{ - int i; - - if (getBitsLeft(pvBuf) < 2+1+5 + 32 + 4 + 16 + 16 + 12) - return -1; - - tPtl->u8ProfileSpace = getBits(pvBuf, 2); - tPtl->u8TierFlag = getOneBit(pvBuf); - tPtl->u8ProfileIdc = getBits(pvBuf, 5); - if (tPtl->u8ProfileIdc == T_PROFILE_HEVC_MAIN) - RPT(RPT_DBG, "Main profile bitstream\n"); - else if (tPtl->u8ProfileIdc == T_PROFILE_HEVC_MAIN_10) - RPT(RPT_DBG, "Main 10 profile bitstream\n"); - else if (tPtl->u8ProfileIdc == T_PROFILE_HEVC_MAIN_STILL_PICTURE) - RPT(RPT_DBG, "Main Still Picture profile bitstream\n"); - else if (tPtl->u8ProfileIdc == T_PROFILE_HEVC_REXT) - RPT(RPT_DBG, "Range Extension profile bitstream\n"); - else - RPT(RPT_WRN, "Unknown HEVC profile: %d\n", tPtl->u8ProfileIdc); - - for (i = 0; i < 32; i++) { - tPtl->au8ProfileCompatibilityFlag[i] = getOneBit(pvBuf); - - if (tPtl->u8ProfileIdc == 0 && i > 0 && tPtl->au8ProfileCompatibilityFlag[i]) - tPtl->u8ProfileIdc = i; - } - tPtl->u8ProgressiveSourceFlag = getOneBit(pvBuf); - tPtl->u8InterlacedSourceFlag = getOneBit(pvBuf); - tPtl->u8NonPackedConstraintFlag = getOneBit(pvBuf); - tPtl->u8FrameOnlyConstraintFlag = getOneBit(pvBuf); - - getBits(pvBuf, 16); // XXX_reserved_zero_44bits[0..15] - getBits(pvBuf, 16); // XXX_reserved_zero_44bits[16..31] - getBits(pvBuf, 12); // XXX_reserved_zero_44bits[32..43] - - return 0; -} - - -static int parsePtl(T_GetBitContext *pvBuf, T_PTL *tPtl, int max_num_sub_layers) -{ - int i; - if (decodeProfileTierLevel(pvBuf, &tPtl->tGeneralPtl) < 0 || - getBitsLeft(pvBuf) < 8 + (8*2 * (max_num_sub_layers - 1 > 0))) { - RPT(RPT_ERR, "PTL information too short\n"); - return -1; - } - - tPtl->tGeneralPtl.u8LevelIdc = getBits(pvBuf, 8); - - for (i = 0; i < max_num_sub_layers - 1; i++) { - tPtl->au8SubLayerProfilePresentFlag[i] = getOneBit(pvBuf); - tPtl->au8SubLayerLevelPresentFlag[i] = getOneBit(pvBuf); - } - - if (max_num_sub_layers - 1> 0) - for (i = max_num_sub_layers - 1; i < 8; i++) - getBits(pvBuf, 2); // reserved_zero_2bits[i] - for (i = 0; i < max_num_sub_layers - 1; i++) { - if (tPtl->au8SubLayerProfilePresentFlag[i] && - decodeProfileTierLevel(pvBuf, &tPtl->atSubLayerPtl[i]) < 0) { - RPT(RPT_ERR, - "PTL information for sublayer %i too short\n", i); - return -1; - } - if (tPtl->au8SubLayerLevelPresentFlag[i]) { - if (getBitsLeft(pvBuf) < 8) { - RPT(RPT_ERR, - "Not enough data for sublayer %i level_idc\n", i); - return -1; - } else - tPtl->atSubLayerPtl[i].u8LevelIdc = getBits(pvBuf, 8); - } - } - - return 0; -} - - -static void setDefaultScalingListData(T_ScalingList *sl) -{ - int matrixId; - - for (matrixId = 0; matrixId < 6; matrixId++) { - // 4x4 default is 16 - memset(sl->aaau8Sl[0][matrixId], 16, 16); - sl->aau8SlDc[0][matrixId] = 16; // default for 16x16 - sl->aau8SlDc[1][matrixId] = 16; // default for 32x32 - } - memcpy(sl->aaau8Sl[1][0], sg_au8DefaultScalingListIntra, 64); - memcpy(sl->aaau8Sl[1][1], sg_au8DefaultScalingListIntra, 64); - memcpy(sl->aaau8Sl[1][2], sg_au8DefaultScalingListIntra, 64); - memcpy(sl->aaau8Sl[1][3], sg_au8DefaultScalingListInter, 64); - memcpy(sl->aaau8Sl[1][4], sg_au8DefaultScalingListInter, 64); - memcpy(sl->aaau8Sl[1][5], sg_au8DefaultScalingListInter, 64); - memcpy(sl->aaau8Sl[2][0], sg_au8DefaultScalingListIntra, 64); - memcpy(sl->aaau8Sl[2][1], sg_au8DefaultScalingListIntra, 64); - memcpy(sl->aaau8Sl[2][2], sg_au8DefaultScalingListIntra, 64); - memcpy(sl->aaau8Sl[2][3], sg_au8DefaultScalingListInter, 64); - memcpy(sl->aaau8Sl[2][4], sg_au8DefaultScalingListInter, 64); - memcpy(sl->aaau8Sl[2][5], sg_au8DefaultScalingListInter, 64); - memcpy(sl->aaau8Sl[3][0], sg_au8DefaultScalingListIntra, 64); - memcpy(sl->aaau8Sl[3][1], sg_au8DefaultScalingListIntra, 64); - memcpy(sl->aaau8Sl[3][2], sg_au8DefaultScalingListIntra, 64); - memcpy(sl->aaau8Sl[3][3], sg_au8DefaultScalingListInter, 64); - memcpy(sl->aaau8Sl[3][4], sg_au8DefaultScalingListInter, 64); - memcpy(sl->aaau8Sl[3][5], sg_au8DefaultScalingListInter, 64); -} - -static int scalingListData(T_GetBitContext *pvBuf, T_ScalingList *sl, T_HEVCSPS *ptSps) -{ - uint8_t scaling_list_pred_mode_flag; - int32_t scaling_list_dc_coef[2][6]; - int size_id, matrix_id, pos; - int i; - - for (size_id = 0; size_id < 4; size_id++) - for (matrix_id = 0; matrix_id < 6; matrix_id += ((size_id == 3) ? 3 : 1)) { - scaling_list_pred_mode_flag = getOneBit(pvBuf); - if (!scaling_list_pred_mode_flag) { - unsigned int delta = parseUe(pvBuf); - /* Only need to handle non-zero delta. Zero means default, - * which should already be in the arrays. */ - if (delta) { - // Copy from previous array. - delta *= (size_id == 3) ? 3 : 1; - if (matrix_id < delta) { - RPT(RPT_ERR, - "Invalid delta in scaling list data: %d.\n", delta); - return -1; - } - - memcpy(sl->aaau8Sl[size_id][matrix_id], - sl->aaau8Sl[size_id][matrix_id - delta], - size_id > 0 ? 64 : 16); - if (size_id > 1) - sl->aau8SlDc[size_id - 2][matrix_id] = sl->aau8SlDc[size_id - 2][matrix_id - delta]; - } - } else { - int next_coef, coef_num; - int32_t scaling_list_delta_coef; - - next_coef = 8; - coef_num = FFMIN(64, 1 << (4 + (size_id << 1))); - if (size_id > 1) { - scaling_list_dc_coef[size_id - 2][matrix_id] = parseSe(pvBuf) + 8; - next_coef = scaling_list_dc_coef[size_id - 2][matrix_id]; - sl->aau8SlDc[size_id - 2][matrix_id] = next_coef; - } - for (i = 0; i < coef_num; i++) { - if (size_id == 0) - pos = 4 * g_au8HevcDiagScan4x4Y[i] + - g_au8HevcDiagScan4x4X[i]; - else - pos = 8 * g_au8HevcDiagScan8x8Y[i] + - g_au8HevcDiagScan8x8X[i]; - - scaling_list_delta_coef = parseSe(pvBuf); - next_coef = (next_coef + 256U + scaling_list_delta_coef) % 256; - sl->aaau8Sl[size_id][matrix_id][pos] = next_coef; - } - } - } - - if (ptSps->iChromaFormatIdc == 3) { - for (i = 0; i < 64; i++) { - sl->aaau8Sl[3][1][i] = sl->aaau8Sl[2][1][i]; - sl->aaau8Sl[3][2][i] = sl->aaau8Sl[2][2][i]; - sl->aaau8Sl[3][4][i] = sl->aaau8Sl[2][4][i]; - sl->aaau8Sl[3][5][i] = sl->aaau8Sl[2][5][i]; - } - sl->aau8SlDc[1][1] = sl->aau8SlDc[0][1]; - sl->aau8SlDc[1][2] = sl->aau8SlDc[0][2]; - sl->aau8SlDc[1][4] = sl->aau8SlDc[0][4]; - sl->aau8SlDc[1][5] = sl->aau8SlDc[0][5]; - } - - - return 0; -} - -int hevcDecodeShortTermRps(T_GetBitContext *pvBuf, - T_ShortTermRPS *rps, const T_HEVCSPS *ptSps, int is_slice_header) -{ - uint8_t rps_predict = 0; - int au32DeltaPoc; - int k0 = 0; - int k1 = 0; - int k = 0; - int i; - - if (rps != ptSps->atStRps && ptSps->uiNbStRps) - rps_predict = getOneBit(pvBuf); - - if (rps_predict) { - const T_ShortTermRPS *ptRpsRidx; - int iDeltaRps; - unsigned int uiAbsDeltaRps; - uint8_t u8UseDeltaFlag = 0; - uint8_t u8DeltaRpsSign = 0; - - if (is_slice_header) { - unsigned int uiDeltaIdx = parseUe(pvBuf) + 1; - if (u8DeltaRpsSign > ptSps->uiNbStRps) { - RPT(RPT_ERR, - "Invalid value of delta_idx in slice header RPS: %d > %d.\n", - u8DeltaRpsSign, ptSps->uiNbStRps); - return -1; - } - ptRpsRidx = &ptSps->atStRps[ptSps->uiNbStRps - u8DeltaRpsSign]; - rps->iRpsIdxNumDeltaPocs = ptRpsRidx->iNumDeltaPocs; - } else - ptRpsRidx = &ptSps->atStRps[rps - ptSps->atStRps - 1]; - - u8DeltaRpsSign = getOneBit(pvBuf); - uiAbsDeltaRps = parseUe(pvBuf) + 1; - if (uiAbsDeltaRps < 1 || uiAbsDeltaRps > 32768) { - RPT(RPT_ERR, - "Invalid value of uiAbsDeltaRps: %d\n", - uiAbsDeltaRps); - return -1; - } - iDeltaRps = (1 - (u8DeltaRpsSign << 1)) * uiAbsDeltaRps; - for (i = 0; i <= ptRpsRidx->iNumDeltaPocs; i++) { - int used = rps->au8Used[k] = getOneBit(pvBuf); - - if (!used) - u8UseDeltaFlag = getOneBit(pvBuf); - - if (used || u8UseDeltaFlag) { - if (i < ptRpsRidx->iNumDeltaPocs) - au32DeltaPoc = iDeltaRps + ptRpsRidx->au32DeltaPoc[i]; - else - au32DeltaPoc = iDeltaRps; - rps->au32DeltaPoc[k] = au32DeltaPoc; - if (au32DeltaPoc < 0) - k0++; - else - k1++; - k++; - } - } - - if (k >= FF_ARRAY_ELEMS(rps->au8Used)) { - RPT(RPT_ERR, - "Invalid iNumDeltaPocs: %d\n", k); - return -1; - } - - rps->iNumDeltaPocs = k; - rps->uiNumNegativePics = k0; - // sort in increasing order (smallest first) - if (rps->iNumDeltaPocs != 0) { - int used, tmp; - for (i = 1; i < rps->iNumDeltaPocs; i++) { - au32DeltaPoc = rps->au32DeltaPoc[i]; - used = rps->au8Used[i]; - for (k = i - 1; k >= 0; k--) { - tmp = rps->au32DeltaPoc[k]; - if (au32DeltaPoc < tmp) { - rps->au32DeltaPoc[k + 1] = tmp; - rps->au8Used[k + 1] = rps->au8Used[k]; - rps->au32DeltaPoc[k] = au32DeltaPoc; - rps->au8Used[k] = used; - } - } - } - } - if ((rps->uiNumNegativePics >> 1) != 0) { - int used; - k = rps->uiNumNegativePics - 1; - // flip the negative values to largest first - for (i = 0; i < rps->uiNumNegativePics >> 1; i++) { - au32DeltaPoc = rps->au32DeltaPoc[i]; - used = rps->au8Used[i]; - rps->au32DeltaPoc[i] = rps->au32DeltaPoc[k]; - rps->au8Used[i] = rps->au8Used[k]; - rps->au32DeltaPoc[k] = au32DeltaPoc; - rps->au8Used[k] = used; - k--; - } - } - } else { - unsigned int uiPrev, uiNbPositivePics; - rps->uiNumNegativePics = parseUe(pvBuf); - uiNbPositivePics = parseUe(pvBuf); - - if (rps->uiNumNegativePics >= HEVC_MAX_REFS || - uiNbPositivePics >= HEVC_MAX_REFS) { - RPT(RPT_ERR, "Too many refs in a short term RPS.\n"); - return -1; - } - - rps->iNumDeltaPocs = rps->uiNumNegativePics + uiNbPositivePics; - if (rps->iNumDeltaPocs) { - uiPrev = 0; - for (i = 0; i < rps->uiNumNegativePics; i++) { - au32DeltaPoc = parseUe(pvBuf) + 1; - if (au32DeltaPoc < 1 || au32DeltaPoc > 32768) { - RPT(RPT_ERR, - "Invalid value of au32DeltaPoc: %d\n", - au32DeltaPoc); - return -1; - } - uiPrev -= au32DeltaPoc; - rps->au32DeltaPoc[i] = uiPrev; - rps->au8Used[i] = getOneBit(pvBuf); - } - uiPrev = 0; - for (i = 0; i < uiNbPositivePics; i++) { - au32DeltaPoc = parseUe(pvBuf) + 1; - if (au32DeltaPoc < 1 || au32DeltaPoc > 32768) { - RPT(RPT_ERR, - "Invalid value of au32DeltaPoc: %d\n", - au32DeltaPoc); - return -1; - } - uiPrev += au32DeltaPoc; - rps->au32DeltaPoc[rps->uiNumNegativePics + i] = uiPrev; - rps->au8Used[rps->uiNumNegativePics + i] = getOneBit(pvBuf); - } - } - } - return 0; -} - -static void decodeSublayerHrd(T_GetBitContext *pvBuf, unsigned int nb_cpb, - int iSubpicParamsPresent) -{ - int i; - - for (i = 0; i < nb_cpb; i++) { - parseUe(pvBuf); // bit_rate_value_minus1 - parseUe(pvBuf); // cpb_size_value_minus1 - - if (iSubpicParamsPresent) { - parseUe(pvBuf); // cpb_size_du_value_minus1 - parseUe(pvBuf); // bit_rate_du_value_minus1 - } - getOneBit(pvBuf); // cbr_flag - } -} - -static int decodeHrd(T_GetBitContext *pvBuf, int common_inf_present, - int max_sublayers) -{ - int iNalParamsPresent = 0, iVclParamsPresent = 0; - int iSubpicParamsPresent = 0; - int i; - - if (common_inf_present) { - iNalParamsPresent = getOneBit(pvBuf); - iVclParamsPresent = getOneBit(pvBuf); - - if (iNalParamsPresent || iVclParamsPresent) { - iSubpicParamsPresent = getOneBit(pvBuf); - - if (iSubpicParamsPresent) { - getBits(pvBuf, 8); // tick_divisor_minus2 - getBits(pvBuf, 5); // du_cpb_removal_delay_increment_length_minus1 - getBits(pvBuf, 1); // sub_pic_cpb_params_in_pic_timing_sei_flag - getBits(pvBuf, 5); // dpb_output_delay_du_length_minus1 - } - - getBits(pvBuf, 4); // bit_rate_scale - getBits(pvBuf, 4); // cpb_size_scale - - if (iSubpicParamsPresent) - getBits(pvBuf, 4); // cpb_size_du_scale - - getBits(pvBuf, 5); // initial_cpb_removal_delay_length_minus1 - getBits(pvBuf, 5); // au_cpb_removal_delay_length_minus1 - getBits(pvBuf, 5); // dpb_output_delay_length_minus1 - } - } - - for (i = 0; i < max_sublayers; i++) { - int low_delay = 0; - unsigned int nb_cpb = 1; - int iFixedRate = getOneBit(pvBuf); - - if (!iFixedRate) - iFixedRate = getOneBit(pvBuf); - - if (iFixedRate) - parseUe(pvBuf); // elemental_duration_in_tc_minus1 - else - low_delay = getOneBit(pvBuf); - - if (!low_delay) { - nb_cpb = parseUe(pvBuf) + 1; - if (nb_cpb < 1 || nb_cpb > 32) { - RPT(RPT_ERR, "nb_cpb %d invalid\n", nb_cpb); - return -1; - } - } - - if (iNalParamsPresent) - decodeSublayerHrd(pvBuf, nb_cpb, iSubpicParamsPresent); - if (iVclParamsPresent) - decodeSublayerHrd(pvBuf, nb_cpb, iSubpicParamsPresent); - } - return 0; -} - - - -static void decodeVui(T_GetBitContext *pvBuf, T_HEVCSPS *ptSps) -{ - T_VUI tBackupVui, *tVui = &ptSps->tVui; - T_GetBitContext tBackup; - int sar_present, alt = 0; - - RPT(RPT_DBG, "Decoding VUI\n"); - - sar_present = getOneBit(pvBuf); - if (sar_present) { - uint8_t sar_idx = getBits(pvBuf, 8); - if (sar_idx < FF_ARRAY_ELEMS(sg_atVuiSar)) - tVui->tSar = sg_atVuiSar[sar_idx]; - else if (sar_idx == 255) { - tVui->tSar.num = getBits(pvBuf, 16); - tVui->tSar.den = getBits(pvBuf, 16); - } else - RPT(RPT_WRN, - "Unknown SAR index: %u.\n", sar_idx); - } - - tVui->iOverscanInfoPresentFlag = getOneBit(pvBuf); - if (tVui->iOverscanInfoPresentFlag) - tVui->iOverscanAppropriateFlag = getOneBit(pvBuf); - - tVui->iVideoSignalTypePresentFlag = getOneBit(pvBuf); - if (tVui->iVideoSignalTypePresentFlag) { - tVui->iVideoFormat = getBits(pvBuf, 3); - tVui->iVideoFullRangeFlag = getOneBit(pvBuf); - tVui->iColourDescriptionPresentFlag = getOneBit(pvBuf); -// if (tVui->iVideoFullRangeFlag && ptSps->pix_fmt == AV_PIX_FMT_YUV420P) -// ptSps->pix_fmt = AV_PIX_FMT_YUVJ420P; - if (tVui->iColourDescriptionPresentFlag) { - tVui->u8ColourPrimaries = getBits(pvBuf, 8); - tVui->u8TransferCharacteristic = getBits(pvBuf, 8); - tVui->u8MatrixCoeffs = getBits(pvBuf, 8); - } - } - - tVui->iChromaLocInfoPresentFlag = getOneBit(pvBuf); - if (tVui->iChromaLocInfoPresentFlag) { - tVui->iChromaSampleLocTypeTopField = parseUe(pvBuf); - tVui->iChromaSampleLocTypeBottomField = parseUe(pvBuf); - } - - tVui->iNeutraChromaIndicationFlag = getOneBit(pvBuf); - tVui->iFieldSeqFlag = getOneBit(pvBuf); - tVui->iFrameFieldInfoPresentFlag = getOneBit(pvBuf); - - // Backup context in case an alternate header is detected - memcpy(&tBackup, pvBuf, sizeof(tBackup)); - memcpy(&tBackupVui, tVui, sizeof(tBackupVui)); - if (getBitsLeft(pvBuf) >= 68 && showBitsLong(pvBuf, 21) == 0x100000) { - tVui->iDefaultDisplayWindowFlag = 0; - RPT(RPT_WRN, "Invalid default display window\n"); - } else - tVui->iDefaultDisplayWindowFlag = getOneBit(pvBuf); - - if (tVui->iDefaultDisplayWindowFlag) { - int vert_mult = sg_au8HevcSubHeightC[ptSps->iChromaFormatIdc]; - int horiz_mult = sg_au8HevcSubWidthC[ptSps->iChromaFormatIdc]; - tVui->tDefDispWin.uiLeftOffset = parseUe(pvBuf) * horiz_mult; - tVui->tDefDispWin.uiRightOffset = parseUe(pvBuf) * horiz_mult; - tVui->tDefDispWin.uiTopOffset = parseUe(pvBuf) * vert_mult; - tVui->tDefDispWin.uiBottomOffset = parseUe(pvBuf) * vert_mult; - } - -timing_info: - tVui->iVuiTimingInfoPresentFlag = getOneBit(pvBuf); - - if (tVui->iVuiTimingInfoPresentFlag) { - if( getBitsLeft(pvBuf) < 66 && !alt) { - // The alternate syntax seem to have timing info located - // at where tDefDispWin is normally located - RPT(RPT_WRN, - "Strange VUI timing information, retrying...\n"); - memcpy(tVui, &tBackupVui, sizeof(tBackupVui)); - memcpy(pvBuf, &tBackup, sizeof(tBackup)); - alt = 1; - goto timing_info; - } - tVui->u32VuiNumUnitsInTick = getBits(pvBuf, 32); - tVui->u32VuiTimeScale = getBits(pvBuf, 32); - if (alt) { - RPT(RPT_INF, "Retry got %u/%u fps\n", - tVui->u32VuiTimeScale, tVui->u32VuiNumUnitsInTick); - } - tVui->iVuiPocProportionalToTimingFlag = getOneBit(pvBuf); - if (tVui->iVuiPocProportionalToTimingFlag) - tVui->iVuiNumTicksPocDiffOneMinus1 = parseUe(pvBuf); - tVui->iVuiHrdParametersPresentFlag = getOneBit(pvBuf); - if (tVui->iVuiHrdParametersPresentFlag) - decodeHrd(pvBuf, 1, ptSps->iMaxSubLayers); - } - - tVui->iBitstreamRestrictionFlag = getOneBit(pvBuf); - if (tVui->iBitstreamRestrictionFlag) { - if (getBitsLeft(pvBuf) < 8 && !alt) { - RPT(RPT_WRN, - "Strange VUI bitstream restriction information, retrying" - " from timing information...\n"); - memcpy(tVui, &tBackupVui, sizeof(tBackupVui)); - memcpy(pvBuf, &tBackup, sizeof(tBackup)); - alt = 1; - goto timing_info; - } - tVui->iTilesFixedStructureFlag = getOneBit(pvBuf); - tVui->iMotionVectorsOverPicBoundariesFlag = getOneBit(pvBuf); - tVui->iRestrictedRefPicListsFlag = getOneBit(pvBuf); - tVui->iMinSpatialSegmentationIdc = parseUe(pvBuf); - tVui->iMaxBytesPerPicDenom = parseUe(pvBuf); - tVui->iMaxBitsPerMinCuDenom = parseUe(pvBuf); - tVui->iLog2MaxMvLengthHorizontal = parseUe(pvBuf); - tVui->iLog2MaxMvLengthVertical = parseUe(pvBuf); - } - - if (getBitsLeft(pvBuf) < 1 && !alt) { - // XXX: Alternate syntax when iSpsRangeExtensionFlag != 0? - RPT(RPT_WRN, - "Overread in VUI, retrying from timing information...\n"); - memcpy(tVui, &tBackupVui, sizeof(tBackupVui)); - memcpy(pvBuf, &tBackup, sizeof(tBackup)); - alt = 1; - goto timing_info; - } -} - -static unsigned avModUintp2c(unsigned a, unsigned p) -{ - return a & ((1 << p) - 1); -} - - -int h265DecSeqParameterSet( void *pvBufSrc, T_HEVCSPS *ptSps ) -{ - T_HEVCWindow *ow; - int iLog2DiffMaxMinTransformBlockSize; - int iBitDepthChroma, iStart, iVuiPresent, iSublayerOrderingInfo; - int i; - int iRet = 0; - - void *pvBuf = NULL; - if(NULL == pvBufSrc || NULL == ptSps) - { - RPT(RPT_ERR,"ERR null pointer\n"); - iRet = -1; - goto exit; - } - - memset((void *)ptSps, 0, sizeof(T_HEVCSPS)); - - pvBuf = deEmulationPrevention(pvBufSrc); - if(NULL == pvBuf) - { - RPT(RPT_ERR,"ERR null pointer\n"); - iRet = -1; - goto exit; - } - - // Coded parameters - - ptSps->uiVpsId = getBits(pvBuf, 4); - if (ptSps->uiVpsId >= HEVC_MAX_VPS_COUNT) { - RPT(RPT_ERR, "VPS id out of range: %d\n", ptSps->uiVpsId); - iRet = -1; - goto exit; - } - - ptSps->iMaxSubLayers = getBits(pvBuf, 3) + 1; - if (ptSps->iMaxSubLayers > HEVC_MAX_SUB_LAYERS) { - RPT(RPT_ERR, "sps_max_sub_layers out of range: %d\n", - ptSps->iMaxSubLayers); - iRet = -1; - goto exit; - } - - ptSps->u8temporalIdNestingFlag = getBits(pvBuf, 1); - if ((iRet = parsePtl(pvBuf, &ptSps->tPtl, ptSps->iMaxSubLayers)) < 0) - goto exit; - - int sps_id = parseUe(pvBuf); - if (sps_id >= HEVC_MAX_SPS_COUNT) { - RPT(RPT_ERR, "SPS id out of range: %d\n", sps_id); - iRet = -1; - goto exit; - } - - ptSps->iChromaFormatIdc = parseUe(pvBuf); - if (ptSps->iChromaFormatIdc > 3U) { - RPT(RPT_ERR, "iChromaFormatIdc %d is invalid\n", ptSps->iChromaFormatIdc); - iRet = -1; - goto exit; - } - - if (ptSps->iChromaFormatIdc == 3) - ptSps->u8SeparateColourPlaneFlag = getOneBit(pvBuf); - - if (ptSps->u8SeparateColourPlaneFlag) - ptSps->iChromaFormatIdc = 0; - - ptSps->iWidth = parseUe(pvBuf); - ptSps->iHeight = parseUe(pvBuf); - - if (getOneBit(pvBuf)) { // pic_conformance_flag - int vert_mult = sg_au8HevcSubHeightC[ptSps->iChromaFormatIdc]; - int horiz_mult = sg_au8HevcSubWidthC[ptSps->iChromaFormatIdc]; - ptSps->tPicConfWin.uiLeftOffset = parseUe(pvBuf) * horiz_mult; - ptSps->tPicConfWin.uiRightOffset = parseUe(pvBuf) * horiz_mult; - ptSps->tPicConfWin.uiTopOffset = parseUe(pvBuf) * vert_mult; - ptSps->tPicConfWin.uiBottomOffset = parseUe(pvBuf) * vert_mult; - - ptSps->tOutputWindow = ptSps->tPicConfWin; - } - - ptSps->iBitDepth = parseUe(pvBuf) + 8; - iBitDepthChroma = parseUe(pvBuf) + 8; - - if (ptSps->iChromaFormatIdc && iBitDepthChroma != ptSps->iBitDepth) { - RPT(RPT_ERR, - "Luma bit depth (%d) is different from chroma bit depth (%d), " - "this is unsupported.\n", - ptSps->iBitDepth, iBitDepthChroma); - iRet = -1; - goto exit; - } - ptSps->iBitDepthChroma = iBitDepthChroma; - - ptSps->uiLog2MaxPocLsb = parseUe(pvBuf) + 4; - if (ptSps->uiLog2MaxPocLsb > 16) { - RPT(RPT_ERR, "log2_max_pic_order_cnt_lsb_minus4 out range: %d\n", - ptSps->uiLog2MaxPocLsb - 4); - iRet = -1; - goto exit; - } - - iSublayerOrderingInfo = getOneBit(pvBuf); - iStart = iSublayerOrderingInfo ? 0 : ptSps->iMaxSubLayers - 1; - for (i = iStart; i < ptSps->iMaxSubLayers; i++) { - ptSps->stTemporalLayer[i].iMaxDecPicBuffering = parseUe(pvBuf) + 1; - ptSps->stTemporalLayer[i].iNumReorderPics = parseUe(pvBuf); - ptSps->stTemporalLayer[i].iMaxLatencyIncrease = parseUe(pvBuf) - 1; - if (ptSps->stTemporalLayer[i].iMaxDecPicBuffering > (unsigned)HEVC_MAX_DPB_SIZE) { - RPT(RPT_ERR, "sps_max_dec_pic_buffering_minus1 out of range: %d\n", - ptSps->stTemporalLayer[i].iMaxDecPicBuffering - 1U); - iRet = -1; - goto exit; - } - if (ptSps->stTemporalLayer[i].iNumReorderPics > ptSps->stTemporalLayer[i].iMaxDecPicBuffering - 1) { - RPT(RPT_WRN, "sps_max_num_reorder_pics out of range: %d\n", - ptSps->stTemporalLayer[i].iNumReorderPics); - if (ptSps->stTemporalLayer[i].iNumReorderPics > HEVC_MAX_DPB_SIZE - 1) { - iRet = -1; - goto exit; - } - ptSps->stTemporalLayer[i].iMaxDecPicBuffering = ptSps->stTemporalLayer[i].iNumReorderPics + 1; - } - } - - if (!iSublayerOrderingInfo) { - for (i = 0; i < iStart; i++) { - ptSps->stTemporalLayer[i].iMaxDecPicBuffering = ptSps->stTemporalLayer[iStart].iMaxDecPicBuffering; - ptSps->stTemporalLayer[i].iNumReorderPics = ptSps->stTemporalLayer[iStart].iNumReorderPics; - ptSps->stTemporalLayer[i].iMaxLatencyIncrease = ptSps->stTemporalLayer[iStart].iMaxLatencyIncrease; - } - } - - ptSps->uiLog2MinCbSize = parseUe(pvBuf) + 3; - ptSps->uiLog2DiffMaxMinCodingBlockSize = parseUe(pvBuf); - ptSps->uiLog2MinTbSize = parseUe(pvBuf) + 2; - iLog2DiffMaxMinTransformBlockSize = parseUe(pvBuf); - ptSps->uiLog2MaxTrafoSize = iLog2DiffMaxMinTransformBlockSize + - ptSps->uiLog2MinTbSize; - - if (ptSps->uiLog2MinCbSize < 3 || ptSps->uiLog2MinCbSize > 30) { - RPT(RPT_ERR, "Invalid value %d for uiLog2MinCbSize", ptSps->uiLog2MinCbSize); - iRet = -1; - goto exit; - } - - if (ptSps->uiLog2DiffMaxMinCodingBlockSize > 30) { - RPT(RPT_ERR, "Invalid value %d for uiLog2DiffMaxMinCodingBlockSize", ptSps->uiLog2DiffMaxMinCodingBlockSize); - iRet = -1; - goto exit; - } - - if (ptSps->uiLog2MinTbSize >= ptSps->uiLog2MinCbSize || ptSps->uiLog2MinTbSize < 2) { - RPT(RPT_ERR, "Invalid value for uiLog2MinTbSize"); - iRet = -1; - goto exit; - } - - if (iLog2DiffMaxMinTransformBlockSize < 0 || iLog2DiffMaxMinTransformBlockSize > 30) { - RPT(RPT_ERR, "Invalid value %d for iLog2DiffMaxMinTransformBlockSize", iLog2DiffMaxMinTransformBlockSize); - iRet = -1; - goto exit; - } - - ptSps->iMaxTransformHierarchyDepthInter = parseUe(pvBuf); - ptSps->iMaxTransformHierarchyDepthIntra = parseUe(pvBuf); - - ptSps->u8ScalingListEnableFlag = getOneBit(pvBuf); - - if (ptSps->u8ScalingListEnableFlag) { - setDefaultScalingListData(&ptSps->tScalingList); - - if (getOneBit(pvBuf)) { - iRet = scalingListData(pvBuf, &ptSps->tScalingList, ptSps); - if (iRet < 0) - goto exit; - } - } - - ptSps->u8AmpEnabledFlag = getOneBit(pvBuf); - ptSps->u8SaoEnabled = getOneBit(pvBuf); - - ptSps->iPcmEnabledFlag = getOneBit(pvBuf); - - if (ptSps->iPcmEnabledFlag) { - ptSps->pcm.u8BitDepth = getBits(pvBuf, 4) + 1; - ptSps->pcm.u8BitDepthChroma = getBits(pvBuf, 4) + 1; - ptSps->pcm.uiLog2MinPcmCbSize = parseUe(pvBuf) + 3; - ptSps->pcm.uiLog2MaxPcmCbSize = ptSps->pcm.uiLog2MinPcmCbSize + - parseUe(pvBuf); - if (FFMAX(ptSps->pcm.u8BitDepth, ptSps->pcm.u8BitDepthChroma) > ptSps->iBitDepth) { - RPT(RPT_ERR, - "PCM bit depth (%d, %d) is greater than normal bit depth (%d)\n", - ptSps->pcm.u8BitDepth, ptSps->pcm.u8BitDepthChroma, ptSps->iBitDepth); - iRet = -1; - goto exit; - } - - ptSps->pcm.u8LoopFilterDisableFlag = getOneBit(pvBuf); - } - - ptSps->uiNbStRps = parseUe(pvBuf); - if (ptSps->uiNbStRps > HEVC_MAX_SHORT_TERM_REF_PIC_SETS) { - RPT(RPT_ERR, "Too many short term RPS: %d.\n", - ptSps->uiNbStRps); - iRet = -1; - goto exit; - } - for (i = 0; i < ptSps->uiNbStRps; i++) { - if ((iRet = hevcDecodeShortTermRps(pvBuf, &ptSps->atStRps[i], - ptSps, 0)) < 0) - goto exit; - } - - ptSps->u8LongTermRefPicsPresentFlag = getOneBit(pvBuf); - if (ptSps->u8LongTermRefPicsPresentFlag) { - ptSps->u8NumLongTermRefPicsSps = parseUe(pvBuf); - if (ptSps->u8NumLongTermRefPicsSps > HEVC_MAX_LONG_TERM_REF_PICS) { - RPT(RPT_ERR, "Too many long term ref pics: %d.\n", - ptSps->u8NumLongTermRefPicsSps); - iRet = -1; - goto exit; - } - for (i = 0; i < ptSps->u8NumLongTermRefPicsSps; i++) { - ptSps->au16LtRefPicPocLsbSps[i] = getBits(pvBuf, ptSps->uiLog2MaxPocLsb); - ptSps->au8UsedByCurrPicLtSpsFlag[i] = getOneBit(pvBuf); - } - } - - ptSps->u8SpsTemporalMvpEnabledFlag = getOneBit(pvBuf); - ptSps->u8SpsStrongIntraMmoothingEnableFlag = getOneBit(pvBuf); - ptSps->tVui.tSar = (T_AVRational){0, 1}; - ptSps->iVuiPresent = getOneBit(pvBuf); - if (ptSps->iVuiPresent) - decodeVui(pvBuf, ptSps); - - - if (getOneBit(pvBuf)) { // sps_extension_flag - int iSpsRangeExtensionFlag = getOneBit(pvBuf); - getBits(pvBuf, 7); //sps_extension_7bits = getBits(pvBuf, 7); - if (iSpsRangeExtensionFlag) { - int iExtendedPrecisionProcessingFlag; - int iCabacBypassAlignmentEnabledFlag; - - ptSps->iTransformSkipRotationEnabledFlag = getOneBit(pvBuf); - ptSps->iTransformSkipContextEnabledFlag = getOneBit(pvBuf); - ptSps->iImplicitRdpcmEnabledFlag = getOneBit(pvBuf); - - ptSps->iExplicitRdpcmEnabledFlag = getOneBit(pvBuf); - - iExtendedPrecisionProcessingFlag = getOneBit(pvBuf); - if (iExtendedPrecisionProcessingFlag) - RPT(RPT_WRN, - "iExtendedPrecisionProcessingFlag not yet implemented\n"); - - ptSps->iIntraSmoothingDisabledFlag = getOneBit(pvBuf); - ptSps->iHighPrecisionOffsetsEnabledFlag = getOneBit(pvBuf); - if (ptSps->iHighPrecisionOffsetsEnabledFlag) - RPT(RPT_WRN, - "iHighPrecisionOffsetsEnabledFlag not yet implemented\n"); - - ptSps->iPersistentRiceAdaptationEnabledFlag = getOneBit(pvBuf); - - iCabacBypassAlignmentEnabledFlag = getOneBit(pvBuf); - if (iCabacBypassAlignmentEnabledFlag) - RPT(RPT_WRN, - "iCabacBypassAlignmentEnabledFlag not yet implemented\n"); - } - } - - ow = &ptSps->tOutputWindow; - if (ow->uiLeftOffset >= INT_MAX - ow->uiRightOffset || - ow->uiTopOffset >= INT_MAX - ow->uiBottomOffset || - ow->uiLeftOffset + ow->uiRightOffset >= ptSps->iWidth || - ow->uiTopOffset + ow->uiBottomOffset >= ptSps->iHeight) { - RPT(RPT_WRN, "Invalid cropping offsets: %u/%u/%u/%u\n", - ow->uiLeftOffset, ow->uiRightOffset, ow->uiTopOffset, ow->uiBottomOffset); - RPT(RPT_WRN, - "Displaying the whole video surface.\n"); - memset(ow, 0, sizeof(*ow)); - memset(&ptSps->tPicConfWin, 0, sizeof(ptSps->tPicConfWin)); - } - - // Inferred parameters - ptSps->uiLog2CtbSize = ptSps->uiLog2MinCbSize + - ptSps->uiLog2DiffMaxMinCodingBlockSize; - ptSps->uiLog2MinPuSize = ptSps->uiLog2MinCbSize - 1; - - if (ptSps->uiLog2CtbSize > HEVC_MAX_LOG2_CTB_SIZE) { - RPT(RPT_ERR, "CTB size out of range: 2^%d\n", ptSps->uiLog2CtbSize); - iRet = -1; - goto exit; - } - if (ptSps->uiLog2CtbSize < 4) { - RPT(RPT_ERR, - "uiLog2CtbSize %d differs from the bounds of any known profile\n", - ptSps->uiLog2CtbSize); - iRet = -1; - goto exit; - } - - ptSps->iCtbWidth = (ptSps->iWidth + (1 << ptSps->uiLog2CtbSize) - 1) >> ptSps->uiLog2CtbSize; - ptSps->iCtbHeight = (ptSps->iHeight + (1 << ptSps->uiLog2CtbSize) - 1) >> ptSps->uiLog2CtbSize; - ptSps->iCtbSize = ptSps->iCtbWidth * ptSps->iCtbHeight; - - ptSps->iMinCbWidth = ptSps->iWidth >> ptSps->uiLog2MinCbSize; - ptSps->iMinCbHeight = ptSps->iHeight >> ptSps->uiLog2MinCbSize; - ptSps->iMinTbWidth = ptSps->iWidth >> ptSps->uiLog2MinTbSize; - ptSps->iMinTbHeight = ptSps->iHeight >> ptSps->uiLog2MinTbSize; - ptSps->iMinPuWidth = ptSps->iWidth >> ptSps->uiLog2MinPuSize; - ptSps->iMinPuHeight = ptSps->iHeight >> ptSps->uiLog2MinPuSize; - ptSps->iTbMask = (1 << (ptSps->uiLog2CtbSize - ptSps->uiLog2MinTbSize)) - 1; - - ptSps->iQpBdOffset = 6 * (ptSps->iBitDepth - 8); - - if (avModUintp2c(ptSps->iWidth, ptSps->uiLog2MinCbSize) || - avModUintp2c(ptSps->iHeight, ptSps->uiLog2MinCbSize)) { - RPT(RPT_ERR, "Invalid coded frame dimensions.\n"); - iRet = -1; - goto exit; - } - - if (ptSps->iMaxTransformHierarchyDepthInter > ptSps->uiLog2CtbSize - ptSps->uiLog2MinTbSize) { - RPT(RPT_ERR, "iMaxTransformHierarchyDepthInter out of range: %d\n", - ptSps->iMaxTransformHierarchyDepthInter); - iRet = -1; - goto exit; - } - if (ptSps->iMaxTransformHierarchyDepthIntra > ptSps->uiLog2CtbSize - ptSps->uiLog2MinTbSize) { - RPT(RPT_ERR, "iMaxTransformHierarchyDepthIntra out of range: %d\n", - ptSps->iMaxTransformHierarchyDepthIntra); - iRet = -1; - goto exit; - } - if (ptSps->uiLog2MaxTrafoSize > FFMIN(ptSps->uiLog2CtbSize, 5)) { - RPT(RPT_ERR, - "max transform block size out of range: %d\n", - ptSps->uiLog2MaxTrafoSize); - iRet = -1; - goto exit; - } - - if (getBitsLeft(pvBuf) < 0) { - RPT(RPT_ERR, - "Overread SPS by %d bits\n", -getBitsLeft(pvBuf)); - iRet = -1; - goto exit; - } - - -exit: - - getBitContextFree(pvBuf); - return iRet; - -} - - -int h265DecVideoParameterSet( void *pvBufSrc, T_HEVCVPS *ptVps ) -{ - int iRet = 0; - int i,j; - int uiVpsId = 0; - - void *pvBuf = NULL; - if(NULL == pvBufSrc || NULL == ptVps) - { - RPT(RPT_ERR,"ERR null pointer\n"); - iRet = -1; - goto exit; - } - - memset((void *)ptVps, 0, sizeof(T_HEVCVPS)); - - pvBuf = deEmulationPrevention(pvBufSrc); - if(NULL == pvBuf) - { - RPT(RPT_ERR,"ERR null pointer\n"); - iRet = -1; - goto exit; - } - - RPT(RPT_DBG, "Decoding VPS\n"); - - uiVpsId = getBits(pvBuf, 4); - if (uiVpsId >= HEVC_MAX_VPS_COUNT) { - RPT(RPT_ERR, "VPS id out of range: %d\n", uiVpsId); - iRet = -1; - goto exit; - } - - if (getBits(pvBuf, 2) != 3) { // vps_reserved_three_2bits - RPT(RPT_ERR, "vps_reserved_three_2bits is not three\n"); - iRet = -1; - goto exit; - } - - ptVps->iVpsMaxLayers = getBits(pvBuf, 6) + 1; - ptVps->iVpsMaxSubLayers = getBits(pvBuf, 3) + 1; - ptVps->u8VpsTemporalIdNestingFlag = getOneBit(pvBuf); - - if (getBits(pvBuf, 16) != 0xffff) { // vps_reserved_ffff_16bits - RPT(RPT_ERR, "vps_reserved_ffff_16bits is not 0xffff\n"); - iRet = -1; - goto exit; - } - - if (ptVps->iVpsMaxSubLayers > HEVC_MAX_SUB_LAYERS) { - RPT(RPT_ERR, "iVpsMaxSubLayers out of range: %d\n", - ptVps->iVpsMaxSubLayers); - iRet = -1; - goto exit; - } - - if (parsePtl(pvBuf, &ptVps->tPtl, ptVps->iVpsMaxSubLayers) < 0){ - iRet = -1; - goto exit; - } - - ptVps->iVpsSubLayerOrderingInfoPresentFlag = getOneBit(pvBuf); - - i = ptVps->iVpsSubLayerOrderingInfoPresentFlag ? 0 : ptVps->iVpsMaxSubLayers - 1; - for (; i < ptVps->iVpsMaxSubLayers; i++) { - ptVps->uiVpsMaxDecPicBuffering[i] = parseUe(pvBuf) + 1; - ptVps->auiVpsNumReorderPics[i] = parseUe(pvBuf); - ptVps->auiVpsMaxLatencyIncrease[i] = parseUe(pvBuf) - 1; - - if (ptVps->uiVpsMaxDecPicBuffering[i] > HEVC_MAX_DPB_SIZE || !ptVps->uiVpsMaxDecPicBuffering[i]) { - RPT(RPT_ERR, "vps_max_dec_pic_buffering_minus1 out of range: %d\n", - ptVps->uiVpsMaxDecPicBuffering[i] - 1); - iRet = -1; - goto exit; - } - if (ptVps->auiVpsNumReorderPics[i] > ptVps->uiVpsMaxDecPicBuffering[i] - 1) { - RPT(RPT_WRN, "vps_max_num_reorder_pics out of range: %d\n", - ptVps->auiVpsNumReorderPics[i]); - } - } - - ptVps->iVpsMaxLayerId = getBits(pvBuf, 6); - ptVps->iVpsNumLayerSets = parseUe(pvBuf) + 1; - if (ptVps->iVpsNumLayerSets < 1 || ptVps->iVpsNumLayerSets > 1024 || - (ptVps->iVpsNumLayerSets - 1LL) * (ptVps->iVpsMaxLayerId + 1LL) > getBitsLeft(pvBuf)) { - RPT(RPT_ERR, "too many layer_id_included_flags\n"); - iRet = -1; - goto exit; - } - - for (i = 1; i < ptVps->iVpsNumLayerSets; i++) - for (j = 0; j <= ptVps->iVpsMaxLayerId; j++) - getBits(pvBuf, 1); // layer_id_included_flag[i][j] - - ptVps->u8VpsTimingInfoPresentFlag = getOneBit(pvBuf); - if (ptVps->u8VpsTimingInfoPresentFlag) { - ptVps->u32VpsNumUnitsInTick = getBits(pvBuf, 32); - ptVps->u32VpsTimeScale = getBits(pvBuf, 32); - ptVps->u8VpsPocProportionalToTimingFlag = getOneBit(pvBuf); - if (ptVps->u8VpsPocProportionalToTimingFlag) - ptVps->iVpsNumTicksPocDiffOne = parseUe(pvBuf) + 1; - ptVps->iVpsNumHrdParameters = parseUe(pvBuf); - if (ptVps->iVpsNumHrdParameters > (unsigned)ptVps->iVpsNumLayerSets) { - RPT(RPT_ERR, - "iVpsNumHrdParameters %d is invalid\n", ptVps->iVpsNumHrdParameters); - iRet = -1; - goto exit; - } - for (i = 0; i < ptVps->iVpsNumHrdParameters; i++) { - int common_inf_present = 1; - - parseUe(pvBuf); // hrd_layer_set_idx - if (i) - common_inf_present = getOneBit(pvBuf); - decodeHrd(pvBuf, common_inf_present, ptVps->iVpsMaxSubLayers); - } - } - getOneBit(pvBuf); /* vps_extension_flag */ - - if (getBitsLeft(pvBuf) < 0) { - RPT(RPT_ERR, - "Overread VPS by %d bits\n", -getBitsLeft(pvBuf)); - - iRet = -1; - goto exit; - } - - -exit: - - getBitContextFree(pvBuf); - return iRet; - -} - -void h264GetWidthHeight(T_SPS *ptSps, int *piWidth, int *piHeight) -{ - // ¿í¸ß¼ÆË㹫ʽ [AUTO-TRANSLATED:e2d61727] - // What is the frame rate - int iCodeWidth = 0; - int iCodedHeight = 0; - iCodeWidth = 16 * ptSps->iMbWidth; - iCodedHeight = 16 * ptSps->iMbHeight; - *piWidth = iCodeWidth - (ptSps->uiCropRight + ptSps->uiCropLeft); - *piHeight = iCodedHeight - (ptSps->uiCropTop + ptSps->uiCropBottom); - if (*piWidth <= 0 || *piHeight <= 0) { - *piWidth = iCodeWidth; - *piHeight = iCodedHeight; - } - - RPT(RPT_DBG, "iCodeWidth:%d, iCodedHeight:%d\n", iCodeWidth, iCodedHeight); - - RPT(RPT_DBG, "*piWidth:%d, *piHeight:%d\n", *piWidth, *piHeight); - - RPT(RPT_DBG, "ptSps->uiCropRight:%d, ptSps->uiCropLeft:%d\n", ptSps->uiCropRight, ptSps->uiCropLeft); - - RPT(RPT_DBG, "ptSps->uiCropTop:%d, ptSps->uiCropBottom:%d\n", ptSps->uiCropTop, ptSps->uiCropBottom); - -} - -int h264GetFormat(T_SPS *ptSps) -{ - return ptSps->iFrameMbsOnlyFlag; -} - - -void h264GeFramerate(T_SPS *ptSps, float *pfFramerate) -{ - - if(ptSps->iTimingInfoPresentFlag) - { - *pfFramerate = (float)ptSps->u32TimeScale / (float)ptSps->u32NumUnitsInTick / 2.0; - }else{ - *pfFramerate = 0; - } - switch((int)*pfFramerate) - { - case 23:// 23.98 - RPT(RPT_DBG, "frame rate:23.98"); - break; - case 24: - RPT(RPT_DBG, "frame rate:24"); - break; - case 25: - RPT(RPT_DBG, "frame rate:25"); - break; - case 29://29.97 - RPT(RPT_DBG, "frame rate:29.97"); - break; - case 30: - RPT(RPT_DBG, "frame rate:30"); - break; - case 50: - RPT(RPT_DBG, "frame rate:50"); - break; - case 59://59.94 - RPT(RPT_DBG, "frame rate:59.94"); - break; - case 60: - RPT(RPT_DBG, "frame rate:60"); - break; - case 6: - RPT(RPT_DBG, "frame rate:6"); - break; - case 8: - RPT(RPT_DBG, "frame rate:8"); - break; - case 12: - RPT(RPT_DBG, "frame rate:12"); - break; - case 15: - RPT(RPT_DBG, "frame rate:15"); - break; - case 10: - RPT(RPT_DBG, "frame rate:10"); - break; - - default: - RPT(RPT_DBG, "frame rate:0"); - break; - } - - return; -} - - - - -void h265GetWidthHeight(T_HEVCSPS *ptSps, int *piWidth, int *piHeight) -{ -#if 1 - int iCodeWidth = 0; - int iCodedHeight = 0; - iCodeWidth = ptSps->iWidth; - iCodedHeight = ptSps->iHeight; - *piWidth = ptSps->iWidth - ptSps->tOutputWindow.uiLeftOffset - ptSps->tOutputWindow.uiRightOffset; - *piHeight = ptSps->iHeight - ptSps->tOutputWindow.uiTopOffset - ptSps->tOutputWindow.uiBottomOffset; - - - RPT(RPT_DBG, "iCodeWidth:%d, iCodedHeight:%d\n", iCodeWidth, iCodedHeight); - - RPT(RPT_DBG, "*piWidth:%d, *piHeight:%d\n", *piWidth, *piHeight); - - RPT(RPT_DBG, "ptSps->tOutputWindow.uiRightOffset:%d, ptSps->tOutputWindow.uiLeftOffset:%d\n", ptSps->tOutputWindow.uiRightOffset, ptSps->tOutputWindow.uiLeftOffset); - - RPT(RPT_DBG, "ptSps->tOutputWindow.uiTopOffset:%d, ptSps->tOutputWindow.uiBottomOffset:%d\n", ptSps->tOutputWindow.uiTopOffset, ptSps->tOutputWindow.uiBottomOffset); -#endif - -} - - - -void h265GeFramerate(T_HEVCVPS *ptVps, T_HEVCSPS *ptSps,float *pfFramerate) -{ - if (ptVps && ptVps->u8VpsTimingInfoPresentFlag) { - *pfFramerate = (float)(ptVps->u32VpsTimeScale) / (float)(ptVps->u32VpsNumUnitsInTick); - - } else if (ptSps && ptSps->tVui.iVuiTimingInfoPresentFlag && ptSps->iVuiPresent) { - *pfFramerate = (float)(ptSps->tVui.u32VuiTimeScale) / (float)(ptSps->tVui.u32VuiNumUnitsInTick); - } - else{ - // vps sps可能不包含帧率 [AUTO-TRANSLATED:15424320] - // vps sps may not contain frame rate - *pfFramerate = 0.0F; - RPT(RPT_WRN, "frame rate:0"); - } -} - -int h265ParsePps(T_GetBitContext *ptGetBitContext, T_HEVC_PPS *ptPps) -{ - int iRet = 0; - - ptPps->pps_pic_parameter_set_id = parseUe(ptGetBitContext); - ptPps->pps_seq_parameter_set_id = parseUe(ptGetBitContext); - ptPps->dependent_slice_segments_enabled_flag = getOneBit(ptGetBitContext); - ptPps->output_flag_present_flag = getOneBit(ptGetBitContext); - ptPps->num_extra_slice_header_bits = getBits(ptGetBitContext, 3); - ptPps->sign_data_hiding_enabled_flag = getOneBit(ptGetBitContext); - ptPps->cabac_init_present_flag = getOneBit(ptGetBitContext); - - ptPps->num_ref_idx_l0_default_active_minus1 = parseUe(ptGetBitContext); - ptPps->num_ref_idx_l1_default_active_minus1 = parseUe(ptGetBitContext); - ptPps->init_qp_minus26 = parseSe(ptGetBitContext); - - ptPps->constrained_intra_pred_flag = getOneBit(ptGetBitContext); - ptPps->transform_skip_enabled_flag = getOneBit(ptGetBitContext); - ptPps->cu_qp_delta_enabled_flag = getOneBit(ptGetBitContext); - - if (ptPps->cu_qp_delta_enabled_flag) { - ptPps->diff_cu_qp_delta_depth = parseUe(ptGetBitContext); - } - ptPps->pps_cb_qp_offset = parseSe(ptGetBitContext); - ptPps->pps_cr_qp_offset = parseSe(ptGetBitContext); - ptPps->pps_slice_chroma_qp_offsets_present_flag = getOneBit(ptGetBitContext); - ptPps->weighted_pred_flag = getOneBit(ptGetBitContext); - ptPps->weighted_bipred_flag = getOneBit(ptGetBitContext); - ptPps->transquant_bypass_enabled_flag = getOneBit(ptGetBitContext); - ptPps->tiles_enabled_flag = getOneBit(ptGetBitContext); - ptPps->entropy_coding_sync_enabled_flag = getOneBit(ptGetBitContext); - return iRet; -} \ No newline at end of file diff --git a/ext-codec/SPSParser.h b/ext-codec/SPSParser.h deleted file mode 100644 index 1671d5c3..00000000 --- a/ext-codec/SPSParser.h +++ /dev/null @@ -1,526 +0,0 @@ -#ifndef _SPSPARSER_H_ -#define _SPSPARSER_H_ - -#if defined (__cplusplus) - extern "C" { -#endif - -#define QP_MAX_NUM (51 + 6*6) // The maximum supported qp - -#define HEVC_MAX_SHORT_TERM_RPS_COUNT 64 - -#define T_PROFILE_HEVC_MAIN 1 -#define T_PROFILE_HEVC_MAIN_10 2 -#define T_PROFILE_HEVC_MAIN_STILL_PICTURE 3 -#define T_PROFILE_HEVC_REXT 4 - - - -/** - * Chromaticity coordinates of the source primaries. - */ -enum T_AVColorPrimaries { - AVCOL_PRI_RESERVED0 = 0, - AVCOL_PRI_BT709 = 1, ///< also ITU-R BT1361 / IEC 61966-2-4 / SMPTE RP177 Annex B - AVCOL_PRI_UNSPECIFIED = 2, - AVCOL_PRI_RESERVED = 3, - AVCOL_PRI_BT470M = 4, ///< also FCC Title 47 Code of Federal Regulations 73.682 (a)(20) - - AVCOL_PRI_BT470BG = 5, ///< also ITU-R BT601-6 625 / ITU-R BT1358 625 / ITU-R BT1700 625 PAL & SECAM - AVCOL_PRI_SMPTE170M = 6, ///< also ITU-R BT601-6 525 / ITU-R BT1358 525 / ITU-R BT1700 NTSC - AVCOL_PRI_SMPTE240M = 7, ///< functionally identical to above - AVCOL_PRI_FILM = 8, ///< colour filters using Illuminant C - AVCOL_PRI_BT2020 = 9, ///< ITU-R BT2020 - AVCOL_PRI_NB, ///< Not part of ABI -}; - -/** - * Color Transfer Characteristic. - */ -enum T_AVColorTransferCharacteristic { - AVCOL_TRC_RESERVED0 = 0, - AVCOL_TRC_BT709 = 1, ///< also ITU-R BT1361 - AVCOL_TRC_UNSPECIFIED = 2, - AVCOL_TRC_RESERVED = 3, - AVCOL_TRC_GAMMA22 = 4, ///< also ITU-R BT470M / ITU-R BT1700 625 PAL & SECAM - AVCOL_TRC_GAMMA28 = 5, ///< also ITU-R BT470BG - AVCOL_TRC_SMPTE170M = 6, ///< also ITU-R BT601-6 525 or 625 / ITU-R BT1358 525 or 625 / ITU-R BT1700 NTSC - AVCOL_TRC_SMPTE240M = 7, - AVCOL_TRC_LINEAR = 8, ///< "Linear transfer characteristics" - AVCOL_TRC_LOG = 9, ///< "Logarithmic transfer characteristic (100:1 range)" - AVCOL_TRC_LOG_SQRT = 10, ///< "Logarithmic transfer characteristic (100 * Sqrt(10) : 1 range)" - AVCOL_TRC_IEC61966_2_4 = 11, ///< IEC 61966-2-4 - AVCOL_TRC_BT1361_ECG = 12, ///< ITU-R BT1361 Extended Colour Gamut - AVCOL_TRC_IEC61966_2_1 = 13, ///< IEC 61966-2-1 (sRGB or sYCC) - AVCOL_TRC_BT2020_10 = 14, ///< ITU-R BT2020 for 10 bit system - AVCOL_TRC_BT2020_12 = 15, ///< ITU-R BT2020 for 12 bit system - AVCOL_TRC_NB, ///< Not part of ABI -}; - -/** - * YUV tColorspace type. - */ -enum T_AVColorSpace { - AVCOL_SPC_RGB = 0, ///< order of coefficients is actually GBR, also IEC 61966-2-1 (sRGB) - AVCOL_SPC_BT709 = 1, ///< also ITU-R BT1361 / IEC 61966-2-4 xvYCC709 / SMPTE RP177 Annex B - AVCOL_SPC_UNSPECIFIED = 2, - AVCOL_SPC_RESERVED = 3, - AVCOL_SPC_FCC = 4, ///< FCC Title 47 Code of Federal Regulations 73.682 (a)(20) - AVCOL_SPC_BT470BG = 5, ///< also ITU-R BT601-6 625 / ITU-R BT1358 625 / ITU-R BT1700 625 PAL & SECAM / IEC 61966-2-4 xvYCC601 - AVCOL_SPC_SMPTE170M = 6, ///< also ITU-R BT601-6 525 / ITU-R BT1358 525 / ITU-R BT1700 NTSC / functionally identical to above - AVCOL_SPC_SMPTE240M = 7, - AVCOL_SPC_YCOCG = 8, ///< Used by Dirac / VC-2 and H.264 FRext, see ITU-T SG16 - AVCOL_SPC_BT2020_NCL = 9, ///< ITU-R BT2020 non-constant luminance system - AVCOL_SPC_BT2020_CL = 10, ///< ITU-R BT2020 constant luminance system - AVCOL_SPC_NB, ///< Not part of ABI -}; - - -enum { - // 7.4.3.1: vps_max_layers_minus1 is in [0, 62]. - HEVC_MAX_LAYERS = 63, - // 7.4.3.1: vps_max_sub_layers_minus1 is in [0, 6]. - HEVC_MAX_SUB_LAYERS = 7, - // 7.4.3.1: vps_num_layer_sets_minus1 is in [0, 1023]. - HEVC_MAX_LAYER_SETS = 1024, - - // 7.4.2.1: vps_video_parameter_set_id is u(4). - HEVC_MAX_VPS_COUNT = 16, - // 7.4.3.2.1: sps_seq_parameter_set_id is in [0, 15]. - HEVC_MAX_SPS_COUNT = 16, - // 7.4.3.3.1: pps_pic_parameter_set_id is in [0, 63]. - HEVC_MAX_PPS_COUNT = 64, - - // A.4.2: MaxDpbSize is bounded above by 16. - HEVC_MAX_DPB_SIZE = 16, - // 7.4.3.1: vps_max_dec_pic_buffering_minus1[i] is in [0, MaxDpbSize - 1]. - HEVC_MAX_REFS = HEVC_MAX_DPB_SIZE, - - // 7.4.3.2.1: num_short_term_ref_pic_sets is in [0, 64]. - HEVC_MAX_SHORT_TERM_REF_PIC_SETS = 64, - // 7.4.3.2.1: num_long_term_ref_pics_sps is in [0, 32]. - HEVC_MAX_LONG_TERM_REF_PICS = 32, - - // A.3: all profiles require that CtbLog2SizeY is in [4, 6]. - HEVC_MIN_LOG2_CTB_SIZE = 4, - HEVC_MAX_LOG2_CTB_SIZE = 6, - - // E.3.2: cpb_cnt_minus1[i] is in [0, 31]. - HEVC_MAX_CPB_CNT = 32, - - // A.4.1: in table A.6 the highest level allows a MaxLumaPs of 35 651 584. - HEVC_MAX_LUMA_PS = 35651584, - // A.4.1: pic_width_in_luma_samples and pic_height_in_luma_samples are - // constrained to be not greater than sqrt(MaxLumaPs * 8). Hence height/ - // width are bounded above by sqrt(8 * 35651584) = 16888.2 samples. - HEVC_MAX_WIDTH = 16888, - HEVC_MAX_HEIGHT = 16888, - - // A.4.1: table A.6 allows at most 22 tile rows for any level. - HEVC_MAX_TILE_ROWS = 22, - // A.4.1: table A.6 allows at most 20 tile columns for any level. - HEVC_MAX_TILE_COLUMNS = 20, - - // 7.4.7.1: in the worst case (tiles_enabled_flag and - // entropy_coding_sync_enabled_flag are both set), entry points can be - // placed at the beginning of every Ctb row in every tile, giving an - // upper bound of (num_tile_columns_minus1 + 1) * PicHeightInCtbsY - 1. - // Only a stream with very high resolution and perverse parameters could - // get near that, though, so set a lower limit here with the maximum - // possible value for 4K video (at most 135 16x16 Ctb rows). - HEVC_MAX_ENTRY_POINT_OFFSETS = HEVC_MAX_TILE_COLUMNS * 135, -}; - - -/** - * rational number numerator/denominator - */ -typedef struct T_AVRational{ - int num; ///< numerator - int den; ///< denominator -} T_AVRational; - - -/*** - * Sequence parameter set - * ¿É²Î¿¼H264±ê×¼µÚ7½ÚºÍ¸½Â¼D E - /*** - * Sequence parameter set - * H.264 sequence parameter set, version 7 and above, D E - - * [AUTO-TRANSLATED:bd590cb8] - */ -#define Extended_SAR 255 - -/** - * Sequence parameter set - */ -typedef struct T_SPS { - unsigned int uiSpsId; - int iProfileIdc; - int iLevelIdc; - int iChromaFormatIdc; - int iTransformBypass; ///< qpprime_y_zero_transform_bypass_flag - int iLog2MaxFrameNum; ///< log2_max_frame_num_minus4 + 4 - int iPocType; ///< pic_order_cnt_type - int iLog2MaxPocLsb; ///< log2_max_pic_order_cnt_lsb_minus4 - int iDeltaPicOrderAlwaysZeroFlag; - int iOffsetForNonRefPic; - int iOffsetForTopToBottomField; - int iPocCycleLength; ///< num_ref_frames_in_pic_order_cnt_cycle - int iRefFrameCount; ///< num_ref_frames - int iGapsInFrameNumAllowedFlag; - int iMbWidth; ///< pic_width_in_mbs_minus1 + 1 - int iMbHeight; ///< pic_height_in_map_units_minus1 + 1 - int iFrameMbsOnlyFlag; - int iMbAff; ///< mb_adaptive_frame_field_flag - int iDirect8x8InferenceFlag; - int iCrop; ///< frame_cropping_flag - - /* those 4 are already in luma samples */ - unsigned int uiCropLeft; ///< frame_cropping_rect_left_offset - unsigned int uiCropRight; ///< frame_cropping_rect_right_offset - unsigned int uiCropTop; ///< frame_cropping_rect_top_offset - unsigned int uiCropBottom; ///< frame_cropping_rect_bottom_offset - int iVuiParametersPresentFlag; - T_AVRational tSar; - int iVideoSignalTypePresentFlag; - int iFullRange; - int iColourDescriptionPresentFlag; - enum T_AVColorPrimaries tColorPrimaries; - enum T_AVColorTransferCharacteristic tColorTrc; - enum T_AVColorSpace tColorspace; - int iTimingInfoPresentFlag; - uint32_t u32NumUnitsInTick; - uint32_t u32TimeScale; - int iFixedFrameRateFlag; - short asOffsetForRefFrame[256]; // FIXME dyn aloc? - int iBitstreamRestrictionFlag; - int iNumReorderFrames; - int iScalingMatrixPresent; - uint8_t aau8ScalingMatrix4[6][16]; - uint8_t aau8ScalingMatrix8[6][64]; - int iNalHrdParametersPresentFlag; - int iVclHrdParametersPresentFlag; - int iPicStructPresentFlag; - int iTimeOffsetLength; - int iCpbCnt; ///< See H.264 E.1.2 - int iInitialCpbRemovalDelayLength; ///< initial_cpb_removal_delay_length_minus1 + 1 - int iCpbRemovalDelayLength; ///< cpb_removal_delay_length_minus1 + 1 - int iDpbOutputDelayLength; ///< dpb_output_delay_length_minus1 + 1 - int iBitDepthLuma; ///< bit_depth_luma_minus8 + 8 - int iBitDepthChroma; ///< bit_depth_chroma_minus8 + 8 - int iResidualColorTransformFlag; ///< residual_colour_transform_flag - int iConstraintSetFlags; ///< constraint_set[0-3]_flag - int iNew; ///< flag to keep track if the decoder context needs re-init due to changed SPS -} T_SPS; - -/** - * Picture parameter set - */ -typedef struct T_PPS { - unsigned int uiSpsId; - int iCabac; ///< entropy_coding_mode_flag - int iPicOrderPresent; ///< pic_order_present_flag - int iSliceGroupCount; ///< num_slice_groups_minus1 + 1 - int iMbSliceGroupMapType; - unsigned int auiRefCount[2]; ///< num_ref_idx_l0/1_active_minus1 + 1 - int iWeightedPred; ///< weighted_pred_flag - int iWeightedBipredIdc; - int iInitQp; ///< pic_init_qp_minus26 + 26 - int iInitQs; ///< pic_init_qs_minus26 + 26 - int aiChromaQpIndexOffset[2]; - int iDeblockingFilterParametersPresent; ///< deblocking_filter_parameters_present_flag - int iConstrainedIntraPred; ///< constrained_intra_pred_flag - int iRedundantPicCntPresent; ///< redundant_pic_cnt_present_flag - int iTransform8x8Mode; ///< transform_8x8_mode_flag - uint8_t aau8ScalingMatrix4[6][16]; - uint8_t aau8ScalingMatrix8[6][64]; - uint8_t u8ChromaQpTable[2][QP_MAX_NUM+1]; ///< pre-scaled (with aiChromaQpIndexOffset) version of qp_table - int iChromaQpDiff; -} T_PPS; - - -typedef struct T_HEVCWindow { - unsigned int uiLeftOffset; - unsigned int uiRightOffset; - unsigned int uiTopOffset; - unsigned int uiBottomOffset; -} T_HEVCWindow; - - -typedef struct T_VUI { - T_AVRational tSar; - - int iOverscanInfoPresentFlag; - int iOverscanAppropriateFlag; - - int iVideoSignalTypePresentFlag; - int iVideoFormat; - int iVideoFullRangeFlag; - int iColourDescriptionPresentFlag; - uint8_t u8ColourPrimaries; - uint8_t u8TransferCharacteristic; - uint8_t u8MatrixCoeffs; - - int iChromaLocInfoPresentFlag; - int iChromaSampleLocTypeTopField; - int iChromaSampleLocTypeBottomField; - int iNeutraChromaIndicationFlag; - - int iFieldSeqFlag; - int iFrameFieldInfoPresentFlag; - - int iDefaultDisplayWindowFlag; - T_HEVCWindow tDefDispWin; - - int iVuiTimingInfoPresentFlag; - uint32_t u32VuiNumUnitsInTick; - uint32_t u32VuiTimeScale; - int iVuiPocProportionalToTimingFlag; - int iVuiNumTicksPocDiffOneMinus1; - int iVuiHrdParametersPresentFlag; - - int iBitstreamRestrictionFlag; - int iTilesFixedStructureFlag; - int iMotionVectorsOverPicBoundariesFlag; - int iRestrictedRefPicListsFlag; - int iMinSpatialSegmentationIdc; - int iMaxBytesPerPicDenom; - int iMaxBitsPerMinCuDenom; - int iLog2MaxMvLengthHorizontal; - int iLog2MaxMvLengthVertical; -} T_VUI; - -typedef struct T_PTLCommon { - uint8_t u8ProfileSpace; - uint8_t u8TierFlag; - uint8_t u8ProfileIdc; - uint8_t au8ProfileCompatibilityFlag[32]; - uint8_t u8LevelIdc; - uint8_t u8ProgressiveSourceFlag; - uint8_t u8InterlacedSourceFlag; - uint8_t u8NonPackedConstraintFlag; - uint8_t u8FrameOnlyConstraintFlag; -} T_PTLCommon; - -typedef struct T_PTL { - T_PTLCommon tGeneralPtl; - T_PTLCommon atSubLayerPtl[HEVC_MAX_SUB_LAYERS]; - - uint8_t au8SubLayerProfilePresentFlag[HEVC_MAX_SUB_LAYERS]; - uint8_t au8SubLayerLevelPresentFlag[HEVC_MAX_SUB_LAYERS]; -} T_PTL; - -typedef struct T_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 aaau8Sl[4][6][64]; - uint8_t aau8SlDc[2][6]; -} T_ScalingList; - -typedef struct T_ShortTermRPS { - unsigned int uiNumNegativePics; - int iNumDeltaPocs; - int iRpsIdxNumDeltaPocs; - int32_t au32DeltaPoc[32]; - uint8_t au8Used[32]; -} T_ShortTermRPS; - - -typedef struct T_HEVCVPS { - uint8_t u8VpsTemporalIdNestingFlag; - int iVpsMaxLayers; - int iVpsMaxSubLayers; ///< vps_max_temporal_layers_minus1 + 1 - - T_PTL tPtl; - int iVpsSubLayerOrderingInfoPresentFlag; - unsigned int uiVpsMaxDecPicBuffering[HEVC_MAX_SUB_LAYERS]; - unsigned int auiVpsNumReorderPics[HEVC_MAX_SUB_LAYERS]; - unsigned int auiVpsMaxLatencyIncrease[HEVC_MAX_SUB_LAYERS]; - int iVpsMaxLayerId; - int iVpsNumLayerSets; ///< vps_num_layer_sets_minus1 + 1 - uint8_t u8VpsTimingInfoPresentFlag; - uint32_t u32VpsNumUnitsInTick; - uint32_t u32VpsTimeScale; - uint8_t u8VpsPocProportionalToTimingFlag; - int iVpsNumTicksPocDiffOne; ///< vps_num_ticks_poc_diff_one_minus1 + 1 - int iVpsNumHrdParameters; - -} T_HEVCVPS; - -typedef struct T_HEVCSPS { - unsigned int uiVpsId; - int iChromaFormatIdc; - uint8_t u8SeparateColourPlaneFlag; - - ///< output (i.e. cropped) values - int iIutputWidth, iOutputHeight; - T_HEVCWindow tOutputWindow; - - T_HEVCWindow tPicConfWin; - - int iBitDepth; - int iBitDepthChroma; - int iPixelShift; - - unsigned int uiLog2MaxPocLsb; - int iPcmEnabledFlag; - - int iMaxSubLayers; - struct { - int iMaxDecPicBuffering; - int iNumReorderPics; - int iMaxLatencyIncrease; - } stTemporalLayer[HEVC_MAX_SUB_LAYERS]; - uint8_t u8temporalIdNestingFlag; - - T_VUI tVui; - T_PTL tPtl; - - uint8_t u8ScalingListEnableFlag; - T_ScalingList tScalingList; - - unsigned int uiNbStRps; - T_ShortTermRPS atStRps[HEVC_MAX_SHORT_TERM_RPS_COUNT]; - - uint8_t u8AmpEnabledFlag; - uint8_t u8SaoEnabled; - - uint8_t u8LongTermRefPicsPresentFlag; - uint16_t au16LtRefPicPocLsbSps[32]; - uint8_t au8UsedByCurrPicLtSpsFlag[32]; - uint8_t u8NumLongTermRefPicsSps; - - struct { - uint8_t u8BitDepth; - uint8_t u8BitDepthChroma; - unsigned int uiLog2MinPcmCbSize; - unsigned int uiLog2MaxPcmCbSize; - uint8_t u8LoopFilterDisableFlag; - } pcm; - uint8_t u8SpsTemporalMvpEnabledFlag; - uint8_t u8SpsStrongIntraMmoothingEnableFlag; - - unsigned int uiLog2MinCbSize; - unsigned int uiLog2DiffMaxMinCodingBlockSize; - unsigned int uiLog2MinTbSize; - unsigned int uiLog2MaxTrafoSize; - unsigned int uiLog2CtbSize; - unsigned int uiLog2MinPuSize; - - int iMaxTransformHierarchyDepthInter; - int iMaxTransformHierarchyDepthIntra; - - int iTransformSkipRotationEnabledFlag; - int iTransformSkipContextEnabledFlag; - int iImplicitRdpcmEnabledFlag; - int iExplicitRdpcmEnabledFlag; - int iIntraSmoothingDisabledFlag; - int iHighPrecisionOffsetsEnabledFlag; - int iPersistentRiceAdaptationEnabledFlag; - - ///< coded frame dimension in various units - int iWidth; - int iHeight; - int iCtbWidth; - int iCtbHeight; - int iCtbSize; - int iMinCbWidth; - int iMinCbHeight; - int iMinTbWidth; - int iMinTbHeight; - int iMinPuWidth; - int iMinPuHeight; - int iTbMask; - - int aiHshift[3]; - int aiVshift[3]; - - int iQpBdOffset; - - int iVuiPresent; -}T_HEVCSPS; - -typedef struct { - int pps_pic_parameter_set_id; - int pps_seq_parameter_set_id; - int dependent_slice_segments_enabled_flag; - int output_flag_present_flag; - int num_extra_slice_header_bits; - int sign_data_hiding_enabled_flag; - int cabac_init_present_flag; - int num_ref_idx_l0_default_active_minus1; - int num_ref_idx_l1_default_active_minus1; - int init_qp_minus26; - int constrained_intra_pred_flag; - int transform_skip_enabled_flag; - int cu_qp_delta_enabled_flag; - int diff_cu_qp_delta_depth; - int pps_cb_qp_offset; - int pps_cr_qp_offset; - int pps_slice_chroma_qp_offsets_present_flag; - int weighted_pred_flag; - int weighted_bipred_flag; - int transquant_bypass_enabled_flag; - int tiles_enabled_flag; - int entropy_coding_sync_enabled_flag; - int uniform_spacing_flag; - int loop_filter_across_tiles_enabled_flag; - int pps_loop_filter_across_slices_enabled_flag; - int deblocking_filter_control_present_flag; - int deblocking_filter_override_enabled_flag; - int pps_deblocking_filter_disabled_flag; - int pps_beta_offset_div2; - int pps_tc_offset_div2; - int pps_scaling_list_data_present_flag; - int lists_modification_present_flag; - int log2_parallel_merge_level_minus2; - int slice_segment_header_extension_present_flag; - int pps_extension_present_flag; - int pps_range_extension_flag; - int pps_multilayer_extension_flag; - int pps_3d_extension_flag; - int pps_scc_extension_flag; - int pps_extension_4bits; - - // PPS range extension fields - int log2_max_transform_skip_block_size_minus2; - int cross_component_prediction_enabled_flag; - int chroma_qp_offset_list_enabled_flag; - int diff_cu_chroma_qp_offset_depth; - int chroma_qp_offset_list_len_minus1; - int cb_qp_offset_list[6]; - int cr_qp_offset_list[6]; - int log2_sao_offset_scale_luma; - int log2_sao_offset_scale_chroma; - - // 可以根据需要添加更多字段 [AUTO-TRANSLATED:57b9a7c1] - // You can add more fields as needed -} T_HEVC_PPS; - -typedef struct T_GetBitContext{ - uint8_t *pu8Buf; // buf - int iBufSize; // buf size - int iBitPos; // bit position - int iTotalBit; // bit number - int iCurBitPos; // current bit position -}T_GetBitContext; - -int h265ParsePps(T_GetBitContext *ptGetBitContext, T_HEVC_PPS *ptPps); -int h264DecSeqParameterSet(void *pvBuf, T_SPS *ptSps); -int h265DecSeqParameterSet( void *pvBufSrc, T_HEVCSPS *ptSps ); -int h265DecVideoParameterSet( void *pvBufSrc, T_HEVCVPS *ptVps ); - - -void h264GetWidthHeight(T_SPS *ptSps, int *piWidth, int *piHeight); -void h265GetWidthHeight(T_HEVCSPS *ptSps, int *piWidth, int *piHeight); - -void h264GeFramerate(T_SPS *ptSps, float *pfFramerate); -void h265GeFramerate(T_HEVCVPS *ptVps, T_HEVCSPS *ptSps,float *pfFramerate); - -#if defined (__cplusplus) -} -#endif - -#endif //_SPS_PPS_H_