From ae68d9dda5d343b844032f33bd7ccf0c2d3fcfdc Mon Sep 17 00:00:00 2001
From: He Junyan <junyan.he@intel.com>
Date: Mon, 22 Apr 2024 15:03:56 +0800
Subject: [PATCH] vah265enc: Set the correct buffer flag for output

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/6703>
---
 subprojects/gst-plugins-bad/sys/va/gstvah265enc.c | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/subprojects/gst-plugins-bad/sys/va/gstvah265enc.c b/subprojects/gst-plugins-bad/sys/va/gstvah265enc.c
index cfbf5dd462..ba3a8c5217 100644
--- a/subprojects/gst-plugins-bad/sys/va/gstvah265enc.c
+++ b/subprojects/gst-plugins-bad/sys/va/gstvah265enc.c
@@ -4714,6 +4714,16 @@ gst_va_h265_enc_prepare_output (GstVaBaseEnc * base,
     return FALSE;
   }
 
+  GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_MARKER);
+  if (frame_enc->poc == 0) {
+    GST_VIDEO_CODEC_FRAME_SET_SYNC_POINT (frame);
+    GST_BUFFER_FLAG_UNSET (buf, GST_BUFFER_FLAG_DELTA_UNIT);
+    GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_HEADER);
+  } else {
+    GST_VIDEO_CODEC_FRAME_UNSET_SYNC_POINT (frame);
+    GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_DELTA_UNIT);
+  }
+
   gst_buffer_replace (&frame->output_buffer, buf);
   gst_clear_buffer (&buf);