/* Gstreamer H.266 bitstream parser
*
* Copyright (C) 2023 Intel Corporation
* Author: Zhong Hongcheng <spartazhc@gmail.com>
* Author: He Junyan <junyan.he@intel.com>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
/**
* SECTION:gsth266parser
* @title: GstH266Parser
* @short_description: Convenience library for h266 video bitstream parsing.
*
* To identify Nals in a bitstream and parse its headers, you should call:
*
* * gst_h266_parser_identify_nalu() to identify the following nalu in
* VVC bitstreams
*
* Then, depending on the #GstH266NalUnitType of the newly parsed #GstH266NalUnit,
* you should call the different functions to parse the structure:
*
* * From #GST_H266_NAL_SLICE_TRAIL to #GST_H266_NAL_SLICE_GDR:
* gst_h266_parser_parse_slice_hdr()
*
* * `GST_H266_NAL_*_SEI`: gst_h266_parser_parse_sei()
*
* * #GST_H266_NAL_VPS: gst_h266_parser_parse_vps()
*
* * #GST_H266_NAL_SPS: gst_h266_parser_parse_sps()
*
* * #GST_H266_NAL_PPS: #gst_h266_parser_parse_pps()
*
* * Any other: gst_h266_parser_parse_nal()
*
* Note: You should always call gst_h266_parser_parse_nal() if you don't
* actually need #GstH266NalUnitType to be parsed for your personal use, in
* order to guarantee that the #GstH266Parser is always up to date.
*
* For more details about the structures, look at the ITU-T H.266
* specifications, you can download them from:
*
* * ITU-T H.266: http://www.itu.int/rec/T-REC-H.266
*
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "gsth266parser.h"
#include "nalutils.h"
#include <gst/base/gstbytereader.h>
#include <gst/base/gstbitreader.h>
#ifndef GST_DISABLE_GST_DEBUG
#define GST_CAT_DEFAULT gst_h266_debug_category_get()
static GstDebugCategory *
gst_h266_debug_category_get (void)
{
static gsize cat_gonce = 0;
if (g_once_init_enter (&cat_gonce)) {
GstDebugCategory *cat = NULL;
GST_DEBUG_CATEGORY_INIT (cat, "codecparsers_h266", 0, "h266 parse library");
g_once_init_leave (&cat_gonce, (gsize) cat);
}
return (GstDebugCategory *) cat_gonce;
}
#endif /* GST_DISABLE_GST_DEBUG */
const guint8 scaling_pred_all_8[8 * 8] = {
8, 8, 8, 8, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 8, 8,
};
const guint8 scaling_pred_all_16[8 * 8] = {
16, 16, 16, 16, 16, 16, 16, 16,
16, 16, 16, 16, 16, 16, 16, 16,
16, 16, 16, 16, 16, 16, 16, 16,
16, 16, 16, 16, 16, 16, 16, 16,
16, 16, 16, 16, 16, 16, 16, 16,
16, 16, 16, 16, 16, 16, 16, 16,
16, 16, 16, 16, 16, 16, 16, 16,
16, 16, 16, 16, 16, 16, 16, 16,
};
/* *INDENT-OFF* */
/* Only need square matrix at most 8x8 for syntax level. */
static const guint8 square_DiagScanOrder_x[4][8 * 8] = {
/* 1x1 */
{ 0, },
/* 2x2 */
{ 0, 0, 1, 1, },
/* 4x4 */
{ 0, 0, 1, 0, 1, 2, 0, 1, 2, 3, 1, 2, 3, 2, 3, 3, },
/* 8x8 */
{ 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, },
};
/* Only need square matrix at most 8x8 for syntax level. */
static const guint8 square_DiagScanOrder_y[4][8 * 8] = {
/* 1x1 */
{ 0, },
/* 2x2 */
{ 0, 1, 0, 1, },
/* 4x4 */
{ 0, 1, 0, 2, 1, 0, 3, 2, 1, 0, 3, 2, 1, 3, 2, 3, },
/* 8x8 */
{ 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, }
};
/* *INDENT-ON* */
typedef struct
{
guint par_n, par_d;
} PAR;
/* ITU-T Rec. H.273 | ISO/IEC 23091-2. Table 7
* Meaning of sample aspect ratio indicator (SampleAspectRatio) */
static const PAR aspect_ratios[17] = {
{0, 0},
{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}
};
/***** Utils ****/
#define EXTENDED_SAR 255
static inline GstH266VPS *
gst_h266_parser_get_vps (GstH266Parser * parser, guint8 id)
{
return (parser->vps[id].valid) ? &parser->vps[id] : NULL;
}
static inline GstH266SPS *
gst_h266_parser_get_sps (GstH266Parser * parser, guint8 id)
{
return (parser->sps[id].valid) ? &parser->sps[id] : NULL;
}
static inline GstH266PPS *
gst_h266_parser_get_pps (GstH266Parser * parser, guint8 id)
{
return (parser->pps[id].valid) ? &parser->pps[id] : NULL;
}
static gboolean
gst_h266_parse_nalu_header (GstH266NalUnit * nalu)
{
guint8 *data = nalu->data + nalu->offset;
GstBitReader br;
if (nalu->size < 2)
return FALSE;
gst_bit_reader_init (&br, data, nalu->size - nalu->offset);
/* skip the forbidden_zero_bit and nuh_reserved_zero_bit */
gst_bit_reader_skip_unchecked (&br, 2);
nalu->layer_id = gst_bit_reader_get_bits_uint8_unchecked (&br, 6);
nalu->type = gst_bit_reader_get_bits_uint8_unchecked (&br, 5);
nalu->temporal_id_plus1 = gst_bit_reader_get_bits_uint8_unchecked (&br, 3);
nalu->header_bytes = 2;
if (nalu->layer_id > 55) {
GST_WARNING ("The value of nuh_layer_id shall be in the "
"range of 0 to 55, inclusive");
return FALSE;
}
/* rules for base layer */
if (nalu->layer_id == 0) {
if (nalu->temporal_id_plus1 - 1 == 0 &&
nalu->type == GST_H266_NAL_SLICE_STSA) {
GST_WARNING ("When NAL unit type is equal to STSA_NUT, "
"TemporalId shall not be equal to 0");
return FALSE;
}
}
return TRUE;
}
struct h266_profile_string
{
GstH266Profile profile;
const gchar *name;
};
static const struct h266_profile_string h266_profiles[] = {
/* keep in sync with definition in the header */
{GST_H266_PROFILE_STILL_PICTURE, "still-picture"},
{GST_H266_PROFILE_MAIN_10, "main-10"},
{GST_H266_PROFILE_MAIN_10_STILL_PICTURE, "main-10-still-picture"},
{GST_H266_PROFILE_MULTILAYER_MAIN_10, "multilayer-main-10"},
{GST_H266_PROFILE_MULTILAYER_MAIN_10_STILL_PICTURE,
"multilayer-main-10-still-picture"},
{GST_H266_PROFILE_MAIN_10_444, "main-10-444"},
{GST_H266_PROFILE_MAIN_10_444_STILL_PICTURE, "main-10-444-still-picture"},
{GST_H266_PROFILE_MULTILAYER_MAIN_10_444, "multilayer-main-10-444"},
{GST_H266_PROFILE_MULTILAYER_MAIN_10_444_STILL_PICTURE,
"multilayer-main-10-444-still-picture"},
{GST_H266_PROFILE_MAIN_12, "main-12"},
{GST_H266_PROFILE_MAIN_12_444, "main-12-444"},
{GST_H266_PROFILE_MAIN_16_444, "main-16-444"},
{GST_H266_PROFILE_MAIN_12_INTRA, "main-12-intra"},
{GST_H266_PROFILE_MAIN_12_444_INTRA, "main-12-444-intra"},
{GST_H266_PROFILE_MAIN_16_444_INTRA, "main-16-444-intra"},
{GST_H266_PROFILE_MAIN_12_STILL_PICTURE, "main-12-still-picture"},
{GST_H266_PROFILE_MAIN_12_444_STILL_PICTURE, "main-12-444-still-picture"},
{GST_H266_PROFILE_MAIN_16_444_STILL_PICTURE, "main-16-444-still-picture"},
};
static gboolean
gst_h266_parse_general_constraints_info (GstH266GeneralConstraintsInfo * gci,
NalReader * nr)
{
guint8 num_additional_bits = 0;
guint8 num_additional_bits_used = 0;
GST_LOG ("parsing \"General Constraints Info Parameters\"");
READ_UINT8 (nr, gci->present_flag, 1);
if (gci->present_flag) {
/* general */
READ_UINT8 (nr, gci->intra_only_constraint_flag, 1);
READ_UINT8 (nr, gci->all_layers_independent_constraint_flag, 1);
READ_UINT8 (nr, gci->one_au_only_constraint_flag, 1);
/* picture format */
READ_UINT8 (nr, gci->sixteen_minus_max_bitdepth_constraint_idc, 4);
CHECK_ALLOWED_MAX (gci->sixteen_minus_max_bitdepth_constraint_idc, 8);
READ_UINT8 (nr, gci->three_minus_max_chroma_format_constraint_idc, 2);
/* NAL unit type related */
READ_UINT8 (nr, gci->no_mixed_nalu_types_in_pic_constraint_flag, 1);
READ_UINT8 (nr, gci->no_trail_constraint_flag, 1);
READ_UINT8 (nr, gci->no_stsa_constraint_flag, 1);
READ_UINT8 (nr, gci->no_rasl_constraint_flag, 1);
READ_UINT8 (nr, gci->no_radl_constraint_flag, 1);
READ_UINT8 (nr, gci->no_idr_constraint_flag, 1);
READ_UINT8 (nr, gci->no_cra_constraint_flag, 1);
READ_UINT8 (nr, gci->no_gdr_constraint_flag, 1);
READ_UINT8 (nr, gci->no_aps_constraint_flag, 1);
READ_UINT8 (nr, gci->no_idr_rpl_constraint_flag, 1);
/* tile, slice, subpicture partitioning */
READ_UINT8 (nr, gci->one_tile_per_pic_constraint_flag, 1);
READ_UINT8 (nr, gci->pic_header_in_slice_header_constraint_flag, 1);
READ_UINT8 (nr, gci->one_slice_per_pic_constraint_flag, 1);
READ_UINT8 (nr, gci->no_rectangular_slice_constraint_flag, 1);
READ_UINT8 (nr, gci->one_slice_per_subpic_constraint_flag, 1);
READ_UINT8 (nr, gci->no_subpic_info_constraint_flag, 1);
/* CTU and block partitioning */
READ_UINT8 (nr, gci->three_minus_max_log2_ctu_size_constraint_idc, 2);
READ_UINT8 (nr, gci->no_partition_constraints_override_constraint_flag, 1);
READ_UINT8 (nr, gci->no_mtt_constraint_flag, 1);
READ_UINT8 (nr, gci->no_qtbtt_dual_tree_intra_constraint_flag, 1);
/* intra */
READ_UINT8 (nr, gci->no_palette_constraint_flag, 1);
READ_UINT8 (nr, gci->no_ibc_constraint_flag, 1);
READ_UINT8 (nr, gci->no_isp_constraint_flag, 1);
READ_UINT8 (nr, gci->no_mrl_constraint_flag, 1);
READ_UINT8 (nr, gci->no_mip_constraint_flag, 1);
READ_UINT8 (nr, gci->no_cclm_constraint_flag, 1);
/* inter */
READ_UINT8 (nr, gci->no_ref_pic_resampling_constraint_flag, 1);
READ_UINT8 (nr, gci->no_res_change_in_clvs_constraint_flag, 1);
READ_UINT8 (nr, gci->no_weighted_prediction_constraint_flag, 1);
READ_UINT8 (nr, gci->no_ref_wraparound_constraint_flag, 1);
READ_UINT8 (nr, gci->no_temporal_mvp_constraint_flag, 1);
READ_UINT8 (nr, gci->no_sbtmvp_constraint_flag, 1);
READ_UINT8 (nr, gci->no_amvr_constraint_flag, 1);
READ_UINT8 (nr, gci->no_bdof_constraint_flag, 1);
READ_UINT8 (nr, gci->no_smvd_constraint_flag, 1);
READ_UINT8 (nr, gci->no_dmvr_constraint_flag, 1);
READ_UINT8 (nr, gci->no_mmvd_constraint_flag, 1);
READ_UINT8 (nr, gci->no_affine_motion_constraint_flag, 1);
READ_UINT8 (nr, gci->no_prof_constraint_flag, 1);
READ_UINT8 (nr, gci->no_bcw_constraint_flag, 1);
READ_UINT8 (nr, gci->no_ciip_constraint_flag, 1);
READ_UINT8 (nr, gci->no_gpm_constraint_flag, 1);
/* transform, quantization, residual */
READ_UINT8 (nr, gci->no_luma_transform_size_64_constraint_flag, 1);
READ_UINT8 (nr, gci->no_transform_skip_constraint_flag, 1);
READ_UINT8 (nr, gci->no_bdpcm_constraint_flag, 1);
READ_UINT8 (nr, gci->no_mts_constraint_flag, 1);
READ_UINT8 (nr, gci->no_lfnst_constraint_flag, 1);
READ_UINT8 (nr, gci->no_joint_cbcr_constraint_flag, 1);
READ_UINT8 (nr, gci->no_sbt_constraint_flag, 1);
READ_UINT8 (nr, gci->no_act_constraint_flag, 1);
READ_UINT8 (nr, gci->no_explicit_scaling_list_constraint_flag, 1);
READ_UINT8 (nr, gci->no_dep_quant_constraint_flag, 1);
READ_UINT8 (nr, gci->no_sign_data_hiding_constraint_flag, 1);
READ_UINT8 (nr, gci->no_cu_qp_delta_constraint_flag, 1);
READ_UINT8 (nr, gci->no_chroma_qp_offset_constraint_flag, 1);
/* loop fitler */
READ_UINT8 (nr, gci->no_sao_constraint_flag, 1);
READ_UINT8 (nr, gci->no_alf_constraint_flag, 1);
READ_UINT8 (nr, gci->no_ccalf_constraint_flag, 1);
READ_UINT8 (nr, gci->no_lmcs_constraint_flag, 1);
READ_UINT8 (nr, gci->no_ladf_constraint_flag, 1);
READ_UINT8 (nr, gci->no_virtual_boundaries_constraint_flag, 1);
READ_UINT8 (nr, num_additional_bits, 8);
if (num_additional_bits > 5) {
READ_UINT8 (nr, gci->all_rap_pictures_constraint_flag, 1);
READ_UINT8 (nr, gci->no_extended_precision_processing_constraint_flag, 1);
READ_UINT8 (nr, gci->no_ts_residual_coding_rice_constraint_flag, 1);
READ_UINT8 (nr, gci->no_rrc_rice_extension_constraint_flag, 1);
READ_UINT8 (nr, gci->no_persistent_rice_adaptation_constraint_flag, 1);
READ_UINT8 (nr, gci->no_reverse_last_sig_coeff_constraint_flag, 1);
num_additional_bits_used = 6;
} else if (num_additional_bits > 0) {
GST_WARNING ("Invalid bitstream: gci_num_additional_bits set "
"to value %d (must be 0 or >= 6)\n", num_additional_bits);
goto error;
}
/* skip the reserved zero bits */
if (!nal_reader_skip (nr, num_additional_bits - num_additional_bits_used))
goto error;
}
while (!nal_reader_is_byte_aligned (nr)) {
if (!nal_reader_skip (nr, 1))
goto error;
}
return TRUE;
error:
GST_WARNING ("error parsing \"General Constraints Info Parameters\"");
return FALSE;
}
static gboolean
gst_h266_parse_profile_tier_level (GstH266ProfileTierLevel * ptl,
NalReader * nr, guint8 profileTierPresentFlag, guint8 MaxNumSubLayersMinus1)
{
gint8 i;
GST_LOG ("parsing \"Profile Tier Level parameters\"");
if (profileTierPresentFlag) {
guint8 profile_idc;
READ_UINT8 (nr, profile_idc, 7);
READ_UINT8 (nr, ptl->tier_flag, 1);
ptl->profile_idc = profile_idc;
}
READ_UINT8 (nr, ptl->level_idc, 8);
if (ptl->profile_idc != GST_H266_PROFILE_NONE &&
ptl->level_idc < /* level 4 */ 64 && ptl->tier_flag) {
GST_WARNING ("High tier not defined for levels below 4");
goto error;
}
READ_UINT8 (nr, ptl->frame_only_constraint_flag, 1);
READ_UINT8 (nr, ptl->multilayer_enabled_flag, 1);
if ((ptl->profile_idc == GST_H266_PROFILE_MAIN_10 ||
ptl->profile_idc == GST_H266_PROFILE_MAIN_10_444 ||
ptl->profile_idc == GST_H266_PROFILE_MAIN_10_STILL_PICTURE ||
ptl->profile_idc == GST_H266_PROFILE_MAIN_10_444_STILL_PICTURE)
&& ptl->multilayer_enabled_flag) {
GST_WARNING ("ptl_multilayer_enabled_flag shall be"
" equal to 0 for non-multilayer profiles");
goto error;
}
if (profileTierPresentFlag) {
if (!gst_h266_parse_general_constraints_info
(&ptl->general_constraints_info, nr))
goto error;
}
for (i = MaxNumSubLayersMinus1 - 1; i >= 0; i--)
READ_UINT8 (nr, ptl->sublayer_level_present_flag[i], 1);
/* skip the reserved zero bits */
while (!nal_reader_is_byte_aligned (nr)) {
if (!nal_reader_skip (nr, 1))
goto error;
}
for (i = MaxNumSubLayersMinus1 - 1; i >= 0; i--)
if (ptl->sublayer_level_present_flag[i])
READ_UINT8 (nr, ptl->sublayer_level_idc[i], 8);
if (profileTierPresentFlag) {
READ_UINT8 (nr, ptl->num_sub_profiles, 8);
for (i = 0; i < ptl->num_sub_profiles; i++) {
guint32 sub_profile_idc;
READ_UINT32 (nr, sub_profile_idc, 32);
ptl->sub_profile_idc[i] = sub_profile_idc;
}
}
return TRUE;
error:
GST_WARNING ("error parsing \"Profile Tier Level Parameters\"");
return FALSE;
}
static void
gst_h266_vui_parameters_set_default (GstH266VUIParams * vui)
{
GST_LOG ("setting \"VUI parameters set default\"");
/* Annex D.8 */
/* *INDENT-OFF* */
*vui = (GstH266VUIParams) {
.colour_primaries = 2,
.transfer_characteristics = 2,
.matrix_coeffs = 2,
.chroma_sample_loc_type_frame = 6,
.chroma_sample_loc_type_top_field = 6,
.chroma_sample_loc_type_bottom_field = 6,
};
/* *INDENT-ON* */
}
static gboolean
gst_h266_parse_vui_parameters (GstH266VUIParams * vui, NalReader * nr)
{
GST_LOG ("parsing \"VUI parameters\"");
READ_UINT8 (nr, vui->progressive_source_flag, 1);
READ_UINT8 (nr, vui->interlaced_source_flag, 1);
READ_UINT8 (nr, vui->non_packed_constraint_flag, 1);
READ_UINT8 (nr, vui->non_projected_constraint_flag, 1);
READ_UINT8 (nr, vui->aspect_ratio_info_present_flag, 1);
if (vui->aspect_ratio_info_present_flag) {
READ_UINT8 (nr, vui->aspect_ratio_constant_flag, 1);
READ_UINT8 (nr, vui->aspect_ratio_idc, 8);
if (vui->aspect_ratio_idc == EXTENDED_SAR) {
READ_UINT16 (nr, vui->sar_width, 16);
READ_UINT16 (nr, vui->sar_height, 16);
vui->par_n = vui->sar_width;
vui->par_d = vui->sar_height;
} else {
vui->par_n = aspect_ratios[vui->aspect_ratio_idc].par_n;
vui->par_d = aspect_ratios[vui->aspect_ratio_idc].par_d;
}
} else {
vui->aspect_ratio_constant_flag = 0;
vui->aspect_ratio_idc = 0;
}
READ_UINT8 (nr, vui->overscan_info_present_flag, 1);
if (vui->overscan_info_present_flag)
READ_UINT8 (nr, vui->overscan_appropriate_flag, 1);
READ_UINT8 (nr, vui->colour_description_present_flag, 1);
if (vui->colour_description_present_flag) {
READ_UINT8 (nr, vui->colour_primaries, 8);
READ_UINT8 (nr, vui->transfer_characteristics, 8);
READ_UINT8 (nr, vui->matrix_coeffs, 8);
READ_UINT8 (nr, vui->full_range_flag, 1);
} else {
vui->colour_primaries = 2;
vui->transfer_characteristics = 2;
vui->matrix_coeffs = 2;
vui->full_range_flag = 0;
}
READ_UINT8 (nr, vui->chroma_loc_info_present_flag, 1);
if (vui->chroma_loc_info_present_flag) {
if (vui->progressive_source_flag && !vui->interlaced_source_flag) {
READ_UE_MAX (nr, vui->chroma_sample_loc_type_frame, 6);
} else {
READ_UE_MAX (nr, vui->chroma_sample_loc_type_top_field, 6);
READ_UE_MAX (nr, vui->chroma_sample_loc_type_bottom_field, 6);
}
} else {
vui->chroma_sample_loc_type_frame = 6;
vui->chroma_sample_loc_type_top_field = vui->chroma_sample_loc_type_frame;
vui->chroma_sample_loc_type_bottom_field =
vui->chroma_sample_loc_type_frame;
}
return TRUE;
error:
GST_WARNING ("error parsing \"VUI parameters\"");
return FALSE;
}
static gboolean
nal_reader_has_more_data_in_payload (NalReader * nr,
guint32 payload_start_pos_bit, guint32 payloadSize)
{
if (nal_reader_is_byte_aligned (nr) &&
(nal_reader_get_pos (nr) >= (payload_start_pos_bit + 8 * payloadSize)))
return FALSE;
return TRUE;
}
static gboolean
gst_h266_parse_vui_payload (GstH266VUIParams * vui, NalReader * nr,
guint16 vui_payload_size)
{
guint32 payload_start_pos;
GST_LOG ("parsing \"VUI payload\"");
payload_start_pos = nal_reader_get_pos (nr);
if (!gst_h266_parse_vui_parameters (vui, nr))
goto error;
if (nal_reader_has_more_data_in_payload (nr, payload_start_pos,
vui_payload_size)) {
if (!nal_reader_skip (nr, 1))
goto error;
while (!nal_reader_is_byte_aligned (nr)) {
if (!nal_reader_skip (nr, 1))
goto error;
}
}
return TRUE;
error:
GST_WARNING ("error parsing \"VUI payload\"");
return FALSE;
}
static gboolean
gst_h266_parse_dpb_parameters (GstH266DPBParameters * dpb,
NalReader * nr, guint8 MaxSubLayersMinus1, guint8 subLayerInfoFlag)
{
gint i;
GST_LOG ("parsing \"DPB Parameters\"");
for (i = (subLayerInfoFlag ? 0 : MaxSubLayersMinus1);
i <= MaxSubLayersMinus1; i++) {
READ_UE_MAX (nr, dpb->max_dec_pic_buffering_minus1[i],
GST_H266_MAX_DPB_SIZE - 1);
READ_UE_MAX (nr, dpb->max_num_reorder_pics[i],
dpb->max_dec_pic_buffering_minus1[i]);
READ_UE_MAX (nr, dpb->max_latency_increase_plus1[i], G_MAXINT32 - 1);
}
return TRUE;
error:
GST_WARNING ("error parsing \"DPB Parameters\"");
return FALSE;
}
static gboolean
gst_h266_ref_pic_list_struct (GstH266RefPicListStruct * rpls, NalReader * nr,
guint8 list_idx, guint8 rpls_idx, const GstH266SPS * sps)
{
gint i;
GST_LOG ("parsing \"ref_pic_list_struct\"");
memset (rpls, 0, sizeof (*rpls));
READ_UE_MAX (nr, rpls->num_ref_entries, GST_H266_MAX_REF_ENTRIES);
if (sps->long_term_ref_pics_flag &&
rpls_idx < sps->num_ref_pic_lists[list_idx] &&
rpls->num_ref_entries > 0) {
READ_UINT8 (nr, rpls->ltrp_in_header_flag, 1);
} else if (sps->long_term_ref_pics_flag)
rpls->ltrp_in_header_flag = 1;
for (i = 0; i < rpls->num_ref_entries; i++) {
if (sps->inter_layer_prediction_enabled_flag) {
READ_UINT8 (nr, rpls->inter_layer_ref_pic_flag[i], 1);
} else
rpls->inter_layer_ref_pic_flag[i] = 0;
if (rpls->inter_layer_ref_pic_flag[i]) {
READ_UE_MAX (nr, rpls->ilrp_idx[i], GST_H266_MAX_REF_PIC_LISTS);
rpls->num_inter_layer_pic++;
continue;
}
if (sps->long_term_ref_pics_flag) {
READ_UINT8 (nr, rpls->st_ref_pic_flag[i], 1);
} else
rpls->st_ref_pic_flag[i] = 1;
if (rpls->st_ref_pic_flag[i]) {
gint abs_delta_poc_st;
READ_UE_MAX (nr, rpls->abs_delta_poc_st[i], G_MAXUINT16 - 1);
if ((sps->weighted_pred_flag || sps->weighted_bipred_flag) && i != 0)
abs_delta_poc_st = rpls->abs_delta_poc_st[i];
else
abs_delta_poc_st = rpls->abs_delta_poc_st[i] + 1;
if (abs_delta_poc_st > 0) {
READ_UINT8 (nr, rpls->strp_entry_sign_flag[i], 1);
}
rpls->delta_poc_val_st[i] =
(1 - 2 * rpls->strp_entry_sign_flag[i]) * abs_delta_poc_st;
rpls->num_short_term_pic++;
} else {
if (!rpls->ltrp_in_header_flag) {
READ_UINT8 (nr, rpls->rpls_poc_lsb_lt[i],
sps->log2_max_pic_order_cnt_lsb_minus4 + 4);
}
rpls->num_long_term_pic++;
}
}
return TRUE;
error:
GST_WARNING ("error parsing \"ref_pic_list_struct \"");
return FALSE;
}
static gboolean
gst_h266_ref_pic_lists (GstH266RefPicLists * rpls, NalReader * nr,
const GstH266SPS * sps, const GstH266PPS * pps)
{
const GstH266RefPicListStruct *ref_list;
gint i, j, num_ltrp_entries;
GST_LOG ("parsing \"ref_pic_lists\"");
for (i = 0; i < 2; i++) {
if (sps->num_ref_pic_lists[i] == 0) {
rpls->rpl_sps_flag[i] = 0;
} else if (i == 0 || (i == 1 && pps->rpl1_idx_present_flag)) {
READ_UINT8 (nr, rpls->rpl_sps_flag[i], 1);
} else {
/* Only the case of (i == 1 && !pps->rpl1_idx_present_flag)
comes to here. */
rpls->rpl_sps_flag[1] = rpls->rpl_sps_flag[0];
}
if (rpls->rpl_sps_flag[i]) {
g_assert (sps->num_ref_pic_lists[i] > 0);
if (sps->num_ref_pic_lists[i] == 1) {
rpls->rpl_idx[i] = 0;
} else if (i == 0 || (i == 1 && pps->rpl1_idx_present_flag)) {
READ_UINT8 (nr, rpls->rpl_idx[i],
gst_util_ceil_log2 (sps->num_ref_pic_lists[i]));
CHECK_ALLOWED_MAX (rpls->rpl_idx[i], sps->num_ref_pic_lists[i] - 1);
} else {
/* Only the case of (i == 1 && !pps->rpl1_idx_present_flag)
comes to here. */
rpls->rpl_idx[1] = rpls->rpl_idx[0];
}
memcpy (&rpls->rpl_ref_list[i],
&sps->ref_pic_list_struct[i][rpls->rpl_idx[i]],
sizeof (rpls->rpl_ref_list[i]));
} else {
gst_h266_ref_pic_list_struct (&rpls->rpl_ref_list[i], nr,
i, sps->num_ref_pic_lists[i], sps);
}
ref_list = &rpls->rpl_ref_list[i];
num_ltrp_entries = 0;
for (j = 0; j < ref_list->num_ref_entries; j++) {
if (ref_list->inter_layer_ref_pic_flag[j]
|| ref_list->st_ref_pic_flag[j])
continue;
if (ref_list->ltrp_in_header_flag)
READ_UINT16 (nr, rpls->poc_lsb_lt[i][j],
sps->log2_max_pic_order_cnt_lsb_minus4 + 4);
READ_UINT8 (nr, rpls->delta_poc_msb_cycle_present_flag[i][j], 1);
if (rpls->delta_poc_msb_cycle_present_flag[i][j])
READ_UE_MAX (nr, rpls->delta_poc_msb_cycle_lt[i][j],
1 << (32 - sps->log2_max_pic_order_cnt_lsb_minus4 - 4));
num_ltrp_entries++;
}
g_assert (num_ltrp_entries == ref_list->num_long_term_pic);
}
return TRUE;
error:
GST_WARNING ("error parsing \"ref_pic_lists \"");
return FALSE;
}
static gboolean
gst_h266_parse_general_timing_hrd_parameters (GstH266GeneralHRDParameters * hrd,
NalReader * nr)
{
GST_LOG ("parsing \"General timing HRD Parameters\"");
READ_UINT32 (nr, hrd->num_units_in_tick, 32);
READ_UINT32 (nr, hrd->time_scale, 32);
READ_UINT8 (nr, hrd->general_nal_hrd_params_present_flag, 1);
READ_UINT8 (nr, hrd->general_vcl_hrd_params_present_flag, 1);
if (hrd->general_nal_hrd_params_present_flag
|| hrd->general_vcl_hrd_params_present_flag) {
READ_UINT8 (nr, hrd->general_same_pic_timing_in_all_ols_flag, 1);
READ_UINT8 (nr, hrd->general_du_hrd_params_present_flag, 1);
if (hrd->general_du_hrd_params_present_flag)
READ_UINT8 (nr, hrd->tick_divisor_minus2, 8);
READ_UINT8 (nr, hrd->bit_rate_scale, 4);
READ_UINT8 (nr, hrd->cpb_size_scale, 4);
if (hrd->general_du_hrd_params_present_flag)
READ_UINT8 (nr, hrd->cpb_size_du_scale, 4);
READ_UE_MAX (nr, hrd->hrd_cpb_cnt_minus1, 31);
}
return TRUE;
error:
GST_WARNING ("error parsing \"General timing HRD Parameters\"");
return FALSE;
}
static gboolean
gst_h266_parse_sublayer_hrd_parameters (GstH266SubLayerHRDParameters * sub_hrd,
NalReader * nr, guint8 subLayerId,
const GstH266GeneralHRDParameters * general)
{
guint i;
GST_LOG ("parsing \"SubLayer HRD Parameters\"");
for (i = 0; i <= general->hrd_cpb_cnt_minus1; i++) {
READ_UE_MAX (nr, sub_hrd->bit_rate_value_minus1[i], G_MAXUINT32 - 1);
READ_UE_MAX (nr, sub_hrd->cpb_size_value_minus1[i], G_MAXUINT32 - 1);
sub_hrd->bit_rate[i] = (sub_hrd->bit_rate_value_minus1[i] + 1) *
(2 << (6 + general->bit_rate_scale));
sub_hrd->cpb_size[i] = (sub_hrd->cpb_size_value_minus1[i] + 1) *
(2 << (4 + general->cpb_size_scale));
if (general->general_du_hrd_params_present_flag) {
READ_UE_MAX (nr, sub_hrd->cpb_size_du_value_minus1[i], G_MAXUINT32 - 1);
READ_UE_MAX (nr, sub_hrd->bit_rate_du_value_minus1[i], G_MAXUINT32 - 1);
}
READ_UINT8 (nr, sub_hrd->cbr_flag[i], 1);
}
for (i = 1; i <= general->hrd_cpb_cnt_minus1; i++) {
if (sub_hrd->bit_rate[i] <= sub_hrd->bit_rate[i - 1]) {
GST_WARNING ("bit_rate_value_minus1[i][j] shall be greater than "
"bit_rate_value_minus1[i][j-1], i=%u, j=%u", subLayerId, i);
goto error;
}
if (sub_hrd->cpb_size[i] <= sub_hrd->cpb_size[i - 1]) {
GST_WARNING ("cpb_size_value_minus1[i][j] shall be less than or equal "
"to cpb_size_value_minus1[i][j-1], i=%u, j=%u", subLayerId, i);
goto error;
}
}
return TRUE;
error:
GST_WARNING ("error parsing \"SubLayer HRD Parameters\"");
return FALSE;
}
static gboolean
gst_h266_parse_ols_timing_hrd_parameters (GstH266OLSHRDParameters * olsHrd,
NalReader * nr, const GstH266GeneralHRDParameters * general,
guint8 firstSubLayer, guint8 MaxSubLayersVal)
{
guint i, j;
GST_LOG ("parsing \"ols timing HRD Parameters\"");
for (i = firstSubLayer; i <= MaxSubLayersVal; i++) {
READ_UINT8 (nr, olsHrd->fixed_pic_rate_general_flag[i], 1);
if (!olsHrd->fixed_pic_rate_general_flag[i]) {
READ_UINT8 (nr, olsHrd->fixed_pic_rate_within_cvs_flag[i], 1);
} else
olsHrd->fixed_pic_rate_within_cvs_flag[i] = 1;
if (olsHrd->fixed_pic_rate_within_cvs_flag[i]) {
READ_UE_MAX (nr, olsHrd->elemental_duration_in_tc_minus1[i], 2047);
olsHrd->low_delay_hrd_flag[i] = 0;
} else if ((general->general_nal_hrd_params_present_flag ||
general->general_vcl_hrd_params_present_flag)
&& general->hrd_cpb_cnt_minus1 == 0) {
READ_UINT8 (nr, olsHrd->low_delay_hrd_flag[i], 1);
} else {
olsHrd->low_delay_hrd_flag[i] = 0;
}
if (general->general_nal_hrd_params_present_flag) {
if (!gst_h266_parse_sublayer_hrd_parameters
(&olsHrd->nal_sub_layer_hrd_parameters[i], nr, i, general))
goto error;
}
if (general->general_vcl_hrd_params_present_flag) {
if (!gst_h266_parse_sublayer_hrd_parameters
(&olsHrd->vcl_sub_layer_hrd_parameters[i], nr, i, general))
goto error;
}
}
for (i = 0; i < firstSubLayer; i++) {
GstH266SubLayerHRDParameters *sub_hrd, *max_sub_hrd;
sub_hrd = &olsHrd->nal_sub_layer_hrd_parameters[i];
max_sub_hrd = &olsHrd->nal_sub_layer_hrd_parameters[MaxSubLayersVal];
if (general->general_nal_hrd_params_present_flag) {
for (j = 0; j <= general->hrd_cpb_cnt_minus1; j++) {
sub_hrd->bit_rate_value_minus1[j] =
max_sub_hrd->bit_rate_value_minus1[j];
if (general->general_du_hrd_params_present_flag) {
sub_hrd->cpb_size_du_value_minus1[j] =
max_sub_hrd->cpb_size_du_value_minus1[j];
sub_hrd->bit_rate_du_value_minus1[j] =
max_sub_hrd->bit_rate_du_value_minus1[j];
}
sub_hrd->cbr_flag[j] = max_sub_hrd->cbr_flag[j];
}
}
}
return TRUE;
error:
GST_WARNING ("error parsing \"ols timing HRD Parameters\"");
return FALSE;
}
static gboolean
gst_h266_parse_alf (GstH266ALF * alf, NalReader * nr,
guint8 aps_chroma_present_flag)
{
gint j, k;
gint filtIdx, sfIdx, altIdx;
GST_LOG ("parsing \"ALF\"");
READ_UINT8 (nr, alf->luma_filter_signal_flag, 1);
if (aps_chroma_present_flag) {
READ_UINT8 (nr, alf->chroma_filter_signal_flag, 1);
READ_UINT8 (nr, alf->cc_cb_filter_signal_flag, 1);
READ_UINT8 (nr, alf->cc_cr_filter_signal_flag, 1);
}
if (alf->luma_filter_signal_flag == 0 &&
alf->chroma_filter_signal_flag == 0 &&
alf->cc_cb_filter_signal_flag == 0 &&
alf->cc_cr_filter_signal_flag == 0) {
GST_WARNING ("At least one of the values of alf_luma_filter_signal_flag, "
"alf_chroma_filter_signal_flag, alf_cc_cb_filter_signal_flag, and "
"alf_cc_cr_filter_signal_flag shall be equal to 1");
goto error;
}
if (alf->luma_filter_signal_flag) {
READ_UINT8 (nr, alf->luma_clip_flag, 1);
READ_UE_MAX (nr, alf->luma_num_filters_signalled_minus1,
GST_H266_NUM_ALF_FILTERS - 1);
if (alf->luma_num_filters_signalled_minus1 > 0) {
for (filtIdx = 0; filtIdx < GST_H266_NUM_ALF_FILTERS; filtIdx++) {
guint length =
gst_util_ceil_log2 (alf->luma_num_filters_signalled_minus1 + 1);
READ_UINT8 (nr, alf->luma_coeff_delta_idx[filtIdx], length);
CHECK_ALLOWED_MAX (alf->luma_coeff_delta_idx[filtIdx],
alf->luma_num_filters_signalled_minus1 + 1);
}
}
for (sfIdx = 0; sfIdx <= alf->luma_num_filters_signalled_minus1; sfIdx++) {
for (j = 0; j < 12; j++) {
READ_UE_MAX (nr, alf->luma_coeff_abs[sfIdx][j], 128);
if (alf->luma_coeff_abs[sfIdx][j])
READ_UINT8 (nr, alf->luma_coeff_sign[sfIdx][j], 1);
}
}
if (alf->luma_clip_flag) {
for (sfIdx = 0; sfIdx <= alf->luma_num_filters_signalled_minus1; sfIdx++) {
for (j = 0; j < 12; j++)
READ_UINT8 (nr, alf->luma_clip_idx[sfIdx][j], 2);
}
}
}
if (alf->chroma_filter_signal_flag) {
READ_UINT8 (nr, alf->chroma_clip_flag, 1);
READ_UE_MAX (nr, alf->chroma_num_alt_filters_minus1, 7);
for (altIdx = 0; altIdx <= alf->chroma_num_alt_filters_minus1; altIdx++) {
for (j = 0; j < 6; j++) {
READ_UE_MAX (nr, alf->chroma_coeff_abs[altIdx][j], 128);
if (alf->chroma_coeff_abs[altIdx][j] > 0)
READ_UINT8 (nr, alf->chroma_coeff_sign[altIdx][j], 1);
}
if (alf->chroma_clip_flag)
for (j = 0; j < 6; j++)
READ_UINT8 (nr, alf->chroma_clip_idx[altIdx][j], 2);
}
}
if (alf->cc_cb_filter_signal_flag) {
READ_UE_MAX (nr, alf->cc_cb_filters_signalled_minus1, 3);
for (k = 0; k < alf->cc_cb_filters_signalled_minus1 + 1; k++) {
for (j = 0; j < 7; j++) {
READ_UINT8 (nr, alf->cc_cb_mapped_coeff_abs[k][j], 3);
if (alf->cc_cb_mapped_coeff_abs[k][j])
READ_UINT8 (nr, alf->cc_cb_coeff_sign[k][j], 1);
}
}
}
if (alf->cc_cr_filter_signal_flag) {
READ_UE_MAX (nr, alf->cc_cr_filters_signalled_minus1, 3);
for (k = 0; k < alf->cc_cr_filters_signalled_minus1 + 1; k++) {
for (j = 0; j < 7; j++) {
READ_UINT8 (nr, alf->cc_cr_mapped_coeff_abs[k][j], 3);
if (alf->cc_cr_mapped_coeff_abs[k][j])
READ_UINT8 (nr, alf->cc_cr_coeff_sign[k][j], 1);
}
}
}
return TRUE;
error:
GST_WARNING ("error parsing \"ALF\"");
return FALSE;
}
static gboolean
gst_h266_parse_lmcs (GstH266LMCS * lmcs, NalReader * nr,
guint8 aps_chroma_present_flag)
{
gint i, LmcsMaxBinIdx;
GST_LOG ("parsing \"LMCS\"");
READ_UE_MAX (nr, lmcs->min_bin_idx, 15);
READ_UE_MAX (nr, lmcs->delta_max_bin_idx, 15);
LmcsMaxBinIdx = 15 - lmcs->delta_max_bin_idx;
if (LmcsMaxBinIdx < lmcs->min_bin_idx) {
GST_WARNING ("The value of LmcsMaxBinIdx(%d) shall be "
">= lmcs_min_bin_idx(%d)", LmcsMaxBinIdx, lmcs->min_bin_idx);
goto error;
}
READ_UE_MAX (nr, lmcs->delta_cw_prec_minus1, 14);
for (i = lmcs->min_bin_idx; i <= LmcsMaxBinIdx; i++) {
READ_UINT8 (nr, lmcs->delta_abs_cw[i], lmcs->delta_cw_prec_minus1 + 1);
if (lmcs->delta_abs_cw[i] > 0)
READ_UINT8 (nr, lmcs->delta_sign_cw_flag[i], 1);
}
if (aps_chroma_present_flag) {
READ_UINT8 (nr, lmcs->delta_abs_crs, 3);
if (lmcs->delta_abs_crs > 0)
READ_UINT8 (nr, lmcs->delta_sign_crs_flag, 1);
}
return TRUE;
error:
GST_WARNING ("error parsing \"LMCS\"");
return FALSE;
}
static gboolean
gst_h266_parse_scaling_list (GstH266ScalingList * scaling_list, NalReader * nr,
guint8 aps_chroma_present_flag)
{
guint ScalingList[8 * 8];
const guint8 *scalingMatrixPred;
gint dc;
gint i, x, y;
gint id, nextCoef, matrixSize, maxIdDelta;
guint refId, log2_size;
GST_LOG ("parsing \"Scaling List\"");
for (id = 0; id < 28; id++) {
matrixSize = id < 2 ? 2 : (id < 8 ? 4 : 8);
log2_size = id < 2 ? 1 : (id < 8 ? 2 : 3);
maxIdDelta = (id < 2) ? id : ((id < 8) ? (id - 2) : (id - 8));
dc = 0;
memset (ScalingList, 0, sizeof (ScalingList));
scaling_list->copy_mode_flag[id] = 1;
scaling_list->pred_mode_flag[id] = 0;
scaling_list->pred_id_delta[id] = 0;
if (aps_chroma_present_flag || id % 3 == 2 || id == 27) {
READ_UINT8 (nr, scaling_list->copy_mode_flag[id], 1);
if (!scaling_list->copy_mode_flag[id])
READ_UINT8 (nr, scaling_list->pred_mode_flag[id], 1);
if ((scaling_list->copy_mode_flag[id] || scaling_list->pred_mode_flag[id])
&& id != 0 && id != 2 && id != 8)
READ_UE_MAX (nr, scaling_list->pred_id_delta[id], maxIdDelta);
if (!scaling_list->copy_mode_flag[id]) {
nextCoef = 0;
if (id > 13) {
READ_SE_ALLOWED (nr, scaling_list->dc_coef[id - 14], -128, 127);
nextCoef = scaling_list->dc_coef[id - 14];
dc = scaling_list->dc_coef[id - 14];
}
for (i = 0; i < matrixSize * matrixSize; i++) {
x = square_DiagScanOrder_x[3][i];
y = square_DiagScanOrder_y[3][i];
if (!(id >= 25 && x >= 4 && y >= 4)) {
READ_SE_ALLOWED (nr, scaling_list->delta_coef[id][i], -128, 127);
nextCoef += scaling_list->delta_coef[id][i];
}
ScalingList[i] = nextCoef;
}
}
}
/* DC */
if (id > 13) {
if (!scaling_list->copy_mode_flag[id]
&& !scaling_list->pred_mode_flag[id]) {
scaling_list->scaling_list_DC[id - 14] = 8;
} else if (!scaling_list->pred_id_delta[id]) {
scaling_list->scaling_list_DC[id - 14] = 16;
} else {
if (id < scaling_list->pred_id_delta[id]) {
GST_WARNING ("Wrong pred_id_delta for scaling list");
goto error;
}
refId = id - scaling_list->pred_id_delta[id];
if (refId >= 14) {
dc += scaling_list->scaling_list_DC[refId - 14];
} else {
dc += scaling_list->scaling_list[refId][0];
}
scaling_list->scaling_list_DC[id - 14] = dc & 255;
}
}
/* AC */
if (!scaling_list->copy_mode_flag[id] && !scaling_list->pred_mode_flag[id]) {
scalingMatrixPred = scaling_pred_all_8;
} else if (!scaling_list->pred_id_delta[id]) {
scalingMatrixPred = scaling_pred_all_16;
} else {
if (id < scaling_list->pred_id_delta[id]) {
GST_WARNING ("Wrong pred_id_delta for scaling list");
goto error;
}
refId = id - scaling_list->pred_id_delta[id];
scalingMatrixPred = scaling_list->scaling_list[refId];
}
for (i = 0; i < matrixSize * matrixSize; i++) {
guint offset;
x = square_DiagScanOrder_x[log2_size][i];
y = square_DiagScanOrder_y[log2_size][i];
offset = y * matrixSize + x;
if (offset > matrixSize * matrixSize) {
GST_WARNING ("Wrong matrix coeff array index:%d", offset);
goto error;
}
scaling_list->scaling_list[id][offset] =
(scalingMatrixPred[offset] + ScalingList[i]) & 255;
}
}
return TRUE;
error:
GST_WARNING ("error parsing \"Scaling List\"");
return FALSE;
}
static gboolean
gst_h266_parse_range_extension (GstH266SPSRangeExtensionParams * range_params,
NalReader * nr, guint8 transform_skip_enabled_flag)
{
GST_LOG ("parsing \"Range Extension\"");
READ_UINT8 (nr, range_params->extended_precision_flag, 1);
if (transform_skip_enabled_flag)
READ_UINT8 (nr,
range_params->ts_residual_coding_rice_present_in_sh_flag, 1);
READ_UINT8 (nr, range_params->rrc_rice_extension_flag, 1);
READ_UINT8 (nr, range_params->persistent_rice_adaptation_enabled_flag, 1);
READ_UINT8 (nr, range_params->reverse_last_sig_coeff_enabled_flag, 1);
return TRUE;
error:
GST_WARNING ("error parsing \"Range Extension\"");
return FALSE;
}
static gboolean
gst_h266_parse_chroma_qp_table (GstH266SPS * sps, NalReader * nr)
{
guint numQpTables;
guint num_points_in_qp_table;
gint qp_in[GST_H266_MAX_POINTS_IN_QP_TABLE + 1];
gint qp_out[GST_H266_MAX_POINTS_IN_QP_TABLE + 1];
guint delta_qp_in[GST_H266_MAX_POINTS_IN_QP_TABLE];
gint qp_bd_offset = 6 * sps->bitdepth_minus8;
gint i, j, k, m, index;
GST_LOG ("parsing \"Chroma QP Table\"");
READ_UINT8 (nr, sps->joint_cbcr_enabled_flag, 1);
READ_UINT8 (nr, sps->same_qp_table_for_chroma_flag, 1);
numQpTables = sps->same_qp_table_for_chroma_flag ?
1 : (sps->joint_cbcr_enabled_flag ? 3 : 2);
for (i = 0; i < numQpTables; i++) {
gint QpBdOffset = 6 * sps->bitdepth_minus8;
READ_SE_ALLOWED (nr, sps->qp_table_start_minus26[i], -26 - QpBdOffset, 36);
READ_UE_MAX (nr, sps->num_points_in_qp_table_minus1[i],
36 - sps->qp_table_start_minus26[i]);
num_points_in_qp_table = sps->num_points_in_qp_table_minus1[i] + 1;
if (num_points_in_qp_table > GST_H266_MAX_POINTS_IN_QP_TABLE) {
GST_WARNING ("num_points_in_qp_table %d out of range",
num_points_in_qp_table);
goto error;
}
qp_out[0] = qp_in[0] = sps->qp_table_start_minus26[i] + 26;
for (j = 0; j < num_points_in_qp_table; j++) {
READ_UE_MAX (nr, sps->delta_qp_in_val_minus1[i][j], 128);
READ_UE_MAX (nr, sps->delta_qp_diff_val[i][j], 128);
delta_qp_in[j] = sps->delta_qp_in_val_minus1[i][j] + 1;
qp_in[j + 1] = qp_in[j] + delta_qp_in[j];
qp_out[j + 1] = qp_out[j] +
(sps->delta_qp_in_val_minus1[i][j] ^ sps->delta_qp_diff_val[i][j]);
}
index = qp_in[0] + qp_bd_offset;
if (index < 0 || index >= GST_H266_MAX_POINTS_IN_QP_TABLE) {
GST_WARNING ("Invalid qp index %d", index);
goto error;
}
sps->chroma_qp_table[i][index] = qp_out[0];
k = qp_in[0] - 1 + qp_bd_offset;
if (k < 0 || k >= GST_H266_MAX_POINTS_IN_QP_TABLE) {
GST_WARNING ("Invalid qp index %d", k);
goto error;
}
for (; k >= 0; k--) {
sps->chroma_qp_table[i][k] =
MAX (sps->chroma_qp_table[i][k + 1] - 1, -qp_bd_offset);
sps->chroma_qp_table[i][k] = MIN (sps->chroma_qp_table[i][k], 63);
}
for (j = 0; j < num_points_in_qp_table; j++) {
gint sh = delta_qp_in[j] >> 1;
index = qp_in[j] + 1 + qp_bd_offset;
if (index < 0 || index >= GST_H266_MAX_POINTS_IN_QP_TABLE) {
GST_WARNING ("Invalid qp index %d", index);
goto error;
}
index = qp_in[j + 1] + qp_bd_offset;
if (index < 0 || index >= GST_H266_MAX_POINTS_IN_QP_TABLE) {
GST_WARNING ("Invalid qp index %d", index);
goto error;
}
for (k = qp_in[j] + 1 + qp_bd_offset, m = 1;
k <= qp_in[j + 1] + qp_bd_offset; k++, m++) {
index = qp_in[j] + qp_bd_offset;
if (index < 0 || index >= GST_H266_MAX_POINTS_IN_QP_TABLE) {
GST_WARNING ("Invalid qp index %d", index);
goto error;
}
sps->chroma_qp_table[i][k] = sps->chroma_qp_table[i][index] +
((qp_out[j + 1] - qp_out[j]) * m + sh) / delta_qp_in[j];
}
}
k = qp_in[num_points_in_qp_table] + 1 + qp_bd_offset;
if (k < 1 || k >= GST_H266_MAX_POINTS_IN_QP_TABLE) {
GST_WARNING ("Invalid qp index %d", k);
goto error;
}
for (; k <= 63 + qp_bd_offset; k++) {
sps->chroma_qp_table[i][k] =
MAX (sps->chroma_qp_table[i][k - 1] + 1, -qp_bd_offset);
sps->chroma_qp_table[i][k] = MIN (sps->chroma_qp_table[i][k], 63);
}
}
if (sps->same_qp_table_for_chroma_flag) {
memcpy (&sps->chroma_qp_table[1], &sps->chroma_qp_table[0],
sizeof (sps->chroma_qp_table[0]));
memcpy (&sps->chroma_qp_table[2], &sps->chroma_qp_table[0],
sizeof (sps->chroma_qp_table[0]));
}
return TRUE;
error:
GST_WARNING ("error parsing \"Chroma QP Table\"");
return FALSE;
}
static GstH266ParserResult
gst_h266_parser_parse_buffering_period (GstH266BufferingPeriod * bp,
NalReader * nr)
{
guint i, j;
GST_LOG ("parsing \"Buffering period\"");
READ_UINT8 (nr, bp->nal_hrd_params_present_flag, 1);
READ_UINT8 (nr, bp->vcl_hrd_params_present_flag, 1);
READ_UINT8 (nr, bp->cpb_initial_removal_delay_length_minus1, 5);
READ_UINT8 (nr, bp->cpb_removal_delay_length_minus1, 5);
READ_UINT8 (nr, bp->dpb_output_delay_length_minus1, 5);
READ_UINT8 (nr, bp->du_hrd_params_present_flag, 1);
if (bp->du_hrd_params_present_flag) {
READ_UINT8 (nr, bp->du_cpb_removal_delay_increment_length_minus1, 5);
READ_UINT8 (nr, bp->dpb_output_delay_du_length_minus1, 5);
READ_UINT8 (nr, bp->du_cpb_params_in_pic_timing_sei_flag, 1);
READ_UINT8 (nr, bp->du_dpb_params_in_pic_timing_sei_flag, 1);
}
READ_UINT8 (nr, bp->concatenation_flag, 1);
READ_UINT8 (nr, bp->additional_concatenation_info_present_flag, 1);
if (bp->additional_concatenation_info_present_flag)
READ_UINT8 (nr, bp->max_initial_removal_delay_for_concatenation,
bp->cpb_removal_delay_length_minus1 + 1);
READ_UINT8 (nr, bp->cpb_removal_delay_delta_minus1,
bp->cpb_removal_delay_length_minus1 + 1);
READ_UINT8 (nr, bp->max_sublayers_minus1, 3);
if (bp->max_sublayers_minus1 > 0)
READ_UINT8 (nr, bp->cpb_removal_delay_deltas_present_flag, 1);
if (bp->cpb_removal_delay_deltas_present_flag) {
READ_UE_MAX (nr, bp->num_cpb_removal_delay_deltas_minus1, 15);
for (i = 0; i <= bp->num_cpb_removal_delay_deltas_minus1; i++)
READ_UINT8 (nr, bp->cpb_removal_delay_delta_val[i], 1);
}
READ_UE_MAX (nr, bp->cpb_cnt_minus1, 31);
if (bp->max_sublayers_minus1 > 0)
READ_UINT8 (nr, bp->sublayer_initial_cpb_removal_delay_present_flag, 1);
for (i = (bp->sublayer_initial_cpb_removal_delay_present_flag ?
0 : bp->max_sublayers_minus1); i <= bp->max_sublayers_minus1; i++) {
if (bp->nal_hrd_params_present_flag) {
for (j = 0; j < bp->cpb_cnt_minus1 + 1; j++) {
/* shall be 0 < x <= 90000 * (CpbSize[i][j] / BitRate[i][j]) */
READ_UINT8 (nr, bp->nal_initial_cpb_removal_delay[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
READ_UINT8 (nr, bp->nal_initial_cpb_removal_offset[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
if (bp->du_hrd_params_present_flag) {
READ_UINT8 (nr, bp->nal_initial_alt_cpb_removal_delay[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
READ_UINT8 (nr, bp->nal_initial_alt_cpb_removal_offset[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
}
}
}
if (bp->vcl_hrd_params_present_flag) {
for (j = 0; j < bp->cpb_cnt_minus1 + 1; j++) {
READ_UINT8 (nr, bp->vcl_initial_cpb_removal_delay[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
READ_UINT8 (nr, bp->vcl_initial_cpb_removal_offset[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
if (bp->du_hrd_params_present_flag) {
READ_UINT8 (nr, bp->vcl_initial_alt_cpb_removal_delay[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
READ_UINT8 (nr, bp->vcl_initial_alt_cpb_removal_offset[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
}
}
}
}
if (bp->max_sublayers_minus1 > 0)
READ_UINT8 (nr, bp->sublayer_dpb_output_offsets_present_flag, 1);
if (bp->sublayer_dpb_output_offsets_present_flag) {
for (i = 0; i < bp->max_sublayers_minus1; i++)
READ_UE (nr, bp->dpb_output_tid_offset[i]);
}
READ_UINT8 (nr, bp->alt_cpb_params_present_flag, 1);
if (bp->alt_cpb_params_present_flag)
READ_UINT8 (nr, bp->use_alt_cpb_params_flag, 1);
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Buffering period\"");
return GST_H266_PARSER_ERROR;
}
static GstH266ParserResult
gst_h266_parser_parse_pic_timing (GstH266PicTiming * pt,
NalReader * nr, const GstH266BufferingPeriod * bp, guint8 TemporalId)
{
guint i, j;
GST_LOG ("parsing \"Picture timing\"");
READ_UINT8 (nr, pt->cpb_removal_delay_minus1[bp->max_sublayers_minus1],
bp->cpb_removal_delay_length_minus1 + 1);
pt->sublayer_delays_present_flag[bp->max_sublayers_minus1] = 1;
for (i = TemporalId; i < bp->max_sublayers_minus1; i++) {
READ_UINT8 (nr, pt->sublayer_delays_present_flag[i], 1);
if (pt->sublayer_delays_present_flag[i]) {
if (bp->cpb_removal_delay_deltas_present_flag)
READ_UINT8 (nr, pt->cpb_removal_delay_delta_enabled_flag[i], 1);
if (pt->cpb_removal_delay_delta_enabled_flag[i]) {
if (bp->num_cpb_removal_delay_deltas_minus1 > 0)
READ_UINT8 (nr, pt->cpb_removal_delay_delta_idx[i],
gst_util_ceil_log2 (bp->num_cpb_removal_delay_deltas_minus1 + 1));
} else {
READ_UINT8 (nr, pt->cpb_removal_delay_minus1[i],
bp->cpb_removal_delay_length_minus1 + 1);
}
}
}
READ_UINT8 (nr, pt->dpb_output_delay, bp->dpb_output_delay_length_minus1 + 1);
if (bp->alt_cpb_params_present_flag) {
READ_UINT8 (nr, pt->cpb_alt_timing_info_present_flag, 1);
if (pt->cpb_alt_timing_info_present_flag) {
if (bp->nal_hrd_params_present_flag) {
for (i = (bp->sublayer_initial_cpb_removal_delay_present_flag ? 0 :
bp->max_sublayers_minus1); i <= bp->max_sublayers_minus1; i++) {
for (j = 0; j < bp->cpb_cnt_minus1 + 1; j++) {
READ_UINT8 (nr, pt->nal_cpb_alt_initial_removal_delay_delta[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
READ_UINT8 (nr, pt->nal_cpb_alt_initial_removal_offset_delta[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
}
READ_UINT8 (nr, pt->nal_cpb_delay_offset[i],
bp->cpb_removal_delay_length_minus1 + 1);
READ_UINT8 (nr, pt->nal_dpb_delay_offset[i],
bp->cpb_removal_delay_length_minus1 + 1);
}
}
if (bp->vcl_hrd_params_present_flag) {
for (i = (bp->sublayer_initial_cpb_removal_delay_present_flag ? 0 :
bp->max_sublayers_minus1); i <= bp->max_sublayers_minus1; i++) {
for (j = 0; j < bp->cpb_cnt_minus1 + 1; j++) {
READ_UINT8 (nr, pt->vcl_cpb_alt_initial_removal_delay_delta[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
READ_UINT8 (nr, pt->vcl_cpb_alt_initial_removal_offset_delta[i][j],
bp->cpb_initial_removal_delay_length_minus1 + 1);
}
READ_UINT8 (nr, pt->vcl_cpb_delay_offset[i],
bp->cpb_removal_delay_length_minus1 + 1);
READ_UINT8 (nr, pt->vcl_dpb_delay_offset[i],
bp->cpb_removal_delay_length_minus1 + 1);
}
}
}
}
if (bp->du_hrd_params_present_flag &&
bp->du_dpb_params_in_pic_timing_sei_flag)
READ_UINT8 (nr, pt->dpb_output_du_delay,
bp->dpb_output_delay_du_length_minus1 + 1);
if (bp->du_hrd_params_present_flag &&
bp->du_cpb_params_in_pic_timing_sei_flag) {
READ_UE (nr, pt->num_decoding_units_minus1);
if (pt->num_decoding_units_minus1 > 0) {
READ_UINT8 (nr, pt->du_common_cpb_removal_delay_flag, 1);
if (pt->du_common_cpb_removal_delay_flag) {
for (i = TemporalId; i <= bp->max_sublayers_minus1; i++) {
if (pt->sublayer_delays_present_flag[i]) {
READ_UINT8 (nr, pt->du_common_cpb_removal_delay_increment_minus1[i],
bp->du_cpb_removal_delay_increment_length_minus1 + 1);
} else {
pt->du_common_cpb_removal_delay_increment_minus1[i] =
pt->du_common_cpb_removal_delay_increment_minus1
[bp->max_sublayers_minus1];
}
}
}
for (i = 0; i <= pt->num_decoding_units_minus1; i++) {
READ_UE (nr, pt->num_nalus_in_du_minus1[i]);
if (!pt->du_common_cpb_removal_delay_flag &&
i < pt->num_decoding_units_minus1)
for (j = TemporalId; j <= bp->max_sublayers_minus1; j++) {
if (pt->sublayer_delays_present_flag[j]) {
READ_UINT8 (nr, pt->du_cpb_removal_delay_increment_minus1[i][j],
bp->du_cpb_removal_delay_increment_length_minus1 + 1);
} else {
pt->du_cpb_removal_delay_increment_minus1[i][j] =
pt->du_cpb_removal_delay_increment_minus1[i]
[bp->max_sublayers_minus1];
}
for (j = 0; j < TemporalId; j++)
pt->du_cpb_removal_delay_increment_minus1[i][j] =
pt->du_cpb_removal_delay_increment_minus1[i]
[bp->max_sublayers_minus1];
}
}
}
if (bp->additional_concatenation_info_present_flag)
READ_UINT8 (nr, pt->delay_for_concatenation_ensured_flag, 1);
READ_UINT8 (nr, pt->display_elemental_periods_minus1, 8);
}
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Picture timing\"");
return GST_H266_PARSER_ERROR;
}
static GstH266ParserResult
gst_h266_parser_parse_du_info (GstH266DUInfo * dui, NalReader * nr,
const GstH266BufferingPeriod * bp, guint8 TemporalId)
{
guint i;
GST_LOG ("parsing \"DU info\"");
READ_UE (nr, dui->decoding_unit_idx);
if (!bp->du_cpb_params_in_pic_timing_sei_flag) {
for (i = TemporalId; i <= bp->max_sublayers_minus1; i++) {
if (i < bp->max_sublayers_minus1)
READ_UINT8 (nr, dui->sublayer_delays_present_flag[i], 1);
if (dui->sublayer_delays_present_flag[i])
READ_UINT8 (nr, dui->du_cpb_removal_delay_increment[i],
bp->du_cpb_removal_delay_increment_length_minus1 + 1);
}
}
if (!bp->du_cpb_params_in_pic_timing_sei_flag)
dui->sublayer_delays_present_flag[bp->max_sublayers_minus1] = 1;
for (i = 0; i <= bp->max_sublayers_minus1; i++) {
if (i < bp->max_sublayers_minus1)
dui->du_cpb_removal_delay_increment[i] =
dui->du_cpb_removal_delay_increment[bp->max_sublayers_minus1];
}
if (!bp->du_dpb_params_in_pic_timing_sei_flag)
READ_UINT8 (nr, dui->dpb_output_du_delay_present_flag, 1);
if (dui->dpb_output_du_delay_present_flag)
READ_UINT8 (nr, dui->dpb_output_du_delay,
bp->dpb_output_delay_du_length_minus1 + 1);
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"DU info\"");
return GST_H266_PARSER_ERROR;
}
static GstH266ParserResult
gst_h266_parser_parse_scalable_nesting (GstH266ScalableNesting * sn,
NalReader * nr)
{
guint i;
GST_LOG ("parsing \"Scalable nesting\"");
/* check: max values here are not right */
READ_UINT8 (nr, sn->ols_flag, 1);
READ_UINT8 (nr, sn->subpic_flag, 1);
if (sn->ols_flag) {
READ_UE_MAX (nr, sn->num_olss_minus1, GST_H266_MAX_TOTAL_NUM_OLSS - 1);
for (i = 0; i <= sn->num_olss_minus1; i++)
READ_UE_MAX (nr, sn->ols_idx_delta_minus1[i],
GST_H266_MAX_TOTAL_NUM_OLSS - 2);
} else {
READ_UINT8 (nr, sn->all_layers_flag, 1);
if (!sn->all_layers_flag) {
READ_UE_MAX (nr, sn->num_layers_minus1, GST_H266_MAX_LAYERS);
for (i = 1; i <= sn->num_layers_minus1; i++)
READ_UINT8 (nr, sn->layer_id[i], 6);
}
}
if (sn->subpic_flag) {
READ_UE_MAX (nr, sn->num_subpics_minus1, GST_H266_MAX_SLICES_PER_AU - 1);
READ_UE_MAX (nr, sn->subpic_id_len_minus1, 15);
for (i = 0; i <= sn->num_subpics_minus1; i++)
READ_UINT8 (nr, sn->subpic_id[i], sn->subpic_id_len_minus1 + 1);
}
READ_UE_MAX (nr, sn->num_seis_minus1, 63);
while (!nal_reader_is_byte_aligned (nr))
if (!nal_reader_skip (nr, 1))
goto error;
/* not implemented yet
for( i = 0; i <= sn->num_seis_minus1; i++ )
sei_message()
*/
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Scalable nesting\"");
return GST_H266_PARSER_ERROR;
}
static GstH266ParserResult
gst_h266_parser_parse_subpic_level_info (GstH266SubPicLevelInfo * sli,
NalReader * nr)
{
guint i, j, k;
GST_LOG ("parsing \"Subpic level info\"");
READ_UINT8 (nr, sli->num_ref_levels_minus1, 3);
READ_UINT8 (nr, sli->cbr_constraint_flag, 1);
READ_UINT8 (nr, sli->explicit_fraction_present_flag, 1);
if (sli->explicit_fraction_present_flag)
READ_UE_MAX (nr, sli->num_subpics_minus1, GST_H266_MAX_SLICES_PER_AU - 1);
READ_UINT8 (nr, sli->max_sublayers_minus1, 3);
READ_UINT8 (nr, sli->sublayer_info_present_flag, 1);
while (!nal_reader_is_byte_aligned (nr))
if (!nal_reader_skip (nr, 1))
goto error;
for (k = sli->sublayer_info_present_flag ? 0 : sli->max_sublayers_minus1;
k <= sli->max_sublayers_minus1; k++) {
for (i = 0; i <= sli->num_ref_levels_minus1; i++) {
READ_UINT8 (nr, sli->non_subpic_layers_fraction[i][k], 8);
READ_UINT8 (nr, sli->ref_level_idc[i][k], 8);
if (sli->explicit_fraction_present_flag) {
for (j = 0; j <= sli->num_subpics_minus1; j++)
READ_UINT8 (nr, sli->ref_level_fraction_minus1[i][j][k], 8);
}
}
}
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Subpic level info\"");
return GST_H266_PARSER_ERROR;
}
static GstH266ParserResult
gst_h266_parser_parse_frame_field_info (GstH266FrameFieldInfo * ffi,
NalReader * nr)
{
GST_LOG ("parsing \"Frame field info\"");
READ_UINT8 (nr, ffi->field_pic_flag, 1);
if (ffi->field_pic_flag) {
READ_UINT8 (nr, ffi->bottom_field_flag, 1);
READ_UINT8 (nr, ffi->pairing_indicated_flag, 1);
if (ffi->pairing_indicated_flag)
READ_UINT8 (nr, ffi->paired_with_next_field_flag, 1);
} else {
READ_UINT8 (nr, ffi->display_fields_from_frame_flag, 1);
if (ffi->display_fields_from_frame_flag)
READ_UINT8 (nr, ffi->top_field_first_flag, 1);
READ_UINT8 (nr, ffi->display_elemental_periods_minus1, 8);
}
READ_UINT8 (nr, ffi->source_scan_type, 2);
READ_UINT8 (nr, ffi->duplicate_flag, 1);
ffi->valid = TRUE;
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Frame field info\"");
return GST_H266_PARSER_ERROR;
}
/**
* gst_h266_parser_new:
*
* Creates a new #GstH266Parser. It should be freed with
* gst_h266_parser_free after use.
*
* Returns: a new #GstH266Parser
*
* Since: 1.26
*/
GstH266Parser *
gst_h266_parser_new (void)
{
GstH266Parser *parser;
parser = g_new0 (GstH266Parser, 1);
return parser;
}
/**
* gst_h266_parser_free:
* @parser: the #GstH266Parser to free
*
* Frees the @parser
*
* Since: 1.26
*/
void
gst_h266_parser_free (GstH266Parser * parser)
{
g_free (parser);
}
/**
* gst_h266_parser_identify_nalu_unchecked:
* @parser: a #GstH266Parser
* @data: The data to parse
* @offset: the offset from which to parse @data
* @size: the size of @data
* @nalu: The #GstH266NalUnit where to store parsed nal headers
*
* Parses @data and fills @nalu from the next nalu data from @data.
*
* This differs from @gst_h266_parser_identify_nalu in that it doesn't
* check whether the packet is complete or not.
*
* Note: Only use this function if you already know the provided @data
* is a complete NALU, else use @gst_h266_parser_identify_nalu.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_identify_nalu_unchecked (GstH266Parser * parser,
const guint8 * data, guint offset, gsize size, GstH266NalUnit * nalu)
{
gint off1;
memset (nalu, 0, sizeof (*nalu));
if (size < offset + 4) {
GST_DEBUG ("Can't parse, buffer has too small size %" G_GSIZE_FORMAT
", offset %u", size, offset);
return GST_H266_PARSER_ERROR;
}
off1 = scan_for_start_codes (data + offset, size - offset);
if (off1 < 0) {
GST_DEBUG ("No start code prefix in this buffer");
return GST_H266_PARSER_NO_NAL;
}
nalu->sc_offset = offset + off1;
/* The scanner ensures one byte passed the start code but to
* identify an VVC NAL, we need 2. */
if (size - nalu->sc_offset - 3 < 2) {
GST_DEBUG ("Not enough bytes after start code to identify");
return GST_H266_PARSER_NO_NAL;
}
/* sc might have 2 or 3 0-bytes */
if (nalu->sc_offset > 0 && data[nalu->sc_offset - 1] == 00)
nalu->sc_offset--;
nalu->offset = offset + off1 + 3;
nalu->data = (guint8 *) data;
nalu->size = size - nalu->offset;
if (!gst_h266_parse_nalu_header (nalu)) {
GST_WARNING ("error parsing \"NAL unit header\"");
nalu->size = 0;
return GST_H266_PARSER_BROKEN_DATA;
}
nalu->valid = TRUE;
if (nalu->type == GST_H266_NAL_EOS || nalu->type == GST_H266_NAL_EOB) {
GST_LOG ("end-of-seq or end-of-stream nal found");
nalu->size = 2;
return GST_H266_PARSER_OK;
}
return GST_H266_PARSER_OK;
}
/**
* gst_h266_parser_identify_nalu:
* @parser: a #GstH266Parser
* @data: The data to parse
* @offset: the offset from which to parse @data
* @size: the size of @data
* @nalu: The #GstH266NalUnit where to store parsed nal headers
*
* Parses @data and fills @nalu from the next nalu data from @data
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_identify_nalu (GstH266Parser * parser,
const guint8 * data, guint offset, gsize size, GstH266NalUnit * nalu)
{
GstH266ParserResult res;
gint off2;
res = gst_h266_parser_identify_nalu_unchecked (parser, data,
offset, size, nalu);
if (res != GST_H266_PARSER_OK)
goto beach;
/* The two NALs are exactly 2 bytes size and are placed at the end of an AU,
* there is no need to wait for the following */
if (nalu->type == GST_H266_NAL_EOS || nalu->type == GST_H266_NAL_EOB)
goto beach;
off2 = scan_for_start_codes (data + nalu->offset, size - nalu->offset);
if (off2 < 0) {
GST_DEBUG ("Nal start %d, No end found", nalu->offset);
return GST_H266_PARSER_NO_NAL_END;
}
/* Callers assumes that enough data will available to identify the next NAL,
* but scan_for_start_codes() only ensure 1 extra byte is available. Ensure
* we have the required two header bytes (3 bytes start code and 2 byte
* header). */
if (size - (nalu->offset + off2) < 5) {
GST_DEBUG ("Not enough bytes identify the next NAL.");
return GST_H266_PARSER_NO_NAL_END;
}
/* Mini performance improvement:
* We could have a way to store how many 0s were skipped to avoid
* parsing them again on the next NAL */
while (off2 > 0 && data[nalu->offset + off2 - 1] == 00)
off2--;
nalu->size = off2;
if (nalu->size < 3)
return GST_H266_PARSER_BROKEN_DATA;
GST_LOG ("Complete nal found. Off: %d, Size: %d", nalu->offset, nalu->size);
beach:
return res;
}
/**
* gst_h266_parser_identify_nalu_vvc:
* @parser: a #GstH266Parser
* @data: The data to parse, must be the beging of the Nal unit
* @offset: the offset from which to parse @data
* @size: the size of @data
* @nal_length_size: the size in bytes of the VVC nal length prefix.
* @nalu: The #GstH266NalUnit where to store parsed nal headers
*
* Parses @data and sets @nalu.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_identify_nalu_vvc (GstH266Parser * parser,
const guint8 * data, guint offset, gsize size, guint8 nal_length_size,
GstH266NalUnit * nalu)
{
GstBitReader br;
memset (nalu, 0, sizeof (*nalu));
/* Would overflow guint below otherwise: the callers needs to ensure that
* this never happens */
if (offset > G_MAXUINT32 - nal_length_size) {
GST_WARNING ("offset + nal_length_size overflow");
nalu->size = 0;
return GST_H266_PARSER_BROKEN_DATA;
}
if (size < offset + nal_length_size) {
GST_DEBUG ("Can't parse, buffer has too small size %" G_GSIZE_FORMAT
", offset %u", size, offset);
return GST_H266_PARSER_ERROR;
}
size = size - offset;
gst_bit_reader_init (&br, data + offset, size);
nalu->size = gst_bit_reader_get_bits_uint32_unchecked (&br,
nal_length_size * 8);
nalu->sc_offset = offset;
nalu->offset = offset + nal_length_size;
if (nalu->size > G_MAXUINT32 - nal_length_size) {
GST_WARNING ("NALU size + nal_length_size overflow");
nalu->size = 0;
return GST_H266_PARSER_BROKEN_DATA;
}
if (size < (gsize) nalu->size + nal_length_size) {
nalu->size = 0;
return GST_H266_PARSER_NO_NAL_END;
}
nalu->data = (guint8 *) data;
if (!gst_h266_parse_nalu_header (nalu)) {
GST_WARNING ("error parsing \"NAL unit header\"");
nalu->size = 0;
return GST_H266_PARSER_BROKEN_DATA;
}
if (nalu->size < 2)
return GST_H266_PARSER_BROKEN_DATA;
nalu->valid = TRUE;
return GST_H266_PARSER_OK;
}
/**
* gst_h266_parser_parse_nal:
* @parser: a #GstH266Parser
* @nalu: The #GstH266NalUnit to parse
*
* This function should be called in the case one doesn't need to
* parse a specific structure. It is necessary to do so to make
* sure @parser is up to date.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_nal (GstH266Parser * parser, GstH266NalUnit * nalu)
{
GstH266VPS vps;
GstH266SPS sps;
GstH266PPS pps;
GstH266APS aps;
switch (nalu->type) {
case GST_H266_NAL_VPS:
return gst_h266_parser_parse_vps (parser, nalu, &vps);
break;
case GST_H266_NAL_SPS:
return gst_h266_parser_parse_sps (parser, nalu, &sps);
break;
case GST_H266_NAL_PPS:
return gst_h266_parser_parse_pps (parser, nalu, &pps);
break;
case GST_H266_NAL_PREFIX_APS:
case GST_H266_NAL_SUFFIX_APS:
return gst_h266_parser_parse_aps (parser, nalu, &aps);
break;
default:
break;
}
return GST_H266_PARSER_OK;
}
/**
* gst_h266_parser_parse_vps:
* @parser: a #GstH266Parser
* @nalu: The #GST_H266_NAL_VPS #GstH266NalUnit to parse
* @vps: The #GstH266VPS to fill.
*
* Parses @data, and fills the @vps structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_vps (GstH266Parser * parser, GstH266NalUnit * nalu,
GstH266VPS * vps)
{
GstH266ParserResult res = gst_h266_parse_vps (nalu, vps);
if (res == GST_H266_PARSER_OK) {
GST_LOG ("adding video parameter set with id: %d to array", vps->vps_id);
if (parser->active_vps && parser->active_vps->vps_id == vps->vps_id)
parser->active_vps = NULL;
parser->vps[vps->vps_id] = *vps;
parser->last_vps = &parser->vps[vps->vps_id];
}
return res;
}
static gboolean
gst_h266_parser_derive_output_layer_sets (GstH266VPS * vps)
{
guint8 dependency_flag[GST_H266_MAX_LAYERS][GST_H266_MAX_LAYERS];
guint reference_layer_idx[GST_H266_MAX_LAYERS][GST_H266_MAX_LAYERS];
guint layer_included_in_ols_flag[GST_H266_MAX_TOTAL_NUM_OLSS]
[GST_H266_MAX_LAYERS];
guint num_ref_layers[GST_H266_MAX_LAYERS];
guint8 output_layer_idx[GST_H266_MAX_TOTAL_NUM_OLSS][GST_H266_MAX_LAYERS];
guint8 layer_used_as_ref_layer_flag[GST_H266_MAX_LAYERS];
guint8 layer_used_as_output_layer_flag[GST_H266_MAX_LAYERS];
guint i, j, m, r;
gint k;
GST_LOG ("deriving output layer sets");
if (vps->max_layers_minus1 == 0) {
g_assert (vps->total_num_olss == 1);
vps->num_multi_layer_olss = 0;
return TRUE;
}
memset (dependency_flag, 0, sizeof (dependency_flag));
memset (reference_layer_idx, 0, sizeof (reference_layer_idx));
memset (layer_included_in_ols_flag, 0, sizeof (layer_included_in_ols_flag));
memset (num_ref_layers, 0, sizeof (num_ref_layers));
memset (output_layer_idx, 0, sizeof (output_layer_idx));
memset (layer_used_as_ref_layer_flag,
0, sizeof (layer_used_as_ref_layer_flag));
memset (layer_used_as_output_layer_flag,
0, sizeof (layer_used_as_output_layer_flag));
/* 7.4.3.3 vps_direct_ref_layer_flag section */
for (i = 0; i <= vps->max_layers_minus1; i++) {
for (j = 0; j <= vps->max_layers_minus1; j++) {
dependency_flag[i][j] = vps->direct_ref_layer_flag[i][j];
for (k = 0; k < i; k++) {
if (vps->direct_ref_layer_flag[i][k] && dependency_flag[k][j])
dependency_flag[i][j] = 1;
}
if (vps->direct_ref_layer_flag[i][j])
layer_used_as_ref_layer_flag[j] = 1;
}
}
for (i = 0; i <= vps->max_layers_minus1; i++) {
for (j = 0, r = 0; j <= vps->max_layers_minus1; j++) {
if (dependency_flag[i][j])
reference_layer_idx[i][r++] = j;
}
num_ref_layers[i] = r;
}
/* 7.4.3.3 vps_ols_output_layer_flag section */
vps->num_output_layers_in_ols[0] = 1;
vps->num_sub_layers_in_layer_in_ols[0][0] =
vps->ptl_max_tid[vps->ols_ptl_idx[0]] + 1;
layer_used_as_output_layer_flag[0] = 1;
for (i = 1; i <= vps->max_layers_minus1; i++) {
if (vps->each_layer_is_an_ols_flag || vps->ols_mode_idc < 2)
layer_used_as_output_layer_flag[i] = 1;
else
layer_used_as_output_layer_flag[i] = 0;
}
for (i = 1; i < vps->total_num_olss; i++) {
if (vps->each_layer_is_an_ols_flag || vps->ols_mode_idc == 0) {
vps->num_output_layers_in_ols[i] = 1;
vps->output_layer_id_in_ols[i][0] = vps->layer_id[i];
if (vps->each_layer_is_an_ols_flag) {
vps->num_sub_layers_in_layer_in_ols[i][0] =
vps->ptl_max_tid[vps->ols_ptl_idx[i]] + 1;
} else {
vps->num_sub_layers_in_layer_in_ols[i][i] =
vps->ptl_max_tid[vps->ols_ptl_idx[i]] + 1;
for (k = i - 1; k >= 0; k--) {
vps->num_sub_layers_in_layer_in_ols[i][k] = 0;
for (m = k + 1; m <= i; m++) {
guint8 max_sublayer_needed =
MIN (vps->num_sub_layers_in_layer_in_ols[i][m],
vps->max_tid_il_ref_pics_plus1[m][k]);
if (vps->direct_ref_layer_flag[m][k] &&
vps->num_sub_layers_in_layer_in_ols[i][k] < max_sublayer_needed)
vps->num_sub_layers_in_layer_in_ols[i][k] = max_sublayer_needed;
}
}
}
} else if (vps->ols_mode_idc == 1) {
vps->num_output_layers_in_ols[i] = i + 1;
for (j = 0; j < vps->num_output_layers_in_ols[i]; j++) {
vps->output_layer_id_in_ols[i][j] = vps->layer_id[j];
vps->num_sub_layers_in_layer_in_ols[i][j] =
vps->ptl_max_tid[vps->ols_ptl_idx[i]] + 1;
}
} else if (vps->ols_mode_idc == 2) {
gint8 highest_included_layer = 0;
for (j = 0; j <= vps->max_layers_minus1; j++)
vps->num_sub_layers_in_layer_in_ols[i][j] = 0;
j = 0;
for (k = 0; k <= vps->max_layers_minus1; k++) {
if (vps->ols_output_layer_flag[i][k]) {
layer_included_in_ols_flag[i][k] = 1;
highest_included_layer = k;
layer_used_as_output_layer_flag[k] = 1;
output_layer_idx[i][j] = k;
vps->output_layer_id_in_ols[i][j] = vps->layer_id[j];
j++;
vps->num_sub_layers_in_layer_in_ols[i][k] =
vps->ptl_max_tid[vps->ols_ptl_idx[i]] + 1;
}
}
vps->num_output_layers_in_ols[i] = j;
for (j = 0; j < vps->num_output_layers_in_ols[i]; j++) {
gint idx = output_layer_idx[i][j];
for (k = 0; k < num_ref_layers[idx]; k++)
layer_included_in_ols_flag[i][reference_layer_idx[idx][k]] = 1;
}
for (k = highest_included_layer - 1; k >= 0; k--) {
if (layer_included_in_ols_flag[i][k]
&& !vps->ols_output_layer_flag[i][k]) {
for (m = k + 1; m <= highest_included_layer; m++) {
guint max_sublayer_needed =
MIN (vps->num_sub_layers_in_layer_in_ols[i][m],
vps->max_tid_il_ref_pics_plus1[m][k]);
if (vps->direct_ref_layer_flag[m][k] &&
layer_included_in_ols_flag[i][m] &&
vps->num_sub_layers_in_layer_in_ols[i][k] < max_sublayer_needed)
vps->num_sub_layers_in_layer_in_ols[i][k] = max_sublayer_needed;
}
}
}
}
}
for (i = 0; i <= vps->max_layers_minus1; i++) {
if (layer_used_as_ref_layer_flag[i] == 0 &&
layer_used_as_output_layer_flag[i] == 0) {
GST_WARNING ("There shall be no layer that is neither an output"
" layer nor a direct reference layer");
return FALSE;
}
}
vps->num_layers_in_ols[0] = 1;
vps->layer_id_in_ols[0][0] = vps->layer_id[0];
vps->num_multi_layer_olss = 0;
for (i = 1; i < vps->total_num_olss; i++) {
if (vps->each_layer_is_an_ols_flag) {
vps->num_layers_in_ols[i] = 1;
vps->layer_id_in_ols[i][0] = vps->layer_id[i];
} else if (vps->ols_mode_idc == 0 || vps->ols_mode_idc == 1) {
vps->num_layers_in_ols[i] = i + 1;
for (j = 0; j < vps->num_layers_in_ols[i]; j++)
vps->layer_id_in_ols[i][j] = vps->layer_id[j];
} else if (vps->ols_mode_idc == 2) {
j = 0;
for (k = 0; k <= vps->max_layers_minus1; k++) {
if (layer_included_in_ols_flag[i][k]) {
vps->layer_id_in_ols[i][j] = vps->layer_id[k];
j++;
}
}
vps->num_layers_in_ols[i] = j;
}
if (vps->num_layers_in_ols[i] > 1) {
vps->multi_layer_ols_idx[i] = vps->num_multi_layer_olss;
vps->num_multi_layer_olss++;
}
}
return TRUE;
}
static gboolean
gst_h266_parser_check_vps (GstH266VPS * vps)
{
guint index;
guint olsIdx, olsTimingHrdIdx, olsPtlIdx, olsDpbParamsIdx;
for (index = 0; index < vps->num_multi_layer_olss; index++) {
olsIdx = vps->multi_layer_ols_idx[index];
olsTimingHrdIdx = vps->ols_timing_hrd_idx[index];
olsPtlIdx = vps->ols_ptl_idx[olsIdx];
if (vps->hrd_max_tid[olsTimingHrdIdx] < vps->ptl_max_tid[olsPtlIdx]) {
GST_WARNING ("The value of vps_hrd_max_tid[vps_ols_timing_hrd_idx[m]] "
"shall be greater than or equal to "
"vps_ptl_max_tid[vps_ols_ptl_idx[n]] for each m-th multi-layer "
"OLS for m from 0 to NumMultiLayerOlss - 1, inclusive, and n "
"being the OLS index of the m-th multi-layer OLS among all OLSs.");
return FALSE;
}
olsDpbParamsIdx = vps->ols_dpb_params_idx[olsIdx];
if (vps->dpb_max_tid[olsDpbParamsIdx] < vps->ptl_max_tid[olsPtlIdx]) {
GST_WARNING ("The value of vps_dpb_max_tid[vps_ols_dpb_params_idx[m]] "
"shall be greater than or equal to "
"vps_ptl_max_tid[vps_ols_ptl_idx[n]] for each m-th multi-layer "
"OLS for m from 0 to NumMultiLayerOlss - 1, inclusive, and n "
"being the OLS index of the m-th multi-layer OLS among all OLSs.");
return FALSE;
}
}
return TRUE;
}
/**
* gst_h266_parse_vps:
* @nalu: The #GST_H266_NAL_VPS #GstH266NalUnit to parse
* @vps: The #GstH266VPS to fill.
*
* Parses @data, and fills the @vps structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parse_vps (GstH266NalUnit * nalu, GstH266VPS * vps)
{
NalReader nr;
gint i, j;
gboolean isPTLReferred[GST_H266_MAX_PTLS];
GST_LOG ("parsing \"Video parameter set\"");
nal_reader_init (&nr, nalu->data + nalu->offset + nalu->header_bytes,
nalu->size - nalu->header_bytes);
memset (vps, 0, sizeof (*vps));
memset (isPTLReferred, 0, sizeof (isPTLReferred));
READ_UINT8 (&nr, vps->vps_id, 4);
if (vps->vps_id == 0) {
GST_WARNING ("vps_id equal to zero is reserved and shall"
" not be used in a bitstream");
goto error;
}
READ_UINT8 (&nr, vps->max_layers_minus1, 6);
if (vps->max_layers_minus1 == 0)
vps->each_layer_is_an_ols_flag = 1;
READ_UINT8 (&nr, vps->max_sublayers_minus1, 3);
CHECK_ALLOWED_MAX (vps->max_sublayers_minus1, GST_H266_MAX_SUBLAYERS - 1);
if (vps->max_layers_minus1 > 0 && vps->max_sublayers_minus1 > 0) {
READ_UINT8 (&nr, vps->default_ptl_dpb_hrd_max_tid_flag, 1);
} else
vps->default_ptl_dpb_hrd_max_tid_flag = 1;
if (vps->max_layers_minus1 > 0) {
READ_UINT8 (&nr, vps->all_independent_layers_flag, 1);
if (vps->all_independent_layers_flag == 0)
vps->each_layer_is_an_ols_flag = 0;
} else
vps->all_independent_layers_flag = 1;
for (i = 0; i <= vps->max_layers_minus1; i++) {
READ_UINT8 (&nr, vps->layer_id[i], 6);
/* 7.4.3.2: For any two non-negative integer values of m and n,
when m is less than n, the value of vps_layer_id[ m ] shall be
less than vps_layer_id[ n ]. */
if (i > 0 && vps->layer_id[i] <= vps->layer_id[i - 1]) {
GST_WARNING ("vps_layer_id[%d](%d) should > vps_layer_id[%d](%d).\n",
i, vps->layer_id[i], i - 1, vps->layer_id[i - 1]);
goto error;
}
if (i > 0 && !vps->all_independent_layers_flag) {
guint count = 0;
READ_UINT8 (&nr, vps->independent_layer_flag[i], 1);
if (!vps->independent_layer_flag[i]) {
READ_UINT8 (&nr, vps->max_tid_ref_present_flag[i], 1);
for (j = 0; j < i; j++) {
READ_UINT8 (&nr, vps->direct_ref_layer_flag[i][j], 1);
if (vps->direct_ref_layer_flag[i][j])
count++;
if (vps->max_tid_ref_present_flag[i]
&& vps->direct_ref_layer_flag[i][j]) {
READ_UINT8 (&nr, vps->max_tid_il_ref_pics_plus1[i][j], 3);
} else {
vps->max_tid_il_ref_pics_plus1[i][j] =
vps->max_sublayers_minus1 + 1;
}
}
if (count == 0) {
GST_WARNING ("There has to be at least one value of j such "
"that the value of vps_direct_dependency_flag[%d][j] is "
"equal to 1, when vps_independent_layer_flag[%d] is equal "
"to 0", i, i);
goto error;
}
}
} else {
vps->independent_layer_flag[i] = 1;
}
}
if (vps->max_layers_minus1 > 0) {
if (vps->all_independent_layers_flag)
READ_UINT8 (&nr, vps->each_layer_is_an_ols_flag, 1);
if (!vps->each_layer_is_an_ols_flag) {
if (!vps->all_independent_layers_flag) {
READ_UINT8 (&nr, vps->ols_mode_idc, 2);
CHECK_ALLOWED_MAX (vps->ols_mode_idc, 2);
} else
vps->ols_mode_idc = 2;
if (vps->ols_mode_idc == 2) {
READ_UINT8 (&nr, vps->num_output_layer_sets_minus2, 8);
for (i = 1; i <= vps->num_output_layer_sets_minus2 + 1; i++)
for (j = 0; j <= vps->max_layers_minus1; j++)
READ_UINT8 (&nr, vps->ols_output_layer_flag[i][j], 1);
}
}
if (vps->each_layer_is_an_ols_flag ||
vps->ols_mode_idc == 0 || vps->ols_mode_idc == 1) {
vps->total_num_olss = vps->max_layers_minus1 + 1;
} else if (vps->ols_mode_idc == 2) {
vps->total_num_olss = vps->num_output_layer_sets_minus2 + 2;
} else {
g_assert_not_reached ();
}
READ_UINT8 (&nr, vps->num_ptls_minus1, 8);
CHECK_ALLOWED_MAX (vps->num_ptls_minus1, vps->total_num_olss - 1);
} else {
vps->each_layer_is_an_ols_flag = 1;
vps->num_ptls_minus1 = 0;
vps->total_num_olss = 1;
}
if (!gst_h266_parser_derive_output_layer_sets (vps)) {
GST_WARNING ("Fail to derive vps layer sets parameters");
goto error;
}
if (vps->num_ptls_minus1 + 1 > vps->total_num_olss) {
GST_WARNING ("The value of vps_num_ptls_minus1 shall"
" be less than TotalNumOlss");
goto error;
}
for (i = 0; i <= vps->num_ptls_minus1; i++) {
if (i > 0) {
READ_UINT8 (&nr, vps->pt_present_flag[i], 1);
} else
vps->pt_present_flag[i] = 1;
if (!vps->default_ptl_dpb_hrd_max_tid_flag) {
READ_UINT8 (&nr, vps->ptl_max_tid[i], 3);
CHECK_ALLOWED_MAX (vps->ptl_max_tid[i], vps->max_sublayers_minus1);
} else
vps->ptl_max_tid[i] = vps->max_sublayers_minus1;
}
while (!nal_reader_is_byte_aligned (&nr))
if (!nal_reader_skip (&nr, 1))
goto error;
for (i = 0; i <= vps->num_ptls_minus1; i++) {
if (i == 0 && !vps->pt_present_flag[i]) {
GST_WARNING ("Profile/Tier should always be present for the first entry");
goto error;
}
if (!gst_h266_parse_profile_tier_level (&vps->profile_tier_level[i], &nr,
vps->pt_present_flag[i], vps->ptl_max_tid[i]))
goto error;
}
for (i = 0; i < vps->total_num_olss; i++) {
if (vps->num_ptls_minus1 > 0 &&
vps->num_ptls_minus1 + 1 != vps->total_num_olss) {
READ_UINT8 (&nr, vps->ols_ptl_idx[i], 8);
} else if (vps->num_ptls_minus1 + 1 == vps->total_num_olss)
vps->ols_ptl_idx[i] = i;
else
vps->ols_ptl_idx[i] = 0;
isPTLReferred[vps->ols_ptl_idx[i]] = TRUE;
}
for (i = 0; i <= vps->num_ptls_minus1; i++) {
if (!isPTLReferred[i]) {
GST_WARNING ("Each profile_tier_level() syntax structure in the "
"VPS shall be referred to by at least one value of "
"vps_ols_ptl_idx[i] for i in the range of 0 to TotalNumOlss ? 1, "
"inclusive.");
goto error;
}
}
if (!vps->each_layer_is_an_ols_flag) {
READ_UE_MAX (&nr, vps->num_dpb_params_minus1,
vps->num_multi_layer_olss - 1);
if (vps->max_sublayers_minus1 > 0)
READ_UINT8 (&nr, vps->sublayer_dpb_params_present_flag, 1);
for (i = 0; i <= vps->num_dpb_params_minus1; i++) {
if (!vps->default_ptl_dpb_hrd_max_tid_flag) {
READ_UINT8 (&nr, vps->dpb_max_tid[i], 3);
CHECK_ALLOWED_MAX (vps->dpb_max_tid[i], vps->max_sublayers_minus1);
} else
vps->dpb_max_tid[i] = vps->max_sublayers_minus1;
if (!gst_h266_parse_dpb_parameters (&vps->dpb[i], &nr,
vps->dpb_max_tid[i], vps->sublayer_dpb_params_present_flag))
goto error;
for (j = (vps->sublayer_dpb_params_present_flag ?
vps->dpb_max_tid[i] : 0); j < vps->dpb_max_tid[i]; j++) {
/* When dpb_max_dec_pic_buffering_minus1[i] is not present for i in
the range of 0 to maxSubLayersMinus1 - 1, inclusive, due to
subLayerInfoFlag being equal to 0, it is inferred to be equal to
dpb_max_dec_pic_buffering_minus1[maxSubLayersMinus1]. */
vps->dpb[i].max_dec_pic_buffering_minus1[j] =
vps->dpb[i].max_dec_pic_buffering_minus1[vps->dpb_max_tid[i]];
/* When dpb_max_num_reorder_pics[i] is not present for i in
the range of 0 to maxSubLayersMinus1 - 1, inclusive, due to
subLayerInfoFlag being equal to 0, it is inferred to be equal
to dpb_max_num_reorder_pics[maxSubLayersMinus1]. */
vps->dpb[i].max_num_reorder_pics[j] =
vps->dpb[i].max_num_reorder_pics[vps->dpb_max_tid[i]];
/* When dpb_max_latency_increase_plus1[i] is not present for i in
the range of 0 to maxSubLayersMinus1 - 1, inclusive, due to
subLayerInfoFlag being equal to 0, it is inferred to be equal to
dpb_max_latency_increase_plus1[maxSubLayersMinus1]. */
vps->dpb[i].max_latency_increase_plus1[j] =
vps->dpb[i].max_latency_increase_plus1[vps->dpb_max_tid[i]];
}
}
for (i = 0; i < vps->num_multi_layer_olss; i++) {
READ_UE_MAX (&nr, vps->ols_dpb_pic_width[i], G_MAXUINT16);
READ_UE_MAX (&nr, vps->ols_dpb_pic_height[i], G_MAXUINT16);
READ_UINT8 (&nr, vps->ols_dpb_chroma_format[i], 2);
READ_UE_MAX (&nr, vps->ols_dpb_bitdepth_minus8[i], 2);
if (vps->num_dpb_params_minus1 > 0
&& vps->num_dpb_params_minus1 + 1 != vps->num_multi_layer_olss) {
READ_UE_MAX (&nr, vps->ols_dpb_params_idx[i],
vps->num_dpb_params_minus1);
} else if (vps->num_dpb_params_minus1 == 0)
vps->ols_dpb_params_idx[i] = 0;
else
vps->ols_dpb_params_idx[i] = i;
}
}
if (!vps->each_layer_is_an_ols_flag)
READ_UINT8 (&nr, vps->timing_hrd_params_present_flag, 1);
if (vps->timing_hrd_params_present_flag) {
gboolean is_dpb_param_referred[GST_H266_MAX_TOTAL_NUM_OLSS];
memset (is_dpb_param_referred, 0, sizeof (is_dpb_param_referred));
if (!gst_h266_parse_general_timing_hrd_parameters
(&vps->general_hrd_params, &nr))
goto error;
if (vps->max_sublayers_minus1 > 0) {
READ_UINT8 (&nr, vps->sublayer_cpb_params_present_flag, 1);
} else
vps->sublayer_cpb_params_present_flag = 0;
READ_UE_MAX (&nr, vps->num_ols_timing_hrd_params_minus1,
vps->num_multi_layer_olss - 1);
for (i = 0; i <= vps->num_ols_timing_hrd_params_minus1; i++) {
guint firstSubLayer;
if (!vps->default_ptl_dpb_hrd_max_tid_flag) {
READ_UINT8 (&nr, vps->hrd_max_tid[i], 3);
CHECK_ALLOWED_MAX (vps->hrd_max_tid[i], vps->max_sublayers_minus1);
} else
vps->hrd_max_tid[i] = vps->max_sublayers_minus1;
firstSubLayer =
vps->sublayer_cpb_params_present_flag ? 0 : vps->hrd_max_tid[i];
if (!gst_h266_parse_ols_timing_hrd_parameters (&vps->ols_hrd_params[i],
&nr, &vps->general_hrd_params, firstSubLayer,
vps->hrd_max_tid[i]))
goto error;
}
for (i = vps->num_ols_timing_hrd_params_minus1 + 1;
i < vps->total_num_olss; i++) {
vps->hrd_max_tid[i] = vps->max_sublayers_minus1;
}
for (i = 0; i < vps->num_multi_layer_olss; i++) {
if (vps->num_ols_timing_hrd_params_minus1 > 0 &&
vps->num_ols_timing_hrd_params_minus1 + 1 !=
vps->num_multi_layer_olss) {
READ_UE_MAX (&nr, vps->ols_timing_hrd_idx[i],
vps->num_ols_timing_hrd_params_minus1);
} else if (vps->num_ols_timing_hrd_params_minus1 == 0) {
vps->ols_timing_hrd_idx[i] = 0;
} else {
vps->ols_timing_hrd_idx[i] = i;
}
is_dpb_param_referred[vps->ols_timing_hrd_idx[i]] = TRUE;
}
for (i = 0; i <= vps->num_ols_timing_hrd_params_minus1; i++) {
if (!is_dpb_param_referred[i]) {
GST_WARNING ("Each vps_ols_timing_hrd_parameters( ) syntax structure "
"in the VPS shall be referred to by at least one value of "
"vps_ols_timing_hrd_idx[i] for i in the range of 1 to "
"NumMultiLayerOlss - 1, inclusive");
goto error;
}
}
} else {
for (int i = 0; i < vps->total_num_olss; i++)
vps->hrd_max_tid[i] = vps->max_sublayers_minus1;
}
READ_UINT8 (&nr, vps->extension_flag, 1);
if (vps->extension_flag)
GST_WARNING ("extension_flag not supported. Ignoring extension data.");
if (!gst_h266_parser_check_vps (vps))
goto error;
vps->valid = TRUE;
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Video parameter set\"");
vps->valid = FALSE;
return GST_H266_PARSER_ERROR;
}
/**
* gst_h266_parser_parse_sps:
* @parser: a #GstH266Parser
* @nalu: The #GST_H266_NAL_SPS #GstH266NalUnit to parse
* @sps: The #GstH266SPS to fill.
*
* Parses @data, and fills the @sps structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_sps (GstH266Parser * parser, GstH266NalUnit * nalu,
GstH266SPS * sps)
{
GstH266ParserResult res = gst_h266_parse_sps (parser, nalu, sps);
if (res == GST_H266_PARSER_OK) {
GST_LOG ("adding sequence parameter set with id: %d to array", sps->sps_id);
if (parser->active_sps && parser->active_sps->sps_id == sps->sps_id)
parser->active_sps = NULL;
parser->sps[sps->sps_id] = *sps;
parser->last_sps = &parser->sps[sps->sps_id];
}
return res;
}
/**
* gst_h266_parse_sps:
* @parser: The #GstH266Parser
* @nalu: The #GST_H266_NAL_SPS #GstH266NalUnit to parse
* @sps: The #GstH266SPS to fill.
*
* Parses @data, and fills the @sps structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parse_sps (GstH266Parser * parser, GstH266NalUnit * nalu,
GstH266SPS * sps)
{
NalReader nr;
GstH266VPS *vps;
GstH266ProfileTierLevel *ptl;
guint i, j;
guint8 ctbLog2SizeY;
guint minQT[3] = { 0, 0, 0 };
guint maxBTSize[3] = { 0, 0, 0 };
guint maxTTSize[3] = { 0, 0, 0 };
guint minCbLog2SizeY, minCuSize, MaxNumMergeCand,
minQtLog2SizeIntraY, minQtLog2SizeInterY;
guint8 sub_width_c, sub_height_c;
const guint8 h266_sub_width_c[] = { 1, 2, 2, 1 };
const guint8 h266_sub_height_c[] = { 1, 2, 1, 1 };
GST_LOG ("parsing \"Sequence parameter set\"");
nal_reader_init (&nr, nalu->data + nalu->offset + nalu->header_bytes,
nalu->size - nalu->header_bytes);
memset (sps, 0, sizeof (*sps));
sps->nuh_layer_id = nalu->layer_id;
READ_UINT8 (&nr, sps->sps_id, 4);
READ_UINT8 (&nr, sps->vps_id, 4);
/* 7.4.3.4: When sps_video_parameter_set_id is equal to 0, SPS
does not refer to a VPS.
We just make vps[0] as the default vps with default flags,
and let the SPS refer to it when vps_id is 0. */
if (sps->vps_id == 0) {
memset (&parser->vps[0], 0, sizeof (*vps));
parser->vps[0].vps_id = 0;
parser->vps[0].max_layers_minus1 = 0;
/* 7.4.3.4:
The value of GeneralLayerIdx[nuh_layer_id] is set equal to 0.
The value of vps_independent_layer_flag[GeneralLayerIdx[nuh_layer_id]]
is inferred to be equal to 1.
The value of TotalNumOlss is set equal to 1,
the value of NumLayersInOls[0] is set equal to 1, and value of
vps_layer_id[0] is inferred to be equal to the value of nuh_layer_id
of all the VCL NAL units, and the value of LayerIdInOls[0][0]
is set equal to vps_layer_id[0]. */
parser->vps[0].independent_layer_flag[0] = 1;
parser->vps[0].total_num_olss = 1;
parser->vps[0].num_layers_in_ols[0] = 1;
parser->vps[0].layer_id[0] = sps->nuh_layer_id;
parser->vps[0].layer_id_in_ols[0][0] = parser->vps[0].layer_id[0];
parser->vps[0].valid = TRUE;
vps = &parser->vps[0];
} else {
vps = gst_h266_parser_get_vps (parser, sps->vps_id);
if (!vps) {
GST_DEBUG ("couldn't find associated video parameter set with id: %d",
sps->vps_id);
}
}
sps->vps = vps;
READ_UINT8 (&nr, sps->max_sublayers_minus1, 3);
CHECK_ALLOWED_MAX (sps->max_sublayers_minus1, GST_H266_MAX_SUBLAYERS - 1);
READ_UINT8 (&nr, sps->chroma_format_idc, 2);
READ_UINT8 (&nr, sps->log2_ctu_size_minus5, 2);
CHECK_ALLOWED_MAX (sps->log2_ctu_size_minus5, 2);
ctbLog2SizeY = sps->log2_ctu_size_minus5 + 5;
sps->ctu_size = 1 << ctbLog2SizeY;
READ_UINT8 (&nr, sps->ptl_dpb_hrd_params_present_flag, 1);
if (!sps->vps_id && !sps->ptl_dpb_hrd_params_present_flag) {
GST_WARNING ("When vps_id is equal to 0, the value of "
"ptl_dpb_hrd_params_present_flag shall be equal to 1");
goto error;
}
if (sps->ptl_dpb_hrd_params_present_flag) {
if (!gst_h266_parse_profile_tier_level (&sps->profile_tier_level, &nr,
1, sps->max_sublayers_minus1))
goto error;
}
ptl = &sps->profile_tier_level;
READ_UINT8 (&nr, sps->gdr_enabled_flag, 1);
if (ptl->general_constraints_info.no_gdr_constraint_flag
&& sps->gdr_enabled_flag) {
GST_WARNING ("When gci_no_gdr_constraint_flag equal to 1 , "
"the value of gdr_enabled_flag shall be equal to 0");
goto error;
}
READ_UINT8 (&nr, sps->ref_pic_resampling_enabled_flag, 1);
if (ptl->general_constraints_info.no_ref_pic_resampling_constraint_flag
&& sps->ref_pic_resampling_enabled_flag) {
GST_WARNING ("When gci_no_ref_pic_resampling_constraint_flag is "
"equal to 1, ref_pic_resampling_enabled_flag shall be equal to 0");
goto error;
}
if (sps->ref_pic_resampling_enabled_flag)
READ_UINT8 (&nr, sps->res_change_in_clvs_allowed_flag, 1);
if (ptl->general_constraints_info.no_res_change_in_clvs_constraint_flag
&& sps->res_change_in_clvs_allowed_flag) {
GST_WARNING ("When no_res_change_in_clvs_constraint_flag is equal "
"to 1, res_change_in_clvs_allowed_flag shall be equal to 0");
goto error;
}
READ_UE_MAX (&nr, sps->pic_width_max_in_luma_samples, GST_H266_MAX_WIDTH);
READ_UE_MAX (&nr, sps->pic_height_max_in_luma_samples, GST_H266_MAX_HEIGHT);
sub_width_c = h266_sub_width_c[sps->chroma_format_idc];
sub_height_c = h266_sub_height_c[sps->chroma_format_idc];
READ_UINT8 (&nr, sps->conformance_window_flag, 1);
if (sps->conformance_window_flag) {
guint width, height;
width = sps->pic_width_max_in_luma_samples / sub_width_c;
height = sps->pic_height_max_in_luma_samples / sub_height_c;
READ_UE_MAX (&nr, sps->conf_win_left_offset, width);
READ_UE_MAX (&nr, sps->conf_win_right_offset, width);
READ_UE_MAX (&nr, sps->conf_win_top_offset, height);
READ_UE_MAX (&nr, sps->conf_win_bottom_offset, height);
if (sub_width_c * (sps->conf_win_left_offset +
sps->conf_win_right_offset) >= sps->pic_width_max_in_luma_samples
|| sub_height_c * (sps->conf_win_top_offset +
sps->conf_win_bottom_offset) >=
sps->pic_height_max_in_luma_samples) {
GST_WARNING ("Invalid sps conformance window: (%u, %u, %u, %u), "
"resolution is %ux%u, sub WxH is %ux%u.\n", sps->conf_win_left_offset,
sps->conf_win_right_offset, sps->conf_win_top_offset,
sps->conf_win_bottom_offset, sps->pic_width_max_in_luma_samples,
sps->pic_height_max_in_luma_samples, sub_width_c, sub_height_c);
goto error;
}
}
READ_UINT8 (&nr, sps->subpic_info_present_flag, 1);
if (ptl->general_constraints_info.no_subpic_info_constraint_flag
&& sps->subpic_info_present_flag) {
GST_WARNING ("When gci_no_subpic_info_constraint_flag is equal to 1, "
"the value of subpic_info_present_flag shall be equal to 0");
goto error;
}
if (sps->subpic_info_present_flag) {
guint32 maxPicWidthInCtus =
((sps->pic_width_max_in_luma_samples - 1) / sps->ctu_size) + 1;
guint32 maxPicHeightInCtus =
((sps->pic_height_max_in_luma_samples - 1) / sps->ctu_size) + 1;
READ_UE_MAX (&nr, sps->num_subpics_minus1, GST_H266_MAX_SLICES_PER_AU - 1);
if (sps->num_subpics_minus1 == 0) {
sps->subpic_ctu_top_left_x[0] = 0;
sps->subpic_ctu_top_left_y[0] = 0;
sps->subpic_width_minus1[0] = maxPicWidthInCtus;
sps->subpic_height_minus1[0] = maxPicHeightInCtus;
sps->independent_subpics_flag = 1;
sps->subpic_same_size_flag = 0;
sps->subpic_treated_as_pic_flag[0] = 1;
sps->loop_filter_across_subpic_enabled_flag[0] = 0;
} else {
guint32 tmpWidthVal, tmpHeightVal, numSubpicCols;
READ_UINT8 (&nr, sps->independent_subpics_flag, 1);
READ_UINT8 (&nr, sps->subpic_same_size_flag, 1);
tmpWidthVal = maxPicWidthInCtus;
tmpHeightVal = maxPicHeightInCtus;
numSubpicCols = 1;
for (i = 0; i <= sps->num_subpics_minus1; i++) {
if (!sps->subpic_same_size_flag || i == 0) {
if (i > 0 && sps->pic_width_max_in_luma_samples > sps->ctu_size) {
READ_UINT16 (&nr, sps->subpic_ctu_top_left_x[i],
gst_util_ceil_log2 (tmpWidthVal));
} else
sps->subpic_ctu_top_left_x[i] = 0;
if (i > 0 && sps->pic_height_max_in_luma_samples > sps->ctu_size) {
READ_UINT16 (&nr, sps->subpic_ctu_top_left_y[i],
gst_util_ceil_log2 (tmpHeightVal));
} else
sps->subpic_ctu_top_left_y[i] = 0;
if (i < sps->num_subpics_minus1
&& sps->pic_width_max_in_luma_samples > sps->ctu_size) {
READ_UINT16 (&nr, sps->subpic_width_minus1[i],
gst_util_ceil_log2 (tmpWidthVal));
} else
sps->subpic_width_minus1[i] =
tmpWidthVal - sps->subpic_ctu_top_left_x[i] - 1;
if (i < sps->num_subpics_minus1
&& sps->pic_height_max_in_luma_samples > sps->ctu_size) {
READ_UINT16 (&nr, sps->subpic_height_minus1[i],
gst_util_ceil_log2 (tmpHeightVal));
} else
sps->subpic_height_minus1[i] =
tmpHeightVal - sps->subpic_ctu_top_left_y[i] - 1;
if (sps->subpic_same_size_flag) {
numSubpicCols = tmpWidthVal / (sps->subpic_width_minus1[0] + 1);
if (!(tmpWidthVal % (sps->subpic_width_minus1[0] + 1) == 0)) {
GST_WARNING ("subpic_width_minus1[0] is invalid.");
goto error;
}
if (!(tmpHeightVal % (sps->subpic_height_minus1[0] + 1) == 0)) {
GST_WARNING ("subpic_height_minus1[0] is invalid.");
goto error;
}
if (!(numSubpicCols * (tmpHeightVal /
(sps->subpic_height_minus1[0] + 1)) ==
(sps->num_subpics_minus1 + 1))) {
GST_WARNING ("when subpic_same_size_flag is equal to, "
"num_subpics_minus1 is invalid");
goto error;
}
}
} else {
numSubpicCols = tmpWidthVal / (sps->subpic_width_minus1[0] + 1);
sps->subpic_ctu_top_left_x[i] =
(i % numSubpicCols) * (sps->subpic_width_minus1[0] + 1);
sps->subpic_ctu_top_left_y[i] =
(i / numSubpicCols) * (sps->subpic_height_minus1[0] + 1);
sps->subpic_width_minus1[i] = sps->subpic_width_minus1[0];
sps->subpic_height_minus1[i] = sps->subpic_height_minus1[0];
}
if (!sps->independent_subpics_flag) {
READ_UINT8 (&nr, sps->subpic_treated_as_pic_flag[i], 1);
READ_UINT8 (&nr, sps->loop_filter_across_subpic_enabled_flag[i], 1);
} else {
sps->subpic_treated_as_pic_flag[i] = 1;
sps->loop_filter_across_subpic_enabled_flag[i] = 0;
}
}
}
READ_UE_MAX (&nr, sps->subpic_id_len_minus1, 15);
if ((1 << (sps->subpic_id_len_minus1 + 1)) < sps->num_subpics_minus1 + 1) {
GST_WARNING ("Invalid subpic_id_len_minus1(%d) value",
sps->subpic_id_len_minus1);
goto error;
}
READ_UINT8 (&nr, sps->subpic_id_mapping_explicitly_signalled_flag, 1);
if (sps->subpic_id_mapping_explicitly_signalled_flag) {
READ_UINT8 (&nr, sps->subpic_id_mapping_present_flag, 1);
if (sps->subpic_id_mapping_present_flag)
for (i = 0; i <= sps->num_subpics_minus1; i++)
READ_UINT32 (&nr, sps->subpic_id[i], sps->subpic_id_len_minus1 + 1);
}
} else {
sps->subpic_id_mapping_explicitly_signalled_flag = 0;
sps->num_subpics_minus1 = 0;
sps->independent_subpics_flag = 1;
sps->subpic_ctu_top_left_x[0] = 0;
sps->subpic_ctu_top_left_y[0] = 0;
sps->subpic_width_minus1[0] =
(sps->pic_width_max_in_luma_samples + sps->ctu_size - 1) >>
gst_util_floor_log2 (sps->ctu_size);
sps->subpic_height_minus1[0] =
(sps->pic_height_max_in_luma_samples + sps->ctu_size - 1) >>
gst_util_floor_log2 (sps->ctu_size);
}
if (!sps->subpic_id_mapping_explicitly_signalled_flag ||
!sps->subpic_id_mapping_present_flag)
for (i = 0; i <= sps->num_subpics_minus1; i++)
sps->subpic_id[i] = i;
READ_UE_MAX (&nr, sps->bitdepth_minus8, 8);
READ_UINT8 (&nr, sps->entropy_coding_sync_enabled_flag, 1);
READ_UINT8 (&nr, sps->entry_point_offsets_present_flag, 1);
READ_UINT8 (&nr, sps->log2_max_pic_order_cnt_lsb_minus4, 4);
CHECK_ALLOWED_MAX (sps->log2_max_pic_order_cnt_lsb_minus4, 12);
READ_UINT8 (&nr, sps->poc_msb_cycle_flag, 1);
if (sps->poc_msb_cycle_flag)
READ_UE_MAX (&nr, sps->poc_msb_cycle_len_minus1,
32 - sps->log2_max_pic_order_cnt_lsb_minus4 - 5);
READ_UINT8 (&nr, sps->num_extra_ph_bytes, 2);
CHECK_ALLOWED_MAX (sps->num_extra_ph_bytes, 2);
for (i = 0; i < (sps->num_extra_ph_bytes * 8); i++)
READ_UINT8 (&nr, sps->extra_ph_bit_present_flag[i], 1);
READ_UINT8 (&nr, sps->num_extra_sh_bytes, 2);
CHECK_ALLOWED_MAX (sps->num_extra_sh_bytes, 2);
for (i = 0; i < (sps->num_extra_sh_bytes * 8); i++)
READ_UINT8 (&nr, sps->extra_sh_bit_present_flag[i], 1);
if (sps->ptl_dpb_hrd_params_present_flag) {
if (sps->max_sublayers_minus1 > 0)
READ_UINT8 (&nr, sps->sublayer_dpb_params_flag, 1);
if (!gst_h266_parse_dpb_parameters (&sps->dpb, &nr,
sps->max_sublayers_minus1, sps->sublayer_dpb_params_flag))
goto error;
}
READ_UE_MAX (&nr, sps->log2_min_luma_coding_block_size_minus2, MIN (4,
sps->log2_ctu_size_minus5 + 3));
minCbLog2SizeY = sps->log2_min_luma_coding_block_size_minus2 + 2;
CHECK_ALLOWED_MAX (minCbLog2SizeY, MIN (6, ctbLog2SizeY));
minCuSize = 1 << minCbLog2SizeY;
if (sps->pic_width_max_in_luma_samples % MAX (8, minCuSize) != 0) {
GST_WARNING ("Coded frame width must be a multiple of "
"Max(8, the minimum unit size)");
goto error;
}
if (sps->pic_height_max_in_luma_samples % MAX (8, minCuSize) != 0) {
GST_WARNING ("Coded frame height must be a multiple of "
"Max(8, the minimum unit size)");
goto error;
}
READ_UINT8 (&nr, sps->partition_constraints_override_enabled_flag, 1);
READ_UE_MAX (&nr, sps->log2_diff_min_qt_min_cb_intra_slice_luma, MIN (6,
ctbLog2SizeY) - minCbLog2SizeY);
minQtLog2SizeIntraY =
sps->log2_diff_min_qt_min_cb_intra_slice_luma + minCbLog2SizeY;
minQT[0] = 1 << minQtLog2SizeIntraY;
CHECK_ALLOWED_MAX (minQT[0], 64);
CHECK_ALLOWED_MAX (minQT[0], 1 << ctbLog2SizeY);
READ_UE_MAX (&nr, sps->max_mtt_hierarchy_depth_intra_slice_luma,
2 * (ctbLog2SizeY - minCbLog2SizeY));
maxTTSize[0] = maxBTSize[0] = minQT[0];
if (sps->max_mtt_hierarchy_depth_intra_slice_luma != 0) {
READ_UE_MAX (&nr, sps->log2_diff_max_bt_min_qt_intra_slice_luma,
ctbLog2SizeY - minQtLog2SizeIntraY);
maxBTSize[0] <<= sps->log2_diff_max_bt_min_qt_intra_slice_luma;
READ_UE_MAX (&nr, sps->log2_diff_max_tt_min_qt_intra_slice_luma, MIN (6,
ctbLog2SizeY) - minQtLog2SizeIntraY);
maxTTSize[0] <<= sps->log2_diff_max_tt_min_qt_intra_slice_luma;
}
if (sps->chroma_format_idc != 0)
READ_UINT8 (&nr, sps->qtbtt_dual_tree_intra_flag, 1);
if (sps->qtbtt_dual_tree_intra_flag) {
READ_UE_MAX (&nr, sps->log2_diff_min_qt_min_cb_intra_slice_chroma, MIN (6,
ctbLog2SizeY) - minCbLog2SizeY);
minQT[2] =
1 << (sps->log2_diff_min_qt_min_cb_intra_slice_chroma + minCbLog2SizeY);
READ_UE_MAX (&nr, sps->max_mtt_hierarchy_depth_intra_slice_chroma,
2 * (ctbLog2SizeY - minCbLog2SizeY));
maxTTSize[2] = maxBTSize[2] = minQT[2];
if (sps->max_mtt_hierarchy_depth_intra_slice_chroma != 0) {
guint minQtLog2SizeIntraC =
sps->log2_diff_min_qt_min_cb_intra_slice_chroma + minCbLog2SizeY;
READ_UE_MAX (&nr, sps->log2_diff_max_bt_min_qt_intra_slice_chroma,
MIN (6, ctbLog2SizeY) - minQtLog2SizeIntraC);
maxBTSize[2] <<= sps->log2_diff_max_bt_min_qt_intra_slice_chroma;
READ_UE_MAX (&nr, sps->log2_diff_max_tt_min_qt_intra_slice_chroma,
MIN (6, ctbLog2SizeY) - minQtLog2SizeIntraC);
maxTTSize[2] <<= sps->log2_diff_max_tt_min_qt_intra_slice_chroma;
}
}
READ_UE_MAX (&nr, sps->log2_diff_min_qt_min_cb_inter_slice, MIN (6,
ctbLog2SizeY) - minCbLog2SizeY);
minQtLog2SizeInterY =
sps->log2_diff_min_qt_min_cb_inter_slice + minCbLog2SizeY;
minQT[1] = 1 << minQtLog2SizeInterY;
READ_UE_MAX (&nr, sps->max_mtt_hierarchy_depth_inter_slice,
2 * (ctbLog2SizeY - minCbLog2SizeY));
maxTTSize[1] = maxBTSize[1] = minQT[1];
if (sps->max_mtt_hierarchy_depth_inter_slice != 0) {
READ_UE_MAX (&nr, sps->log2_diff_max_bt_min_qt_inter_slice,
ctbLog2SizeY - minQtLog2SizeInterY);
maxBTSize[1] <<= sps->log2_diff_max_bt_min_qt_inter_slice;
READ_UE_MAX (&nr, sps->log2_diff_max_tt_min_qt_inter_slice,
MIN (6, ctbLog2SizeY) - minQtLog2SizeInterY);
maxTTSize[1] <<= sps->log2_diff_max_bt_min_qt_inter_slice;
}
if (sps->ctu_size > 32)
READ_UINT8 (&nr, sps->max_luma_transform_size_64_flag, 1);
READ_UINT8 (&nr, sps->transform_skip_enabled_flag, 1);
if (sps->transform_skip_enabled_flag) {
READ_UE_MAX (&nr, sps->log2_transform_skip_max_size_minus2, 3);
READ_UINT8 (&nr, sps->bdpcm_enabled_flag, 1);
}
READ_UINT8 (&nr, sps->mts_enabled_flag, 1);
if (sps->mts_enabled_flag) {
READ_UINT8 (&nr, sps->explicit_mts_intra_enabled_flag, 1);
READ_UINT8 (&nr, sps->explicit_mts_inter_enabled_flag, 1);
}
READ_UINT8 (&nr, sps->lfnst_enabled_flag, 1);
if (sps->chroma_format_idc != 0) {
if (!gst_h266_parse_chroma_qp_table (sps, &nr))
goto error;
} else {
sps->joint_cbcr_enabled_flag = 0;
sps->same_qp_table_for_chroma_flag = 0;
}
READ_UINT8 (&nr, sps->sao_enabled_flag, 1);
READ_UINT8 (&nr, sps->alf_enabled_flag, 1);
if (sps->alf_enabled_flag && sps->chroma_format_idc != 0) {
READ_UINT8 (&nr, sps->ccalf_enabled_flag, 1);
} else
sps->ccalf_enabled_flag = 0;
READ_UINT8 (&nr, sps->lmcs_enabled_flag, 1);
READ_UINT8 (&nr, sps->weighted_pred_flag, 1);
READ_UINT8 (&nr, sps->weighted_bipred_flag, 1);
READ_UINT8 (&nr, sps->long_term_ref_pics_flag, 1);
if (sps->vps_id > 0) {
READ_UINT8 (&nr, sps->inter_layer_prediction_enabled_flag, 1);
} else
sps->inter_layer_prediction_enabled_flag = 0;
READ_UINT8 (&nr, sps->idr_rpl_present_flag, 1);
if (ptl->general_constraints_info.no_idr_constraint_flag &&
sps->idr_rpl_present_flag) {
GST_WARNING ("When gci_no_idr_rpl_constraint_flag equal to 1, "
"the value of sps_idr_rpl_present_flag shall be equal to 0.");
goto error;
}
READ_UINT8 (&nr, sps->rpl1_same_as_rpl0_flag, 1);
for (i = 0; i < (sps->rpl1_same_as_rpl0_flag ? 1 : 2); i++) {
READ_UE_MAX (&nr, sps->num_ref_pic_lists[i], GST_H266_MAX_REF_PIC_LISTS);
for (j = 0; j < sps->num_ref_pic_lists[i]; j++)
gst_h266_ref_pic_list_struct (&sps->ref_pic_list_struct[i][j], &nr,
i, j, sps);
}
if (sps->rpl1_same_as_rpl0_flag) {
sps->num_ref_pic_lists[1] = sps->num_ref_pic_lists[0];
memcpy (&sps->ref_pic_list_struct[1], &sps->ref_pic_list_struct[0],
sizeof (sps->ref_pic_list_struct[0]));
}
READ_UINT8 (&nr, sps->ref_wraparound_enabled_flag, 1);
if (sps->ref_wraparound_enabled_flag) {
for (i = 0; i <= sps->num_subpics_minus1; i++) {
if (sps->subpic_treated_as_pic_flag[i] &&
(sps->subpic_width_minus1[i] + 1 !=
(sps->pic_width_max_in_luma_samples + sps->ctu_size - 1) /
sps->ctu_size)) {
GST_WARNING ("sps_ref_wraparound_enabled_flag cannot be equal "
"to 1 when there is at least one subpicture with "
"SubPicTreatedAsPicFlag equal to 1 and the subpicture's width "
"is not equal to picture's width");
goto error;
}
}
}
READ_UINT8 (&nr, sps->temporal_mvp_enabled_flag, 1);
if (sps->temporal_mvp_enabled_flag)
READ_UINT8 (&nr, sps->sbtmvp_enabled_flag, 1);
READ_UINT8 (&nr, sps->amvr_enabled_flag, 1);
READ_UINT8 (&nr, sps->bdof_enabled_flag, 1);
if (sps->bdof_enabled_flag)
READ_UINT8 (&nr, sps->bdof_control_present_in_ph_flag, 1);
READ_UINT8 (&nr, sps->smvd_enabled_flag, 1);
READ_UINT8 (&nr, sps->dmvr_enabled_flag, 1);
if (sps->dmvr_enabled_flag)
READ_UINT8 (&nr, sps->dmvr_control_present_in_ph_flag, 1);
READ_UINT8 (&nr, sps->mmvd_enabled_flag, 1);
if (sps->mmvd_enabled_flag)
READ_UINT8 (&nr, sps->mmvd_fullpel_only_enabled_flag, 1);
READ_UE_MAX (&nr, sps->six_minus_max_num_merge_cand, 5);
MaxNumMergeCand = 6 - sps->six_minus_max_num_merge_cand;
READ_UINT8 (&nr, sps->sbt_enabled_flag, 1);
READ_UINT8 (&nr, sps->affine_enabled_flag, 1);
if (sps->affine_enabled_flag) {
READ_UE_MAX (&nr, sps->five_minus_max_num_subblock_merge_cand,
5 - sps->sbtmvp_enabled_flag);
READ_UINT8 (&nr, sps->sps_6param_affine_enabled_flag, 1);
if (sps->amvr_enabled_flag)
READ_UINT8 (&nr, sps->affine_amvr_enabled_flag, 1);
READ_UINT8 (&nr, sps->affine_prof_enabled_flag, 1);
if (sps->affine_prof_enabled_flag)
READ_UINT8 (&nr, sps->prof_control_present_in_ph_flag, 1);
}
READ_UINT8 (&nr, sps->bcw_enabled_flag, 1);
READ_UINT8 (&nr, sps->ciip_enabled_flag, 1);
if (MaxNumMergeCand >= 2) {
READ_UINT8 (&nr, sps->gpm_enabled_flag, 1);
if (sps->gpm_enabled_flag && MaxNumMergeCand >= 3)
READ_UE_MAX (&nr, sps->max_num_merge_cand_minus_max_num_gpm_cand,
MaxNumMergeCand - 2);
}
READ_UE_MAX (&nr, sps->log2_parallel_merge_level_minus2, ctbLog2SizeY - 2);
READ_UINT8 (&nr, sps->isp_enabled_flag, 1);
READ_UINT8 (&nr, sps->mrl_enabled_flag, 1);
READ_UINT8 (&nr, sps->mip_enabled_flag, 1);
if (sps->chroma_format_idc != 0)
READ_UINT8 (&nr, sps->cclm_enabled_flag, 1);
if (sps->chroma_format_idc == 1) {
READ_UINT8 (&nr, sps->chroma_horizontal_collocated_flag, 1);
READ_UINT8 (&nr, sps->chroma_vertical_collocated_flag, 1);
} else {
sps->chroma_horizontal_collocated_flag = 1;
sps->chroma_vertical_collocated_flag = 1;
}
READ_UINT8 (&nr, sps->palette_enabled_flag, 1);
if ((ptl->profile_idc == GST_H266_PROFILE_MAIN_12 ||
ptl->profile_idc == GST_H266_PROFILE_MAIN_12_INTRA ||
ptl->profile_idc == GST_H266_PROFILE_MAIN_12_STILL_PICTURE) &&
sps->palette_enabled_flag) {
GST_WARNING ("sps_palette_enabled_flag shall be equal to 0 "
"for Main 12 (420) profiles");
goto error;
}
if (sps->chroma_format_idc == 3 && !sps->max_luma_transform_size_64_flag)
READ_UINT8 (&nr, sps->act_enabled_flag, 1);
if (sps->transform_skip_enabled_flag || sps->palette_enabled_flag)
READ_UE_MAX (&nr, sps->min_qp_prime_ts, 8);
READ_UINT8 (&nr, sps->ibc_enabled_flag, 1);
if (sps->ibc_enabled_flag)
READ_UE_MAX (&nr, sps->six_minus_max_num_ibc_merge_cand, 5);
READ_UINT8 (&nr, sps->ladf_enabled_flag, 1);
if (sps->ladf_enabled_flag) {
READ_UINT8 (&nr, sps->num_ladf_intervals_minus2, 2);
READ_SE_ALLOWED (&nr, sps->ladf_lowest_interval_qp_offset, -63, 63);
for (i = 0; i < sps->num_ladf_intervals_minus2 + 1; i++) {
READ_SE_ALLOWED (&nr, sps->ladf_qp_offset[i], -63, 63);
READ_UE_MAX (&nr, sps->ladf_delta_threshold_minus1[i],
(2 << (8 + sps->bitdepth_minus8)) - 3);
}
}
READ_UINT8 (&nr, sps->explicit_scaling_list_enabled_flag, 1);
if (sps->lfnst_enabled_flag && sps->explicit_scaling_list_enabled_flag)
READ_UINT8 (&nr, sps->scaling_matrix_for_lfnst_disabled_flag, 1);
if (sps->act_enabled_flag && sps->explicit_scaling_list_enabled_flag)
READ_UINT8 (&nr,
sps->scaling_matrix_for_alternative_colour_space_disabled_flag, 1);
if (sps->scaling_matrix_for_alternative_colour_space_disabled_flag)
READ_UINT8 (&nr, sps->scaling_matrix_designated_colour_space_flag, 1);
READ_UINT8 (&nr, sps->dep_quant_enabled_flag, 1);
READ_UINT8 (&nr, sps->sign_data_hiding_enabled_flag, 1);
READ_UINT8 (&nr, sps->virtual_boundaries_enabled_flag, 1);
if (ptl->general_constraints_info.no_virtual_boundaries_constraint_flag &&
sps->virtual_boundaries_enabled_flag) {
GST_WARNING ("When gci_no_virtual_boundaries_constraint_flag is "
"equal to 1, sps_virtual_boundaries_enabled_flag shall be "
"equal to 0");
goto error;
}
if (sps->virtual_boundaries_enabled_flag) {
READ_UINT8 (&nr, sps->virtual_boundaries_present_flag, 1);
if (sps->virtual_boundaries_present_flag) {
READ_UE (&nr, sps->num_ver_virtual_boundaries);
if (sps->pic_width_max_in_luma_samples <= 8 &&
sps->num_ver_virtual_boundaries != 0) {
GST_WARNING ("SPS: When picture width is less than or equal to 8, "
"the number of vertical virtual boundaries shall be equal to 0");
goto error;
}
if (sps->num_ver_virtual_boundaries > 3) {
GST_WARNING ("SPS: The number of vertical virtual boundaries "
"shall be in the range of 0 to 3");
goto error;
}
for (i = 0; i < sps->num_ver_virtual_boundaries; i++)
READ_UE_MAX (&nr, sps->virtual_boundary_pos_x_minus1[i],
(sps->pic_width_max_in_luma_samples + 7) / 8 - 2);
READ_UE (&nr, sps->num_hor_virtual_boundaries);
if (sps->pic_height_max_in_luma_samples <= 8 &&
sps->num_hor_virtual_boundaries != 0) {
GST_WARNING ("SPS: When picture height is less than or equal to 8, "
"the number of horizontal virtual boundaries shall be equal to 0");
goto error;
}
if (sps->num_hor_virtual_boundaries > 3) {
GST_WARNING ("SPS: The number of horizontal virtual boundaries "
"shall be in the range of 0 to 3");
goto error;
}
for (i = 0; i < sps->num_hor_virtual_boundaries; i++)
READ_UE_MAX (&nr, sps->virtual_boundary_pos_y_minus1[i],
(sps->pic_height_max_in_luma_samples + 7) / 8 - 2);
}
}
if (sps->ptl_dpb_hrd_params_present_flag) {
READ_UINT8 (&nr, sps->timing_hrd_params_present_flag, 1);
if (sps->timing_hrd_params_present_flag) {
guint8 firstSubLayer;
gst_h266_parse_general_timing_hrd_parameters (&sps->general_hrd_params,
&nr);
if (sps->max_sublayers_minus1 > 0)
READ_UINT8 (&nr, sps->sublayer_cpb_params_present_flag, 1);
firstSubLayer =
sps->sublayer_cpb_params_present_flag ? 0 : sps->max_sublayers_minus1;
gst_h266_parse_ols_timing_hrd_parameters (&sps->ols_hrd_params, &nr,
&sps->general_hrd_params, firstSubLayer, sps->max_sublayers_minus1);
}
}
READ_UINT8 (&nr, sps->field_seq_flag, 1);
READ_UINT8 (&nr, sps->vui_parameters_present_flag, 1);
if (sps->vui_parameters_present_flag) {
READ_UE_MAX (&nr, sps->vui_payload_size_minus1, 1023);
while (!nal_reader_is_byte_aligned (&nr))
if (!nal_reader_skip (&nr, 1))
goto error;
if (!gst_h266_parse_vui_payload (&sps->vui_params, &nr,
sps->vui_payload_size_minus1 + 1))
goto error;
} else {
gst_h266_vui_parameters_set_default (&sps->vui_params);
}
READ_UINT8 (&nr, sps->extension_flag, 1);
if (sps->extension_flag) {
READ_UINT8 (&nr, sps->range_extension_flag, 1);
for (i = 0; i < 7; i++) {
READ_UINT8 (&nr, sps->extension_7_flags[i], 1);
if (sps->extension_7_flags[i]) {
GST_WARNING ("The value of sps_extension_7bits shall be equal "
"to 0 in bitstreams conforming to this version of this document");
goto error;
}
}
if (sps->range_extension_flag) {
if (sps->bitdepth_minus8 + 8 <= 10) {
GST_WARNING ("The value of sps_range_extension_flag shall be 0 "
"when BitDepth is less than or equal to 10.");
goto error;
}
if (!gst_h266_parse_range_extension (&sps->range_params,
&nr, sps->transform_skip_enabled_flag))
goto error;
}
}
sps->max_width = sps->pic_width_max_in_luma_samples;
sps->max_height = sps->pic_height_max_in_luma_samples;
if (sps->conformance_window_flag) {
sps->crop_rect_width = sps->max_width -
(sps->conf_win_left_offset + sps->conf_win_right_offset) * sub_width_c;
sps->crop_rect_height = sps->max_height -
(sps->conf_win_top_offset + sps->conf_win_bottom_offset) * sub_height_c;
sps->crop_rect_x = sps->conf_win_left_offset * sub_width_c;
sps->crop_rect_y = sps->conf_win_top_offset * sub_height_c;
GST_LOG ("crop_rectangle x=%u y=%u width=%u, height=%u", sps->crop_rect_x,
sps->crop_rect_y, sps->crop_rect_width, sps->crop_rect_height);
}
/* calculate fps_num fps_den */
sps->fps_num = 0;
sps->fps_den = 1;
if (sps->ptl_dpb_hrd_params_present_flag
&& sps->timing_hrd_params_present_flag) {
sps->fps_num = sps->general_hrd_params.time_scale;
sps->fps_den = sps->general_hrd_params.num_units_in_tick;
GST_LOG ("framerate %d/%d in SPS", sps->fps_num, sps->fps_den);
} else if (vps && vps->timing_hrd_params_present_flag) {
sps->fps_num = vps->general_hrd_params.time_scale;
sps->fps_den = vps->general_hrd_params.num_units_in_tick;
GST_LOG ("framerate %d/%d in VPS", sps->fps_num, sps->fps_den);
} else {
GST_LOG ("unknown framerate");
}
sps->valid = TRUE;
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Sequence parameter set\"");
sps->valid = FALSE;
return GST_H266_PARSER_ERROR;
}
static gboolean
gst_h266_parser_parse_tile_info (GstH266SPS * sps,
GstH266PPS * pps, NalReader * nr)
{
guint exp_tile_width = 0, exp_tile_height = 0;
guint unified_size, remaining_size;
gint i;
GST_LOG ("parsing \"Tile Info\"");
READ_UE_MAX (nr, pps->num_exp_tile_columns_minus1,
MIN (pps->pic_width_in_ctbs_y - 1, GST_H266_MAX_TILE_COLUMNS - 1));
READ_UE_MAX (nr, pps->num_exp_tile_rows_minus1,
MIN (pps->pic_height_in_ctbs_y - 1, GST_H266_MAX_TILE_ROWS - 1));
for (i = 0; i <= pps->num_exp_tile_columns_minus1; i++) {
READ_UE_MAX (nr, pps->tile_column_width_minus1[i],
pps->pic_width_in_ctbs_y - exp_tile_width - 1);
exp_tile_width += pps->tile_column_width_minus1[i] + 1;
}
remaining_size = pps->pic_width_in_ctbs_y - exp_tile_width;
unified_size = (i == 0 ? pps->pic_width_in_ctbs_y :
(pps->tile_column_width_minus1[i - 1] + 1));
pps->num_tile_columns =
i + ((remaining_size + unified_size - 1) / unified_size);
if (pps->num_tile_columns > GST_H266_MAX_TILE_COLUMNS) {
GST_WARNING ("NumTileColumns(%d) large than max tile columns %d.\n",
pps->num_tile_columns, GST_H266_MAX_TILE_COLUMNS);
goto error;
}
while (remaining_size > unified_size) {
pps->tile_column_width_minus1[i] = unified_size - 1;
remaining_size -= unified_size;
i++;
}
if (remaining_size > 0)
pps->tile_column_width_minus1[i] = remaining_size - 1;
for (i = 0; i <= pps->num_exp_tile_rows_minus1; i++) {
READ_UE_MAX (nr, pps->tile_row_height_minus1[i],
pps->pic_height_in_ctbs_y - exp_tile_height - 1);
exp_tile_height += pps->tile_row_height_minus1[i] + 1;
}
remaining_size = pps->pic_height_in_ctbs_y - exp_tile_height;
unified_size = (i == 0 ? pps->pic_height_in_ctbs_y :
(pps->tile_row_height_minus1[i - 1] + 1));
pps->num_tile_rows = i + ((remaining_size + unified_size - 1) / unified_size);
if (pps->num_tile_rows > GST_H266_MAX_TILE_ROWS) {
GST_WARNING ("NumTileRows(%d) large than max tile rows %d.\n",
pps->num_tile_rows, GST_H266_MAX_TILE_ROWS);
goto error;
}
while (remaining_size > unified_size) {
pps->tile_row_height_minus1[i] = unified_size - 1;
remaining_size -= unified_size;
i++;
}
if (remaining_size > 0)
pps->tile_row_height_minus1[i] = remaining_size - 1;
pps->num_tiles_in_pic = pps->num_tile_columns * pps->num_tile_rows;
if (pps->num_tiles_in_pic > GST_H266_MAX_TILES_PER_AU) {
GST_WARNING ("NumTilesInPic(%d) large than max tiles per AU %d.\n",
pps->num_tiles_in_pic, GST_H266_MAX_TILES_PER_AU);
goto error;
}
if (pps->num_tiles_in_pic > 1) {
READ_UINT8 (nr, pps->loop_filter_across_tiles_enabled_flag, 1);
READ_UINT8 (nr, pps->rect_slice_flag, 1);
} else {
pps->loop_filter_across_tiles_enabled_flag = 0;
pps->rect_slice_flag = 1;
}
pps->tile_col_bd_val[0] = 0;
for (i = 0; i < pps->num_tile_columns; i++) {
pps->tile_col_bd_val[i + 1] =
pps->tile_col_bd_val[i] + pps->tile_column_width_minus1[i] + 1;
}
pps->tile_row_bd_val[0] = 0;
for (i = 0; i < pps->num_tile_rows; i++) {
pps->tile_row_bd_val[i + 1] =
pps->tile_row_bd_val[i] + pps->tile_row_height_minus1[i] + 1;
}
return TRUE;
error:
GST_WARNING ("error parsing \"Tile Info\"");
return FALSE;
}
static gboolean
gst_h266_parser_parse_picture_partition (GstH266SPS * sps,
GstH266PPS * pps, NalReader * nr)
{
guint16 tile_idx, tile_x, tile_y, ctu_x, ctu_y;
gint i, j;
GST_LOG ("parsing \"Picture Partition\"");
READ_UINT8 (nr, pps->log2_ctu_size_minus5, 2);
if (pps->log2_ctu_size_minus5 != sps->log2_ctu_size_minus5) {
GST_WARNING ("pps_log2_ctu_size_minus5 shall be equal "
"to sps_log2_ctu_size_minus5");
goto error;
}
if (!gst_h266_parser_parse_tile_info (sps, pps, nr))
goto error;
if (pps->rect_slice_flag) {
READ_UINT8 (nr, pps->single_slice_per_subpic_flag, 1);
} else {
pps->single_slice_per_subpic_flag = 0;
}
if (pps->rect_slice_flag) {
if (!pps->single_slice_per_subpic_flag) {
tile_idx = 0;
READ_UE_MAX (nr, pps->num_slices_in_pic_minus1,
GST_H266_MAX_SLICES_PER_AU - 1);
if (pps->num_slices_in_pic_minus1 > 1) {
READ_UINT8 (nr, pps->tile_idx_delta_present_flag, 1);
} else {
pps->tile_idx_delta_present_flag = 0;
}
/* Handle the last one after this loop */
for (i = 0; i < pps->num_slices_in_pic_minus1; i++) {
pps->slice_top_left_tile_idx[i] = tile_idx;
tile_x = tile_idx % pps->num_tile_columns;
tile_y = tile_idx / pps->num_tile_columns;
if (tile_x != pps->num_tile_columns - 1) {
READ_UE_MAX (nr, pps->slice_width_in_tiles_minus1[i],
pps->num_tile_columns - 1);
} else {
pps->slice_width_in_tiles_minus1[i] = 0;
}
if (tile_y != pps->num_tile_rows - 1 &&
(pps->tile_idx_delta_present_flag || tile_x == 0)) {
READ_UE_MAX (nr, pps->slice_height_in_tiles_minus1[i],
pps->num_tile_rows - 1);
} else {
if (tile_y == pps->num_tile_rows - 1) {
pps->slice_height_in_tiles_minus1[i] = 0;
} else {
/* tile_x != 0, so i should be > 0 when we get here. */
pps->slice_height_in_tiles_minus1[i] =
pps->slice_height_in_tiles_minus1[i - 1];
}
}
ctu_x = pps->tile_col_bd_val[tile_x];
ctu_y = pps->tile_row_bd_val[tile_y];
/* slice is no bigger than tile */
if (pps->slice_width_in_tiles_minus1[i] == 0 &&
pps->slice_height_in_tiles_minus1[i] == 0 &&
pps->tile_row_height_minus1[tile_y] > 0) {
gint num_slices_in_tile, uniform_slice_height,
remaining_height_in_ctbs_y;
gint k;
remaining_height_in_ctbs_y = pps->tile_row_height_minus1[tile_y] + 1;
READ_UE_MAX (nr, pps->num_exp_slices_in_tile[i],
pps->tile_row_height_minus1[tile_y]);
/* slice is equal to tile */
if (pps->num_exp_slices_in_tile[i] == 0) {
num_slices_in_tile = 1;
pps->slice_top_left_ctu_x[i] = ctu_x;
pps->slice_top_left_ctu_y[i] = ctu_y;
pps->slice_height_in_ctus[i] =
pps->tile_row_height_minus1[tile_y] + 1;
} else { /* tile contains multi slices */
guint16 slice_height_in_ctus;
for (j = 0; j < pps->num_exp_slices_in_tile[i]; j++) {
READ_UE_MAX (nr, pps->exp_slice_height_in_ctus_minus1[i][j],
pps->tile_row_height_minus1[tile_y]);
slice_height_in_ctus =
pps->exp_slice_height_in_ctus_minus1[i][j] + 1;
pps->slice_height_in_ctus[i + j] = slice_height_in_ctus;
pps->slice_top_left_ctu_x[i + j] = ctu_x;
pps->slice_top_left_ctu_y[i + j] = ctu_y;
ctu_y += slice_height_in_ctus;
remaining_height_in_ctbs_y -= slice_height_in_ctus;
}
uniform_slice_height =
1 + pps->exp_slice_height_in_ctus_minus1[i][j - 1];
/* Assign the remaining CTBs to slices */
while (remaining_height_in_ctbs_y > uniform_slice_height) {
if (i + j > pps->num_slices_in_pic_minus1) {
GST_WARNING ("Too may slices %d", i + j + 1);
goto error;
}
pps->slice_height_in_ctus[i + j] = uniform_slice_height;
pps->slice_top_left_ctu_x[i + j] = ctu_x;
pps->slice_top_left_ctu_y[i + j] = ctu_y;
ctu_y += uniform_slice_height;
remaining_height_in_ctbs_y -= uniform_slice_height;
j++;
}
if (remaining_height_in_ctbs_y > 0) {
if (i + j > pps->num_slices_in_pic_minus1) {
GST_WARNING ("Too may slices %d", i + j + 1);
goto error;
}
pps->slice_height_in_ctus[i + j] = remaining_height_in_ctbs_y;
pps->slice_top_left_ctu_x[i + j] = ctu_x;
pps->slice_top_left_ctu_y[i + j] = ctu_y;
j++;
}
num_slices_in_tile = j;
}
/* slice_top_left_tile_idx[0] already set */
for (k = 1; k < num_slices_in_tile; k++)
pps->slice_top_left_tile_idx[i + k] = tile_idx;
i += num_slices_in_tile - 1;
} else { /* Slice may contain multi tiles. */
guint16 height = 0;
pps->num_exp_slices_in_tile[i] = 0;
for (j = 0; j <= pps->slice_height_in_tiles_minus1[i]; j++)
height += pps->tile_row_height_minus1[tile_y + j] + 1;
pps->slice_height_in_ctus[i] = height;
pps->slice_top_left_ctu_x[i] = ctu_x;
pps->slice_top_left_ctu_y[i] = ctu_y;
}
if (i < pps->num_slices_in_pic_minus1) {
if (pps->tile_idx_delta_present_flag) {
gint num_tiles_in_pic = pps->num_tiles_in_pic;
READ_SE_ALLOWED (nr, pps->tile_idx_delta_val[i],
-num_tiles_in_pic + 1, num_tiles_in_pic - 1);
if (pps->tile_idx_delta_val[i] == 0) {
/* When present, the value of pps_tile_idx_delta_val[i]
shall not be equal to 0. */
GST_WARNING ("pps->tile_idx_delta_val[i] shall not be"
" equal to 0.\n");
goto error;
}
tile_idx += pps->tile_idx_delta_val[i];
} else {
pps->tile_idx_delta_val[i] = 0;
tile_idx += pps->slice_width_in_tiles_minus1[i] + 1;
if (tile_idx % pps->num_tile_columns == 0) {
tile_idx += pps->slice_height_in_tiles_minus1[i] *
pps->num_tile_columns;
}
}
}
}
if (i > pps->num_slices_in_pic_minus1 + 1) {
GST_WARNING ("wrong slice num %d, bigger than total slice num %d",
i, pps->num_slices_in_pic_minus1 + 1);
goto error;
} else if (i == pps->num_slices_in_pic_minus1) {
/* Assign the left to the last slice if not explicitly assigned. */
guint16 height = 0;
pps->slice_top_left_tile_idx[i] = tile_idx;
tile_x = tile_idx % pps->num_tile_columns;
tile_y = tile_idx / pps->num_tile_columns;
ctu_x = 0, ctu_y = 0;
for (j = 0; j < tile_x; j++)
ctu_x += pps->tile_column_width_minus1[j] + 1;
for (j = 0; j < tile_y; j++)
ctu_y += pps->tile_row_height_minus1[j] + 1;
pps->slice_top_left_ctu_x[i] = ctu_x;
pps->slice_top_left_ctu_y[i] = ctu_y;
pps->slice_width_in_tiles_minus1[i] =
pps->num_tile_columns - tile_x - 1;
pps->slice_height_in_tiles_minus1[i] = pps->num_tile_rows - tile_y - 1;
for (j = 0; j <= pps->slice_height_in_tiles_minus1[i]; j++)
height += pps->tile_row_height_minus1[tile_y + j] + 1;
pps->slice_height_in_ctus[i] = height;
pps->num_exp_slices_in_tile[i] = 0;
}
/* calculate NumSlicesInSubpic */
for (i = 0; i <= sps->num_subpics_minus1; i++) {
pps->num_slices_in_subpic[i] = 0;
for (j = 0; j <= pps->num_slices_in_pic_minus1; j++) {
guint16 pos_x = pps->slice_top_left_ctu_x[j];
guint16 pos_y = pps->slice_top_left_ctu_y[j];
if ((pos_x >= sps->subpic_ctu_top_left_x[i]) &&
(pos_x < sps->subpic_ctu_top_left_x[i] +
sps->subpic_width_minus1[i] + 1) &&
(pos_y >= sps->subpic_ctu_top_left_y[i])
&& (pos_y < sps->subpic_ctu_top_left_y[i] +
sps->subpic_height_minus1[i] + 1))
pps->num_slices_in_subpic[i]++;
}
}
} else {
gint start_x, start_y;
pps->num_slices_in_pic_minus1 = sps->num_subpics_minus1;
for (i = 0; i <= sps->num_subpics_minus1; i++) {
start_x = -1;
start_y = -1;
pps->num_slices_in_subpic[i] = 1;
for (tile_y = 0; tile_y < pps->num_tile_rows; tile_y++) {
for (tile_x = 0; tile_x < pps->num_tile_columns; tile_x++) {
if ((pps->tile_col_bd_val[tile_x] >= sps->subpic_ctu_top_left_x[i])
&& (pps->tile_col_bd_val[tile_x] < sps->subpic_ctu_top_left_x[i]
+ sps->subpic_width_minus1[i] + 1)
&& (pps->tile_row_bd_val[tile_y] >=
sps->subpic_ctu_top_left_y[i])
&& (pps->tile_row_bd_val[tile_y] < sps->subpic_ctu_top_left_y[i]
+ sps->subpic_height_minus1[i] + 1)) {
if (start_x == -1) {
start_x = tile_x;
start_y = tile_y;
}
pps->slice_width_in_tiles_minus1[i] = tile_x - start_x;
pps->slice_height_in_tiles_minus1[i] = tile_y - start_y;
}
}
}
if (start_x == -1) {
GST_WARNING ("No tile found for subpic start at: [%d, %d], "
"size: [%d, %d] in CTUs", sps->subpic_ctu_top_left_x[i],
sps->subpic_ctu_top_left_y[i], sps->subpic_width_minus1[i] + 1,
sps->subpic_height_minus1[i] + 1);
goto error;
}
pps->slice_top_left_tile_idx[i] =
start_x + start_y * pps->num_tile_columns;
pps->slice_top_left_ctu_x[i] = sps->subpic_ctu_top_left_x[i];
pps->slice_top_left_ctu_y[i] = sps->subpic_ctu_top_left_y[i];
pps->slice_height_in_ctus[i] = sps->subpic_height_minus1[i] + 1;
}
}
}
if (!pps->rect_slice_flag || pps->single_slice_per_subpic_flag ||
pps->num_slices_in_pic_minus1 > 0) {
READ_UINT8 (nr, pps->loop_filter_across_slices_enabled_flag, 1);
} else {
pps->loop_filter_across_slices_enabled_flag = 0;
}
return TRUE;
error:
GST_WARNING ("error parsing \"Picture Partition\"");
return FALSE;
}
static guint16
gst_h266_parser_add_slice_ctus_map (GstH266Parser * parser,
const GstH266PPS * pps, guint slice_start_offset,
guint ctu_x, guint ctu_y, guint width, guint height)
{
guint x, y;
guint16 ctb_count = 0;
for (y = ctu_y; y < ctu_y + height; y++) {
for (x = ctu_x; x < ctu_x + width; x++) {
/* CtbAddrInRs */
parser->ctb_addr_in_slice[slice_start_offset + ctb_count] =
y * pps->pic_width_in_ctbs_y + x;
ctb_count++;
}
}
return ctb_count;
}
static gboolean
gst_h266_parser_generate_ctb_map (GstH266Parser * parser,
const GstH266PPS * pps)
{
gint tile_y, tile_x, ctu_y, ctu_x;
guint ctb_addr_x, ctb_addr_y;
gint ctu_idx = 0;
gint i, j, k;
memset (parser->ctb_addr_in_slice, 0, sizeof (parser->ctb_addr_in_slice));
memset (parser->slice_start_offset, 0, sizeof (parser->slice_start_offset));
memset (parser->num_ctus_in_slice, 0, sizeof (parser->num_ctus_in_slice));
memset (parser->ctb_to_tile_col_bd, 0, sizeof (parser->ctb_to_tile_col_bd));
memset (parser->ctb_to_tile_row_bd, 0, sizeof (parser->ctb_to_tile_row_bd));
if (pps->pic_size_in_ctbs_y >= GST_H266_MAX_CTUS_IN_PICTURE) {
GST_WARNING ("Too many CTBs %d", pps->pic_size_in_ctbs_y);
return FALSE;
}
tile_x = 0, tile_y = 0;
for (ctb_addr_x = 0; ctb_addr_x < pps->pic_width_in_ctbs_y; ctb_addr_x++) {
if (ctb_addr_x == pps->tile_col_bd_val[tile_x + 1])
tile_x++;
parser->ctb_to_tile_col_bd[ctb_addr_x] = pps->tile_col_bd_val[tile_x];
}
parser->ctb_to_tile_col_bd[pps->pic_width_in_ctbs_y] =
pps->pic_width_in_ctbs_y;
for (ctb_addr_y = 0; ctb_addr_y < pps->pic_height_in_ctbs_y; ctb_addr_y++) {
if (ctb_addr_y == pps->tile_row_bd_val[tile_y + 1])
tile_y++;
parser->ctb_to_tile_row_bd[ctb_addr_y] = pps->tile_row_bd_val[tile_y];
}
parser->ctb_to_tile_row_bd[pps->pic_height_in_ctbs_y] =
pps->pic_height_in_ctbs_y;
/* Map between raster scan address and CTU address.
* For non rect slice mode, the slice number for each picture is not
* fixed, we only need to establish the map based on tile info.
*
* For rect slice mode, the slice structure for each picture is fixed
* based on the PPS info. So beside the map, we can also know the
* slice_start_offset and num_ctus_in_slice for each slice.
*/
if (!pps->no_pic_partition_flag && pps->rect_slice_flag) {
guint16 ctb_count;
guint16 slice_start_offset = 0;
for (i = 0; i <= pps->num_slices_in_pic_minus1; i++) {
tile_x = pps->slice_top_left_tile_idx[i] % pps->num_tile_columns;
tile_y = pps->slice_top_left_tile_idx[i] / pps->num_tile_columns;
if (pps->slice_width_in_tiles_minus1[i] == 0 &&
pps->slice_height_in_tiles_minus1[i] == 0) {
/* Slice contains no more than one tile, the slice_top_left_ctu_x/y
and slice_height_in_ctus give all the info. */
ctb_count = gst_h266_parser_add_slice_ctus_map (parser,
pps, slice_start_offset, pps->slice_top_left_ctu_x[i],
pps->slice_top_left_ctu_y[i],
pps->tile_column_width_minus1[tile_x] + 1,
pps->slice_height_in_ctus[i]);
parser->slice_start_offset[i] = slice_start_offset;
parser->num_ctus_in_slice[i] = ctb_count;
slice_start_offset += ctb_count;
} else {
guint16 ctu_x, ctu_y, ctu_width, ctu_height;
g_assert (pps->tile_col_bd_val[tile_x] == pps->slice_top_left_ctu_x[i]);
g_assert (pps->tile_row_bd_val[tile_y] == pps->slice_top_left_ctu_y[i]);
parser->slice_start_offset[i] = slice_start_offset;
for (j = 0; j <= pps->slice_height_in_tiles_minus1[i]; j++) {
for (k = 0; k <= pps->slice_width_in_tiles_minus1[i]; k++) {
ctu_x = pps->tile_col_bd_val[tile_x + k];
ctu_y = pps->tile_row_bd_val[tile_y + j];
ctu_width = pps->tile_column_width_minus1[tile_x + k] + 1;
ctu_height = pps->tile_row_height_minus1[tile_y + j] + 1;
ctb_count = gst_h266_parser_add_slice_ctus_map (parser,
pps, slice_start_offset, ctu_x, ctu_y, ctu_width, ctu_height);
slice_start_offset += ctb_count;
}
}
parser->num_ctus_in_slice[i] =
slice_start_offset - parser->slice_start_offset[i];
}
}
} else {
for (tile_y = 0; tile_y < pps->num_tile_rows; tile_y++) {
for (tile_x = 0; tile_x < pps->num_tile_columns; tile_x++) {
for (ctu_y = pps->tile_row_bd_val[tile_y];
ctu_y < pps->tile_row_bd_val[tile_y] +
pps->tile_row_height_minus1[tile_y] + 1; ctu_y++) {
for (ctu_x = pps->tile_col_bd_val[tile_x];
ctu_x < pps->tile_col_bd_val[tile_x] +
pps->tile_column_width_minus1[tile_x] + 1; ctu_x++) {
g_assert (ctu_idx < pps->pic_size_in_ctbs_y);
parser->ctb_addr_in_slice[ctu_idx++] =
ctu_y * pps->pic_width_in_ctbs_y + ctu_x;
}
}
}
}
}
return TRUE;
}
static gboolean
gst_h266_parser_set_active_sps_pps (GstH266Parser * parser,
const GstH266SPS * sps, const GstH266PPS * pps)
{
gboolean need_update = FALSE;
g_assert (sps->vps);
g_assert (sps->valid);
g_assert (pps->valid);
if (sps->vps != parser->active_vps) {
parser->active_vps = (GstH266VPS *) sps->vps;
need_update = TRUE;
}
if (sps != parser->active_sps) {
parser->active_sps = (GstH266SPS *) sps;
need_update = TRUE;
}
if (pps != parser->active_pps) {
parser->active_pps = (GstH266PPS *) pps;
need_update = TRUE;
}
if (need_update) {
if (!gst_h266_parser_generate_ctb_map (parser, pps))
return FALSE;
GST_LOG ("set active VPS:%d, SPS:%d, PPS:%d",
sps->vps->vps_id, sps->sps_id, pps->pps_id);
}
return TRUE;
}
/**
* gst_h266_parser_parse_pps:
* @parser: a #GstH266Parser
* @nalu: The #GST_H266_NAL_PPS #GstH266NalUnit to parse
* @pps: The #GstH266PPS to fill.
*
* Parses @data, and fills the @pps structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_pps (GstH266Parser * parser,
GstH266NalUnit * nalu, GstH266PPS * pps)
{
GstH266ParserResult res = gst_h266_parse_pps (parser, nalu, pps);
if (res == GST_H266_PARSER_OK) {
GST_LOG ("adding picture parameter set with id: %d to array", pps->pps_id);
if (parser->active_pps && parser->active_pps->pps_id == pps->pps_id)
parser->active_pps = NULL;
parser->pps[pps->pps_id] = *pps;
parser->last_pps = &parser->pps[pps->pps_id];
}
return res;
}
/**
* gst_h266_parse_pps:
* @parser: a #GstH266Parser
* @nalu: The #GST_H266_NAL_PPS #GstH266NalUnit to parse
* @pps: The #GstH266PPS to fill.
*
* Parses @data, and fills the @pps structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parse_pps (GstH266Parser * parser, GstH266NalUnit * nalu,
GstH266PPS * pps)
{
NalReader nr;
GstH266SPS *sps;
gint min_cb_size_y, ctb_size_y;
guint8 sub_width_c, sub_height_c;
const guint8 h266_sub_width_c[] = { 1, 2, 2, 1 };
const guint8 h266_sub_height_c[] = { 1, 2, 1, 1 };
gint QpBdOffset, i;
GST_LOG ("parsing \"Picture parameter set\"");
nal_reader_init (&nr, nalu->data + nalu->offset + nalu->header_bytes,
nalu->size - nalu->header_bytes);
memset (pps, 0, sizeof (*pps));
READ_UINT8 (&nr, pps->pps_id, 6);
READ_UINT8 (&nr, pps->sps_id, 4);
sps = gst_h266_parser_get_sps (parser, pps->sps_id);
if (!sps) {
GST_WARNING ("couldn't find associated sequence parameter set with id: %d",
pps->sps_id);
return GST_H266_PARSER_BROKEN_LINK;
}
pps->sps = sps;
ctb_size_y = 1 << (sps->log2_ctu_size_minus5 + 5);
min_cb_size_y = 1 << (sps->log2_min_luma_coding_block_size_minus2 + 2);
sub_width_c = h266_sub_width_c[sps->chroma_format_idc];
sub_height_c = h266_sub_height_c[sps->chroma_format_idc];
READ_UINT8 (&nr, pps->mixed_nalu_types_in_pic_flag, 1);
READ_UE_ALLOWED (&nr, pps->pic_width_in_luma_samples, 1,
sps->pic_width_max_in_luma_samples);
READ_UE_ALLOWED (&nr, pps->pic_height_in_luma_samples, 1,
sps->pic_height_max_in_luma_samples);
if (pps->pic_width_in_luma_samples % MAX (min_cb_size_y, 8) ||
pps->pic_height_in_luma_samples % MAX (min_cb_size_y, 8)) {
GST_WARNING ("Invalid dimensions: %ux%u not divisible "
"by %u, MinCbSizeY = %u.\n", pps->pic_width_in_luma_samples,
pps->pic_height_in_luma_samples, MAX (min_cb_size_y, 8), min_cb_size_y);
goto error;
}
if (!sps->res_change_in_clvs_allowed_flag &&
(pps->pic_width_in_luma_samples !=
sps->pic_width_max_in_luma_samples ||
pps->pic_height_in_luma_samples !=
sps->pic_height_max_in_luma_samples)) {
GST_WARNING ("Resoltuion change is not allowed, "
"resolution sps(%ux%u) mismatched with pps(%ux%u).\n",
sps->pic_width_max_in_luma_samples,
sps->pic_height_max_in_luma_samples,
pps->pic_width_in_luma_samples, pps->pic_height_in_luma_samples);
goto error;
}
if (sps->ref_wraparound_enabled_flag) {
if ((ctb_size_y / min_cb_size_y + 1) >
(pps->pic_width_in_luma_samples / min_cb_size_y - 1)) {
GST_WARNING ("The value %d of (CtbSizeY / MinCbSizeY + 1) shall be "
"less than or equal to the value %d of "
"(pps_pic_width_in_luma_samples / MinCbSizeY - 1).",
ctb_size_y / min_cb_size_y + 1,
pps->pic_width_in_luma_samples / min_cb_size_y - 1);
goto error;
}
}
READ_UINT8 (&nr, pps->conformance_window_flag, 1);
if (pps->conformance_window_flag &&
pps->pic_width_in_luma_samples == sps->pic_width_max_in_luma_samples &&
pps->pic_height_in_luma_samples == sps->pic_height_max_in_luma_samples) {
GST_WARNING ("When pps_pic_width_in_luma_samples is equal to "
"sps_pic_width_max_in_luma_samples and pps_pic_height_in_luma_samples "
"is equal to sps_pic_height_max_in_luma_samples, the value of "
"pps_conformance_window_flag shall be equal to 0");
goto error;
}
if (pps->conformance_window_flag) {
guint width, height;
width = pps->pic_width_in_luma_samples / sub_width_c;
height = pps->pic_height_in_luma_samples / sub_height_c;
READ_UE_MAX (&nr, pps->conf_win_left_offset, width);
READ_UE_MAX (&nr, pps->conf_win_right_offset, width);
READ_UE_MAX (&nr, pps->conf_win_top_offset, height);
READ_UE_MAX (&nr, pps->conf_win_bottom_offset, height);
if (sub_width_c * (pps->conf_win_left_offset +
pps->conf_win_right_offset) >= pps->pic_width_in_luma_samples
|| sub_height_c * (pps->conf_win_top_offset +
pps->conf_win_bottom_offset) >= pps->pic_height_in_luma_samples) {
GST_WARNING ("Invalid pps conformance window: (%u, %u, %u, %u), "
"resolution is %ux%u, sub WxH is %ux%u.\n", pps->conf_win_left_offset,
pps->conf_win_right_offset, pps->conf_win_top_offset,
pps->conf_win_bottom_offset, pps->pic_width_in_luma_samples,
pps->pic_height_in_luma_samples, sub_width_c, sub_height_c);
goto error;
}
} else {
if (pps->pic_width_in_luma_samples ==
sps->pic_width_max_in_luma_samples &&
pps->pic_height_in_luma_samples ==
sps->pic_height_max_in_luma_samples) {
pps->conf_win_left_offset = sps->conf_win_left_offset;
pps->conf_win_right_offset = sps->conf_win_right_offset;
pps->conf_win_top_offset = sps->conf_win_top_offset;
pps->conf_win_bottom_offset = sps->conf_win_bottom_offset;
} else {
pps->conf_win_left_offset = 0;
pps->conf_win_right_offset = 0;
pps->conf_win_top_offset = 0;
pps->conf_win_bottom_offset = 0;
}
}
READ_UINT8 (&nr, pps->scaling_window_explicit_signalling_flag, 1);
if (pps->scaling_window_explicit_signalling_flag) {
if (!sps->ref_pic_resampling_enabled_flag) {
GST_WARNING ("When sps_ref_pic_resampling_enabled_flag is equal to 0, "
"the value of pps_scaling_window_explicit_signalling_flag "
"shall be equal to 0");
goto error;
}
READ_SE (&nr, pps->scaling_win_left_offset);
CHECK_ALLOWED (pps->scaling_win_left_offset * sub_width_c,
-pps->pic_width_in_luma_samples * 15, pps->pic_width_in_luma_samples);
READ_SE (&nr, pps->scaling_win_right_offset);
CHECK_ALLOWED (pps->scaling_win_right_offset * sub_width_c,
-pps->pic_width_in_luma_samples * 15, pps->pic_width_in_luma_samples);
READ_SE (&nr, pps->scaling_win_top_offset);
CHECK_ALLOWED (pps->scaling_win_top_offset * sub_height_c,
-pps->pic_height_in_luma_samples * 15, pps->pic_height_in_luma_samples);
READ_SE (&nr, pps->scaling_win_bottom_offset);
CHECK_ALLOWED (pps->scaling_win_bottom_offset * sub_height_c,
-pps->pic_height_in_luma_samples * 15, pps->pic_height_in_luma_samples);
CHECK_ALLOWED ((pps->scaling_win_left_offset +
pps->scaling_win_right_offset) * sub_width_c,
-pps->pic_width_in_luma_samples * 15, pps->pic_width_in_luma_samples);
CHECK_ALLOWED ((pps->scaling_win_top_offset +
pps->scaling_win_bottom_offset) * sub_height_c,
-pps->pic_height_in_luma_samples * 15, pps->pic_height_in_luma_samples);
} else {
pps->scaling_win_left_offset = pps->conf_win_left_offset;
pps->scaling_win_right_offset = pps->conf_win_right_offset;
pps->scaling_win_top_offset = pps->conf_win_top_offset;
pps->scaling_win_bottom_offset = pps->conf_win_bottom_offset;
}
READ_UINT8 (&nr, pps->output_flag_present_flag, 1);
READ_UINT8 (&nr, pps->no_pic_partition_flag, 1);
READ_UINT8 (&nr, pps->subpic_id_mapping_present_flag, 1);
if (pps->subpic_id_mapping_present_flag) {
if (!pps->no_pic_partition_flag) {
READ_UE (&nr, pps->num_subpics_minus1);
if (pps->num_subpics_minus1 != sps->num_subpics_minus1) {
GST_WARNING ("pps_num_subpics_minus1 shall be equal "
"to sps_num_subpics_minus1");
goto error;
}
} else {
pps->num_subpics_minus1 = 0;
}
READ_UE (&nr, pps->subpic_id_len_minus1);
if (pps->subpic_id_len_minus1 != sps->subpic_id_len_minus1) {
GST_WARNING ("pps_subpic_id_len_minus1 shall be equal "
"to sps_subpic_id_len_minus1");
goto error;
}
for (i = 0; i <= pps->num_subpics_minus1; i++)
READ_UINT16 (&nr, pps->subpic_id[i], pps->subpic_id_len_minus1 + 1);
}
pps->pic_width_in_ctbs_y =
(pps->pic_width_in_luma_samples + ctb_size_y - 1) / ctb_size_y;
pps->pic_height_in_ctbs_y =
(pps->pic_height_in_luma_samples + ctb_size_y - 1) / ctb_size_y;
pps->pic_size_in_ctbs_y =
pps->pic_width_in_ctbs_y * pps->pic_height_in_ctbs_y;
if (!pps->no_pic_partition_flag) {
if (!gst_h266_parser_parse_picture_partition (sps, pps, &nr))
goto error;
} else {
pps->single_slice_per_subpic_flag = 0;
pps->num_exp_tile_columns_minus1 = 0;
pps->tile_column_width_minus1[0] = pps->pic_width_in_ctbs_y - 1;
pps->num_exp_tile_rows_minus1 = 0;
pps->tile_row_height_minus1[0] = pps->pic_height_in_ctbs_y - 1;
pps->num_tile_columns = 1;
pps->num_tile_rows = 1;
pps->num_tiles_in_pic = 1;
pps->rect_slice_flag = 0;
pps->tile_col_bd_val[0] = 0;
for (i = 0; i < pps->num_tile_columns; i++) {
pps->tile_col_bd_val[i + 1] =
pps->tile_col_bd_val[i] + pps->tile_column_width_minus1[i] + 1;
}
pps->tile_row_bd_val[0] = 0;
for (i = 0; i < pps->num_tile_rows; i++) {
pps->tile_row_bd_val[i + 1] =
pps->tile_row_bd_val[i] + pps->tile_row_height_minus1[i] + 1;
}
}
READ_UINT8 (&nr, pps->cabac_init_present_flag, 1);
for (i = 0; i < 2; i++)
READ_UE_MAX (&nr, pps->num_ref_idx_default_active_minus1[i], 14);
READ_UINT8 (&nr, pps->rpl1_idx_present_flag, 1);
READ_UINT8 (&nr, pps->weighted_pred_flag, 1);
READ_UINT8 (&nr, pps->weighted_bipred_flag, 1);
READ_UINT8 (&nr, pps->ref_wraparound_enabled_flag, 1);
if (pps->ref_wraparound_enabled_flag)
READ_UE_MAX (&nr, pps->pic_width_minus_wraparound_offset,
(pps->pic_width_in_luma_samples / min_cb_size_y)
- (ctb_size_y / min_cb_size_y) - 2);
QpBdOffset = 6 * sps->bitdepth_minus8;
READ_SE_ALLOWED (&nr, pps->init_qp_minus26, -(26 + QpBdOffset), 37);
READ_UINT8 (&nr, pps->cu_qp_delta_enabled_flag, 1);
READ_UINT8 (&nr, pps->chroma_tool_offsets_present_flag, 1);
if (pps->chroma_tool_offsets_present_flag) {
READ_SE_ALLOWED (&nr, pps->cb_qp_offset, -12, 12);
READ_SE_ALLOWED (&nr, pps->cr_qp_offset, -12, 12);
READ_UINT8 (&nr, pps->joint_cbcr_qp_offset_present_flag, 1);
if (pps->joint_cbcr_qp_offset_present_flag) {
READ_SE_ALLOWED (&nr, pps->joint_cbcr_qp_offset_value, -12, 12);
} else {
pps->joint_cbcr_qp_offset_value = 0;
}
READ_UINT8 (&nr, pps->slice_chroma_qp_offsets_present_flag, 1);
READ_UINT8 (&nr, pps->cu_chroma_qp_offset_list_enabled_flag, 1);
if (pps->cu_chroma_qp_offset_list_enabled_flag) {
READ_UE_MAX (&nr, pps->chroma_qp_offset_list_len_minus1, 5);
for (i = 0; i <= pps->chroma_qp_offset_list_len_minus1; i++) {
READ_SE_ALLOWED (&nr, pps->cb_qp_offset_list[i], -12, 12);
READ_SE_ALLOWED (&nr, pps->cr_qp_offset_list[i], -12, 12);
if (pps->joint_cbcr_qp_offset_present_flag) {
READ_SE_ALLOWED (&nr, pps->joint_cbcr_qp_offset_list[i], -12, 12);
} else {
pps->joint_cbcr_qp_offset_list[i] = 0;
}
}
}
} else {
pps->cb_qp_offset = 0;
pps->cr_qp_offset = 0;
pps->joint_cbcr_qp_offset_present_flag = 0;
pps->joint_cbcr_qp_offset_value = 0;
pps->slice_chroma_qp_offsets_present_flag = 0;
pps->cu_chroma_qp_offset_list_enabled_flag = 0;
}
READ_UINT8 (&nr, pps->deblocking_filter_control_present_flag, 1);
if (pps->deblocking_filter_control_present_flag) {
READ_UINT8 (&nr, pps->deblocking_filter_override_enabled_flag, 1);
READ_UINT8 (&nr, pps->deblocking_filter_disabled_flag, 1);
if (!pps->no_pic_partition_flag &&
pps->deblocking_filter_override_enabled_flag) {
READ_UINT8 (&nr, pps->dbf_info_in_ph_flag, 1);
} else {
pps->dbf_info_in_ph_flag = 0;
}
if (!pps->deblocking_filter_disabled_flag) {
READ_SE_ALLOWED (&nr, pps->luma_beta_offset_div2, -12, 12);
READ_SE_ALLOWED (&nr, pps->luma_tc_offset_div2, -12, 12);
if (pps->chroma_tool_offsets_present_flag) {
READ_SE_ALLOWED (&nr, pps->cb_beta_offset_div2, -12, 12);
READ_SE_ALLOWED (&nr, pps->cb_tc_offset_div2, -12, 12);
READ_SE_ALLOWED (&nr, pps->cr_beta_offset_div2, -12, 12);
READ_SE_ALLOWED (&nr, pps->cr_tc_offset_div2, -12, 12);
} else {
pps->cb_beta_offset_div2 = 0;
pps->cb_tc_offset_div2 = 0;
pps->cr_beta_offset_div2 = pps->luma_beta_offset_div2;
pps->cr_tc_offset_div2 = pps->luma_tc_offset_div2;
}
} else {
pps->luma_beta_offset_div2 = 0;
pps->luma_tc_offset_div2 = 0;
pps->cb_beta_offset_div2 = 0;
pps->cb_tc_offset_div2 = 0;
pps->cr_beta_offset_div2 = pps->luma_beta_offset_div2;
pps->cr_tc_offset_div2 = pps->luma_tc_offset_div2;
}
} else {
pps->deblocking_filter_override_enabled_flag = 0;
pps->deblocking_filter_disabled_flag = 0;
pps->dbf_info_in_ph_flag = 0;
pps->luma_beta_offset_div2 = 0;
pps->luma_tc_offset_div2 = 0;
pps->cb_beta_offset_div2 = 0;
pps->cb_tc_offset_div2 = 0;
pps->cr_beta_offset_div2 = 0;
pps->cr_tc_offset_div2 = 0;
}
if (!pps->no_pic_partition_flag) {
READ_UINT8 (&nr, pps->rpl_info_in_ph_flag, 1);
READ_UINT8 (&nr, pps->sao_info_in_ph_flag, 1);
READ_UINT8 (&nr, pps->alf_info_in_ph_flag, 1);
if ((pps->weighted_pred_flag || pps->weighted_bipred_flag) &&
pps->rpl_info_in_ph_flag)
READ_UINT8 (&nr, pps->wp_info_in_ph_flag, 1);
READ_UINT8 (&nr, pps->qp_delta_info_in_ph_flag, 1);
}
READ_UINT8 (&nr, pps->picture_header_extension_present_flag, 1);
READ_UINT8 (&nr, pps->slice_header_extension_present_flag, 1);
READ_UINT8 (&nr, pps->extension_flag, 1);
if (pps->extension_flag)
GST_WARNING ("extension_flag not supported. Ignoring extension data.");
/* calculate width and height */
pps->width = pps->pic_width_in_luma_samples;
pps->height = pps->pic_height_in_luma_samples;
if (pps->conformance_window_flag) {
pps->crop_rect_width = pps->width -
(pps->conf_win_left_offset + pps->conf_win_right_offset) * sub_width_c;
pps->crop_rect_height = pps->height -
(pps->conf_win_top_offset + pps->conf_win_bottom_offset) * sub_height_c;
pps->crop_rect_x = pps->conf_win_left_offset * sub_width_c;
pps->crop_rect_y = pps->conf_win_top_offset * sub_height_c;
GST_LOG ("crop_rectangle x=%u y=%u width=%u, height=%u", pps->crop_rect_x,
pps->crop_rect_y, pps->crop_rect_width, pps->crop_rect_height);
}
pps->valid = TRUE;
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Picture parameter set\"");
pps->valid = FALSE;
return GST_H266_PARSER_ERROR;
}
/**
* gst_h266_parser_parse_aps:
* @parser: a #GstH266Parser
* @nalu: The APS #GstH266NalUnit to parse
* @aps: The #GstH266APS to fill.
*
* Parses @data, and fills the @aps structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_aps (GstH266Parser * parser,
GstH266NalUnit * nalu, GstH266APS * aps)
{
GstH266ParserResult res = gst_h266_parse_aps (parser, nalu, aps);
if (res == GST_H266_PARSER_OK) {
GST_LOG ("adding adaptation parameter set with id: %d to array",
aps->aps_id);
parser->aps[aps->params_type][aps->aps_id] = *aps;
parser->last_aps[aps->params_type] =
&parser->aps[aps->params_type][aps->aps_id];
}
return res;
}
/**
* gst_h266_parse_aps:
* @parser: a #GstH266Parser
* @nalu: The APS #GstH266NalUnit to parse
* @aps: The #GstH266APS to fill.
*
* Parses @data, and fills the @aps structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parse_aps (GstH266Parser * parser, GstH266NalUnit * nalu,
GstH266APS * aps)
{
NalReader nr;
guint8 params_type;
GST_LOG ("parsing APS");
nal_reader_init (&nr, nalu->data + nalu->offset + nalu->header_bytes,
nalu->size - nalu->header_bytes);
memset (aps, 0, sizeof (*aps));
READ_UINT8 (&nr, params_type, 3);
aps->params_type = params_type;
READ_UINT8 (&nr, aps->aps_id, 5);
CHECK_ALLOWED_MAX (aps->aps_id, GST_H266_MAX_APS_COUNT);
READ_UINT8 (&nr, aps->chroma_present_flag, 1);
switch (aps->params_type) {
case GST_H266_ALF_APS:
if (!gst_h266_parse_alf (&aps->alf, &nr, aps->chroma_present_flag))
goto error;
break;
case GST_H266_LMCS_APS:
if (!gst_h266_parse_lmcs (&aps->lmcs, &nr, aps->chroma_present_flag))
goto error;
break;
case GST_H266_SCALING_APS:
if (!gst_h266_parse_scaling_list (&aps->sl, &nr,
aps->chroma_present_flag))
goto error;
break;
default:
GST_WARNING ("unknown APS params_type %d", aps->params_type);
goto error;
break;
}
READ_UINT8 (&nr, aps->extension_flag, 1);
if (aps->extension_flag)
GST_WARNING ("extension_flag not supported. Ignoring extension data.");
aps->valid = TRUE;
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Adaptation parameter set\"");
aps->valid = FALSE;
return GST_H266_PARSER_ERROR;
}
/**
* gst_h266_parser_parse_aud:
* @parser: a #GstH266Parser
* @nalu: The AUD #GstH266NalUnit to parse
* @aud: The #GstH266AUD to fill.
*
* Parses @data, and fills the @aud structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_aud (GstH266Parser * parser,
GstH266NalUnit * nalu, GstH266AUD * aud)
{
NalReader nr;
GST_LOG ("parsing Access Unit Delimiter");
nal_reader_init (&nr, nalu->data + nalu->offset + nalu->header_bytes,
nalu->size - nalu->header_bytes);
memset (aud, 0, sizeof (*aud));
READ_UINT8 (&nr, aud->irap_or_gdr_flag, 1);
READ_UINT8 (&nr, aud->pic_type, 3);
CHECK_ALLOWED_MAX (aud->pic_type, 2);
/* Skip the byte alignment bits */
if (!nal_reader_skip (&nr, 1))
goto error;
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Access Unit Delimiter\"");
return GST_H266_PARSER_ERROR;
}
/**
* gst_h266_parser_parse_opi:
* @parser: a #GstH266Parser
* @nalu: The OPI #GstH266NalUnit to parse
* @opi: The #GstH266OPI to fill.
*
* Parses @data, and fills the @opi structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_opi (GstH266Parser * parser,
GstH266NalUnit * nalu, GstH266OPI * opi)
{
NalReader nr;
GST_LOG ("parsing Operating Point Information");
nal_reader_init (&nr, nalu->data + nalu->offset + nalu->header_bytes,
nalu->size - nalu->header_bytes);
memset (opi, 0, sizeof (*opi));
READ_UINT8 (&nr, opi->ols_info_present_flag, 1);
READ_UINT8 (&nr, opi->htid_info_present_flag, 1);
if (opi->ols_info_present_flag)
READ_UE (&nr, opi->ols_idx);
if (opi->htid_info_present_flag)
READ_UINT8 (&nr, opi->htid_plus1, 3);
READ_UINT8 (&nr, opi->extension_flag, 1);
if (opi->extension_flag)
GST_WARNING ("extension_flag not supported. Ignoring extension data.");
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Operating Point Information\"");
return GST_H266_PARSER_ERROR;
}
/**
* gst_h266_parser_parse_dci:
* @parser: a #GstH266Parser
* @nalu: The DCI #GstH266NalUnit to parse
* @dci: The #GstH266DCI to fill.
*
* Parses @data, and fills the @dci structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_dci (GstH266Parser * parser,
GstH266NalUnit * nalu, GstH266DCI * dci)
{
NalReader nr;
guint8 dci_reserved_zero_4bits;
guint i;
GST_LOG ("parsing Decoding Capability Information");
nal_reader_init (&nr, nalu->data + nalu->offset + nalu->header_bytes,
nalu->size - nalu->header_bytes);
memset (dci, 0, sizeof (*dci));
READ_UINT8 (&nr, dci_reserved_zero_4bits, 4);
READ_UINT8 (&nr, dci->num_ptls_minus1, 4);
CHECK_ALLOWED_MAX (dci->num_ptls_minus1, 15);
for (i = 0; i <= dci->num_ptls_minus1; i++) {
if (!gst_h266_parse_profile_tier_level (&dci->profile_tier_level[i],
&nr, 1, 0))
goto error;
}
READ_UINT8 (&nr, dci->extension_flag, 1);
if (dci->extension_flag)
GST_WARNING ("extension_flag not supported. Ignoring extension data.");
return GST_H266_PARSER_OK;
error:
GST_WARNING ("error parsing \"Decoding Capability Information\"");
return GST_H266_PARSER_ERROR;
}
static gboolean
gst_h266_parse_pred_weight_table (GstH266PredWeightTable * pwt, NalReader * nr,
const GstH266SPS * sps, const GstH266PPS * pps,
const GstH266RefPicLists * ref_lists, const guint8 num_ref_idx_active[2])
{
gint i, j;
GST_LOG ("parsing Pred Weight Table");
memset (pwt, 0, sizeof (*pwt));
READ_UE_MAX (nr, pwt->luma_log2_weight_denom, 7);
if (sps->chroma_format_idc != 0) {
gint luma_log2_weight_denom = pwt->luma_log2_weight_denom;
READ_SE_ALLOWED (nr, pwt->delta_chroma_log2_weight_denom,
-luma_log2_weight_denom, 7 - luma_log2_weight_denom);
} else {
pwt->delta_chroma_log2_weight_denom = 0;
}
if (pps->wp_info_in_ph_flag) {
READ_UE_MAX (nr, pwt->num_l0_weights, MIN (15,
ref_lists->rpl_ref_list[0].num_ref_entries));
} else {
pwt->num_l0_weights = num_ref_idx_active[0];
}
for (i = 0; i < pwt->num_l0_weights; i++)
READ_UINT8 (nr, pwt->luma_weight_l0_flag[i], 1);
if (sps->chroma_format_idc != 0) {
for (i = 0; i < pwt->num_l0_weights; i++)
READ_UINT8 (nr, pwt->chroma_weight_l0_flag[i], 1);
}
for (i = 0; i < pwt->num_l0_weights; i++) {
if (pwt->luma_weight_l0_flag[i]) {
READ_SE_ALLOWED (nr, pwt->delta_luma_weight_l0[i], -128, 127);
READ_SE_ALLOWED (nr, pwt->luma_offset_l0[i], -128, 127);
} else {
pwt->delta_luma_weight_l0[i] = 0;
pwt->luma_offset_l0[i] = 0;
}
if (pwt->chroma_weight_l0_flag[i]) {
for (j = 0; j < 2; j++) {
READ_SE_ALLOWED (nr, pwt->delta_chroma_weight_l0[i][j], -128, 127);
READ_SE_ALLOWED (nr, pwt->delta_chroma_offset_l0[i][j], -4 * 128,
4 * 127);
}
}
}
if (pps->weighted_bipred_flag && pps->wp_info_in_ph_flag &&
ref_lists->rpl_ref_list[1].num_ref_entries > 0) {
}
if (!pps->weighted_bipred_flag
|| ref_lists->rpl_ref_list[1].num_ref_entries == 0) {
pwt->num_l1_weights = 0;
} else if (pps->wp_info_in_ph_flag) {
READ_UE_MAX (nr, pwt->num_l1_weights, MIN (15,
ref_lists->rpl_ref_list[1].num_ref_entries));
} else {
pwt->num_l1_weights = num_ref_idx_active[1];
}
for (i = 0; i < pwt->num_l1_weights; i++)
READ_UINT8 (nr, pwt->luma_weight_l1_flag[i], 1);
if (sps->chroma_format_idc != 0) {
for (i = 0; i < pwt->num_l1_weights; i++)
READ_UINT8 (nr, pwt->chroma_weight_l1_flag[i], 1);
}
for (i = 0; i < pwt->num_l1_weights; i++) {
if (pwt->luma_weight_l1_flag[i]) {
READ_SE_ALLOWED (nr, pwt->delta_luma_weight_l1[i], -128, 127);
READ_SE_ALLOWED (nr, pwt->luma_offset_l1[i], -128, 127);
} else {
pwt->delta_luma_weight_l1[i] = 0;
pwt->luma_offset_l1[i] = 0;
}
if (pwt->chroma_weight_l1_flag[i]) {
for (j = 0; j < 2; j++) {
READ_SE_ALLOWED (nr, pwt->delta_chroma_weight_l1[i][j], -128, 127);
READ_SE_ALLOWED (nr, pwt->delta_chroma_offset_l1[i][j], -4 * 128,
4 * 127);
}
}
}
return TRUE;
error:
GST_WARNING ("error parsing \"Pred Weight Table\"");
return FALSE;
}
static void
gst_h266_parse_inherit_deblock_param_from_pps (GstH266PicHdr * ph)
{
ph->luma_beta_offset_div2 = ph->pps->luma_beta_offset_div2;
ph->luma_tc_offset_div2 = ph->pps->luma_tc_offset_div2;
ph->cb_beta_offset_div2 = ph->pps->cb_beta_offset_div2;
ph->cb_tc_offset_div2 = ph->pps->cb_tc_offset_div2;
ph->cr_beta_offset_div2 = ph->pps->cr_beta_offset_div2;
ph->cr_tc_offset_div2 = ph->pps->cr_tc_offset_div2;
}
static GstH266ParserResult
gst_h266_parse_picture_hdr_structure (GstH266PicHdr * ph,
NalReader * nr, GstH266Parser * parser)
{
const GstH266SPS *sps;
guint ctb_log2_size_y, min_cb_log2_size_y, i;
GstH266ParserResult ret = GST_H266_PARSER_OK;
GST_LOG ("parsing Picture Header Structure");
READ_UINT8 (nr, ph->gdr_or_irap_pic_flag, 1);
READ_UINT8 (nr, ph->non_ref_pic_flag, 1);
if (ph->gdr_or_irap_pic_flag) {
READ_UINT8 (nr, ph->gdr_pic_flag, 1);
} else {
ph->gdr_pic_flag = 0;
}
READ_UINT8 (nr, ph->inter_slice_allowed_flag, 1);
if (ph->inter_slice_allowed_flag) {
READ_UINT8 (nr, ph->intra_slice_allowed_flag, 1);
} else {
ph->intra_slice_allowed_flag = 1;
}
READ_UE_MAX (nr, ph->pps_id, GST_H266_MAX_PPS_COUNT - 1);
ph->pps = gst_h266_parser_get_pps (parser, ph->pps_id);
if (!ph->pps) {
GST_WARNING ("PPS id %d not available.", ph->pps_id);
ret = GST_H266_PARSER_BROKEN_LINK;
goto error_with_ret;
}
sps = gst_h266_parser_get_sps (parser, ph->pps->sps_id);
if (!sps) {
GST_WARNING ("SPS id %d not available.", ph->pps->sps_id);
ret = GST_H266_PARSER_BROKEN_LINK;
goto error_with_ret;
}
READ_UINT16 (nr, ph->pic_order_cnt_lsb,
sps->log2_max_pic_order_cnt_lsb_minus4 + 4);
if (ph->gdr_pic_flag)
READ_UE_MAX (nr, ph->recovery_poc_cnt,
1 << (sps->log2_max_pic_order_cnt_lsb_minus4 + 4));
if ((!ph->gdr_or_irap_pic_flag || ph->gdr_pic_flag) &&
(!ph->gdr_pic_flag || ph->recovery_poc_cnt != 0)) {
const GstH266GeneralConstraintsInfo *general_constraints_info =
&sps->profile_tier_level.general_constraints_info;
if (sps->profile_tier_level.profile_idc & GST_H266_PROFILE_INTRA) {
GST_WARNING ("Invalid non-irap pictures or gdr "
"pictures with ph_recovery_poc_cnt!=0 for Intra profile");
goto error;
}
if (general_constraints_info->all_rap_pictures_constraint_flag) {
GST_WARNING ("gci_all_rap_pictures_flag equal to 1 specifies that "
"all pictures in OlsInScope are IRAP pictures or GDR pictures "
"with ph_recovery_poc_cnt equal to 0");
goto error;
}
}
for (i = 0; i < sps->num_extra_ph_bytes * 8; i++) {
/* extra bits are ignored now */
if (sps->extra_ph_bit_present_flag[i])
READ_UINT8 (nr, ph->extra_bit[i], 1);
}
if (sps->poc_msb_cycle_flag) {
READ_UINT8 (nr, ph->poc_msb_cycle_present_flag, 1);
if (ph->poc_msb_cycle_present_flag)
READ_UINT8 (nr, ph->poc_msb_cycle_val, sps->poc_msb_cycle_len_minus1 + 1);
}
if (sps->alf_enabled_flag && ph->pps->alf_info_in_ph_flag) {
READ_UINT8 (nr, ph->alf_enabled_flag, 1);
if (ph->alf_enabled_flag) {
READ_UINT8 (nr, ph->num_alf_aps_ids_luma, 3);
for (i = 0; i < ph->num_alf_aps_ids_luma; i++)
READ_UINT8 (nr, ph->alf_aps_id_luma[i], 3);
if (sps->chroma_format_idc != 0) {
READ_UINT8 (nr, ph->alf_cb_enabled_flag, 1);
READ_UINT8 (nr, ph->alf_cr_enabled_flag, 1);
} else {
ph->alf_cb_enabled_flag = 0;
ph->alf_cr_enabled_flag = 0;
}
if (ph->alf_cb_enabled_flag || ph->alf_cr_enabled_flag)
READ_UINT8 (nr, ph->alf_aps_id_chroma, 3);
if (sps->ccalf_enabled_flag) {
READ_UINT8 (nr, ph->alf_cc_cb_enabled_flag, 1);
if (ph->alf_cc_cb_enabled_flag)
READ_UINT8 (nr, ph->alf_cc_cb_aps_id, 3);
READ_UINT8 (nr, ph->alf_cc_cr_enabled_flag, 1);
if (ph->alf_cc_cr_enabled_flag)
READ_UINT8 (nr, ph->alf_cc_cr_aps_id, 3);
}
}
} else {
ph->alf_enabled_flag = 0;
}
if (sps->lmcs_enabled_flag) {
READ_UINT8 (nr, ph->lmcs_enabled_flag, 1);
if (ph->lmcs_enabled_flag) {
READ_UINT8 (nr, ph->lmcs_aps_id, 2);
if (sps->chroma_format_idc != 0) {
READ_UINT8 (nr, ph->chroma_residual_scale_flag, 1);
} else {
ph->chroma_residual_scale_flag = 0;
}
}
} else {
ph->lmcs_enabled_flag = 0;
ph->chroma_residual_scale_flag = 0;
}
if (sps->explicit_scaling_list_enabled_flag) {
READ_UINT8 (nr, ph->explicit_scaling_list_enabled_flag, 1);
if (ph->explicit_scaling_list_enabled_flag)
READ_UINT8 (nr, ph->scaling_list_aps_id, 3);
} else {
ph->explicit_scaling_list_enabled_flag = 0;
}
if (sps->virtual_boundaries_enabled_flag &&
!sps->virtual_boundaries_present_flag) {
READ_UINT8 (nr, ph->virtual_boundaries_present_flag, 1);
if (ph->virtual_boundaries_present_flag) {
READ_UE_MAX (nr, ph->num_ver_virtual_boundaries,
((ph->pps->pic_width_in_luma_samples <= 8) ? 0 : 3));
for (i = 0; i < ph->num_ver_virtual_boundaries; i++)
READ_UE_MAX (nr, ph->virtual_boundary_pos_x_minus1[i],
(ph->pps->pic_width_in_luma_samples + 7) / 8 - 2);
READ_UE_MAX (nr, ph->num_hor_virtual_boundaries,
((ph->pps->pic_height_in_luma_samples <= 8) ? 0 : 3));
for (i = 0; i < ph->num_hor_virtual_boundaries; i++) {
READ_UE_MAX (nr, ph->virtual_boundary_pos_y_minus1[i],
(ph->pps->pic_height_in_luma_samples + 7) / 8 - 2);
}
} else {
ph->num_ver_virtual_boundaries = 0;
ph->num_hor_virtual_boundaries = 0;
}
}
if (ph->pps->output_flag_present_flag && !ph->non_ref_pic_flag) {
READ_UINT8 (nr, ph->pic_output_flag, 1);
} else {
ph->pic_output_flag = 1;
}
if (ph->pps->rpl_info_in_ph_flag) {
if (!gst_h266_ref_pic_lists (&ph->ref_pic_lists, nr, sps, ph->pps))
goto error;
}
if (sps->partition_constraints_override_enabled_flag) {
READ_UINT8 (nr, ph->partition_constraints_override_flag, 1);
} else {
ph->partition_constraints_override_flag = 0;
}
ctb_log2_size_y = sps->log2_ctu_size_minus5 + 5;
min_cb_log2_size_y = sps->log2_min_luma_coding_block_size_minus2 + 2;
ph->log2_diff_min_qt_min_cb_intra_slice_luma =
sps->log2_diff_min_qt_min_cb_intra_slice_luma;
ph->max_mtt_hierarchy_depth_intra_slice_luma =
sps->max_mtt_hierarchy_depth_intra_slice_luma;
ph->log2_diff_max_bt_min_qt_intra_slice_luma =
sps->log2_diff_max_bt_min_qt_intra_slice_luma;
ph->log2_diff_max_tt_min_qt_intra_slice_luma =
sps->log2_diff_max_tt_min_qt_intra_slice_luma;
ph->log2_diff_min_qt_min_cb_intra_slice_chroma =
sps->log2_diff_min_qt_min_cb_intra_slice_chroma;
ph->max_mtt_hierarchy_depth_intra_slice_chroma =
sps->max_mtt_hierarchy_depth_intra_slice_chroma;
ph->log2_diff_max_bt_min_qt_intra_slice_chroma =
sps->log2_diff_max_bt_min_qt_intra_slice_chroma;
ph->log2_diff_max_tt_min_qt_intra_slice_chroma =
sps->log2_diff_max_tt_min_qt_intra_slice_chroma;
ph->log2_diff_min_qt_min_cb_inter_slice =
sps->log2_diff_min_qt_min_cb_inter_slice;
ph->max_mtt_hierarchy_depth_inter_slice =
sps->max_mtt_hierarchy_depth_inter_slice;
ph->log2_diff_max_bt_min_qt_inter_slice =
sps->log2_diff_max_bt_min_qt_inter_slice;
ph->log2_diff_max_tt_min_qt_inter_slice =
sps->log2_diff_max_tt_min_qt_inter_slice;
ph->collocated_from_l0_flag = 1;
if (ph->intra_slice_allowed_flag) {
guint min_qt_log2_size_intra_y;
if (ph->partition_constraints_override_flag) {
READ_UE_MAX (nr, ph->log2_diff_min_qt_min_cb_intra_slice_luma,
MIN (6, ctb_log2_size_y) - min_cb_log2_size_y);
min_qt_log2_size_intra_y =
ph->log2_diff_min_qt_min_cb_intra_slice_luma + min_cb_log2_size_y;
READ_UE_MAX (nr, ph->max_mtt_hierarchy_depth_intra_slice_luma,
2 * (ctb_log2_size_y - min_cb_log2_size_y));
if (ph->max_mtt_hierarchy_depth_intra_slice_luma != 0) {
READ_UE_MAX (nr, ph->log2_diff_max_bt_min_qt_intra_slice_luma,
ctb_log2_size_y - min_qt_log2_size_intra_y);
READ_UE_MAX (nr, ph->log2_diff_max_tt_min_qt_intra_slice_luma,
MIN (6, ctb_log2_size_y) - min_qt_log2_size_intra_y);
} else {
ph->log2_diff_max_bt_min_qt_intra_slice_luma =
sps->log2_diff_max_bt_min_qt_intra_slice_luma;
ph->log2_diff_max_tt_min_qt_intra_slice_luma =
sps->log2_diff_max_tt_min_qt_intra_slice_luma;
}
if (sps->qtbtt_dual_tree_intra_flag) {
READ_UE_MAX (nr, ph->log2_diff_min_qt_min_cb_intra_slice_chroma,
MIN (6, ctb_log2_size_y) - min_cb_log2_size_y);
READ_UE_MAX (nr, ph->max_mtt_hierarchy_depth_intra_slice_chroma,
2 * (ctb_log2_size_y - min_cb_log2_size_y));
if (sps->max_mtt_hierarchy_depth_intra_slice_chroma != 0) {
gint32 min_qt_log2_size_intra_c =
sps->log2_diff_min_qt_min_cb_intra_slice_chroma +
min_cb_log2_size_y;
READ_UE_MAX (nr, ph->log2_diff_max_bt_min_qt_intra_slice_chroma,
MIN (6, ctb_log2_size_y) - min_qt_log2_size_intra_c);
READ_UE_MAX (nr, ph->log2_diff_max_tt_min_qt_intra_slice_chroma,
MIN (6, ctb_log2_size_y) - min_qt_log2_size_intra_c);
} else {
ph->log2_diff_max_bt_min_qt_intra_slice_chroma =
sps->log2_diff_max_bt_min_qt_intra_slice_chroma;
ph->log2_diff_max_tt_min_qt_intra_slice_chroma =
sps->log2_diff_max_tt_min_qt_intra_slice_chroma;
}
}
}
min_qt_log2_size_intra_y =
ph->log2_diff_min_qt_min_cb_intra_slice_luma + ctb_log2_size_y;
if (ph->pps->cu_qp_delta_enabled_flag) {
READ_UE_MAX (nr, ph->cu_qp_delta_subdiv_intra_slice,
2 * (ctb_log2_size_y - min_qt_log2_size_intra_y +
ph->max_mtt_hierarchy_depth_intra_slice_luma));
} else {
ph->cu_qp_delta_subdiv_intra_slice = 0;
}
if (ph->pps->cu_chroma_qp_offset_list_enabled_flag) {
READ_UE_MAX (nr, ph->cu_chroma_qp_offset_subdiv_intra_slice,
2 * (ctb_log2_size_y - min_qt_log2_size_intra_y +
ph->max_mtt_hierarchy_depth_intra_slice_luma));
} else {
ph->cu_chroma_qp_offset_subdiv_intra_slice = 0;
}
}
if (ph->inter_slice_allowed_flag) {
guint min_qt_log2_size_inter_y;
if (ph->partition_constraints_override_flag) {
READ_UE_MAX (nr, ph->log2_diff_min_qt_min_cb_inter_slice,
MIN (6, ctb_log2_size_y) - min_cb_log2_size_y);
min_qt_log2_size_inter_y =
ph->log2_diff_min_qt_min_cb_inter_slice + min_cb_log2_size_y;
READ_UE_MAX (nr, ph->max_mtt_hierarchy_depth_inter_slice,
2 * (ctb_log2_size_y - min_cb_log2_size_y));
if (ph->max_mtt_hierarchy_depth_inter_slice != 0) {
READ_UE_MAX (nr, ph->log2_diff_max_bt_min_qt_inter_slice,
ctb_log2_size_y - min_qt_log2_size_inter_y);
READ_UE_MAX (nr, ph->log2_diff_max_tt_min_qt_inter_slice,
MIN (6, ctb_log2_size_y) - min_qt_log2_size_inter_y);
}
}
min_qt_log2_size_inter_y =
ph->log2_diff_min_qt_min_cb_inter_slice + min_cb_log2_size_y;
if (ph->pps->cu_qp_delta_enabled_flag) {
READ_UE_MAX (nr, ph->cu_qp_delta_subdiv_inter_slice,
2 * (ctb_log2_size_y - min_qt_log2_size_inter_y +
ph->max_mtt_hierarchy_depth_inter_slice));
} else {
ph->cu_qp_delta_subdiv_inter_slice = 0;
}
if (ph->pps->cu_chroma_qp_offset_list_enabled_flag) {
READ_UE_MAX (nr, ph->cu_chroma_qp_offset_subdiv_inter_slice,
2 * (ctb_log2_size_y - min_qt_log2_size_inter_y +
ph->max_mtt_hierarchy_depth_inter_slice));
} else {
ph->cu_chroma_qp_offset_subdiv_inter_slice = 0;
}
if (sps->temporal_mvp_enabled_flag) {
READ_UINT8 (nr, ph->temporal_mvp_enabled_flag, 1);
if (ph->temporal_mvp_enabled_flag && ph->pps->rpl_info_in_ph_flag) {
if (ph->ref_pic_lists.rpl_ref_list[1].num_ref_entries > 0) {
READ_UINT8 (nr, ph->collocated_from_l0_flag, 1);
} else {
ph->collocated_from_l0_flag = 1;
}
if ((ph->collocated_from_l0_flag &&
ph->ref_pic_lists.rpl_ref_list[0].num_ref_entries > 1) ||
(!ph->collocated_from_l0_flag &&
ph->ref_pic_lists.rpl_ref_list[1].num_ref_entries > 1)) {
guint idx = ph->collocated_from_l0_flag ? 0 : 1;
READ_UE_MAX (nr, ph->collocated_ref_idx,
ph->ref_pic_lists.rpl_ref_list[idx].num_ref_entries - 1);
} else {
ph->collocated_ref_idx = 0;
}
}
}
if (sps->mmvd_fullpel_only_enabled_flag) {
READ_UINT8 (nr, ph->mmvd_fullpel_only_flag, 1);
} else {
ph->mmvd_fullpel_only_flag = 0;
}
if (!ph->pps->rpl_info_in_ph_flag ||
ph->ref_pic_lists.rpl_ref_list[1].num_ref_entries > 0) {
READ_UINT8 (nr, ph->mvd_l1_zero_flag, 1);
if (sps->bdof_control_present_in_ph_flag) {
READ_UINT8 (nr, ph->bdof_disabled_flag, 1);
} else {
ph->bdof_disabled_flag = !sps->bdof_enabled_flag;
}
if (sps->dmvr_control_present_in_ph_flag) {
READ_UINT8 (nr, ph->dmvr_disabled_flag, 1);
} else {
ph->dmvr_disabled_flag = !sps->dmvr_enabled_flag;
}
} else {
ph->mvd_l1_zero_flag = 1;
}
if (sps->prof_control_present_in_ph_flag) {
READ_UINT8 (nr, ph->prof_disabled_flag, 1);
} else {
ph->prof_disabled_flag = !sps->affine_prof_enabled_flag;
}
if ((ph->pps->weighted_pred_flag ||
ph->pps->weighted_bipred_flag) && ph->pps->wp_info_in_ph_flag) {
guint8 num_ref_idx_active[2];
num_ref_idx_active[0] = ph->ref_pic_lists.rpl_ref_list[0].num_ref_entries;
num_ref_idx_active[1] = ph->ref_pic_lists.rpl_ref_list[1].num_ref_entries;
if (!gst_h266_parse_pred_weight_table (&ph->pred_weight_table,
nr, sps, ph->pps, &ph->ref_pic_lists, num_ref_idx_active))
goto error;
}
}
if (ph->pps->qp_delta_info_in_ph_flag) {
/* SliceQpy = 26 + pps_init_qp_minus26 + ph_qp_delta,
The value of SliceQp Y shall be in the range of
-QpBdOffset to +63, inclusive */
gint qp_bd_offset = 6 * sps->bitdepth_minus8;
READ_SE_ALLOWED (nr, ph->qp_delta,
-qp_bd_offset - (26 + ph->pps->init_qp_minus26),
63 - (26 + ph->pps->init_qp_minus26));
}
if (sps->joint_cbcr_enabled_flag) {
READ_UINT8 (nr, ph->joint_cbcr_sign_flag, 1);
} else {
ph->joint_cbcr_sign_flag = 0;
}
if (sps->sao_enabled_flag && ph->pps->sao_info_in_ph_flag) {
READ_UINT8 (nr, ph->sao_luma_enabled_flag, 1);
if (sps->chroma_format_idc != 0) {
READ_UINT8 (nr, ph->sao_chroma_enabled_flag, 1);
} else
ph->sao_chroma_enabled_flag = 0;
} else {
ph->sao_luma_enabled_flag = 0;
ph->sao_chroma_enabled_flag = 0;
}
if (ph->pps->dbf_info_in_ph_flag) {
READ_UINT8 (nr, ph->deblocking_params_present_flag, 1);
if (ph->deblocking_params_present_flag) {
if (!ph->pps->deblocking_filter_disabled_flag) {
READ_UINT8 (nr, ph->deblocking_filter_disabled_flag, 1);
} else {
ph->deblocking_filter_disabled_flag = 0;
}
if (!ph->deblocking_filter_disabled_flag) {
READ_SE_ALLOWED (nr, ph->luma_beta_offset_div2, -12, 12);
READ_SE_ALLOWED (nr, ph->luma_tc_offset_div2, -12, 12);
if (ph->pps->chroma_tool_offsets_present_flag) {
READ_SE_ALLOWED (nr, ph->cb_beta_offset_div2, -12, 12);
READ_SE_ALLOWED (nr, ph->cb_tc_offset_div2, -12, 12);
READ_SE_ALLOWED (nr, ph->cr_beta_offset_div2, -12, 12);
READ_SE_ALLOWED (nr, ph->cr_tc_offset_div2, -12, 12);
} else {
ph->cb_beta_offset_div2 = ph->luma_beta_offset_div2;
ph->cb_tc_offset_div2 = ph->luma_tc_offset_div2;
ph->cr_beta_offset_div2 = ph->luma_beta_offset_div2;
ph->cr_tc_offset_div2 = ph->luma_tc_offset_div2;
}
} else {
if (ph->pps->chroma_tool_offsets_present_flag) {
gst_h266_parse_inherit_deblock_param_from_pps (ph);
} else {
ph->luma_beta_offset_div2 = ph->pps->luma_beta_offset_div2;
ph->luma_tc_offset_div2 = ph->pps->luma_tc_offset_div2;
ph->cb_beta_offset_div2 = ph->luma_beta_offset_div2;
ph->cb_tc_offset_div2 = ph->luma_tc_offset_div2;
ph->cr_beta_offset_div2 = ph->luma_beta_offset_div2;
ph->cr_tc_offset_div2 = ph->luma_tc_offset_div2;
}
}
} else {
ph->deblocking_filter_disabled_flag =
ph->pps->deblocking_filter_disabled_flag;
gst_h266_parse_inherit_deblock_param_from_pps (ph);
}
} else {
ph->deblocking_filter_disabled_flag =
ph->pps->deblocking_filter_disabled_flag;
gst_h266_parse_inherit_deblock_param_from_pps (ph);
}
if (ph->pps->picture_header_extension_present_flag) {
READ_UE_MAX (nr, ph->extension_length, 256);
for (i = 0; i < ph->extension_length; i++)
READ_UINT8 (nr, ph->extension_data_byte[i], 8);
}
return ret;
error:
ret = GST_H266_PARSER_ERROR;
error_with_ret:
GST_WARNING ("error parsing \"Picture Header\"");
return FALSE;
}
/**
* gst_h266_parser_parse_picture_hdr:
* @parser: a #GstH266Parser
* @nalu: The picture header #GstH266NalUnit to parse
* @ph: The #GstH266PicHdr to fill.
*
* Parses @data, and fills the @ph structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_picture_hdr (GstH266Parser * parser,
GstH266NalUnit * nalu, GstH266PicHdr * ph)
{
NalReader nr;
GstH266ParserResult ret;
GST_LOG ("parsing Picture Header");
nal_reader_init (&nr, nalu->data + nalu->offset + nalu->header_bytes,
nalu->size - nalu->header_bytes);
memset (ph, 0, sizeof (*ph));
ret = gst_h266_parse_picture_hdr_structure (ph, &nr, parser);
if (ret != GST_H266_PARSER_OK)
goto error;
ph->valid = TRUE;
parser->ph = *ph;
return ret;
error:
GST_WARNING ("error parsing \"Picture Header\"");
return ret;
}
static gboolean
gst_h266_parser_parse_slice_address (GstH266Parser * parser, NalReader * nr,
const GstH266PPS * pps, GstH266SliceHdr * sh, guint16 curr_subpic_idx,
const guint16 ** ctb_addr_in_curr_slice, guint16 * num_ctus_in_curr_slice)
{
GST_LOG ("parsing Slice Address");
if (!pps->no_pic_partition_flag && pps->rect_slice_flag) {
guint16 pic_level_slice_idx = sh->slice_address;
gint j;
for (j = 0; j < curr_subpic_idx; j++)
pic_level_slice_idx += pps->num_slices_in_subpic[j];
*ctb_addr_in_curr_slice = parser->ctb_addr_in_slice +
parser->slice_start_offset[pic_level_slice_idx];
*num_ctus_in_curr_slice = parser->num_ctus_in_slice[pic_level_slice_idx];
} else {
guint16 tile_idx;
guint16 tile_x = sh->slice_address % pps->num_tile_columns;
gint16 tile_y = sh->slice_address / pps->num_tile_columns;
guint16 slice_start_ctb =
pps->tile_row_bd_val[tile_y] * pps->pic_width_in_ctbs_y +
pps->tile_col_bd_val[tile_x] * (pps->tile_row_height_minus1[tile_y] +
1);
if (pps->num_tiles_in_pic - sh->slice_address > 1) {
READ_UE_MAX (nr, sh->num_tiles_in_slice_minus1,
pps->num_tiles_in_pic - 1);
} else {
sh->num_tiles_in_slice_minus1 = 0;
}
*ctb_addr_in_curr_slice = parser->ctb_addr_in_slice + slice_start_ctb;
*num_ctus_in_curr_slice = 0;
for (tile_idx = sh->slice_address;
tile_idx <= sh->slice_address + sh->num_tiles_in_slice_minus1;
tile_idx++) {
tile_x = tile_idx % pps->num_tile_columns;
tile_y = tile_idx / pps->num_tile_columns;
*num_ctus_in_curr_slice += (pps->tile_row_height_minus1[tile_y] + 1) *
(pps->tile_column_width_minus1[tile_x] + 1);
}
}
return TRUE;
error:
GST_WARNING ("error parsing Slice Address");
return FALSE;
}
/**
* gst_h266_parser_parse_slice_hdr:
* @parser: a #GstH266Parser
* @nalu: The slice #GstH266NalUnit to parse
* @sh: The #GstH266SliceHdr to fill.
*
* Parses @data, and fills the @sh structure.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_slice_hdr (GstH266Parser * parser,
GstH266NalUnit * nalu, GstH266SliceHdr * sh)
{
NalReader nr;
const GstH266SPS *sps;
const GstH266PPS *pps;
const GstH266PicHdr *ph;
const GstH266RefPicLists *ref_pic_lists;
const GstH266GeneralConstraintsInfo *constraints_info;
guint8 nal_unit_type;
guint16 curr_subpic_idx, num_slices_in_subpic;
const guint16 *ctb_addr_in_curr_slice; /* CtbAddrInCurrSlice */
guint16 num_ctus_in_curr_slice; /* NumCtusInCurrSlice */
GstH266ParserResult ret = GST_H266_PARSER_OK;
gint i;
GST_LOG ("parsing Slice Header");
nal_reader_init (&nr, nalu->data + nalu->offset + nalu->header_bytes,
nalu->size - nalu->header_bytes);
memset (sh, 0, sizeof (*sh));
READ_UINT8 (&nr, sh->picture_header_in_slice_header_flag, 1);
if (sh->picture_header_in_slice_header_flag) {
ret = gst_h266_parse_picture_hdr_structure (&sh->picture_header,
&nr, parser);
if (ret != GST_H266_PARSER_OK)
goto error_with_ret;
} else {
if (!parser->ph.valid) {
GST_WARNING ("Picture header not available.\n");
goto error;
}
sh->picture_header = parser->ph;
}
ph = &sh->picture_header;
pps = gst_h266_parser_get_pps (parser, ph->pps_id);
if (!pps) {
GST_WARNING ("PPS id %d not available.", ph->pps_id);
ret = GST_H266_PARSER_BROKEN_LINK;
goto error_with_ret;
}
sps = gst_h266_parser_get_sps (parser, pps->sps_id);
if (!sps) {
GST_WARNING ("SPS id %d not available.", pps->sps_id);
ret = GST_H266_PARSER_BROKEN_LINK;
goto error_with_ret;
}
if (!gst_h266_parser_set_active_sps_pps (parser, sps, pps)) {
GST_WARNING ("PPS id %d not active.", ph->pps_id);
ret = GST_H266_PARSER_ERROR;
goto error_with_ret;
}
constraints_info = &sps->profile_tier_level.general_constraints_info;
if (constraints_info->pic_header_in_slice_header_constraint_flag &&
!sh->picture_header_in_slice_header_flag) {
GST_WARNING ("PH shall be present in SH, when "
"pic_header_in_slice_header_constraint_flag is equal to 1");
goto error;
}
if (sh->picture_header_in_slice_header_flag) {
if (pps->rpl_info_in_ph_flag) {
GST_WARNING ("When sh_picture_header_in_slice_header_flag is equal "
"to 1, rpl_info_in_ph_flag shall be equal to 0");
goto error;
}
if (pps->dbf_info_in_ph_flag) {
GST_WARNING ("When sh_picture_header_in_slice_header_flag is equal "
"to 1, dbf_info_in_ph_flag shall be equal to 0");
goto error;
}
if (pps->sao_info_in_ph_flag) {
GST_WARNING ("When sh_picture_header_in_slice_header_flag is equal "
"to 1, sao_info_in_ph_flag shall be equal to 0");
goto error;
}
if (pps->alf_info_in_ph_flag) {
GST_WARNING ("When sh_picture_header_in_slice_header_flag is equal "
"to 1, alf_info_in_ph_flag shall be equal to 0");
goto error;
}
if (pps->wp_info_in_ph_flag) {
GST_WARNING ("When sh_picture_header_in_slice_header_flag is equal "
"to 1, wp_info_in_ph_flag shall be equal to 0");
goto error;
}
if (pps->qp_delta_info_in_ph_flag) {
GST_WARNING ("When sh_picture_header_in_slice_header_flag is equal "
"to 1, qp_delta_info_in_ph_flag shall be equal to 0");
goto error;
}
if (sps->subpic_info_present_flag) {
GST_WARNING ("When sps_subpic_info_present_flag is equal to 1, "
"the value of sh_picture_header_in_slice_header_flag shall be "
"equal to 0");
goto error;
}
}
curr_subpic_idx = 0;
if (sps->subpic_info_present_flag) {
READ_UINT16 (&nr, sh->subpic_id, sps->subpic_id_len_minus1 + 1);
if (sps->subpic_id_mapping_explicitly_signalled_flag) {
for (i = 0; i <= sps->num_subpics_minus1; i++) {
guint16 subpic_id_val = pps->subpic_id_mapping_present_flag ?
pps->subpic_id[i] : sps->subpic_id[i];
if (subpic_id_val == sh->subpic_id) {
curr_subpic_idx = i;
break;
}
}
} else {
curr_subpic_idx = sh->subpic_id;
if (curr_subpic_idx > sps->num_subpics_minus1) {
GST_WARNING ("sh_subpic_id(%d) should in range [0, %d]",
curr_subpic_idx, sps->num_subpics_minus1);
goto error;
}
}
}
num_slices_in_subpic = pps->num_slices_in_subpic[curr_subpic_idx];
if ((pps->rect_slice_flag && num_slices_in_subpic > 1) ||
(!pps->rect_slice_flag && pps->num_tiles_in_pic > 1)) {
guint bits, max;
if (!pps->rect_slice_flag) {
bits = gst_util_ceil_log2 (pps->num_tiles_in_pic);
max = pps->num_tiles_in_pic - 1;
} else {
bits = gst_util_ceil_log2 (num_slices_in_subpic);
max = num_slices_in_subpic - 1;
}
READ_UINT16 (&nr, sh->slice_address, bits);
CHECK_ALLOWED_MAX (sh->slice_address, max);
} else {
sh->slice_address = 0;
}
for (i = 0; i < sps->num_extra_sh_bytes * 8; i++) {
if (sps->extra_sh_bit_present_flag[i])
READ_UINT8 (&nr, sh->extra_bit[i], 1);
}
if (!gst_h266_parser_parse_slice_address (parser, &nr, pps, sh,
curr_subpic_idx, &ctb_addr_in_curr_slice, &num_ctus_in_curr_slice))
goto error;
if (ph->inter_slice_allowed_flag) {
READ_UE_MAX (&nr, sh->slice_type, 2);
} else {
sh->slice_type = GST_H266_I_SLICE;
}
if (!ph->intra_slice_allowed_flag && sh->slice_type == GST_H266_I_SLICE) {
GST_WARNING ("when ph_intra_slice_allowed_flag = 0, "
"no I_Slice is allowed");
goto error;
}
nal_unit_type = nalu->type;
if (nal_unit_type == GST_H266_NAL_SLICE_IDR_W_RADL ||
nal_unit_type == GST_H266_NAL_SLICE_IDR_N_LP ||
nal_unit_type == GST_H266_NAL_SLICE_CRA ||
nal_unit_type == GST_H266_NAL_SLICE_GDR)
READ_UINT8 (&nr, sh->no_output_of_prior_pics_flag, 1);
if (sps->alf_enabled_flag && !pps->alf_info_in_ph_flag) {
READ_UINT8 (&nr, sh->alf_enabled_flag, 1);
if (sh->alf_enabled_flag) {
READ_UINT8 (&nr, sh->num_alf_aps_ids_luma, 3);
for (i = 0; i < sh->num_alf_aps_ids_luma; i++)
READ_UINT8 (&nr, sh->alf_aps_id_luma[i], 3);
if (sps->chroma_format_idc != 0) {
READ_UINT8 (&nr, sh->alf_cb_enabled_flag, 1);
READ_UINT8 (&nr, sh->alf_cr_enabled_flag, 1);
} else {
sh->alf_cb_enabled_flag = ph->alf_cb_enabled_flag;
sh->alf_cr_enabled_flag = ph->alf_cr_enabled_flag;
}
if (sh->alf_cb_enabled_flag || sh->alf_cr_enabled_flag) {
READ_UINT8 (&nr, sh->alf_aps_id_chroma, 3);
} else {
sh->alf_aps_id_chroma = ph->alf_aps_id_chroma;
}
if (sps->ccalf_enabled_flag) {
READ_UINT8 (&nr, sh->alf_cc_cb_enabled_flag, 1);
if (sh->alf_cc_cb_enabled_flag) {
READ_UINT8 (&nr, sh->alf_cc_cb_aps_id, 3);
} else {
sh->alf_cc_cb_aps_id = ph->alf_cc_cb_aps_id;
}
READ_UINT8 (&nr, sh->alf_cc_cr_enabled_flag, 1);
if (sh->alf_cc_cr_enabled_flag) {
READ_UINT8 (&nr, sh->alf_cc_cr_aps_id, 3);
} else {
sh->alf_cc_cr_aps_id = ph->alf_cc_cr_aps_id;
}
} else {
sh->alf_cc_cb_enabled_flag = ph->alf_cc_cb_enabled_flag;
sh->alf_cc_cr_enabled_flag = ph->alf_cc_cr_enabled_flag;
sh->alf_cc_cb_aps_id = ph->alf_cc_cb_aps_id;
sh->alf_cc_cr_aps_id = ph->alf_cc_cr_aps_id;
}
}
} else {
sh->alf_enabled_flag = ph->alf_enabled_flag;
sh->num_alf_aps_ids_luma = ph->num_alf_aps_ids_luma;
for (i = 0; i < sh->num_alf_aps_ids_luma; i++)
sh->alf_aps_id_luma[i] = ph->alf_aps_id_luma[i];
sh->alf_cb_enabled_flag = ph->alf_cb_enabled_flag;
sh->alf_cr_enabled_flag = ph->alf_cr_enabled_flag;
sh->alf_aps_id_chroma = ph->alf_aps_id_chroma;
sh->alf_cc_cb_enabled_flag = ph->alf_cc_cb_enabled_flag;
sh->alf_cc_cb_aps_id = ph->alf_cc_cb_aps_id;
sh->alf_cc_cr_enabled_flag = ph->alf_cc_cr_enabled_flag;
sh->alf_cc_cr_aps_id = ph->alf_cc_cr_aps_id;
}
if (ph->lmcs_enabled_flag && !sh->picture_header_in_slice_header_flag) {
READ_UINT8 (&nr, sh->lmcs_used_flag, 1);
} else {
sh->lmcs_used_flag =
sh->picture_header_in_slice_header_flag ? ph->lmcs_enabled_flag : 0;
}
if (ph->explicit_scaling_list_enabled_flag &&
!sh->picture_header_in_slice_header_flag) {
READ_UINT8 (&nr, sh->explicit_scaling_list_used_flag, 1);
} else {
sh->explicit_scaling_list_used_flag =
sh->picture_header_in_slice_header_flag ?
ph->explicit_scaling_list_enabled_flag : 0;
}
if (!pps->rpl_info_in_ph_flag &&
((nal_unit_type != GST_H266_NAL_SLICE_IDR_W_RADL &&
nal_unit_type != GST_H266_NAL_SLICE_IDR_N_LP) ||
sps->idr_rpl_present_flag)) {
if (!gst_h266_ref_pic_lists (&sh->ref_pic_lists, &nr, sps, pps))
goto error;
} else {
sh->ref_pic_lists = ph->ref_pic_lists;
}
ref_pic_lists = &sh->ref_pic_lists;
if ((sh->slice_type != GST_H266_I_SLICE &&
ref_pic_lists->rpl_ref_list[0].num_ref_entries > 1) ||
(sh->slice_type == GST_H266_B_SLICE &&
ref_pic_lists->rpl_ref_list[1].num_ref_entries > 1)) {
READ_UINT8 (&nr, sh->num_ref_idx_active_override_flag, 1);
if (sh->num_ref_idx_active_override_flag) {
for (i = 0; i < (sh->slice_type == GST_H266_B_SLICE ? 2 : 1); i++)
if (ref_pic_lists->rpl_ref_list[i].num_ref_entries > 1) {
READ_UE_MAX (&nr, sh->num_ref_idx_active_minus1[i], 14);
} else {
sh->num_ref_idx_active_minus1[i] = 0;
}
}
} else {
sh->num_ref_idx_active_override_flag = 1;
}
for (i = 0; i < 2; i++) {
if (sh->slice_type == GST_H266_B_SLICE
|| (sh->slice_type == GST_H266_P_SLICE && i == 0)) {
if (sh->num_ref_idx_active_override_flag) {
sh->num_ref_idx_active[i] = sh->num_ref_idx_active_minus1[i] + 1;
} else {
sh->num_ref_idx_active[i] =
MIN (ref_pic_lists->rpl_ref_list[i].num_ref_entries,
pps->num_ref_idx_default_active_minus1[i] + 1);
}
} else {
/* sh_slice_type == I || (sh_slice_type == P && i == 1) */
sh->num_ref_idx_active[i] = 0;
}
}
sh->collocated_from_l0_flag = ph->collocated_from_l0_flag;
if (sh->slice_type != GST_H266_I_SLICE) {
if (pps->cabac_init_present_flag) {
READ_UINT8 (&nr, sh->cabac_init_flag, 1);
} else {
sh->cabac_init_flag = 0;
}
if (ph->temporal_mvp_enabled_flag) {
if (ph->temporal_mvp_enabled_flag) {
if (sh->slice_type == GST_H266_P_SLICE) {
sh->collocated_from_l0_flag = 1;
} else if (!pps->rpl_info_in_ph_flag
&& sh->slice_type == GST_H266_B_SLICE) {
READ_UINT8 (&nr, sh->collocated_from_l0_flag, 1);
} else {
sh->collocated_from_l0_flag = ph->collocated_from_l0_flag;
}
}
if (!pps->rpl_info_in_ph_flag) {
if ((sh->collocated_from_l0_flag && sh->num_ref_idx_active[0] > 1) ||
(!sh->collocated_from_l0_flag && sh->num_ref_idx_active[1] > 1)) {
gint idx = sh->collocated_from_l0_flag ? 0 : 1;
READ_UE_MAX (&nr, sh->collocated_ref_idx,
sh->num_ref_idx_active[idx] - 1);
} else {
sh->collocated_ref_idx = 0;
}
} else {
sh->collocated_ref_idx = ph->collocated_ref_idx;
}
}
if (!pps->wp_info_in_ph_flag &&
((pps->weighted_pred_flag && sh->slice_type == GST_H266_P_SLICE) ||
(pps->weighted_bipred_flag &&
sh->slice_type == GST_H266_B_SLICE))) {
if (!gst_h266_parse_pred_weight_table (&sh->pred_weight_table,
&nr, sps, pps, ref_pic_lists, sh->num_ref_idx_active))
goto error;
}
}
if (!pps->qp_delta_info_in_ph_flag) {
READ_SE_ALLOWED (&nr, sh->qp_delta, -63, 63);
} else {
sh->qp_delta = ph->qp_delta;
}
sh->slice_qp_y = 26 + pps->init_qp_minus26 + sh->qp_delta;
if (sh->slice_qp_y < -6 * sps->bitdepth_minus8 || sh->slice_qp_y > 63) {
GST_WARNING ("SliceQpy = %d is our of range.", sh->slice_qp_y);
goto error;
}
if (pps->slice_chroma_qp_offsets_present_flag) {
gint8 off;
READ_SE_ALLOWED (&nr, sh->cb_qp_offset, -12, 12);
off = pps->cb_qp_offset + sh->cb_qp_offset;
if (off < -12 || off > 12) {
GST_WARNING ("pps_cb_qp_offset + sh_cb_qp_offset(%d) = %d, "
"out of range [-12, 12].", sh->cb_qp_offset, off);
goto error;
}
READ_SE_ALLOWED (&nr, sh->cr_qp_offset, -12, 12);
off = pps->cr_qp_offset + sh->cr_qp_offset;
if (off < -12 || off > 12) {
GST_WARNING ("pps_cr_qp_offset + sh_cr_qp_offset(%d) = %d, "
"out of range [-12, 12].", sh->cr_qp_offset, off);
goto error;
}
if (sps->joint_cbcr_enabled_flag) {
READ_SE_ALLOWED (&nr, sh->joint_cbcr_qp_offset, -12, 12);
off = pps->joint_cbcr_qp_offset_value + sh->joint_cbcr_qp_offset;
if (off < -12 || off > 12) {
GST_WARNING ("pps_joint_cbcr_qp_offset_value + sh_joint_cbcr_qp_offset"
"(%d) = %d, out of range [-12, 12].",
sh->joint_cbcr_qp_offset, off);
goto error;
}
} else {
sh->joint_cbcr_qp_offset = 0;
}
} else {
sh->cb_qp_offset = 0;
sh->cr_qp_offset = 0;
sh->joint_cbcr_qp_offset = 0;
}
if (pps->cu_chroma_qp_offset_list_enabled_flag) {
READ_UINT8 (&nr, sh->cu_chroma_qp_offset_enabled_flag, 1);
} else {
sh->cu_chroma_qp_offset_enabled_flag = 0;
}
if (sps->sao_enabled_flag && !pps->sao_info_in_ph_flag) {
READ_UINT8 (&nr, sh->sao_luma_used_flag, 1);
if (sps->chroma_format_idc != 0) {
READ_UINT8 (&nr, sh->sao_chroma_used_flag, 1);
} else {
sh->sao_chroma_used_flag = ph->sao_chroma_enabled_flag;
}
} else {
sh->sao_luma_used_flag = ph->sao_luma_enabled_flag;
sh->sao_chroma_used_flag = ph->sao_chroma_enabled_flag;
}
/* Inherit deblock filter features from picture header */
sh->deblocking_filter_disabled_flag = ph->deblocking_filter_disabled_flag;
sh->luma_beta_offset_div2 = ph->luma_beta_offset_div2;
sh->luma_tc_offset_div2 = ph->luma_tc_offset_div2;
sh->cb_beta_offset_div2 = ph->cb_beta_offset_div2;
sh->cb_tc_offset_div2 = ph->cb_tc_offset_div2;
sh->cr_beta_offset_div2 = ph->cr_beta_offset_div2;
sh->cr_tc_offset_div2 = ph->cr_tc_offset_div2;
if (pps->deblocking_filter_override_enabled_flag && !pps->dbf_info_in_ph_flag) {
READ_UINT8 (&nr, sh->deblocking_params_present_flag, 1);
} else {
sh->deblocking_params_present_flag = 0;
}
if (sh->deblocking_params_present_flag) {
if (!pps->deblocking_filter_disabled_flag) {
READ_UINT8 (&nr, sh->deblocking_filter_disabled_flag, 1);
} else {
sh->deblocking_filter_disabled_flag = 0;
}
if (!sh->deblocking_filter_disabled_flag) {
READ_SE_ALLOWED (&nr, sh->luma_beta_offset_div2, -12, 12);
READ_SE_ALLOWED (&nr, sh->luma_tc_offset_div2, -12, 12);
if (pps->chroma_tool_offsets_present_flag) {
READ_SE_ALLOWED (&nr, sh->cb_beta_offset_div2, -12, 12);
READ_SE_ALLOWED (&nr, sh->cb_tc_offset_div2, -12, 12);
READ_SE_ALLOWED (&nr, sh->cr_beta_offset_div2, -12, 12);
READ_SE_ALLOWED (&nr, sh->cr_tc_offset_div2, -12, 12);
} else {
sh->cb_beta_offset_div2 = sh->luma_beta_offset_div2;
sh->cb_tc_offset_div2 = sh->luma_tc_offset_div2;
sh->cr_beta_offset_div2 = sh->luma_beta_offset_div2;
sh->cr_tc_offset_div2 = sh->luma_tc_offset_div2;
}
}
}
if (sps->dep_quant_enabled_flag) {
READ_UINT8 (&nr, sh->dep_quant_used_flag, 1);
} else {
sh->dep_quant_used_flag = 0;
}
if (sps->sign_data_hiding_enabled_flag && !sh->dep_quant_used_flag) {
READ_UINT8 (&nr, sh->sign_data_hiding_used_flag, 1);
} else {
sh->sign_data_hiding_used_flag = 0;
}
if (sps->transform_skip_enabled_flag &&
!sh->dep_quant_used_flag && !sh->sign_data_hiding_used_flag) {
READ_UINT8 (&nr, sh->ts_residual_coding_disabled_flag, 1);
} else {
sh->ts_residual_coding_disabled_flag = 0;
}
if (sh->ts_residual_coding_disabled_flag &&
sps->range_params.ts_residual_coding_rice_present_in_sh_flag) {
READ_UINT8 (&nr, sh->ts_residual_coding_rice_idx_minus1, 3);
} else {
sh->ts_residual_coding_rice_idx_minus1 = 0;
}
if (sps->range_params.reverse_last_sig_coeff_enabled_flag) {
READ_UINT8 (&nr, sh->reverse_last_sig_coeff_flag, 1);
} else {
sh->reverse_last_sig_coeff_flag = 0;
}
if (pps->slice_header_extension_present_flag) {
READ_UE_MAX (&nr, sh->slice_header_extension_length, 256);
for (i = 0; i < sh->slice_header_extension_length; i++)
READ_UINT8 (&nr, sh->slice_header_extension_data_byte[i], 8);
}
/* (141) */
sh->num_entry_points = 0;
if (sps->entry_point_offsets_present_flag) {
for (i = 1; i < num_ctus_in_curr_slice; i++) {
guint16 ctb_addr_x, ctb_addr_y, pre_ctb_addr_x, pre_ctb_addr_y;
ctb_addr_x = ctb_addr_in_curr_slice[i] % pps->pic_width_in_ctbs_y;
ctb_addr_y = ctb_addr_in_curr_slice[i] / pps->pic_width_in_ctbs_y;
pre_ctb_addr_x = ctb_addr_in_curr_slice[i - 1] % pps->pic_width_in_ctbs_y;
pre_ctb_addr_y = ctb_addr_in_curr_slice[i - 1] / pps->pic_width_in_ctbs_y;
if (parser->ctb_to_tile_row_bd[ctb_addr_y] !=
parser->ctb_to_tile_row_bd[pre_ctb_addr_y]
|| parser->ctb_to_tile_col_bd[ctb_addr_x] !=
parser->ctb_to_tile_col_bd[pre_ctb_addr_x]
|| (ctb_addr_y != pre_ctb_addr_y
&& sps->entropy_coding_sync_enabled_flag)) {
sh->entry_point_start_ctu[sh->num_entry_points] = i;
sh->num_entry_points++;
}
}
if (sh->num_entry_points > GST_H266_MAX_ENTRY_POINTS) {
GST_WARNING ("Too many entry points: %d.", sh->num_entry_points);
goto error;
}
if (sh->num_entry_points > 0) {
READ_UE_MAX (&nr, sh->entry_offset_len_minus1, 31);
for (i = 0; i < sh->num_entry_points; i++) {
READ_UINT32 (&nr, sh->entry_point_offset_minus1[i],
sh->entry_offset_len_minus1 + 1);
}
}
}
/* Skip the byte alignment bits */
if (!nal_reader_skip (&nr, 1))
goto error;
while (!nal_reader_is_byte_aligned (&nr))
if (!nal_reader_skip (&nr, 1))
goto error;
sh->header_size = nal_reader_get_pos (&nr);
sh->n_emulation_prevention_bytes = nal_reader_get_epb_count (&nr);
return ret;
error:
ret = GST_H266_PARSER_ERROR;
error_with_ret:
GST_WARNING ("error parsing \"Slice Header\"");
return ret;
}
static GstH266ParserResult
gst_h266_parser_parse_sei_message (GstH266SEIMessage * sei, NalReader * nr,
GstH266Parser * parser, guint8 nal_type, guint8 nal_tid)
{
guint8 payload_type_byte, payload_size_byte;
guint remaining, payload_size, payloadSize, payload_start_pos_bit;
GstH266ParserResult res = GST_H266_PARSER_OK;
if (nal_type == GST_H266_NAL_PREFIX_SEI)
GST_LOG ("parsing \"Prefix SEI message\"");
else if (nal_type == GST_H266_NAL_SUFFIX_SEI)
GST_LOG ("parsing \"Suffix SEI message\"");
memset (sei, 0, sizeof (*sei));
do {
READ_UINT8 (nr, payload_type_byte, 8);
sei->payloadType += payload_type_byte;
} while (payload_type_byte == 0xff);
payloadSize = 0;
do {
READ_UINT8 (nr, payload_size_byte, 8);
payloadSize += payload_size_byte;
} while (payload_size_byte == 0xff);
remaining = nal_reader_get_remaining (nr);
payload_size = payloadSize * 8 < remaining ? payloadSize * 8 : remaining;
payload_start_pos_bit = nal_reader_get_pos (nr);
GST_LOG ("SEI message received: payloadType %u, payloadSize = %u bits",
sei->payloadType, payload_size);
if (nal_type == GST_H266_NAL_PREFIX_SEI) {
switch (sei->payloadType) {
case GST_H266_SEI_BUF_PERIOD:
res = gst_h266_parser_parse_buffering_period
(&sei->payload.buffering_period, nr);
if (res == GST_H266_PARSER_OK) {
parser->buffering_period = *sei;
parser->last_buffering_period = &parser->buffering_period;
}
break;
case GST_H266_SEI_PIC_TIMING:
if (!parser->last_buffering_period) {
GST_WARNING ("No buffering_period SEI.");
goto error;
}
res = gst_h266_parser_parse_pic_timing (&sei->payload.pic_timing, nr,
&parser->buffering_period.payload.buffering_period, nal_tid);
break;
case GST_H266_SEI_DU_INFO:
if (!parser->last_buffering_period) {
GST_WARNING ("No buffering_period SEI.");
goto error;
}
res = gst_h266_parser_parse_du_info (&sei->payload.du_info, nr,
&parser->buffering_period.payload.buffering_period, nal_tid);
break;
case GST_H266_SEI_FRAME_FIELD_INFO:
res = gst_h266_parser_parse_frame_field_info
(&sei->payload.frame_field_info, nr);
break;
case GST_H266_SEI_SUBPIC_LEVEL_INFO:
res = gst_h266_parser_parse_subpic_level_info
(&sei->payload.subpic_level_info, nr);
break;
default:
/* Just consume payloadSize bytes, which does not account for
emulation prevention bytes */
if (!nal_reader_skip_long (nr, payload_size))
goto error;
res = GST_H266_PARSER_OK;
break;
}
} else if (nal_type == GST_H266_NAL_SUFFIX_SEI) {
switch (sei->payloadType) {
case GST_H266_SEI_SCALABLE_NESTING:
res = gst_h266_parser_parse_scalable_nesting
(&sei->payload.scalable_nesting, nr);
break;
default:
/* Just consume payloadSize bytes, which does not account for
emulation prevention bytes */
if (!nal_reader_skip_long (nr, payload_size))
goto error;
res = GST_H266_PARSER_OK;
break;
}
}
/* Not parsing the reserved_payload_extension, but it shouldn't be
* an issue because of 1: There shall not be any reserved_payload_extension
* present in bitstreams conforming to the specification.2. Even though
* it is present, the size will be less than total PayloadSize since the
* size of reserved_payload_extension is supposed to be
* 8 * payloadSize - nEarlierBits - nPayloadZeroBits -1 which means the
* the current implementation will still skip all unnecessary bits correctly.
* In theory, we can have a more optimized implementation by skipping the
* data left in PayLoadSize without out individually checking for each bits,
* since the totoal size will be always less than payloadSize*/
while (nal_reader_has_more_data_in_payload (nr, payload_start_pos_bit,
payloadSize)) {
/* Skip the byte alignment bits */
if (!nal_reader_skip (nr, 1))
goto error;
while (!nal_reader_is_byte_aligned (nr)) {
if (!nal_reader_skip (nr, 1))
goto error;
}
}
return res;
error:
GST_WARNING ("error parsing \"Sei message\"");
return GST_H266_PARSER_ERROR;
}
/**
* gst_h266_parser_parse_sei:
* @nalparser: a #GstH266Parser
* @nalu: The `GST_H266_NAL_*_SEI` #GstH266NalUnit to parse
* @messages: The GArray of #GstH266SEIMessage to fill. The caller must free
* it when done.
*
* Parses @data, create and fills the @messages array.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_sei (GstH266Parser * nalparser, GstH266NalUnit * nalu,
GArray ** messages)
{
NalReader nr;
GstH266SEIMessage sei;
GstH266ParserResult res;
GST_LOG ("parsing SEI nal");
nal_reader_init (&nr, nalu->data + nalu->offset + nalu->header_bytes,
nalu->size - nalu->header_bytes);
*messages = g_array_new (FALSE, FALSE, sizeof (GstH266SEIMessage));
do {
res = gst_h266_parser_parse_sei_message (&sei, &nr, nalparser, nalu->type,
nalu->temporal_id_plus1 - 1);
if (res == GST_H266_PARSER_OK)
g_array_append_val (*messages, sei);
else
break;
} while (nal_reader_has_more_data (&nr));
return res;
}
/**
* gst_h266_profile_to_string:
* @profile: a #GstH266Profile
*
* Returns the descriptive name for the #GstH266Profile.
*
* Returns: (nullable): the name for @profile or %NULL on error
*
* Since: 1.26
*/
const gchar *
gst_h266_profile_to_string (GstH266Profile profile)
{
guint i;
if (profile <= GST_H266_PROFILE_INVALID || profile >= GST_H266_PROFILE_MAX)
return NULL;
for (i = 0; i < G_N_ELEMENTS (h266_profiles); i++) {
if (profile == h266_profiles[i].profile)
return h266_profiles[i].name;
}
return NULL;
}
/**
* gst_h266_profile_from_string:
* @string: the descriptive name for #GstH266Profile
*
* Returns a #GstH266Profile for the @string.
*
* Returns: the #GstH266Profile of @string or %GST_H266_PROFILE_INVALID on error
*
* Since: 1.26
*/
GstH266Profile
gst_h266_profile_from_string (const gchar * string)
{
guint i;
if (string == NULL)
return GST_H266_PROFILE_INVALID;
for (i = 0; i < G_N_ELEMENTS (h266_profiles); i++) {
if (g_strcmp0 (string, h266_profiles[i].name) == 0) {
return h266_profiles[i].profile;
}
}
return GST_H266_PROFILE_INVALID;
}
/**
* gst_h266_decoder_config_record_free:
* @config: (nullable): a #GstH266DecoderConfigRecord data
*
* Free @config data
*
* Since: 1.26
*/
void
gst_h266_decoder_config_record_free (GstH266DecoderConfigRecord * config)
{
if (!config)
return;
if (config->nalu_array)
g_array_unref (config->nalu_array);
g_free (config);
}
static void
gst_clear_h266_decoder_config_record_nalu_array
(GstH266DecoderConfigRecordNalUnitArray * array)
{
if (!array)
return;
if (array->nalu)
g_array_unref (array->nalu);
}
static GstH266DecoderConfigRecord *
gst_h266_decoder_config_record_new (void)
{
GstH266DecoderConfigRecord *config;
config = g_new0 (GstH266DecoderConfigRecord, 1);
config->nalu_array = g_array_new (FALSE,
FALSE, sizeof (GstH266DecoderConfigRecordNalUnitArray));
g_array_set_clear_func (config->nalu_array,
(GDestroyNotify) gst_clear_h266_decoder_config_record_nalu_array);
return config;
}
#define READ_CONFIG_UINT8(br, val, nbits) G_STMT_START { \
if (!gst_bit_reader_get_bits_uint8 (br, &val, nbits)) { \
GST_WARNING ("Failed to read " G_STRINGIFY (val)); \
goto error; \
} \
} G_STMT_END;
#define READ_CONFIG_UINT16(br, val, nbits) G_STMT_START { \
if (!gst_bit_reader_get_bits_uint16 (br, &val, nbits)) { \
GST_WARNING ("Failed to read " G_STRINGIFY (val)); \
goto error; \
} \
} G_STMT_END;
#define READ_CONFIG_UINT32(br, val, nbits) G_STMT_START { \
if (!gst_bit_reader_get_bits_uint32 (br, &val, nbits)) { \
GST_WARNING ("Failed to read " G_STRINGIFY (val)); \
goto error; \
} \
} G_STMT_END;
#define SKIP_CONFIG_BITS(br, nbits) G_STMT_START { \
if (!gst_bit_reader_skip (br, nbits)) { \
GST_WARNING ("Failed to skip %d bits", nbits); \
goto error; \
} \
} G_STMT_END;
static GstH266ParserResult
gst_h266_parser_parse_ptl_record (GstBitReader * br, GstH266PTLRecord * ptl,
guint8 num_sublayers)
{
GstH266ParserResult result = GST_H266_PARSER_OK;
gint i;
SKIP_CONFIG_BITS (br, 2);
READ_CONFIG_UINT8 (br, ptl->num_bytes_constraint_info, 6);
if (!ptl->num_bytes_constraint_info)
GST_WARNING
("num_bytes_constraint_info is 0, but should be greater than 0");
READ_CONFIG_UINT8 (br, ptl->general_profile_idc, 7);
READ_CONFIG_UINT8 (br, ptl->general_tier_flag, 1);
READ_CONFIG_UINT8 (br, ptl->general_level_idc, 8);
READ_CONFIG_UINT8 (br, ptl->ptl_frame_only_constraint_flag, 1);
READ_CONFIG_UINT8 (br, ptl->ptl_multilayer_enabled_flag, 1);
for (i = 0; i < ptl->num_bytes_constraint_info - 1; i++)
READ_CONFIG_UINT8 (br, ptl->general_constraint_info[i], 8);
READ_CONFIG_UINT8 (br, ptl->general_constraint_info[i], 6);
for (i = num_sublayers - 2; i >= 0; i--)
READ_CONFIG_UINT8 (br, ptl->ptl_sublayer_level_present_flag[i], 1);
if (num_sublayers > 1)
SKIP_CONFIG_BITS (br, 9 - num_sublayers);
for (i = num_sublayers - 2; i >= 0; i--)
if (ptl->ptl_sublayer_level_present_flag[i])
READ_CONFIG_UINT8 (br, ptl->sublayer_level_idc[i], 8);
READ_CONFIG_UINT8 (br, ptl->ptl_num_sub_profiles, 8);
for (i = 0; i < ptl->ptl_num_sub_profiles; i++)
READ_CONFIG_UINT32 (br, ptl->general_sub_profile_idc[i], 32);
return result;
error:
GST_WARNING ("Failed to parse PTL record");
result = GST_H266_PARSER_ERROR;
return result;
}
/**
* gst_h266_parser_parse_decoder_config_record:
* @parser: a #GstH266Parser
* @data: the data to parse
* @size: the size of @data
* @config: (out): parsed #GstH266DecoderConfigRecord data
*
* Parses VVCDecoderConfigurationRecord data and fill into @config.
* The caller must free @config via gst_h266_decoder_config_record_free()
*
* This method does not parse APS, VPS, SPS and PPS and therefore the caller needs to
* parse each NAL unit via appropriate parsing method.
*
* Returns: a #GstH266ParserResult
*
* Since: 1.26
*/
GstH266ParserResult
gst_h266_parser_parse_decoder_config_record (GstH266Parser * parser,
const guint8 * data, gsize size, GstH266DecoderConfigRecord ** config)
{
GstH266DecoderConfigRecord *ret;
GstBitReader br;
GstH266ParserResult result = GST_H266_PARSER_OK;
guint i;
guint8 num_of_arrays;
GST_LOG ("parsing \"Decoder configuration record\"");
g_return_val_if_fail (parser != NULL, GST_H266_PARSER_ERROR);
g_return_val_if_fail (data != NULL, GST_H266_PARSER_ERROR);
g_return_val_if_fail (config != NULL, GST_H266_PARSER_ERROR);
*config = NULL;
if (size < 2) {
GST_WARNING ("Too small size vvcC");
return GST_H266_PARSER_ERROR;
}
gst_bit_reader_init (&br, data, size);
ret = gst_h266_decoder_config_record_new ();
SKIP_CONFIG_BITS (&br, 5);
READ_CONFIG_UINT8 (&br, ret->length_size_minus_one, 2);
if (ret->length_size_minus_one == 2) {
/* "length_size_minus_one + 1" should be 1, 2, or 4 */
GST_WARNING ("Wrong nal-length-size");
}
READ_CONFIG_UINT8 (&br, ret->ptl_present_flag, 1);
if (ret->ptl_present_flag) {
READ_CONFIG_UINT16 (&br, ret->ols_idx, 9);
READ_CONFIG_UINT8 (&br, ret->num_sublayers, 3);
READ_CONFIG_UINT8 (&br, ret->constant_frame_rate, 2);
READ_CONFIG_UINT8 (&br, ret->chroma_format_idc, 2);
READ_CONFIG_UINT8 (&br, ret->bit_depth_minus8, 3);
SKIP_CONFIG_BITS (&br, 5);
if (gst_h266_parser_parse_ptl_record (&br, &ret->native_ptl,
ret->num_sublayers) != GST_H266_PARSER_OK)
goto error;
READ_CONFIG_UINT16 (&br, ret->max_picture_width, 16);
READ_CONFIG_UINT16 (&br, ret->max_picture_height, 16);
READ_CONFIG_UINT16 (&br, ret->avg_frame_rate, 16);
}
READ_CONFIG_UINT8 (&br, num_of_arrays, 8);
for (i = 0; i < num_of_arrays; i++) {
GstH266DecoderConfigRecordNalUnitArray array;
guint8 nalu_type;
GstH266NalUnit nalu;
guint16 num_nalu = 1, j;
guint bit_offset, offset;
READ_CONFIG_UINT8 (&br, array.array_completeness, 1);
SKIP_CONFIG_BITS (&br, 2);
READ_CONFIG_UINT8 (&br, nalu_type, 5);
array.nal_unit_type = nalu_type;
if (nalu_type != GST_H266_NAL_DCI && nalu_type != GST_H266_NAL_OPI)
READ_CONFIG_UINT16 (&br, num_nalu, 16);
bit_offset = gst_bit_reader_get_pos (&br);
g_assert (bit_offset % 8 == 0);
offset = bit_offset / 8;
array.nalu = g_array_sized_new (FALSE, FALSE, sizeof (GstH266NalUnit),
num_nalu);
for (j = 0; j < num_nalu; j++) {
result = gst_h266_parser_identify_nalu_vvc (parser, data, offset, size,
2, &nalu);
if (result != GST_H266_PARSER_OK) {
g_array_unref (array.nalu);
goto error;
}
g_array_append_val (array.nalu, nalu);
offset = nalu.offset + nalu.size;
}
g_array_append_val (ret->nalu_array, array);
if (i != num_of_arrays - 1 && !gst_bit_reader_set_pos (&br, offset * 8)) {
GST_WARNING ("Not enough bytes for NAL reading");
goto error;
}
}
*config = ret;
return GST_H266_PARSER_OK;
error:
result = GST_H266_PARSER_ERROR;
gst_h266_decoder_config_record_free (ret);
return result;
}
#undef READ_CONFIG_UINT8
#undef READ_CONFIG_UINT16
#undef READ_CONFIG_UINT32
#undef SKIP_CONFIG_BITS