commit 65adac65100c0c5209dd2eb98ea9695f9c7452e5
author Song Bing <b06498@freescale.com> Wed May 20 15:15:08 2015 +0800
committer Lyon Wang <lyon.wang@nxp.com> Wed Dec 19 16:09:50 2018 +0800
tree b4ce1564f53d0c1c46c27fd141ccc0d49bf1c378
parent 3e20710126c94aaffc308b72b61d06f6207c745a

mpegvideoparse: Need detect picture coding type when drain

Need detect picture coding type when drain

Upstream Status: Waiting for review.

https://bugzilla.gnome.org/show_bug.cgi?id=749617

gst/videoparsers/gstmpegvideoparse.c

diff --git a/gst/videoparsers/gstmpegvideoparse.c b/gst/videoparsers/gstmpegvideoparse.c
index 6896339..769e0fd 100644
--- a/gst/videoparsers/gstmpegvideoparse.c
+++ b/gst/videoparsers/gstmpegvideoparse.c
@@ -715,6 +715,22 @@
   if (GST_BASE_PARSE_DRAINING (parse)) {
     GST_LOG_OBJECT (mpvparse, "draining, accepting all data");
     off = size;
+    /* decide picture codding type */
+    if (mpvparse->pic_offset >= 0 && mpvparse->pic_offset < off) {
+      GstMpegVideoPacket header;
+
+      header.data = map.data;
+      header.type = GST_MPEG_VIDEO_PACKET_PICTURE;
+      header.offset = mpvparse->pic_offset;
+      header.size = map.size - mpvparse->pic_offset;
+      if (gst_mpeg_video_packet_parse_picture_header (&header, &mpvparse->pichdr))
+        GST_LOG_OBJECT (mpvparse, "picture_coding_type %d (%s), ending"
+            "frame of size %d", mpvparse->pichdr.pic_type,
+            picture_type_name (mpvparse->pichdr.pic_type), off - 4);
+      else
+        GST_LOG_OBJECT (mpvparse, "Couldn't parse picture at offset %d",
+            mpvparse->pic_offset);
+    }
     ret = TRUE;
   } else {
     GST_LOG_OBJECT (mpvparse, "need more data");