mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2024-11-26 11:41:09 +00:00
Code Issues 1 Projects Releases Wiki Activity

vaav1enc: Move repeat header data to a dedicated buffer

Browse source 
When enable parallel encoding, it is possible that the unshown frame
is not output but it is already be marked as a repeated frame header.
So we need to use a dedicated buffer to hold the repeat frame header,
don't mix it with the orignal frame data.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/6867>
This commit is contained in:
He Junyan 2024-05-13 18:52:28 +08:00 committed by GStreamer Marge Bot
parent fca7e5aa23
commit 77f779e86a
1 changed files with 29 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
subprojects/gst-plugins-bad/sys/va/gstvaav1enc.c
View file

@ -180,6 +180,9 @@ struct _GstVaAV1EncFrame
guint cached_frame_header_size; guint cached_frame_header_size;
guint8 cached_frame_header[32]; guint8 cached_frame_header[32];
guint repeat_frame_header_size;
guint8 repeat_frame_header[32];
}; };
struct _GstVaAV1Enc struct _GstVaAV1Enc
@ -515,6 +518,9 @@ gst_va_av1_enc_frame_new (void)
frame->order_hint = -1; frame->order_hint = -1;
frame->repeat_index = -1; frame->repeat_index = -1;
frame->cached_frame_header_size = 0; frame->cached_frame_header_size = 0;
memset (frame->cached_frame_header, 0, sizeof (frame->cached_frame_header));
frame->repeat_frame_header_size = 0;
memset (frame->repeat_frame_header, 0, sizeof (frame->repeat_frame_header));
return frame; return frame;
} }
@ -3815,26 +3821,37 @@ static void
_av1_add_repeat_frame_header (GstVaAV1Enc * self, GstVaAV1EncFrame * va_frame) _av1_add_repeat_frame_header (GstVaAV1Enc * self, GstVaAV1EncFrame * va_frame)
{ {
GstAV1FrameHeaderOBU frame_hdr = { 0, }; GstAV1FrameHeaderOBU frame_hdr = { 0, };
guint frame_hdr_data_size; guint data_size;
data_size = sizeof (va_frame->repeat_frame_header) -
va_frame->repeat_frame_header_size;
/* Repeat frame always shows a frame and so begins with a TD. */ /* Repeat frame always shows a frame and so begins with a TD. */
_av1_add_td (self, va_frame); if (gst_av1_bit_writer_temporal_delimiter_obu (TRUE,
va_frame->repeat_frame_header + va_frame->repeat_frame_header_size,
&data_size) != GST_AV1_BIT_WRITER_OK) {
GST_ERROR_OBJECT (self, "Failed to write temporal delimiter.");
/* The only possible failure is not enough buffer size,
user should ensure that. */
g_assert_not_reached ();
}
va_frame->repeat_frame_header_size += data_size;
data_size = sizeof (va_frame->repeat_frame_header) -
va_frame->repeat_frame_header_size;
frame_hdr.show_existing_frame = 1; frame_hdr.show_existing_frame = 1;
frame_hdr.frame_to_show_map_idx = va_frame->repeat_index; frame_hdr.frame_to_show_map_idx = va_frame->repeat_index;
frame_hdr_data_size = sizeof (va_frame->cached_frame_header) -
va_frame->cached_frame_header_size;
if (gst_av1_bit_writer_frame_header_obu (&frame_hdr, &self->sequence_hdr, if (gst_av1_bit_writer_frame_header_obu (&frame_hdr, &self->sequence_hdr,
va_frame->temporal_id, va_frame->spatial_id, TRUE, va_frame->temporal_id, va_frame->spatial_id, TRUE,
va_frame->cached_frame_header + va_frame->cached_frame_header_size, va_frame->repeat_frame_header + va_frame->repeat_frame_header_size,
&frame_hdr_data_size) != GST_AV1_BIT_WRITER_OK) { &data_size) != GST_AV1_BIT_WRITER_OK) {
GST_ERROR_OBJECT (self, "Failed to write repeat frame header."); GST_ERROR_OBJECT (self, "Failed to write repeat frame header.");
g_assert_not_reached (); g_assert_not_reached ();
} }
va_frame->cached_frame_header_size += frame_hdr_data_size; va_frame->repeat_frame_header_size += data_size;
} }
static GstFlowReturn static GstFlowReturn
@ -3853,10 +3870,6 @@ gst_va_av1_enc_encode_frame (GstVaBaseEnc * base,
return GST_FLOW_ERROR; return GST_FLOW_ERROR;
} }
memset (va_frame->cached_frame_header, 0,
sizeof (va_frame->cached_frame_header));
va_frame->cached_frame_header_size = 0;
if (va_frame->type & FRAME_TYPE_REPEAT) { if (va_frame->type & FRAME_TYPE_REPEAT) {
g_assert (va_frame->flags & FRAME_FLAG_ALREADY_ENCODED); g_assert (va_frame->flags & FRAME_FLAG_ALREADY_ENCODED);
_av1_add_repeat_frame_header (self, va_frame); _av1_add_repeat_frame_header (self, va_frame);
@ -4058,16 +4071,16 @@ gst_va_av1_enc_prepare_output (GstVaBaseEnc * base,
g_assert ((frame_enc->flags & FRAME_FLAG_FRAME_IN_TU_CACHE) == 0); g_assert ((frame_enc->flags & FRAME_FLAG_FRAME_IN_TU_CACHE) == 0);
buf = gst_video_encoder_allocate_output_buffer buf = gst_video_encoder_allocate_output_buffer
(GST_VIDEO_ENCODER_CAST (base), frame_enc->cached_frame_header_size); (GST_VIDEO_ENCODER_CAST (base), frame_enc->repeat_frame_header_size);
if (!buf) { if (!buf) {
GST_ERROR_OBJECT (base, "Failed to create output buffer"); GST_ERROR_OBJECT (base, "Failed to create output buffer");
return FALSE; return FALSE;
} }
sz = gst_buffer_fill (buf, 0, frame_enc->cached_frame_header, sz = gst_buffer_fill (buf, 0, frame_enc->repeat_frame_header,
frame_enc->cached_frame_header_size); frame_enc->repeat_frame_header_size);
if (sz != frame_enc->cached_frame_header_size) { if (sz != frame_enc->repeat_frame_header_size) {
GST_ERROR_OBJECT (base, "Failed to write output buffer for repeat frame"); GST_ERROR_OBJECT (base, "Failed to write output buffer for repeat frame");
gst_clear_buffer (&buf); gst_clear_buffer (&buf);
return FALSE; return FALSE;