FaSTTUBe/GST-Tensordecoder-ov_ep
37
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

nvh264sldec: Reopen decoder object if larger DPB size is required

Browse Source 
Equivalent to the d3d11h264dec fix
https://gitlab.freedesktop.org/gstreamer/gst-plugins-bad/-/merge_requests/1839

Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-bad/-/merge_requests/2059>
This commit is contained in:
Seungha Yang 2021-03-07 16:47:07 +09:00
parent a29c8935b7
commit 9da6da20e7
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
sys/nvcodec/gstnvh264dec.c
View File

@ -112,6 +112,7 @@ struct _GstNvH264Dec
guint coded_width, coded_height; guint coded_width, coded_height;
guint bitdepth; guint bitdepth;
guint chroma_format_idc; guint chroma_format_idc;
gint max_dpb_size;
GstVideoFormat out_format; GstVideoFormat out_format;
/* For OpenGL interop. */ /* For OpenGL interop. */
@ -237,6 +238,20 @@ gst_nv_h264_dec_set_context (GstElement * element, GstContext * context)
GST_ELEMENT_CLASS (parent_class)->set_context (element, context); GST_ELEMENT_CLASS (parent_class)->set_context (element, context);
} }
/* Clear all codec specific (e.g., SPS) data */
static void
gst_d3d11_h264_dec_reset (GstNvH264Dec * self)
{
self->width = 0;
self->height = 0;
self->coded_width = 0;
self->coded_height = 0;
self->bitdepth = 0;
self->chroma_format_idc = 0;
self->out_format = GST_VIDEO_FORMAT_UNKNOWN;
self->max_dpb_size = 0;
}
static gboolean static gboolean
gst_nv_h264_dec_open (GstVideoDecoder * decoder) gst_nv_h264_dec_open (GstVideoDecoder * decoder)
{ {
@ -250,6 +265,8 @@ gst_nv_h264_dec_open (GstVideoDecoder * decoder)
return FALSE; return FALSE;
} }
gst_d3d11_h264_dec_reset (self);
return TRUE; return TRUE;
} }
@ -367,6 +384,12 @@ gst_nv_h264_dec_new_sequence (GstH264Decoder * decoder, const GstH264SPS * sps,
modified = TRUE; modified = TRUE;
} }
if (self->max_dpb_size < max_dpb_size) {
GST_INFO_OBJECT (self, "Requires larger DPB size (%d -> %d)",
self->max_dpb_size, max_dpb_size);
modified = TRUE;
}
if (modified || !self->decoder) { if (modified || !self->decoder) {
GstVideoInfo info; GstVideoInfo info;
@ -396,6 +419,7 @@ gst_nv_h264_dec_new_sequence (GstH264Decoder * decoder, const GstH264SPS * sps,
gst_video_info_set_format (&info, gst_video_info_set_format (&info,
self->out_format, self->width, self->height); self->out_format, self->width, self->height);
self->max_dpb_size = max_dpb_size;
/* FIXME: add support cudaVideoCodec_H264_SVC and cudaVideoCodec_H264_MVC */ /* FIXME: add support cudaVideoCodec_H264_SVC and cudaVideoCodec_H264_MVC */
self->decoder = gst_nv_decoder_new (self->context, cudaVideoCodec_H264, self->decoder = gst_nv_decoder_new (self->context, cudaVideoCodec_H264,
&info, &info,