mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2024-07-07 15:15:53 +00:00
Code Issues 1 Projects Releases Wiki Activity

va: h265dec: Use picture and slide extension parameters.

Browse source 
This is transitional commit to later implement extended and screen
profiles.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-bad/-/merge_requests/2255>
This commit is contained in:
Víctor Manuel Jáquez Leal 2021-05-24 18:18:52 +02:00
parent 8e04651b8b
commit 0edc2f6b98
1 changed files with 41 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

77
sys/va/gstvah265dec.c
View file

@ -73,7 +73,7 @@ struct slice
guint8 *data; guint8 *data;
guint size; guint size;
VASliceParameterBufferHEVC param; VASliceParameterBufferHEVCExtension param;
}; };
typedef struct _GstVaH265Dec GstVaH265Dec; typedef struct _GstVaH265Dec GstVaH265Dec;
@ -94,7 +94,8 @@ struct _GstVaH265Dec
gint coded_height; gint coded_height;
gint dpb_size; gint dpb_size;
VAPictureParameterBufferHEVC pic_param; VAPictureParameterBufferHEVCExtension pic_param;
gint32 WpOffsetHalfRangeC; gint32 WpOffsetHalfRangeC;
struct slice prev_slice; struct slice prev_slice;
@ -112,7 +113,7 @@ static const gchar *sink_caps_str = "video/x-h265";
static inline void static inline void
_set_last_slice_flag (GstVaH265Dec * self) _set_last_slice_flag (GstVaH265Dec * self)
{ {
self->prev_slice.param.LongSliceFlags.fields.LastSliceOfPic = 1; self->prev_slice.param.base.LongSliceFlags.fields.LastSliceOfPic = 1;
} }
static void static void
@ -152,7 +153,7 @@ _submit_previous_slice (GstVaBaseDec * base, GstVaDecodePicture * va_pic)
return FALSE; return FALSE;
ret = gst_va_decoder_add_slice_buffer (base->decoder, va_pic, &slice->param, ret = gst_va_decoder_add_slice_buffer (base->decoder, va_pic, &slice->param,
sizeof (slice->param), slice->data, slice->size); sizeof (slice->param.base), slice->data, slice->size);
return ret; return ret;
} }
@ -279,7 +280,7 @@ _get_reference_index (GstH265Decoder * decoder, GstH265Picture * picture)
guint8 i; guint8 i;
for (i = 0; i < 15; i++) { for (i = 0; i < 15; i++) {
VAPictureHEVC *ref_va_pic = &self->pic_param.ReferenceFrames[i]; VAPictureHEVC *ref_va_pic = &self->pic_param.base.ReferenceFrames[i];
if (ref_va_pic->picture_id == VA_INVALID_ID) if (ref_va_pic->picture_id == VA_INVALID_ID)
break; break;
@ -310,7 +311,7 @@ _fill_ref_pic_list (GstH265Decoder * decoder, GstH265Picture * cur_pic,
static void static void
_fill_pred_weight_table (GstVaH265Dec * self, GstH265SliceHdr * header, _fill_pred_weight_table (GstVaH265Dec * self, GstH265SliceHdr * header,
VASliceParameterBufferHEVC * slice_param) VASliceParameterBufferHEVCExtension * slice_param)
{ {
gint chroma_weight, chroma_log2_weight_denom; gint chroma_weight, chroma_log2_weight_denom;
gint i, j; gint i, j;
@ -321,25 +322,25 @@ _fill_pred_weight_table (GstVaH265Dec * self, GstH265SliceHdr * header,
(!pps->weighted_bipred_flag && GST_H265_IS_B_SLICE (header))) (!pps->weighted_bipred_flag && GST_H265_IS_B_SLICE (header)))
return; return;
slice_param->luma_log2_weight_denom = slice_param->base.luma_log2_weight_denom =
header->pred_weight_table.luma_log2_weight_denom; header->pred_weight_table.luma_log2_weight_denom;
if (pps->sps->chroma_array_type != 0) if (pps->sps->chroma_array_type != 0)
slice_param->delta_chroma_log2_weight_denom = slice_param->base.delta_chroma_log2_weight_denom =
header->pred_weight_table.delta_chroma_log2_weight_denom; header->pred_weight_table.delta_chroma_log2_weight_denom;
for (i = 0; i <= header->num_ref_idx_l0_active_minus1; i++) { for (i = 0; i <= header->num_ref_idx_l0_active_minus1; i++) {
if (!header->pred_weight_table.luma_weight_l0_flag[i]) if (!header->pred_weight_table.luma_weight_l0_flag[i])
continue; continue;
slice_param->delta_luma_weight_l0[i] = slice_param->base.delta_luma_weight_l0[i] =
header->pred_weight_table.delta_luma_weight_l0[i]; header->pred_weight_table.delta_luma_weight_l0[i];
slice_param->luma_offset_l0[i] = slice_param->base.luma_offset_l0[i] =
header->pred_weight_table.luma_offset_l0[i]; header->pred_weight_table.luma_offset_l0[i];
} }
chroma_log2_weight_denom = slice_param->luma_log2_weight_denom + chroma_log2_weight_denom = slice_param->base.luma_log2_weight_denom +
slice_param->delta_chroma_log2_weight_denom; slice_param->base.delta_chroma_log2_weight_denom;
for (i = 0; i <= header->num_ref_idx_l0_active_minus1; i++) { for (i = 0; i <= header->num_ref_idx_l0_active_minus1; i++) {
if (!header->pred_weight_table.chroma_weight_l0_flag[i]) if (!header->pred_weight_table.chroma_weight_l0_flag[i])
@ -348,19 +349,20 @@ _fill_pred_weight_table (GstVaH265Dec * self, GstH265SliceHdr * header,
for (j = 0; j < 2; j++) { for (j = 0; j < 2; j++) {
gint8 delta_chroma_offset_l0 = gint8 delta_chroma_offset_l0 =
header->pred_weight_table.delta_chroma_offset_l0[i][j]; header->pred_weight_table.delta_chroma_offset_l0[i][j];
gint chroma_offset;
slice_param->delta_chroma_weight_l0[i][j] = slice_param->base.delta_chroma_weight_l0[i][j] =
header->pred_weight_table.delta_chroma_weight_l0[i][j]; header->pred_weight_table.delta_chroma_weight_l0[i][j];
/* Find ChromaWeightL0 */ /* Find ChromaWeightL0 */
chroma_weight = (1 << chroma_log2_weight_denom) + chroma_weight = (1 << chroma_log2_weight_denom) +
header->pred_weight_table.delta_chroma_weight_l0[i][j]; header->pred_weight_table.delta_chroma_weight_l0[i][j];
chroma_offset = self->WpOffsetHalfRangeC + delta_chroma_offset_l0
- ((self->WpOffsetHalfRangeC * chroma_weight)
>> chroma_log2_weight_denom);
/* 7-56 */ /* 7-56 */
slice_param->ChromaOffsetL0[i][j] = CLAMP ( slice_param->base.ChromaOffsetL0[i][j] = CLAMP (chroma_offset,
(self->WpOffsetHalfRangeC + delta_chroma_offset_l0 -
((self->WpOffsetHalfRangeC *
chroma_weight) >> chroma_log2_weight_denom)),
-self->WpOffsetHalfRangeC, self->WpOffsetHalfRangeC - 1); -self->WpOffsetHalfRangeC, self->WpOffsetHalfRangeC - 1);
} }
} }
@ -373,9 +375,9 @@ _fill_pred_weight_table (GstVaH265Dec * self, GstH265SliceHdr * header,
if (!header->pred_weight_table.luma_weight_l1_flag[i]) if (!header->pred_weight_table.luma_weight_l1_flag[i])
continue; continue;
slice_param->delta_luma_weight_l1[i] = slice_param->base.delta_luma_weight_l1[i] =
header->pred_weight_table.delta_luma_weight_l1[i]; header->pred_weight_table.delta_luma_weight_l1[i];
slice_param->luma_offset_l1[i] = slice_param->base.luma_offset_l1[i] =
header->pred_weight_table.luma_offset_l1[i]; header->pred_weight_table.luma_offset_l1[i];
} }
@ -386,20 +388,23 @@ _fill_pred_weight_table (GstVaH265Dec * self, GstH265SliceHdr * header,
for (j = 0; j < 2; j++) { for (j = 0; j < 2; j++) {
gint8 delta_chroma_offset_l1 = gint8 delta_chroma_offset_l1 =
header->pred_weight_table.delta_chroma_offset_l1[i][j]; header->pred_weight_table.delta_chroma_offset_l1[i][j];
gint chroma_offset;
slice_param->delta_chroma_weight_l1[i][j] = slice_param->base.delta_chroma_weight_l1[i][j] =
header->pred_weight_table.delta_chroma_weight_l1[i][j]; header->pred_weight_table.delta_chroma_weight_l1[i][j];
/* Find ChromaWeightL1 */ /* Find ChromaWeightL1 */
chroma_weight = (1 << chroma_log2_weight_denom) + chroma_weight = (1 << chroma_log2_weight_denom) +
header->pred_weight_table.delta_chroma_weight_l1[i][j]; header->pred_weight_table.delta_chroma_weight_l1[i][j];
chroma_offset = self->WpOffsetHalfRangeC + delta_chroma_offset_l1
- ((self->WpOffsetHalfRangeC * chroma_weight)
>> chroma_log2_weight_denom);
/* 7-56 */ /* 7-56 */
slice_param->ChromaOffsetL1[i][j] = CLAMP ( slice_param->base.ChromaOffsetL1[i][j] = CLAMP (chroma_offset,
(self->WpOffsetHalfRangeC + delta_chroma_offset_l1 -
((self->WpOffsetHalfRangeC *
chroma_weight) >> chroma_log2_weight_denom)),
-self->WpOffsetHalfRangeC, self->WpOffsetHalfRangeC - 1); -self->WpOffsetHalfRangeC, self->WpOffsetHalfRangeC - 1);
} }
} }
} }
@ -424,7 +429,7 @@ gst_va_h265_dec_decode_slice (GstH265Decoder * decoder,
GstVaBaseDec *base = GST_VA_BASE_DEC (decoder); GstVaBaseDec *base = GST_VA_BASE_DEC (decoder);
GstVaH265Dec *self = GST_VA_H265_DEC (decoder); GstVaH265Dec *self = GST_VA_H265_DEC (decoder);
GstVaDecodePicture *va_pic; GstVaDecodePicture *va_pic;
VASliceParameterBufferHEVC *slice_param; VASliceParameterBufferHEVCExtension *slice_param;
va_pic = gst_h265_picture_get_user_data (picture); va_pic = gst_h265_picture_get_user_data (picture);
if (!_submit_previous_slice (base, va_pic)) { if (!_submit_previous_slice (base, va_pic)) {
@ -436,7 +441,7 @@ gst_va_h265_dec_decode_slice (GstH265Decoder * decoder,
slice_param = &self->prev_slice.param; slice_param = &self->prev_slice.param;
/* *INDENT-OFF* */ /* *INDENT-OFF* */
*slice_param = (VASliceParameterBufferHEVC) { slice_param->base = (VASliceParameterBufferHEVC) {
.slice_data_size = nalu->size, .slice_data_size = nalu->size,
.slice_data_offset = 0, .slice_data_offset = 0,
.slice_data_flag = VA_SLICE_DATA_FLAG_ALL, .slice_data_flag = VA_SLICE_DATA_FLAG_ALL,
@ -473,9 +478,9 @@ gst_va_h265_dec_decode_slice (GstH265Decoder * decoder,
}; };
/* *INDENT-ON* */ /* *INDENT-ON* */
_fill_ref_pic_list (decoder, picture, slice_param->RefPicList[0], _fill_ref_pic_list (decoder, picture, slice_param->base.RefPicList[0],
ref_pic_list0); ref_pic_list0);
_fill_ref_pic_list (decoder, picture, slice_param->RefPicList[1], _fill_ref_pic_list (decoder, picture, slice_param->base.RefPicList[1],
ref_pic_list1); ref_pic_list1);
_fill_pred_weight_table (GST_VA_H265_DEC (decoder), header, slice_param); _fill_pred_weight_table (GST_VA_H265_DEC (decoder), header, slice_param);
@ -497,7 +502,7 @@ gst_va_h265_dec_start_picture (GstH265Decoder * decoder,
GstVaDecodePicture *va_pic; GstVaDecodePicture *va_pic;
GstH265ScalingList *scaling_list = NULL; GstH265ScalingList *scaling_list = NULL;
VAIQMatrixBufferHEVC iq_matrix = { 0, }; VAIQMatrixBufferHEVC iq_matrix = { 0, };
VAPictureParameterBufferHEVC *pic_param = &self->pic_param; VAPictureParameterBufferHEVCExtension *pic_param = &self->pic_param;
guint i; guint i;
va_pic = gst_h265_picture_get_user_data (picture); va_pic = gst_h265_picture_get_user_data (picture);
@ -506,7 +511,7 @@ gst_va_h265_dec_start_picture (GstH265Decoder * decoder,
sps = pps->sps; sps = pps->sps;
/* *INDENT-OFF* */ /* *INDENT-OFF* */
*pic_param = (VAPictureParameterBufferHEVC) { pic_param->base = (VAPictureParameterBufferHEVC) {
.pic_width_in_luma_samples = sps->pic_width_in_luma_samples, .pic_width_in_luma_samples = sps->pic_width_in_luma_samples,
.pic_height_in_luma_samples = sps->pic_height_in_luma_samples, .pic_height_in_luma_samples = sps->pic_height_in_luma_samples,
.sps_max_dec_pic_buffering_minus1 = sps->max_dec_pic_buffering_minus1[sps->max_sub_layers_minus1], .sps_max_dec_pic_buffering_minus1 = sps->max_dec_pic_buffering_minus1[sps->max_sub_layers_minus1],
@ -581,28 +586,28 @@ gst_va_h265_dec_start_picture (GstH265Decoder * decoder,
/* *INDENT-ON* */ /* *INDENT-ON* */
for (i = 0; i <= pps->num_tile_columns_minus1; i++) for (i = 0; i <= pps->num_tile_columns_minus1; i++)
pic_param->column_width_minus1[i] = pps->column_width_minus1[i]; pic_param->base.column_width_minus1[i] = pps->column_width_minus1[i];
for (i = 0; i <= pps->num_tile_rows_minus1; i++) for (i = 0; i <= pps->num_tile_rows_minus1; i++)
pic_param->row_height_minus1[i] = pps->row_height_minus1[i]; pic_param->base.row_height_minus1[i] = pps->row_height_minus1[i];
_fill_vaapi_pic (decoder, &pic_param->CurrPic, picture); _fill_vaapi_pic (decoder, &pic_param->base.CurrPic, picture);
/* reference frames */ /* reference frames */
{ {
GArray *ref_list = gst_h265_dpb_get_pictures_all (dpb); GArray *ref_list = gst_h265_dpb_get_pictures_all (dpb);
for (i = 0; i < 15 && i < ref_list->len; i++) { for (i = 0; i < 15 && i < ref_list->len; i++) {
GstH265Picture *pic = g_array_index (ref_list, GstH265Picture *, i); GstH265Picture *pic = g_array_index (ref_list, GstH265Picture *, i);
_fill_vaapi_pic (decoder, &pic_param->ReferenceFrames[i], pic); _fill_vaapi_pic (decoder, &pic_param->base.ReferenceFrames[i], pic);
} }
g_array_unref (ref_list); g_array_unref (ref_list);
for (; i < 15; i++) for (; i < 15; i++)
_init_vaapi_pic (&pic_param->ReferenceFrames[i]); _init_vaapi_pic (&pic_param->base.ReferenceFrames[i]);
} }
if (!gst_va_decoder_add_param_buffer (base->decoder, va_pic, if (!gst_va_decoder_add_param_buffer (base->decoder, va_pic,
VAPictureParameterBufferType, pic_param, sizeof (*pic_param))) VAPictureParameterBufferType, pic_param, sizeof (pic_param->base)))
return FALSE; return FALSE;
if (pps->scaling_list_data_present_flag || if (pps->scaling_list_data_present_flag ||