gstreamer/subprojects/gst-plugins-bad/gst-libs/gst/codecparsers/gsth265bitwriter.c
He Junyan 2408ca2f18 h265bitwriter: Correct the all API to byte aligned.
In fact, all the h265 bit writer have byte aligned output. So we
change the API from bit size in unit to byte size, which is easy
to use.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/3193>
2022-10-27 12:15:43 +00:00

2307 lines
72 KiB
C

/* GStreamer
* Copyright (C) 2021 Intel Corporation
* 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 the0
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "gsth265bitwriter.h"
#include <gst/codecparsers/nalutils.h>
#include <gst/base/gstbitwriter.h>
#include <math.h>
/******************************** Utils ********************************/
#define SIGNED(val) (2 * ABS(val) - ((val) > 0))
/* Write an unsigned integer Exp-Golomb-coded syntax element. i.e. ue(v) */
static gboolean
_bs_write_ue (GstBitWriter * bs, guint32 value)
{
guint32 size_in_bits = 0;
guint32 tmp_value = ++value;
while (tmp_value) {
++size_in_bits;
tmp_value >>= 1;
}
if (size_in_bits > 1
&& !gst_bit_writer_put_bits_uint32 (bs, 0, size_in_bits - 1))
return FALSE;
if (!gst_bit_writer_put_bits_uint32 (bs, value, size_in_bits))
return FALSE;
return TRUE;
}
#define WRITE_BITS_UNCHECK(bw, val, nbits) \
(nbits <= 8 ? gst_bit_writer_put_bits_uint8 (bw, val, nbits) : \
(nbits <= 16 ? gst_bit_writer_put_bits_uint16 (bw, val, nbits) : \
(nbits <= 32 ? gst_bit_writer_put_bits_uint32 (bw, val, nbits) : \
FALSE)))
#define WRITE_BITS(bw, val, nbits) \
if (!WRITE_BITS_UNCHECK (bw, val, nbits)) { \
g_warning ("Unsupported bit size: %u", nbits); \
have_space = FALSE; \
goto error; \
}
#define WRITE_UE_UNCHECK(bw, val) _bs_write_ue (bw, val)
#ifdef WRITE_UE
#undef WRITE_UE
#endif
#define WRITE_UE(bw, val) \
if (!(have_space = WRITE_UE_UNCHECK (bw, val))) \
goto error; \
#define WRITE_UE_MAX(bw, val, max) \
if ((guint32) val > (max) || !(have_space = WRITE_UE_UNCHECK (bw, val))) \
goto error;
#define WRITE_SE(bw, val) WRITE_UE (bw, SIGNED (val))
#define WRITE_SE_RANGE(bw, val, min, max) \
if (val > max || val < min || \
!(have_space = WRITE_UE_UNCHECK (bw, SIGNED (val)))) \
goto error;
#define WRITE_BYTES_UNCHECK(bw, ptr, nbytes) \
gst_bit_writer_put_bytes(bw, ptr, nbytes)
#ifdef WRITE_BYTES
#undef WRITE_BYTES
#endif
#define WRITE_BYTES(bw, ptr, nbytes) \
if (!(have_space = WRITE_BYTES_UNCHECK (bw, ptr, nbytes))) \
goto error;
/***************************** End of Utils ****************************/
#define EXTENDED_SAR 255
/**** Default scaling_lists according to Table 7-5 and 7-6 *****/
/* Table 7-5 */
static const guint8 default_scaling_list0[16] = {
16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16,
16, 16, 16, 16
};
/* Combined the values in Table 7-6 to make the calculation easier
* Default scaling list of 8x8 and 16x16 matrices for matrixId = 0, 1 and 2
* Default scaling list of 32x32 matrix for matrixId = 0
*/
static const guint8 default_scaling_list1[64] = {
16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 17, 16,
17, 16, 17, 18, 17, 18, 18, 17, 18, 21, 19, 20,
21, 20, 19, 21, 24, 22, 22, 24, 24, 22, 22, 24,
25, 25, 27, 30, 27, 25, 25, 29, 31, 35, 35, 31,
29, 36, 41, 44, 41, 36, 47, 54, 54, 47, 65, 70,
65, 88, 88, 115
};
/* Combined the values in Table 7-6 to make the calculation easier
* Default scaling list of 8x8 and 16x16 matrices for matrixId = 3, 4 and 5
* Default scaling list of 32x32 matrix for matrixId = 1
*/
static const guint8 default_scaling_list2[64] = {
16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 17, 17,
17, 17, 17, 18, 18, 18, 18, 18, 18, 20, 20, 20,
20, 20, 20, 20, 24, 24, 24, 24, 24, 24, 24, 24,
25, 25, 25, 25, 25, 25, 25, 28, 28, 28, 28, 28,
28, 33, 33, 33, 33, 33, 41, 41, 41, 41, 54, 54,
54, 71, 71, 91
};
static gboolean
_h265_bit_writer_profile_tier_level (const GstH265ProfileTierLevel * ptl,
guint8 maxNumSubLayersMinus1, GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
guint i, j;
GST_DEBUG ("writing profile_tier_level");
WRITE_BITS (bw, ptl->profile_space, 2);
WRITE_BITS (bw, ptl->tier_flag, 1);
WRITE_BITS (bw, ptl->profile_idc, 5);
for (j = 0; j < 32; j++)
WRITE_BITS (bw, ptl->profile_compatibility_flag[j], 1);
WRITE_BITS (bw, ptl->progressive_source_flag, 1);
WRITE_BITS (bw, ptl->interlaced_source_flag, 1);
WRITE_BITS (bw, ptl->non_packed_constraint_flag, 1);
WRITE_BITS (bw, ptl->frame_only_constraint_flag, 1);
if (ptl->profile_idc == 4 || ptl->profile_compatibility_flag[4] ||
ptl->profile_idc == 5 || ptl->profile_compatibility_flag[5] ||
ptl->profile_idc == 6 || ptl->profile_compatibility_flag[6] ||
ptl->profile_idc == 7 || ptl->profile_compatibility_flag[7] ||
ptl->profile_idc == 8 || ptl->profile_compatibility_flag[8] ||
ptl->profile_idc == 9 || ptl->profile_compatibility_flag[9] ||
ptl->profile_idc == 10 || ptl->profile_compatibility_flag[10] ||
ptl->profile_idc == 11 || ptl->profile_compatibility_flag[11]) {
WRITE_BITS (bw, ptl->max_12bit_constraint_flag, 1);
WRITE_BITS (bw, ptl->max_10bit_constraint_flag, 1);
WRITE_BITS (bw, ptl->max_8bit_constraint_flag, 1);
WRITE_BITS (bw, ptl->max_422chroma_constraint_flag, 1);
WRITE_BITS (bw, ptl->max_420chroma_constraint_flag, 1);
WRITE_BITS (bw, ptl->max_monochrome_constraint_flag, 1);
WRITE_BITS (bw, ptl->intra_constraint_flag, 1);
WRITE_BITS (bw, ptl->one_picture_only_constraint_flag, 1);
WRITE_BITS (bw, ptl->lower_bit_rate_constraint_flag, 1);
if (ptl->profile_idc == 5 || ptl->profile_compatibility_flag[5] ||
ptl->profile_idc == 9 || ptl->profile_compatibility_flag[9] ||
ptl->profile_idc == 10 || ptl->profile_compatibility_flag[10] ||
ptl->profile_idc == 11 || ptl->profile_compatibility_flag[11]) {
WRITE_BITS (bw, ptl->max_14bit_constraint_flag, 1);
/* general_reserved_zero_33bits */
WRITE_BITS (bw, 0, 32);
WRITE_BITS (bw, 0, 1);
} else {
/* general_reserved_zero_34bits */
WRITE_BITS (bw, 0, 32);
WRITE_BITS (bw, 0, 2);
}
} else if (ptl->profile_idc == 2 || ptl->profile_compatibility_flag[2]) {
/* general_reserved_zero_7bits */
WRITE_BITS (bw, 0, 7);
WRITE_BITS (bw, ptl->one_picture_only_constraint_flag, 1);
/* general_reserved_zero_35bits */
WRITE_BITS (bw, 0, 32);
WRITE_BITS (bw, 0, 3);
} else {
/* general_reserved_zero_43bits */
WRITE_BITS (bw, 0, 32);
WRITE_BITS (bw, 0, 11);
}
/* general_inbld_flag, just set to 0 */
WRITE_BITS (bw, 0, 1);
WRITE_BITS (bw, ptl->level_idc, 8);
for (j = 0; j < maxNumSubLayersMinus1; j++) {
if (ptl->sub_layer_profile_present_flag[j]) {
GST_WARNING ("sub layer profile does not supported now");
goto error;
}
WRITE_BITS (bw, ptl->sub_layer_profile_present_flag[j], 1);
if (ptl->sub_layer_level_present_flag[j]) {
GST_WARNING ("sub layer level does not supported now");
goto error;
}
WRITE_BITS (bw, ptl->sub_layer_level_present_flag[j], 1);
}
if (maxNumSubLayersMinus1 > 0) {
for (i = maxNumSubLayersMinus1; i < 8; i++)
/* reserved_zero_2bits */
WRITE_BITS (bw, 0, 2);
}
/* TODO: Add sub layers profiles. */
*space = TRUE;
return TRUE;
error:
GST_WARNING ("error to write profile_tier_level");
*space = have_space;
return FALSE;
}
static gboolean
_h265_bit_writer_sub_layer_hrd_parameters (const GstH265SubLayerHRDParams *
sub_hrd, guint8 CpbCnt, guint8 sub_pic_hrd_params_present_flag,
GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
guint i;
GST_DEBUG ("writing \"subLayer HRD Parameters\"");
for (i = 0; i <= CpbCnt; i++) {
WRITE_UE_MAX (bw, sub_hrd->bit_rate_value_minus1[i], G_MAXUINT32 - 1);
WRITE_UE_MAX (bw, sub_hrd->cpb_size_value_minus1[i], G_MAXUINT32 - 1);
if (sub_pic_hrd_params_present_flag) {
WRITE_UE_MAX (bw, sub_hrd->cpb_size_du_value_minus1[i], G_MAXUINT32 - 1);
WRITE_UE_MAX (bw, sub_hrd->bit_rate_du_value_minus1[i], G_MAXUINT32 - 1);
}
WRITE_BITS (bw, sub_hrd->cbr_flag[i], 1);
}
*space = TRUE;
return TRUE;
error:
GST_WARNING ("error to write \"subLayer HRD Parameters \"");
*space = have_space;
return FALSE;
}
static gboolean
_h265_bit_writer_hrd_parameters (const GstH265HRDParams * hrd,
guint8 commonInfPresentFlag, guint8 maxNumSubLayersMinus1,
GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
guint i;
GST_DEBUG ("writing \"HRD Parameters\"");
if (commonInfPresentFlag) {
WRITE_BITS (bw, hrd->nal_hrd_parameters_present_flag, 1);
WRITE_BITS (bw, hrd->vcl_hrd_parameters_present_flag, 1);
if (hrd->nal_hrd_parameters_present_flag
|| hrd->vcl_hrd_parameters_present_flag) {
WRITE_BITS (bw, hrd->sub_pic_hrd_params_present_flag, 1);
if (hrd->sub_pic_hrd_params_present_flag) {
WRITE_BITS (bw, hrd->tick_divisor_minus2, 8);
WRITE_BITS (bw, hrd->du_cpb_removal_delay_increment_length_minus1, 5);
WRITE_BITS (bw, hrd->sub_pic_cpb_params_in_pic_timing_sei_flag, 1);
WRITE_BITS (bw, hrd->dpb_output_delay_du_length_minus1, 5);
}
WRITE_BITS (bw, hrd->bit_rate_scale, 4);
WRITE_BITS (bw, hrd->cpb_size_scale, 4);
if (hrd->sub_pic_hrd_params_present_flag)
WRITE_BITS (bw, hrd->cpb_size_du_scale, 4);
WRITE_BITS (bw, hrd->initial_cpb_removal_delay_length_minus1, 5);
WRITE_BITS (bw, hrd->au_cpb_removal_delay_length_minus1, 5);
WRITE_BITS (bw, hrd->dpb_output_delay_length_minus1, 5);
}
}
for (i = 0; i <= maxNumSubLayersMinus1; i++) {
WRITE_BITS (bw, hrd->fixed_pic_rate_general_flag[i], 1);
if (!hrd->fixed_pic_rate_general_flag[i]) {
WRITE_BITS (bw, hrd->fixed_pic_rate_within_cvs_flag[i], 1);
} else {
if (hrd->fixed_pic_rate_within_cvs_flag[i] == 0)
goto error;
}
if (hrd->fixed_pic_rate_within_cvs_flag[i]) {
WRITE_UE_MAX (bw, hrd->elemental_duration_in_tc_minus1[i], 2047);
} else {
WRITE_BITS (bw, hrd->low_delay_hrd_flag[i], 1);
}
if (!hrd->low_delay_hrd_flag[i])
WRITE_UE_MAX (bw, hrd->cpb_cnt_minus1[i], 31);
if (hrd->nal_hrd_parameters_present_flag)
if (!_h265_bit_writer_sub_layer_hrd_parameters
(&hrd->sublayer_hrd_params[i], hrd->cpb_cnt_minus1[i],
hrd->sub_pic_hrd_params_present_flag, bw, &have_space))
goto error;
/* TODO: need to separate nal and vcl from hrd_parameters. */
if (hrd->vcl_hrd_parameters_present_flag)
if (!_h265_bit_writer_sub_layer_hrd_parameters
(&hrd->sublayer_hrd_params[i], hrd->cpb_cnt_minus1[i],
hrd->sub_pic_hrd_params_present_flag, bw, &have_space))
goto error;
}
*space = TRUE;
return TRUE;
error:
GST_WARNING ("error to write \"HRD Parameters\"");
*space = have_space;
return FALSE;
}
static gboolean
_h265_bit_writer_vps (const GstH265VPS * vps, GstBitWriter * bw,
gboolean * space)
{
gboolean have_space = TRUE;
guint i, j;
GST_DEBUG ("writing VPS");
WRITE_BITS (bw, vps->id, 4);
WRITE_BITS (bw, vps->base_layer_internal_flag, 1);
WRITE_BITS (bw, vps->base_layer_available_flag, 1);
WRITE_BITS (bw, vps->max_layers_minus1, 6);
WRITE_BITS (bw, vps->max_sub_layers_minus1, 3);
WRITE_BITS (bw, vps->temporal_id_nesting_flag, 1);
/* reserved_0xffff_16bits */
WRITE_BITS (bw, 0xffff, 16);
if (!_h265_bit_writer_profile_tier_level (&vps->profile_tier_level,
vps->max_sub_layers_minus1, bw, &have_space))
goto error;
WRITE_BITS (bw, vps->sub_layer_ordering_info_present_flag, 1);
for (i = (vps->sub_layer_ordering_info_present_flag ? 0 :
vps->max_sub_layers_minus1); i <= vps->max_sub_layers_minus1; i++) {
WRITE_UE (bw, vps->max_dec_pic_buffering_minus1[i]);
WRITE_UE_MAX (bw, vps->max_num_reorder_pics[i],
vps->max_dec_pic_buffering_minus1[i]);
WRITE_UE_MAX (bw, vps->max_latency_increase_plus1[i], G_MAXUINT32 - 1);
}
/* max_layer_id should be <63, but only support 1 layer now. */
if (vps->max_layer_id > 1) {
GST_WARNING ("multi layers are not supported now");
goto error;
}
WRITE_BITS (bw, vps->max_layer_id, 6);
if (vps->num_layer_sets_minus1 >= 1) {
GST_WARNING ("layer set is not supported now");
goto error;
}
WRITE_UE_MAX (bw, vps->num_layer_sets_minus1, 1023);
/* TODO: support multi-layer. */
for (i = 1; i <= vps->num_layer_sets_minus1; i++) {
for (j = 0; j <= vps->max_layer_id; j++) {
/* layer_id_included_flag[i][j] */
WRITE_BITS (bw, 0, 1);
}
}
WRITE_BITS (bw, vps->timing_info_present_flag, 1);
if (vps->timing_info_present_flag) {
WRITE_BITS (bw, vps->num_units_in_tick, 32);
WRITE_BITS (bw, vps->time_scale, 32);
WRITE_BITS (bw, vps->poc_proportional_to_timing_flag, 1);
if (vps->poc_proportional_to_timing_flag)
WRITE_UE_MAX (bw, vps->num_ticks_poc_diff_one_minus1, G_MAXUINT32 - 1);
/* TODO: VPS can have multiple hrd parameters, and therefore hrd_params
* should be an array (like Garray). Just support 1 hdr parameter now.
*/
if (vps->num_hrd_parameters > 1) {
GST_WARNING ("HRD parameters > 1 is not supported now");
goto error;
}
WRITE_UE_MAX (bw, vps->num_hrd_parameters, vps->num_layer_sets_minus1 + 1);
if (vps->num_hrd_parameters) {
WRITE_UE_MAX (bw, vps->hrd_layer_set_idx, vps->num_layer_sets_minus1);
if (!_h265_bit_writer_hrd_parameters (&vps->hrd_params,
vps->cprms_present_flag, vps->max_sub_layers_minus1,
bw, &have_space))
goto error;
}
}
if (vps->vps_extension) {
GST_WARNING ("vps extension is not supported now");
goto error;
}
WRITE_BITS (bw, 0, 1);
*space = TRUE;
return TRUE;
error:
GST_WARNING ("failed to write VPS");
*space = have_space;
return FALSE;
}
/**
* gst_h265_bit_writer_vps:
* @vps: the vps of #GstH265VPS to write
* @start_code: whether adding the nal start code
* @data: (out): the bit stream generated by the sps
* @size: (inout): the size in bytes of the input and output
*
* Generating the according h265 bit stream by providing the vps.
*
* Returns: a #GstH265BitWriterResult
*
* Since: 1.22
**/
GstH265BitWriterResult
gst_h265_bit_writer_vps (const GstH265VPS * vps, gboolean start_code,
guint8 * data, guint * size)
{
gboolean have_space = TRUE;
GstBitWriter bw;
g_return_val_if_fail (vps != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);
gst_bit_writer_init_with_data (&bw, data, *size, FALSE);
if (start_code)
WRITE_BITS (&bw, 0x00000001, 32);
/* NAL unit header */
/* forbidden_zero_bit */
WRITE_BITS (&bw, 0, 1);
/* nal_unit_type */
WRITE_BITS (&bw, GST_H265_NAL_VPS, 6);
/* nuh_layer_id, only support 0 now */
WRITE_BITS (&bw, 0, 6);
/* nuh_temporal_id_plus1, only support 1 now */
WRITE_BITS (&bw, 1, 3);
if (!_h265_bit_writer_vps (vps, &bw, &have_space))
goto error;
/* Add trailings. */
WRITE_BITS (&bw, 1, 1);
if (!gst_bit_writer_align_bytes (&bw, 0)) {
have_space = FALSE;
goto error;
}
*size = gst_bit_writer_get_size (&bw) / 8;
gst_bit_writer_reset (&bw);
return GST_H265_BIT_WRITER_OK;
error:
gst_bit_writer_reset (&bw);
*size = 0;
if (!have_space)
return GST_H265_BIT_WRITER_NO_MORE_SPACE;
return GST_H265_BIT_WRITER_INVALID_DATA;
}
static gboolean
_get_scaling_list_params (const GstH265ScalingList * dest_scaling_list,
guint8 sizeId, guint8 matrixId, const guint8 ** sl, guint8 * size,
gint16 * scaling_list_dc_coef_minus8)
{
switch (sizeId) {
case GST_H265_QUANT_MATIX_4X4:
*sl = dest_scaling_list->scaling_lists_4x4[matrixId];
if (size)
*size = 16;
break;
case GST_H265_QUANT_MATIX_8X8:
*sl = dest_scaling_list->scaling_lists_8x8[matrixId];
if (size)
*size = 64;
break;
case GST_H265_QUANT_MATIX_16X16:
*sl = dest_scaling_list->scaling_lists_16x16[matrixId];
if (size)
*size = 64;
if (scaling_list_dc_coef_minus8)
*scaling_list_dc_coef_minus8 =
dest_scaling_list->scaling_list_dc_coef_minus8_16x16[matrixId];
break;
case GST_H265_QUANT_MATIX_32X32:
*sl = dest_scaling_list->scaling_lists_32x32[matrixId];
if (size)
*size = 64;
if (scaling_list_dc_coef_minus8)
*scaling_list_dc_coef_minus8 =
dest_scaling_list->scaling_list_dc_coef_minus8_32x32[matrixId];
break;
default:
g_assert_not_reached ();
return FALSE;
}
return TRUE;
}
static const guint8 *
_get_default_scaling_lists (GstH265QuantMatrixSize sizeId, guint8 matrixId)
{
const guint8 *sl;
switch (sizeId) {
case GST_H265_QUANT_MATIX_4X4:
sl = default_scaling_list0;
break;
case GST_H265_QUANT_MATIX_8X8:
case GST_H265_QUANT_MATIX_16X16:
if (matrixId <= 2) {
sl = default_scaling_list1;
} else {
sl = default_scaling_list2;
}
break;
case GST_H265_QUANT_MATIX_32X32:
if (matrixId == 0) {
sl = default_scaling_list1;
} else {
sl = default_scaling_list2;
}
break;
default:
g_assert_not_reached ();
return NULL;
}
return sl;
}
static gboolean
_compare_scaling_list_matrix (GstH265QuantMatrixSize sizeId,
const guint8 * sl0, const guint8 * sl1,
gint16 dc_coef_minus8_0, gint16 dc_coef_minus8_1)
{
guint size = sizeId == GST_H265_QUANT_MATIX_4X4 ? 16 : 64;
if (memcmp (sl0, sl1, size * sizeof (guint8)))
return FALSE;
if (sizeId <= GST_H265_QUANT_MATIX_8X8)
return TRUE;
return dc_coef_minus8_0 == dc_coef_minus8_1;
}
static gboolean
_h265_bit_writer_scaling_lists (const GstH265ScalingList * src_scaling_list,
GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
GstH265QuantMatrixSize sizeId;
guint8 matrixId;
guint8 scaling_list_pred_mode_flag = 0;
guint8 size, i, j;
GST_DEBUG ("writing scaling lists");
for (sizeId = 0; sizeId <= GST_H265_QUANT_MATIX_32X32; sizeId++) {
for (matrixId = 0;
matrixId < ((sizeId == GST_H265_QUANT_MATIX_32X32) ? 2 : 6);
matrixId++) {
gint16 scaling_list_dc_coef_minus8 = 8;
const guint8 *sl;
const guint8 *default_sl;
guint8 nextCoef;
gint8 coef_val;
guint8 scaling_list_pred_matrix_id_delta;
if (!_get_scaling_list_params (src_scaling_list, sizeId, matrixId,
&sl, &size, &scaling_list_dc_coef_minus8))
goto error;
/* Check whether it is the default matrix. */
default_sl = _get_default_scaling_lists (sizeId, matrixId);
if (_compare_scaling_list_matrix (sizeId, sl, default_sl,
scaling_list_dc_coef_minus8, 8)) {
scaling_list_pred_mode_flag = 0;
WRITE_BITS (bw, scaling_list_pred_mode_flag, 1);
scaling_list_pred_matrix_id_delta = 0;
WRITE_UE_MAX (bw, scaling_list_pred_matrix_id_delta, matrixId);
continue;
}
/* If some previous matrix is the same, just ref it. */
scaling_list_pred_matrix_id_delta = 0;
for (j = 0; j < matrixId; j++) {
gboolean ret;
guint8 size2;
const guint8 *sl2;
gint16 scaling_list_dc_coef_minus8_2 = 8;
ret = _get_scaling_list_params (src_scaling_list, sizeId, j,
&sl2, &size2, &scaling_list_dc_coef_minus8_2);
g_assert (ret);
g_assert (size == size2);
if (_compare_scaling_list_matrix (sizeId, sl, sl2,
scaling_list_dc_coef_minus8, scaling_list_dc_coef_minus8_2)) {
scaling_list_pred_matrix_id_delta = matrixId - j;
break;
}
}
if (scaling_list_pred_matrix_id_delta > 0) {
scaling_list_pred_mode_flag = 0;
WRITE_BITS (bw, scaling_list_pred_mode_flag, 1);
WRITE_UE_MAX (bw, scaling_list_pred_matrix_id_delta, matrixId);
continue;
}
/* Just explicitly signal all matrix coef. */
scaling_list_pred_mode_flag = 1;
WRITE_BITS (bw, scaling_list_pred_mode_flag, 1);
nextCoef = 8;
if (sizeId > 1) {
WRITE_SE_RANGE (bw, scaling_list_dc_coef_minus8, -7, 247);
nextCoef = scaling_list_dc_coef_minus8 + 8;
}
for (i = 0; i < size; i++) {
coef_val = sl[i] - nextCoef;
nextCoef = sl[i];
if (coef_val > 127) {
coef_val = coef_val - 256;
}
if (coef_val < -128) {
coef_val = coef_val + 256;
}
WRITE_SE_RANGE (bw, coef_val, -128, 127);
}
}
}
*space = TRUE;
return TRUE;
error:
GST_WARNING ("error to write scaling lists");
*space = have_space;
return FALSE;
}
static gboolean
_h265_bit_writer_short_term_ref_pic_set (const GstH265ShortTermRefPicSet *
stRPS, guint8 stRpsIdx, const GstH265SPS * sps,
GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
gint32 prev;
gint i = 0;
GST_DEBUG ("writing \"ShortTermRefPicSetParameter\"");
if (stRPS->inter_ref_pic_set_prediction_flag) {
/* TODO */
GST_WARNING ("inter_ref_pic_set_prediction_flag mode not supported");
goto error;
}
if (stRpsIdx != 0)
WRITE_BITS (bw, stRPS->inter_ref_pic_set_prediction_flag, 1);
if (stRPS->NumNegativePics + stRPS->NumPositivePics != stRPS->NumDeltaPocs)
goto error;
/* 7-49 */
WRITE_UE_MAX (bw, stRPS->NumNegativePics,
sps->max_dec_pic_buffering_minus1[sps->max_sub_layers_minus1]);
/* 7-50 */
WRITE_UE_MAX (bw, stRPS->NumPositivePics,
(sps->max_dec_pic_buffering_minus1[sps->max_sub_layers_minus1] -
stRPS->NumNegativePics));
prev = 0;
for (i = 0; i < stRPS->NumNegativePics; i++) {
WRITE_UE_MAX (bw, prev - stRPS->DeltaPocS0[i] - 1, 32767);
prev = stRPS->DeltaPocS0[i];
/* 7-51 */
WRITE_BITS (bw, stRPS->UsedByCurrPicS0[i], 1);
}
prev = 0;
for (i = 0; i < stRPS->NumPositivePics; i++) {
WRITE_UE_MAX (bw, stRPS->DeltaPocS1[i] - prev - 1, 32767);
prev = stRPS->DeltaPocS1[i];
/* 7-52 */
WRITE_BITS (bw, stRPS->UsedByCurrPicS1[i], 1);
}
*space = TRUE;
return TRUE;
error:
GST_WARNING ("error to write \"ShortTermRefPicSet Parameters\"");
*space = have_space;
return FALSE;
}
static gboolean
_h265_bit_writer_vui_parameters (const GstH265SPS * sps,
GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
const GstH265VUIParams *vui = &sps->vui_params;
GST_DEBUG ("writing \"VUI Parameters\"");
WRITE_BITS (bw, vui->aspect_ratio_info_present_flag, 1);
if (vui->aspect_ratio_info_present_flag) {
WRITE_BITS (bw, vui->aspect_ratio_idc, 8);
if (vui->aspect_ratio_idc == EXTENDED_SAR) {
WRITE_BITS (bw, vui->sar_width, 16);
WRITE_BITS (bw, vui->sar_height, 16);
}
}
WRITE_BITS (bw, vui->overscan_info_present_flag, 1);
if (vui->overscan_info_present_flag)
WRITE_BITS (bw, vui->overscan_appropriate_flag, 1);
WRITE_BITS (bw, vui->video_signal_type_present_flag, 1);
if (vui->video_signal_type_present_flag) {
WRITE_BITS (bw, vui->video_format, 3);
WRITE_BITS (bw, vui->video_full_range_flag, 1);
WRITE_BITS (bw, vui->colour_description_present_flag, 1);
if (vui->colour_description_present_flag) {
WRITE_BITS (bw, vui->colour_primaries, 8);
WRITE_BITS (bw, vui->transfer_characteristics, 8);
WRITE_BITS (bw, vui->matrix_coefficients, 8);
}
}
WRITE_BITS (bw, vui->chroma_loc_info_present_flag, 1);
if (vui->chroma_loc_info_present_flag) {
WRITE_UE_MAX (bw, vui->chroma_sample_loc_type_top_field, 5);
WRITE_UE_MAX (bw, vui->chroma_sample_loc_type_bottom_field, 5);
}
WRITE_BITS (bw, vui->neutral_chroma_indication_flag, 1);
WRITE_BITS (bw, vui->field_seq_flag, 1);
WRITE_BITS (bw, vui->frame_field_info_present_flag, 1);
WRITE_BITS (bw, vui->default_display_window_flag, 1);
if (vui->default_display_window_flag) {
WRITE_UE (bw, vui->def_disp_win_left_offset);
WRITE_UE (bw, vui->def_disp_win_right_offset);
WRITE_UE (bw, vui->def_disp_win_top_offset);
WRITE_UE (bw, vui->def_disp_win_bottom_offset);
}
WRITE_BITS (bw, vui->timing_info_present_flag, 1);
if (vui->timing_info_present_flag) {
if (vui->num_units_in_tick == 0)
GST_WARNING ("num_units_in_tick = 0 (incompliant to H.265 E.2.1).");
WRITE_BITS (bw, vui->num_units_in_tick, 32);
if (vui->time_scale == 0)
GST_WARNING ("time_scale = 0 (incompliant to H.265 E.2.1).");
WRITE_BITS (bw, vui->time_scale, 32);
WRITE_BITS (bw, vui->poc_proportional_to_timing_flag, 1);
if (vui->poc_proportional_to_timing_flag)
WRITE_UE_MAX (bw, vui->num_ticks_poc_diff_one_minus1, G_MAXUINT32 - 1);
WRITE_BITS (bw, vui->hrd_parameters_present_flag, 1);
if (vui->hrd_parameters_present_flag)
if (!_h265_bit_writer_hrd_parameters (&vui->hrd_params, 1,
sps->max_sub_layers_minus1, bw, &have_space))
goto error;
}
WRITE_BITS (bw, vui->bitstream_restriction_flag, 1);
if (vui->bitstream_restriction_flag) {
WRITE_BITS (bw, vui->tiles_fixed_structure_flag, 1);
WRITE_BITS (bw, vui->motion_vectors_over_pic_boundaries_flag, 1);
WRITE_BITS (bw, vui->restricted_ref_pic_lists_flag, 1);
WRITE_UE_MAX (bw, vui->min_spatial_segmentation_idc, 4096);
WRITE_UE_MAX (bw, vui->max_bytes_per_pic_denom, 16);
WRITE_UE_MAX (bw, vui->max_bits_per_min_cu_denom, 16);
WRITE_UE_MAX (bw, vui->log2_max_mv_length_horizontal, 16);
WRITE_UE_MAX (bw, vui->log2_max_mv_length_vertical, 15);
}
*space = TRUE;
return TRUE;
error:
GST_WARNING ("error to write \"VUI Parameters\"");
*space = have_space;
return FALSE;
}
static gboolean
_h265_bit_writer_sps (const GstH265SPS * sps,
GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
guint i;
GST_DEBUG ("writing SPS");
WRITE_BITS (bw, sps->vps->id, 4);
WRITE_BITS (bw, sps->max_sub_layers_minus1, 3);
WRITE_BITS (bw, sps->temporal_id_nesting_flag, 1);
if (!_h265_bit_writer_profile_tier_level (&sps->profile_tier_level,
sps->max_sub_layers_minus1, bw, &have_space))
goto error;
WRITE_UE_MAX (bw, sps->id, GST_H265_MAX_SPS_COUNT - 1);
WRITE_UE_MAX (bw, sps->chroma_format_idc, 3);
if (sps->chroma_format_idc == 3)
WRITE_BITS (bw, sps->separate_colour_plane_flag, 1);
if (sps->pic_width_in_luma_samples < 1)
goto error;
WRITE_UE_MAX (bw, sps->pic_width_in_luma_samples, 16888);
if (sps->pic_height_in_luma_samples < 1)
goto error;
WRITE_UE_MAX (bw, sps->pic_height_in_luma_samples, 16888);
WRITE_BITS (bw, sps->conformance_window_flag, 1);
if (sps->conformance_window_flag) {
WRITE_UE (bw, sps->conf_win_left_offset);
WRITE_UE (bw, sps->conf_win_right_offset);
WRITE_UE (bw, sps->conf_win_top_offset);
WRITE_UE (bw, sps->conf_win_bottom_offset);
}
WRITE_UE_MAX (bw, sps->bit_depth_luma_minus8, 6);
WRITE_UE_MAX (bw, sps->bit_depth_chroma_minus8, 6);
WRITE_UE_MAX (bw, sps->log2_max_pic_order_cnt_lsb_minus4, 12);
WRITE_BITS (bw, sps->sub_layer_ordering_info_present_flag, 1);
for (i = (sps->sub_layer_ordering_info_present_flag ? 0 :
sps->max_sub_layers_minus1); i <= sps->max_sub_layers_minus1; i++) {
WRITE_UE_MAX (bw, sps->max_dec_pic_buffering_minus1[i], 16);
WRITE_UE_MAX (bw, sps->max_num_reorder_pics[i],
sps->max_dec_pic_buffering_minus1[i]);
WRITE_UE (bw, sps->max_latency_increase_plus1[i]);
}
/* The limits are calculted based on the profile_tier_level constraint
* in Annex-A: CtbLog2SizeY = 4 to 6 */
WRITE_UE_MAX (bw, sps->log2_min_luma_coding_block_size_minus3, 3);
WRITE_UE_MAX (bw, sps->log2_diff_max_min_luma_coding_block_size, 6);
WRITE_UE_MAX (bw, sps->log2_min_transform_block_size_minus2, 3);
WRITE_UE_MAX (bw, sps->log2_diff_max_min_transform_block_size, 3);
WRITE_UE_MAX (bw, sps->max_transform_hierarchy_depth_inter, 4);
WRITE_UE_MAX (bw, sps->max_transform_hierarchy_depth_intra, 4);
WRITE_BITS (bw, sps->scaling_list_enabled_flag, 1);
if (sps->scaling_list_enabled_flag) {
WRITE_BITS (bw, sps->scaling_list_data_present_flag, 1);
if (sps->scaling_list_data_present_flag)
if (!_h265_bit_writer_scaling_lists (&sps->scaling_list, bw, &have_space))
goto error;
}
WRITE_BITS (bw, sps->amp_enabled_flag, 1);
WRITE_BITS (bw, sps->sample_adaptive_offset_enabled_flag, 1);
WRITE_BITS (bw, sps->pcm_enabled_flag, 1);
if (sps->pcm_enabled_flag) {
WRITE_BITS (bw, sps->pcm_sample_bit_depth_luma_minus1, 4);
WRITE_BITS (bw, sps->pcm_sample_bit_depth_chroma_minus1, 4);
WRITE_UE_MAX (bw, sps->log2_min_pcm_luma_coding_block_size_minus3, 2);
WRITE_UE_MAX (bw, sps->log2_diff_max_min_pcm_luma_coding_block_size, 2);
WRITE_BITS (bw, sps->pcm_loop_filter_disabled_flag, 1);
}
WRITE_UE_MAX (bw, sps->num_short_term_ref_pic_sets, 64);
for (i = 0; i < sps->num_short_term_ref_pic_sets; i++) {
if (!_h265_bit_writer_short_term_ref_pic_set
(&sps->short_term_ref_pic_set[i], i, sps, bw, &have_space))
goto error;
}
WRITE_BITS (bw, sps->long_term_ref_pics_present_flag, 1);
if (sps->long_term_ref_pics_present_flag) {
WRITE_UE_MAX (bw, sps->num_long_term_ref_pics_sps, 32);
for (i = 0; i < sps->num_long_term_ref_pics_sps; i++) {
WRITE_BITS (bw, sps->lt_ref_pic_poc_lsb_sps[i],
sps->log2_max_pic_order_cnt_lsb_minus4 + 4);
WRITE_BITS (bw, sps->used_by_curr_pic_lt_sps_flag[i], 1);
}
}
WRITE_BITS (bw, sps->temporal_mvp_enabled_flag, 1);
WRITE_BITS (bw, sps->strong_intra_smoothing_enabled_flag, 1);
WRITE_BITS (bw, sps->vui_parameters_present_flag, 1);
if (sps->vui_parameters_present_flag) {
if (!_h265_bit_writer_vui_parameters (sps, bw, &have_space))
goto error;
}
WRITE_BITS (bw, sps->sps_extension_flag, 1);
if (sps->sps_extension_flag) {
WRITE_BITS (bw, sps->sps_range_extension_flag, 1);
WRITE_BITS (bw, sps->sps_multilayer_extension_flag, 1);
WRITE_BITS (bw, sps->sps_3d_extension_flag, 1);
WRITE_BITS (bw, sps->sps_scc_extension_flag, 1);
WRITE_BITS (bw, sps->sps_extension_4bits, 4);
}
if (sps->sps_range_extension_flag) {
WRITE_BITS (bw,
sps->sps_extension_params.transform_skip_rotation_enabled_flag, 1);
WRITE_BITS (bw,
sps->sps_extension_params.transform_skip_context_enabled_flag, 1);
WRITE_BITS (bw, sps->sps_extension_params.implicit_rdpcm_enabled_flag, 1);
WRITE_BITS (bw, sps->sps_extension_params.explicit_rdpcm_enabled_flag, 1);
WRITE_BITS (bw,
sps->sps_extension_params.extended_precision_processing_flag, 1);
WRITE_BITS (bw, sps->sps_extension_params.intra_smoothing_disabled_flag, 1);
WRITE_BITS (bw,
sps->sps_extension_params.high_precision_offsets_enabled_flag, 1);
WRITE_BITS (bw,
sps->sps_extension_params.persistent_rice_adaptation_enabled_flag, 1);
WRITE_BITS (bw,
sps->sps_extension_params.cabac_bypass_alignment_enabled_flag, 1);
}
if (sps->sps_multilayer_extension_flag) {
GST_WARNING ("do not support multilayer extension");
goto error;
}
if (sps->sps_3d_extension_flag) {
GST_WARNING ("do not support 3d extension");
goto error;
}
if (sps->sps_scc_extension_flag) {
const GstH265SPSSccExtensionParams *scc_params =
&sps->sps_scc_extension_params;
WRITE_BITS (bw, scc_params->sps_curr_pic_ref_enabled_flag, 1);
WRITE_BITS (bw, scc_params->palette_mode_enabled_flag, 1);
if (scc_params->palette_mode_enabled_flag) {
WRITE_UE_MAX (bw, scc_params->palette_max_size, 64);
WRITE_UE_MAX (bw, scc_params->delta_palette_max_predictor_size,
128 - scc_params->palette_max_size);
WRITE_BITS (bw,
scc_params->sps_palette_predictor_initializers_present_flag, 1);
if (scc_params->sps_palette_predictor_initializers_present_flag) {
guint comp;
WRITE_UE_MAX (bw,
scc_params->sps_num_palette_predictor_initializer_minus1,
scc_params->palette_max_size +
scc_params->delta_palette_max_predictor_size - 1);
for (comp = 0; comp < (sps->chroma_format_idc == 0 ? 1 : 3); comp++) {
guint num_bits;
guint num =
scc_params->sps_num_palette_predictor_initializer_minus1 + 1;
num_bits = (comp == 0 ? sps->bit_depth_luma_minus8 + 8 :
sps->bit_depth_chroma_minus8 + 8);
for (i = 0; i < num; i++)
WRITE_BITS (bw,
scc_params->sps_palette_predictor_initializer[comp][i],
num_bits);
}
}
}
WRITE_BITS (bw, scc_params->motion_vector_resolution_control_idc, 2);
WRITE_BITS (bw, scc_params->intra_boundary_filtering_disabled_flag, 1);
}
*space = TRUE;
return TRUE;
error:
GST_WARNING ("failed to write SPS");
*space = have_space;
return FALSE;
}
/**
* gst_h265_bit_writer_sps:
* @sps: the sps of #GstH265SPS to write
* @start_code: whether adding the nal start code
* @data: (out): the bit stream generated by the sps
* @size: (inout): the size in bytes of the input and output
*
* Generating the according h265 bit stream by providing the sps.
*
* Returns: a #GstH265BitWriterResult
*
* Since: 1.22
**/
GstH265BitWriterResult
gst_h265_bit_writer_sps (const GstH265SPS * sps, gboolean start_code,
guint8 * data, guint * size)
{
gboolean have_space = TRUE;
GstBitWriter bw;
g_return_val_if_fail (sps != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (sps->vps != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);
gst_bit_writer_init_with_data (&bw, data, *size, FALSE);
if (start_code)
WRITE_BITS (&bw, 0x00000001, 32);
/* NAL unit header */
/* forbidden_zero_bit */
WRITE_BITS (&bw, 0, 1);
/* nal_unit_type */
WRITE_BITS (&bw, GST_H265_NAL_SPS, 6);
/* nuh_layer_id, only support 0 now */
WRITE_BITS (&bw, 0, 6);
/* nuh_temporal_id_plus1, only support 1 now */
WRITE_BITS (&bw, 1, 3);
if (!_h265_bit_writer_sps (sps, &bw, &have_space))
goto error;
/* Add trailings. */
WRITE_BITS (&bw, 1, 1);
if (!gst_bit_writer_align_bytes (&bw, 0)) {
have_space = FALSE;
goto error;
}
*size = gst_bit_writer_get_size (&bw) / 8;
gst_bit_writer_reset (&bw);
return GST_H265_BIT_WRITER_OK;
error:
gst_bit_writer_reset (&bw);
*size = 0;
if (!have_space)
return GST_H265_BIT_WRITER_NO_MORE_SPACE;
return GST_H265_BIT_WRITER_INVALID_DATA;
}
static gboolean
_h265_bit_writer_pps (const GstH265PPS * pps, GstBitWriter * bw,
gboolean * space)
{
gboolean have_space = TRUE;
GST_DEBUG ("writing PPS");
WRITE_UE_MAX (bw, pps->id, GST_H265_MAX_PPS_COUNT - 1);
WRITE_UE_MAX (bw, pps->sps->id, GST_H265_MAX_SPS_COUNT - 1);
WRITE_BITS (bw, pps->dependent_slice_segments_enabled_flag, 1);
WRITE_BITS (bw, pps->output_flag_present_flag, 1);
WRITE_BITS (bw, pps->num_extra_slice_header_bits, 3);
WRITE_BITS (bw, pps->sign_data_hiding_enabled_flag, 1);
WRITE_BITS (bw, pps->cabac_init_present_flag, 1);
WRITE_UE_MAX (bw, pps->num_ref_idx_l0_default_active_minus1, 14);
WRITE_UE_MAX (bw, pps->num_ref_idx_l1_default_active_minus1, 14);
WRITE_SE_RANGE (bw, pps->init_qp_minus26,
-(26 + 6 * pps->sps->bit_depth_luma_minus8), 25);
WRITE_BITS (bw, pps->constrained_intra_pred_flag, 1);
WRITE_BITS (bw, pps->transform_skip_enabled_flag, 1);
WRITE_BITS (bw, pps->cu_qp_delta_enabled_flag, 1);
if (pps->cu_qp_delta_enabled_flag)
WRITE_UE_MAX (bw, pps->diff_cu_qp_delta_depth,
pps->sps->log2_diff_max_min_luma_coding_block_size);
WRITE_SE_RANGE (bw, pps->cb_qp_offset, -12, 12);
WRITE_SE_RANGE (bw, pps->cr_qp_offset, -12, 12);
WRITE_BITS (bw, pps->slice_chroma_qp_offsets_present_flag, 1);
WRITE_BITS (bw, pps->weighted_pred_flag, 1);
WRITE_BITS (bw, pps->weighted_bipred_flag, 1);
WRITE_BITS (bw, pps->transquant_bypass_enabled_flag, 1);
WRITE_BITS (bw, pps->tiles_enabled_flag, 1);
WRITE_BITS (bw, pps->entropy_coding_sync_enabled_flag, 1);
if (pps->tiles_enabled_flag) {
if (pps->num_tile_columns_minus1 + 1 >
G_N_ELEMENTS (pps->column_width_minus1)) {
GST_WARNING ("Invalid \"num_tile_columns_minus1\" %d",
pps->num_tile_columns_minus1);
goto error;
}
if (pps->num_tile_rows_minus1 + 1 > G_N_ELEMENTS (pps->row_height_minus1)) {
GST_WARNING ("Invalid \"num_tile_rows_minus1\" %d",
pps->num_tile_rows_minus1);
goto error;
}
WRITE_UE_MAX (bw, pps->num_tile_columns_minus1, pps->PicWidthInCtbsY - 1);
WRITE_UE_MAX (bw, pps->num_tile_rows_minus1, pps->PicHeightInCtbsY - 1);
WRITE_BITS (bw, pps->uniform_spacing_flag, 1);
/* 6.5.1, 6-4, 6-5, 7.4.3.3.1 */
if (!pps->uniform_spacing_flag) {
guint i;
for (i = 0; i < pps->num_tile_columns_minus1; i++)
WRITE_UE (bw, pps->column_width_minus1[i]);
for (i = 0; i < pps->num_tile_rows_minus1; i++)
WRITE_UE (bw, pps->row_height_minus1[i]);
}
WRITE_BITS (bw, pps->loop_filter_across_tiles_enabled_flag, 1);
}
WRITE_BITS (bw, pps->loop_filter_across_slices_enabled_flag, 1);
WRITE_BITS (bw, pps->deblocking_filter_control_present_flag, 1);
if (pps->deblocking_filter_control_present_flag) {
WRITE_BITS (bw, pps->deblocking_filter_override_enabled_flag, 1);
WRITE_BITS (bw, pps->deblocking_filter_disabled_flag, 1);
if (!pps->deblocking_filter_disabled_flag) {
WRITE_SE_RANGE (bw, pps->beta_offset_div2, -6, 6);
WRITE_SE_RANGE (bw, pps->tc_offset_div2, -6, +6);
}
}
WRITE_BITS (bw, pps->scaling_list_data_present_flag, 1);
if (pps->scaling_list_data_present_flag)
if (!_h265_bit_writer_scaling_lists (&pps->scaling_list, bw, &have_space))
goto error;
WRITE_BITS (bw, pps->lists_modification_present_flag, 1);
WRITE_UE_MAX (bw, pps->log2_parallel_merge_level_minus2, 4);
/* TODO: slice_segment_header */
if (pps->slice_segment_header_extension_present_flag) {
GST_WARNING
("slice_segment_header_extension_present_flag is not supported");
goto error;
}
WRITE_BITS (bw, pps->slice_segment_header_extension_present_flag, 1);
WRITE_BITS (bw, pps->pps_extension_flag, 1);
if (pps->pps_extension_flag) {
WRITE_BITS (bw, pps->pps_range_extension_flag, 1);
WRITE_BITS (bw, pps->pps_multilayer_extension_flag, 1);
WRITE_BITS (bw, pps->pps_3d_extension_flag, 1);
WRITE_BITS (bw, pps->pps_scc_extension_flag, 1);
WRITE_BITS (bw, pps->pps_extension_4bits, 4);
}
if (pps->pps_range_extension_flag) {
guint i;
guint32 MaxBitDepthY, MaxBitDepthC;
if (pps->transform_skip_enabled_flag)
WRITE_UE (bw,
pps->pps_extension_params.log2_max_transform_skip_block_size_minus2);
WRITE_BITS (bw,
pps->pps_extension_params.cross_component_prediction_enabled_flag, 1);
WRITE_BITS (bw,
pps->pps_extension_params.chroma_qp_offset_list_enabled_flag, 1);
if (pps->pps_extension_params.chroma_qp_offset_list_enabled_flag) {
WRITE_UE_MAX (bw,
pps->pps_extension_params.diff_cu_chroma_qp_offset_depth,
pps->sps->log2_diff_max_min_luma_coding_block_size);
WRITE_UE_MAX (bw,
pps->pps_extension_params.chroma_qp_offset_list_len_minus1, 5);
for (i = 0;
i <= pps->pps_extension_params.chroma_qp_offset_list_len_minus1;
i++) {
WRITE_SE_RANGE (bw, pps->pps_extension_params.cb_qp_offset_list[i],
-12, 12);
WRITE_SE_RANGE (bw, pps->pps_extension_params.cr_qp_offset_list[i],
-12, 12);
}
}
MaxBitDepthY = pps->sps->bit_depth_luma_minus8 > 2 ?
pps->sps->bit_depth_luma_minus8 - 2 : 0;
MaxBitDepthC = pps->sps->bit_depth_chroma_minus8 > 2 ?
pps->sps->bit_depth_chroma_minus8 - 2 : 0;
WRITE_UE_MAX (bw, pps->pps_extension_params.log2_sao_offset_scale_luma,
MaxBitDepthY);
WRITE_UE_MAX (bw, pps->pps_extension_params.log2_sao_offset_scale_chroma,
MaxBitDepthC);
}
if (pps->pps_multilayer_extension_flag) {
GST_WARNING ("do not support multilayer extension");
goto error;
}
if (pps->pps_3d_extension_flag) {
GST_WARNING ("do not support 3d extension");
goto error;
}
if (pps->pps_scc_extension_flag) {
const GstH265PPSSccExtensionParams *pps_scc =
&pps->pps_scc_extension_params;
WRITE_BITS (bw, pps_scc->pps_curr_pic_ref_enabled_flag, 1);
WRITE_BITS (bw,
pps_scc->residual_adaptive_colour_transform_enabled_flag, 1);
if (pps_scc->residual_adaptive_colour_transform_enabled_flag) {
WRITE_BITS (bw, pps_scc->pps_slice_act_qp_offsets_present_flag, 1);
WRITE_SE_RANGE (bw, (gint8) pps_scc->pps_act_y_qp_offset_plus5, -7, 17);
WRITE_SE_RANGE (bw, (gint8) pps_scc->pps_act_cb_qp_offset_plus5, -7, 17);
WRITE_SE_RANGE (bw, (gint8) pps_scc->pps_act_cr_qp_offset_plus3, -9, 15);
}
WRITE_BITS (bw,
pps_scc->pps_palette_predictor_initializers_present_flag, 1);
if (pps_scc->pps_palette_predictor_initializers_present_flag) {
guint i;
WRITE_UE_MAX (bw,
pps_scc->pps_num_palette_predictor_initializer,
pps->sps->sps_scc_extension_params.palette_max_size +
pps->sps->sps_scc_extension_params.delta_palette_max_predictor_size);
if (pps_scc->pps_num_palette_predictor_initializer > 0) {
guint comp;
WRITE_BITS (bw, pps_scc->monochrome_palette_flag, 1);
/* It is a requirement of bitstream conformance that the value of
luma_bit_depth_entry_minus8 shall be equal to the value of
bit_depth_luma_minus8 */
WRITE_UE_MAX (bw, pps_scc->luma_bit_depth_entry_minus8,
pps->sps->bit_depth_luma_minus8);
if (!pps_scc->monochrome_palette_flag) {
/* It is a requirement of bitstream conformance that the value
of chroma_bit_depth_entry_minus8 shall be equal to the value
of bit_depth_chroma_minus8. */
WRITE_UE_MAX (bw, pps_scc->chroma_bit_depth_entry_minus8,
pps->sps->bit_depth_chroma_minus8);
}
for (comp = 0; comp < (pps_scc->monochrome_palette_flag ? 1 : 3);
comp++) {
guint num_bits;
guint num = pps_scc->pps_num_palette_predictor_initializer;
num_bits = (comp == 0 ?
pps_scc->luma_bit_depth_entry_minus8 + 8 :
pps_scc->chroma_bit_depth_entry_minus8 + 8);
for (i = 0; i < num; i++)
WRITE_BITS (bw,
pps_scc->pps_palette_predictor_initializer[comp][i], num_bits);
}
}
}
}
return TRUE;
error:
GST_WARNING ("failed to write PPS");
return FALSE;
}
/**
* gst_h265_bit_writer_pps:
* @pps: the pps of #GstH265PPS to write
* @start_code: whether adding the nal start code
* @data: (out): the bit stream generated by the pps
* @size: (inout): the size in bytes of the input and output
*
* Generating the according h265 bit stream by providing the pps.
*
* Returns: a #GstH265BitWriterResult
*
* Since: 1.22
**/
GstH265BitWriterResult
gst_h265_bit_writer_pps (const GstH265PPS * pps, gboolean start_code,
guint8 * data, guint * size)
{
gboolean have_space = TRUE;
GstBitWriter bw;
g_return_val_if_fail (pps != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (pps->sps != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);
gst_bit_writer_init_with_data (&bw, data, *size, FALSE);
if (start_code)
WRITE_BITS (&bw, 0x00000001, 32);
/* NAL unit header */
/* forbidden_zero_bit */
WRITE_BITS (&bw, 0, 1);
/* nal_unit_type */
WRITE_BITS (&bw, GST_H265_NAL_PPS, 6);
/* nuh_layer_id, only support 0 now */
WRITE_BITS (&bw, 0, 6);
/* nuh_temporal_id_plus1, only support 1 now */
WRITE_BITS (&bw, 1, 3);
if (!_h265_bit_writer_pps (pps, &bw, &have_space))
goto error;
/* Add trailings. */
WRITE_BITS (&bw, 1, 1);
if (!gst_bit_writer_align_bytes (&bw, 0)) {
have_space = FALSE;
goto error;
}
*size = gst_bit_writer_get_size (&bw) / 8;
gst_bit_writer_reset (&bw);
return GST_H265_BIT_WRITER_OK;
error:
gst_bit_writer_reset (&bw);
*size = 0;
if (!have_space)
return GST_H265_BIT_WRITER_NO_MORE_SPACE;
return GST_H265_BIT_WRITER_INVALID_DATA;
}
static gboolean
_h265_slice_bit_writer_ref_pic_list_modification (const GstH265SliceHdr *
slice, gint NumPocTotalCurr, GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
guint i;
const GstH265RefPicListModification *rpl_mod =
&slice->ref_pic_list_modification;
const guint n = ceil_log2 (NumPocTotalCurr);
WRITE_BITS (bw, rpl_mod->ref_pic_list_modification_flag_l0, 1);
if (rpl_mod->ref_pic_list_modification_flag_l0) {
for (i = 0; i <= slice->num_ref_idx_l0_active_minus1; i++) {
WRITE_BITS (bw, rpl_mod->list_entry_l0[i], n);
}
}
if (GST_H265_IS_B_SLICE (slice)) {
WRITE_BITS (bw, rpl_mod->ref_pic_list_modification_flag_l1, 1);
if (rpl_mod->ref_pic_list_modification_flag_l1)
for (i = 0; i <= slice->num_ref_idx_l1_active_minus1; i++) {
WRITE_BITS (bw, rpl_mod->list_entry_l1[i], n);
}
}
*space = TRUE;
return TRUE;
error:
GST_WARNING ("error to write \"Reference picture list modifications\"");
*space = have_space;
return FALSE;
}
static gboolean
_h265_slice_bit_writer_pred_weight_table (const GstH265SliceHdr * slice,
GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
const GstH265PredWeightTable *p;
const GstH265PPS *pps = slice->pps;
const GstH265SPS *sps = pps->sps;
gint i, j;
GST_DEBUG ("writing \"Prediction weight table\"");
p = &slice->pred_weight_table;
WRITE_UE_MAX (bw, p->luma_log2_weight_denom, 7);
if (sps->chroma_format_idc != 0) {
WRITE_SE_RANGE (bw, p->delta_chroma_log2_weight_denom,
(0 - p->luma_log2_weight_denom), (7 - p->luma_log2_weight_denom));
}
for (i = 0; i <= slice->num_ref_idx_l0_active_minus1; i++)
WRITE_BITS (bw, p->luma_weight_l0_flag[i], 1);
if (sps->chroma_format_idc != 0)
for (i = 0; i <= slice->num_ref_idx_l0_active_minus1; i++)
WRITE_BITS (bw, p->chroma_weight_l0_flag[i], 1);
for (i = 0; i <= slice->num_ref_idx_l0_active_minus1; i++) {
if (p->luma_weight_l0_flag[i]) {
WRITE_SE_RANGE (bw, p->delta_luma_weight_l0[i], -128, 127);
WRITE_SE_RANGE (bw, p->luma_offset_l0[i], -128, 127);
}
if (p->chroma_weight_l0_flag[i])
for (j = 0; j < 2; j++) {
WRITE_SE_RANGE (bw, p->delta_chroma_weight_l0[i][j], -128, 127);
WRITE_SE_RANGE (bw, p->delta_chroma_offset_l0[i][j], -512, 511);
}
}
if (GST_H265_IS_B_SLICE (slice)) {
for (i = 0; i <= slice->num_ref_idx_l1_active_minus1; i++)
WRITE_BITS (bw, p->luma_weight_l1_flag[i], 1);
if (sps->chroma_format_idc != 0)
for (i = 0; i <= slice->num_ref_idx_l1_active_minus1; i++)
WRITE_BITS (bw, p->chroma_weight_l1_flag[i], 1);
for (i = 0; i <= slice->num_ref_idx_l1_active_minus1; i++) {
if (p->luma_weight_l1_flag[i]) {
WRITE_SE_RANGE (bw, p->delta_luma_weight_l1[i], -128, 127);
WRITE_SE_RANGE (bw, p->luma_offset_l1[i], -128, 127);
}
if (p->chroma_weight_l1_flag[i])
for (j = 0; j < 2; j++) {
WRITE_SE_RANGE (bw, p->delta_chroma_weight_l1[i][j], -128, 127);
WRITE_SE_RANGE (bw, p->delta_chroma_offset_l1[i][j], -512, 511);
}
}
}
*space = TRUE;
return TRUE;
error:
GST_WARNING ("error to write \"Prediction weight table\"");
*space = have_space;
return FALSE;
}
static gboolean
_h265_bit_writer_slice_header (const GstH265SliceHdr * slice,
guint32 nal_type, GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
guint i;
const GstH265SPS *sps = slice->pps->sps;
const GstH265PPSSccExtensionParams *pps_scc_extension_params =
&slice->pps->pps_scc_extension_params;
const GstH265PPSExtensionParams *pps_extension_params =
&slice->pps->pps_extension_params;
GST_DEBUG ("writing slice header");
WRITE_BITS (bw, slice->first_slice_segment_in_pic_flag, 1);
if (GST_H265_IS_NAL_TYPE_IRAP (nal_type))
WRITE_BITS (bw, slice->no_output_of_prior_pics_flag, 1);
WRITE_UE_MAX (bw, slice->pps->id, GST_H265_MAX_PPS_COUNT - 1);
if (!slice->first_slice_segment_in_pic_flag) {
guint32 PicSizeInCtbsY;
guint32 PicWidthInCtbsY;
guint32 PicHeightInCtbsY;
guint32 CtbSizeY, MinCbLog2SizeY, CtbLog2SizeY;
guint n;
/* We can not directly use slice->pps->PicWidthInCtbsY/PicHeightInCtbsY,
they are calculated value when parsing but may not have value here. */
MinCbLog2SizeY = sps->log2_min_luma_coding_block_size_minus3 + 3;
CtbLog2SizeY =
MinCbLog2SizeY + sps->log2_diff_max_min_luma_coding_block_size;
CtbSizeY = 1 << CtbLog2SizeY;
PicHeightInCtbsY =
ceil ((gdouble) sps->pic_height_in_luma_samples / (gdouble) CtbSizeY);
PicWidthInCtbsY =
ceil ((gdouble) sps->pic_width_in_luma_samples / (gdouble) CtbSizeY);
PicSizeInCtbsY = PicWidthInCtbsY * PicHeightInCtbsY;
n = ceil_log2 (PicSizeInCtbsY);
if (slice->pps->dependent_slice_segments_enabled_flag)
WRITE_BITS (bw, slice->dependent_slice_segment_flag, 1);
/* sice_segment_address parsing */
WRITE_BITS (bw, slice->segment_address, n);
}
if (!slice->dependent_slice_segment_flag) {
for (i = 0; i < slice->pps->num_extra_slice_header_bits; i++) {
/* slice_reserved_flag */
WRITE_BITS (bw, 0, 1);
}
WRITE_UE_MAX (bw, slice->type, 63);
if (slice->pps->output_flag_present_flag)
WRITE_BITS (bw, slice->pic_output_flag, 1);
if (sps->separate_colour_plane_flag == 1)
WRITE_BITS (bw, slice->colour_plane_id, 2);
if (!GST_H265_IS_NAL_TYPE_IDR (nal_type)) {
WRITE_BITS (bw, slice->pic_order_cnt_lsb,
(sps->log2_max_pic_order_cnt_lsb_minus4 + 4));
WRITE_BITS (bw, slice->short_term_ref_pic_set_sps_flag, 1);
if (!slice->short_term_ref_pic_set_sps_flag) {
if (!_h265_bit_writer_short_term_ref_pic_set
(&slice->short_term_ref_pic_sets, sps->num_short_term_ref_pic_sets,
slice->pps->sps, bw, &have_space))
goto error;
} else if (sps->num_short_term_ref_pic_sets > 1) {
const guint n = ceil_log2 (sps->num_short_term_ref_pic_sets);
if (slice->short_term_ref_pic_set_idx >
sps->num_short_term_ref_pic_sets - 1)
goto error;
WRITE_BITS (bw, slice->short_term_ref_pic_set_idx, n);
}
if (sps->long_term_ref_pics_present_flag) {
guint32 limit;
if (sps->num_long_term_ref_pics_sps > 0)
WRITE_UE_MAX (bw, slice->num_long_term_sps,
sps->num_long_term_ref_pics_sps);
WRITE_UE_MAX (bw, slice->num_long_term_pics, 16);
limit = slice->num_long_term_sps + slice->num_long_term_pics;
for (i = 0; i < limit; i++) {
if (i < slice->num_long_term_sps) {
if (sps->num_long_term_ref_pics_sps > 1) {
const guint n = ceil_log2 (sps->num_long_term_ref_pics_sps);
WRITE_BITS (bw, slice->lt_idx_sps[i], n);
}
} else {
WRITE_BITS (bw, slice->poc_lsb_lt[i],
(sps->log2_max_pic_order_cnt_lsb_minus4 + 4));
WRITE_BITS (bw, slice->used_by_curr_pic_lt_flag[i], 1);
}
WRITE_BITS (bw, slice->delta_poc_msb_present_flag[i], 1);
if (slice->delta_poc_msb_present_flag[i])
WRITE_UE (bw, slice->delta_poc_msb_cycle_lt[i]);
}
}
if (sps->temporal_mvp_enabled_flag)
WRITE_BITS (bw, slice->temporal_mvp_enabled_flag, 1);
}
if (sps->sample_adaptive_offset_enabled_flag) {
gboolean ChromaArrayType =
sps->separate_colour_plane_flag == 0 ? sps->chroma_format_idc : 0;
WRITE_BITS (bw, slice->sao_luma_flag, 1);
if (ChromaArrayType)
WRITE_BITS (bw, slice->sao_chroma_flag, 1);
}
if (GST_H265_IS_B_SLICE (slice) || GST_H265_IS_P_SLICE (slice)) {
WRITE_BITS (bw, slice->num_ref_idx_active_override_flag, 1);
if (slice->num_ref_idx_active_override_flag) {
WRITE_UE_MAX (bw, slice->num_ref_idx_l0_active_minus1, 14);
if (GST_H265_IS_B_SLICE (slice))
WRITE_UE_MAX (bw, slice->num_ref_idx_l1_active_minus1, 14);
}
if (slice->pps->lists_modification_present_flag
&& slice->NumPocTotalCurr > 1) {
if (!_h265_slice_bit_writer_ref_pic_list_modification (slice,
slice->NumPocTotalCurr, bw, &have_space))
goto error;
}
if (GST_H265_IS_B_SLICE (slice))
WRITE_BITS (bw, slice->mvd_l1_zero_flag, 1);
if (slice->pps->cabac_init_present_flag)
WRITE_BITS (bw, slice->cabac_init_flag, 1);
if (slice->temporal_mvp_enabled_flag) {
if (GST_H265_IS_B_SLICE (slice))
WRITE_BITS (bw, slice->collocated_from_l0_flag, 1);
if ((slice->collocated_from_l0_flag
&& slice->num_ref_idx_l0_active_minus1 > 0)
|| (!slice->collocated_from_l0_flag
&& slice->num_ref_idx_l1_active_minus1 > 0)) {
if ((GST_H265_IS_P_SLICE (slice))
|| ((GST_H265_IS_B_SLICE (slice))
&& (slice->collocated_from_l0_flag))) {
WRITE_UE_MAX (bw, slice->collocated_ref_idx,
slice->num_ref_idx_l0_active_minus1);
} else if ((GST_H265_IS_B_SLICE (slice))
&& (!slice->collocated_from_l0_flag)) {
WRITE_UE_MAX (bw, slice->collocated_ref_idx,
slice->num_ref_idx_l1_active_minus1);
}
}
}
if ((slice->pps->weighted_pred_flag && GST_H265_IS_P_SLICE (slice)) ||
(slice->pps->weighted_bipred_flag && GST_H265_IS_B_SLICE (slice)))
if (!_h265_slice_bit_writer_pred_weight_table (slice, bw, &have_space))
goto error;
WRITE_UE_MAX (bw, slice->five_minus_max_num_merge_cand, 4);
if (sps->sps_scc_extension_params.motion_vector_resolution_control_idc
== 2)
WRITE_BITS (bw, slice->use_integer_mv_flag, 1);
}
WRITE_SE_RANGE (bw, slice->qp_delta, -87, 77);
if (slice->pps->slice_chroma_qp_offsets_present_flag) {
WRITE_SE_RANGE (bw, slice->cb_qp_offset, -12, 12);
WRITE_SE_RANGE (bw, slice->cr_qp_offset, -12, 12);
}
if (pps_scc_extension_params->pps_slice_act_qp_offsets_present_flag) {
WRITE_SE_RANGE (bw, slice->slice_act_y_qp_offset, -12, 12);
WRITE_SE_RANGE (bw, slice->slice_act_cb_qp_offset, -12, 12);
WRITE_SE_RANGE (bw, slice->slice_act_cr_qp_offset, -12, 12);
}
if (pps_extension_params->chroma_qp_offset_list_enabled_flag)
WRITE_BITS (bw, slice->cu_chroma_qp_offset_enabled_flag, 1);
if (slice->pps->deblocking_filter_override_enabled_flag)
WRITE_BITS (bw, slice->deblocking_filter_override_flag, 1);
if (slice->deblocking_filter_override_flag) {
WRITE_BITS (bw, slice->deblocking_filter_disabled_flag, 1);
if (!slice->deblocking_filter_disabled_flag) {
WRITE_SE_RANGE (bw, slice->beta_offset_div2, -6, 6);
WRITE_SE_RANGE (bw, slice->tc_offset_div2, -6, 6);
}
}
if (slice->pps->loop_filter_across_slices_enabled_flag &&
(slice->sao_luma_flag || slice->sao_chroma_flag ||
!slice->deblocking_filter_disabled_flag))
WRITE_BITS (bw, slice->loop_filter_across_slices_enabled_flag, 1);
}
if (slice->pps->tiles_enabled_flag
|| slice->pps->entropy_coding_sync_enabled_flag) {
guint32 offset_max;
if (!slice->pps->tiles_enabled_flag
&& slice->pps->entropy_coding_sync_enabled_flag) {
offset_max = slice->pps->PicHeightInCtbsY - 1;
} else if (slice->pps->tiles_enabled_flag
&& !slice->pps->entropy_coding_sync_enabled_flag) {
offset_max =
(slice->pps->num_tile_columns_minus1 +
1) * (slice->pps->num_tile_rows_minus1 + 1) - 1;
} else {
offset_max =
(slice->pps->num_tile_columns_minus1 +
1) * slice->pps->PicHeightInCtbsY - 1;
}
WRITE_UE_MAX (bw, slice->num_entry_point_offsets, offset_max);
if (slice->num_entry_point_offsets > 0) {
WRITE_UE_MAX (bw, slice->offset_len_minus1, 31);
for (i = 0; i < slice->num_entry_point_offsets; i++)
WRITE_BITS (bw, slice->entry_point_offset_minus1[i],
(slice->offset_len_minus1 + 1));
}
}
/* TODO */
if (slice->pps->slice_segment_header_extension_present_flag) {
GST_WARNING
("slice_segment_header_extension_present_flag is not supported");
goto error;
}
*space = TRUE;
return TRUE;
error:
GST_WARNING ("error to write slice header");
*space = have_space;
return FALSE;
}
/**
* gst_h265_bit_writer_slice_hdr:
* @slice: the slice header of #GstH265SliceHdr to write
* @start_code: whether adding the nal start code
* @nal_type: the slice's nal type of #GstH265NalUnitType
* @data: (out): the bit stream generated by the slice header
* @size: (inout): the size in bytes of the input and output
*
* Generating the according h265 bit stream by providing the slice header.
*
* Returns: a #GstH265BitWriterResult
*
* Since: 1.22
**/
GstH265BitWriterResult
gst_h265_bit_writer_slice_hdr (const GstH265SliceHdr * slice,
gboolean start_code, guint32 nal_type, guint8 * data, guint * size)
{
gboolean have_space = TRUE;
GstBitWriter bw;
g_return_val_if_fail (slice != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (slice->pps != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (slice->pps->sps != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (nal_type >= GST_H265_NAL_SLICE_TRAIL_N &&
nal_type <= GST_H265_NAL_SLICE_CRA_NUT, GST_H265_BIT_WRITER_ERROR);
gst_bit_writer_init_with_data (&bw, data, *size, FALSE);
if (start_code)
WRITE_BITS (&bw, 0x00000001, 32);
/* NAL unit header */
/* forbidden_zero_bit */
WRITE_BITS (&bw, 0, 1);
/* nal_unit_type */
WRITE_BITS (&bw, nal_type, 6);
/* nuh_layer_id, only support 0 now */
WRITE_BITS (&bw, 0, 6);
/* nuh_temporal_id_plus1, only support 1 now */
WRITE_BITS (&bw, 1, 3);
if (!_h265_bit_writer_slice_header (slice, nal_type, &bw, &have_space))
goto error;
/* Add trailings. */
WRITE_BITS (&bw, 1, 1);
if (!gst_bit_writer_align_bytes (&bw, 0)) {
have_space = FALSE;
goto error;
}
*size = gst_bit_writer_get_size (&bw) / 8;
gst_bit_writer_reset (&bw);
return GST_H265_BIT_WRITER_OK;
error:
gst_bit_writer_reset (&bw);
*size = 0;
if (!have_space)
return GST_H265_BIT_WRITER_NO_MORE_SPACE;
return GST_H265_BIT_WRITER_INVALID_DATA;
}
static gboolean
_h265_bit_writer_sei_registered_user_data (const GstH265RegisteredUserData *
rud, GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
GST_DEBUG ("Writing \"Registered user data\"");
WRITE_BITS (bw, rud->country_code, 8);
if (rud->country_code == 0xff)
WRITE_BITS (bw, rud->country_code_extension, 8);
WRITE_BYTES (bw, rud->data, rud->size);
*space = TRUE;
return TRUE;
error:
GST_DEBUG ("Failed to write \"Registered user data\"");
*space = have_space;
return FALSE;
}
static gboolean
_h265_bit_writer_sei_time_code (const GstH265TimeCode * tc,
GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
gint i;
GST_DEBUG ("Wrtiting \"Time code\"");
WRITE_BITS (bw, tc->num_clock_ts, 2);
for (i = 0; i < tc->num_clock_ts; i++) {
WRITE_BITS (bw, tc->clock_timestamp_flag[i], 1);
if (tc->clock_timestamp_flag[i]) {
WRITE_BITS (bw, tc->units_field_based_flag[i], 1);
WRITE_BITS (bw, tc->counting_type[i], 5);
WRITE_BITS (bw, tc->full_timestamp_flag[i], 1);
WRITE_BITS (bw, tc->discontinuity_flag[i], 1);
WRITE_BITS (bw, tc->cnt_dropped_flag[i], 1);
WRITE_BITS (bw, tc->n_frames[i], 9);
if (tc->full_timestamp_flag[i]) {
WRITE_BITS (bw, tc->seconds_value[i], 6);
WRITE_BITS (bw, tc->minutes_value[i], 6);
WRITE_BITS (bw, tc->hours_value[i], 5);
} else {
WRITE_BITS (bw, tc->seconds_flag[i], 1);
if (tc->seconds_flag[i]) {
WRITE_BITS (bw, tc->seconds_value[i], 6);
WRITE_BITS (bw, tc->minutes_flag[i], 1);
if (tc->minutes_flag[i]) {
WRITE_BITS (bw, tc->minutes_value[i], 6);
WRITE_BITS (bw, tc->hours_flag[i], 1);
if (tc->hours_flag[i]) {
WRITE_BITS (bw, tc->hours_value[i], 5);
}
}
}
}
}
WRITE_BITS (bw, tc->time_offset_length[i], 5);
if (tc->time_offset_length[i] > 0)
WRITE_BITS (bw, tc->time_offset_value[i], tc->time_offset_length[i]);
}
*space = TRUE;
return TRUE;
error:
GST_WARNING ("Failed to write \"Time code\"");
*space = have_space;
return FALSE;
}
static gboolean
_h265_bit_writer_sei_mastering_display_colour_volume (const
GstH265MasteringDisplayColourVolume * mdcv, GstBitWriter * bw,
gboolean * space)
{
gboolean have_space = TRUE;
gint i;
GST_DEBUG ("Wrtiting \"Mastering display colour volume\"");
for (i = 0; i < 3; i++) {
WRITE_BITS (bw, mdcv->display_primaries_x[i], 16);
WRITE_BITS (bw, mdcv->display_primaries_y[i], 16);
}
WRITE_BITS (bw, mdcv->white_point_x, 16);
WRITE_BITS (bw, mdcv->white_point_y, 16);
WRITE_BITS (bw, mdcv->max_display_mastering_luminance, 32);
WRITE_BITS (bw, mdcv->min_display_mastering_luminance, 32);
*space = TRUE;
return TRUE;
error:
GST_WARNING ("Failed to write \"Mastering display colour volume\"");
*space = have_space;
return FALSE;
}
static gboolean
_h265_bit_writer_sei_content_light_level_info (const
GstH265ContentLightLevel * cll, GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
GST_DEBUG ("Writing \"Content light level\"");
WRITE_BITS (bw, cll->max_content_light_level, 16);
WRITE_BITS (bw, cll->max_pic_average_light_level, 16);
*space = TRUE;
return TRUE;
error:
GST_WARNING ("Failed to write \"Content light level\"");
*space = have_space;
return FALSE;
}
static gboolean
_h265_bit_writer_sei_message (const GstH265SEIMessage * msg,
GstBitWriter * bw, gboolean * space)
{
gboolean have_space = TRUE;
GST_DEBUG ("writing SEI message");
switch (msg->payloadType) {
case GST_H265_SEI_REGISTERED_USER_DATA:
if (!_h265_bit_writer_sei_registered_user_data
(&msg->payload.registered_user_data, bw, &have_space))
goto error;
break;
case GST_H265_SEI_TIME_CODE:
if (!_h265_bit_writer_sei_time_code
(&msg->payload.time_code, bw, &have_space))
goto error;
break;
case GST_H265_SEI_MASTERING_DISPLAY_COLOUR_VOLUME:
if (!_h265_bit_writer_sei_mastering_display_colour_volume
(&msg->payload.mastering_display_colour_volume, bw, &have_space))
goto error;
break;
case GST_H265_SEI_CONTENT_LIGHT_LEVEL:
if (!_h265_bit_writer_sei_content_light_level_info
(&msg->payload.content_light_level, bw, &have_space))
goto error;
break;
default:
break;
}
/* Add trailings. */
WRITE_BITS (bw, 1, 1);
gst_bit_writer_align_bytes_unchecked (bw, 0);
*space = TRUE;
return TRUE;
error:
GST_WARNING ("error to write SEI message");
*space = have_space;
return FALSE;
}
/**
* gst_h265_bit_writer_sei:
* @sei_messages: An array of #GstH265SEIMessage to write
* @start_code: whether adding the nal start code
* @data: (out): the bit stream generated by the sei messages
* @size: (inout): the size in bytes of the input and output
*
* Generating the according h265 bit stream by providing sei messages.
*
* Returns: a #GstH265BitWriterResult
*
* Since: 1.22
**/
GstH265BitWriterResult
gst_h265_bit_writer_sei (GArray * sei_messages,
GstH265NalUnitType nal_type, gboolean start_code, guint8 * data,
guint * size)
{
gboolean have_space = TRUE;
GstBitWriter bw;
GstH265SEIMessage *sei;
gboolean have_written_data = FALSE;
guint i;
g_return_val_if_fail (sei_messages != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (nal_type == GST_H265_NAL_PREFIX_SEI
|| nal_type == GST_H265_NAL_SUFFIX_SEI, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);
if (nal_type == GST_H265_NAL_PREFIX_SEI) {
GST_WARNING ("prefix sei is not supported");
return GST_H265_BIT_WRITER_ERROR;
}
gst_bit_writer_init_with_data (&bw, data, *size, FALSE);
if (start_code)
WRITE_BITS (&bw, 0x00000001, 32);
/* NAL unit header */
/* forbidden_zero_bit */
WRITE_BITS (&bw, 0, 1);
/* nal_unit_type */
WRITE_BITS (&bw, nal_type, 6);
/* nuh_layer_id, only support 0 now */
WRITE_BITS (&bw, 0, 6);
/* nuh_temporal_id_plus1, only support 1 now */
WRITE_BITS (&bw, 1, 3);
for (i = 0; i < sei_messages->len; i++) {
guint32 payload_size_data;
guint32 payload_type_data;
gst_bit_writer_init (&bw);
sei = &g_array_index (sei_messages, GstH265SEIMessage, i);
if (!_h265_bit_writer_sei_message (sei, &bw, &have_space))
goto error;
if (gst_bit_writer_get_size (&bw) == 0) {
GST_FIXME ("Unsupported SEI type %d", sei->payloadType);
continue;
}
have_written_data = TRUE;
g_assert (gst_bit_writer_get_size (&bw) % 8 == 0);
payload_size_data = (gst_bit_writer_get_size (&bw) + 7) / 8;
payload_type_data = sei->payloadType;
/* write payload type bytes */
while (payload_type_data >= 0xff) {
WRITE_BITS (&bw, 0xff, 8);
payload_type_data -= -0xff;
}
WRITE_BITS (&bw, payload_type_data, 8);
/* write payload size bytes */
while (payload_size_data >= 0xff) {
WRITE_BITS (&bw, 0xff, 8);
payload_size_data -= -0xff;
}
WRITE_BITS (&bw, payload_size_data, 8);
if (gst_bit_writer_get_size (&bw) / 8)
WRITE_BYTES (&bw, gst_bit_writer_get_data (&bw),
gst_bit_writer_get_size (&bw) / 8);
gst_bit_writer_reset (&bw);
}
if (!have_written_data) {
GST_WARNING ("No written sei data");
goto error;
}
/* Add trailings. */
WRITE_BITS (&bw, 1, 1);
if (!gst_bit_writer_align_bytes (&bw, 0)) {
have_space = FALSE;
goto error;
}
*size = gst_bit_writer_get_size (&bw) / 8;
gst_bit_writer_reset (&bw);
return GST_H265_BIT_WRITER_OK;
error:
gst_bit_writer_reset (&bw);
*size = 0;
if (!have_space)
return GST_H265_BIT_WRITER_NO_MORE_SPACE;
return GST_H265_BIT_WRITER_INVALID_DATA;
}
/**
* gst_h265_bit_writer_aud:
* @pic_type: indicate the possible slice types list just
* as the H265 spec Table 7-2 defines
* @start_code: whether adding the nal start code
* @data: (out): the bit stream generated by the aud
* @size: (inout): the size in bytes of the input and output
*
* Generating the according h265 bit stream of an aud.
*
* Returns: a #GstH265BitWriterResult
*
* Since: 1.22
**/
GstH265BitWriterResult
gst_h265_bit_writer_aud (guint8 pic_type, gboolean start_code,
guint8 * data, guint * size)
{
gboolean have_space = TRUE;
GstBitWriter bw;
g_return_val_if_fail (pic_type <= 2, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (data != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (size != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (*size > 0, GST_H265_BIT_WRITER_ERROR);
gst_bit_writer_init_with_data (&bw, data, *size, FALSE);
if (start_code)
WRITE_BITS (&bw, 0x00000001, 32);
/* NAL unit header */
/* forbidden_zero_bit */
WRITE_BITS (&bw, 0, 1);
/* nal_unit_type */
WRITE_BITS (&bw, GST_H265_NAL_AUD, 6);
/* nuh_layer_id, only support 0 now */
WRITE_BITS (&bw, 0, 6);
/* nuh_temporal_id_plus1, only support 1 now */
WRITE_BITS (&bw, 1, 3);
WRITE_BITS (&bw, pic_type, 3);
/* Add trailings. */
WRITE_BITS (&bw, 1, 1);
if (!gst_bit_writer_align_bytes (&bw, 0)) {
goto error;
}
*size = gst_bit_writer_get_size (&bw) / 8;
gst_bit_writer_reset (&bw);
return GST_H265_BIT_WRITER_OK;
error:
gst_bit_writer_reset (&bw);
*size = 0;
if (!have_space)
return GST_H265_BIT_WRITER_NO_MORE_SPACE;
return GST_H265_BIT_WRITER_INVALID_DATA;
}
/**
* gst_h265_bit_writer_convert_to_nal:
* @nal_prefix_size: the size in bytes for the prefix of a nal, may
* be 2, 3 or 4
* @packetized: whether to write the bit stream in packetized format,
* which does not have the start code but has a @nal_prefix_size bytes'
* size prepending to the real nal data
* @has_startcode: whether the input already has a start code
* @add_trailings: whether to add rbsp trailing bits to make the output
* aligned to byte
* @raw_data: the input bit stream
* @raw_size: the size in bits of the input bit stream
* @nal_data: (out): the output bit stream converted to a real nal
* @nal_size: (inout): the size in bytes of the output
*
* Converting a bit stream into a real nal packet. If the bit stream already
* has a start code, it will be replaced by the new one specified by the
* @nal_prefix_size and @packetized. It is assured that the output aligns to
* the byte and the all the emulations are inserted.
*
* Returns: a #GstH265BitWriterResult
*
* Since: 1.22
**/
GstH265BitWriterResult
gst_h265_bit_writer_convert_to_nal (guint nal_prefix_size,
gboolean packetized, gboolean has_startcode, gboolean add_trailings,
const guint8 * raw_data, gsize raw_size, guint8 * nal_data,
guint * nal_size)
{
NalWriter nw;
guint8 *data;
guint32 size = 0;
gboolean need_more_space = FALSE;
g_return_val_if_fail (
(packetized && nal_prefix_size > 1 && nal_prefix_size < 5) ||
(!packetized && (nal_prefix_size == 3 || nal_prefix_size == 4)),
GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (raw_data != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (raw_size > 0, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (nal_data != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (nal_size != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (*nal_size > 0, GST_H265_BIT_WRITER_ERROR);
if (has_startcode) {
/* Skip the start code, the NalWriter will add it automatically. */
if (raw_size >= 4 && raw_data[0] == 0
&& raw_data[1] == 0 && raw_data[2] == 0 && raw_data[3] == 0x01) {
raw_data += 4;
raw_size -= 4 * 8;
} else if (raw_size >= 3 && raw_data[0] == 0 && raw_data[1] == 0
&& raw_data[2] == 0x01) {
raw_data += 3;
raw_size -= 3 * 8;
} else {
/* Fail to find the start code. */
g_return_val_if_reached (GST_H265_BIT_WRITER_ERROR);
}
}
/* If no RBSP trailing needed, it must align to byte. We assume
that the rbsp trailing bits are already added. */
if (!add_trailings)
g_return_val_if_fail (raw_size % 8 == 0, GST_H265_BIT_WRITER_ERROR);
nal_writer_init (&nw, nal_prefix_size, packetized);
if (!nal_writer_put_bytes (&nw, raw_data, raw_size / 8))
goto error;
if (raw_size % 8) {
guint8 data = *(raw_data + raw_size / 8);
if (!nal_writer_put_bits_uint8 (&nw,
data >> (8 - raw_size % 8), raw_size % 8))
goto error;
}
if (add_trailings) {
if (!nal_writer_do_rbsp_trailing_bits (&nw))
goto error;
}
data = nal_writer_reset_and_get_data (&nw, &size);
if (!data)
goto error;
if (size > *nal_size) {
need_more_space = TRUE;
g_free (data);
goto error;
}
memcpy (nal_data, data, size);
*nal_size = size;
g_free (data);
nal_writer_reset (&nw);
return GST_H265_BIT_WRITER_OK;
error:
nal_writer_reset (&nw);
*nal_size = 0;
GST_WARNING ("Failed to convert nal data");
if (need_more_space)
return GST_H265_BIT_WRITER_NO_MORE_SPACE;
return GST_H265_BIT_WRITER_INVALID_DATA;
}