vdpau: use GstVdpVideoBufferPool in GstVdpVideoSrcPad

We also don't pad_alloc our GstVdpVideoBuffers anymore since we don't support
downstream negotation anyway.

1 files changed, 34 insertions, 58 deletions

diff --git a/sys/vdpau/gstvdp/gstvdpvideosrcpad.c b/sys/vdpau/gstvdp/gstvdpvideosrcpad.c
index 7c8444b8d..6c8a88368 100644
--- a/sys/vdpau/gstvdp/gstvdpvideosrcpad.c
+++ b/sys/vdpau/gstvdp/gstvdpvideosrcpad.c
@@ -19,6 +19,7 @@
  */
 
 #include "gstvdpvideobuffer.h"
+#include "gstvdpvideobufferpool.h"
 
 #include "gstvdpvideosrcpad.h"
 
@@ -35,6 +36,7 @@ struct _GstVdpVideoSrcPad
 {
   GstPad pad;
 
+  GstVdpBufferPool *bpool;
   GstCaps *caps;
 
   gboolean yuv_output;
@@ -133,44 +135,11 @@ gst_vdp_video_src_pad_push (GstVdpVideoSrcPad * vdp_pad,
   return gst_pad_push (pad, out_buf);
 }
 
-static GstFlowReturn
-gst_vdp_video_src_pad_alloc_with_caps (GstVdpVideoSrcPad * vdp_pad,
-    GstCaps * caps, GstVdpVideoBuffer ** video_buf, GError ** error)
-{
-  GstFlowReturn ret;
-
-  ret = gst_pad_alloc_buffer ((GstPad *) vdp_pad, 0, 0, caps,
-      (GstBuffer **) video_buf);
-  if (ret != GST_FLOW_OK)
-    return ret;
-
-  if (!gst_caps_is_equal_fixed (caps, GST_BUFFER_CAPS (*video_buf)))
-    goto wrong_caps;
-
-  if (!GST_IS_VDP_VIDEO_BUFFER (*video_buf))
-    goto invalid_buf;
-
-  return GST_FLOW_OK;
-
-wrong_caps:
-  gst_buffer_unref (GST_BUFFER (*video_buf));
-  g_set_error (error, GST_STREAM_ERROR, GST_STREAM_ERROR_FAILED,
-      "Sink element returned buffer with wrong caps");
-  return GST_FLOW_ERROR;
-
-invalid_buf:
-  gst_buffer_unref (GST_BUFFER (*video_buf));
-  g_set_error (error, GST_STREAM_ERROR, GST_STREAM_ERROR_FAILED,
-      "Sink element returned buffer of wrong type");
-  return GST_FLOW_ERROR;
-}
-
 GstFlowReturn
 gst_vdp_video_src_pad_alloc_buffer (GstVdpVideoSrcPad * vdp_pad,
     GstVdpVideoBuffer ** video_buf, GError ** error)
 {
   GstCaps *caps;
-  GstFlowReturn ret;
 
   g_return_val_if_fail (GST_IS_VDP_VIDEO_SRC_PAD (vdp_pad), GST_FLOW_ERROR);
 
@@ -178,27 +147,13 @@ gst_vdp_video_src_pad_alloc_buffer (GstVdpVideoSrcPad * vdp_pad,
   if (!caps)
     return GST_FLOW_NOT_NEGOTIATED;
 
-  if (vdp_pad->yuv_output) {
-    GstVdpDevice *device = vdp_pad->device;
-
-    *video_buf = gst_vdp_video_buffer_new (device, VDP_CHROMA_TYPE_420,
-        vdp_pad->width, vdp_pad->height, NULL);
-    if (!*video_buf)
-      goto video_buf_error;
-
-  } else {
-    ret = gst_vdp_video_src_pad_alloc_with_caps (vdp_pad, caps, video_buf,
-        error);
-    if (ret != GST_FLOW_OK)
-      return ret;
-  }
+  *video_buf =
+      (GstVdpVideoBuffer *) gst_vdp_buffer_pool_get_buffer (vdp_pad->bpool,
+      error);
+  if (!*video_buf)
+    return GST_FLOW_ERROR;
 
   return GST_FLOW_OK;
-
-video_buf_error:
-  g_set_error (error, GST_RESOURCE_ERROR, GST_RESOURCE_ERROR_READ,
-      "Couldn't create a GstVdpVideoBuffer");
-  return GST_FLOW_ERROR;
 }
 
 static gboolean
@@ -207,6 +162,9 @@ gst_vdp_video_src_pad_setcaps (GstPad * pad, GstCaps * caps)
   GstVdpVideoSrcPad *vdp_pad = GST_VDP_VIDEO_SRC_PAD (pad);
   const GstStructure *structure;
 
+  VdpChromaType chroma_type;
+  GstCaps *video_caps;
+
   structure = gst_caps_get_structure (caps, 0);
   if (gst_structure_has_name (structure, "video/x-raw-yuv")) {
     if (!gst_structure_get_int (structure, "width", &vdp_pad->width))
@@ -216,17 +174,28 @@ gst_vdp_video_src_pad_setcaps (GstPad * pad, GstCaps * caps)
     if (!gst_structure_get_fourcc (structure, "format", &vdp_pad->fourcc))
       return FALSE;
 
+    chroma_type = VDP_CHROMA_TYPE_420;
+
     vdp_pad->yuv_output = TRUE;
   } else if (gst_structure_has_name (structure, "video/x-vdpau-video")) {
     if (!gst_structure_get_int (structure, "width", &vdp_pad->width))
       return FALSE;
     if (!gst_structure_get_int (structure, "height", &vdp_pad->height))
       return FALSE;
+    if (!gst_structure_get_int (structure, "chroma-type",
+            (gint *) & chroma_type))
+      return FALSE;
 
     vdp_pad->yuv_output = FALSE;
   } else
     return FALSE;
 
+  video_caps = gst_caps_new_simple ("video/x-vdpau-video",
+      "chroma-type", G_TYPE_INT, (gint) chroma_type,
+      "width", G_TYPE_INT, vdp_pad->width,
+      "height", G_TYPE_INT, vdp_pad->height, NULL);
+  gst_vdp_buffer_pool_set_caps (vdp_pad->bpool, video_caps);
+
   return TRUE;
 }
 
@@ -265,15 +234,25 @@ gst_vdp_video_src_pad_activate_push (GstPad * pad, gboolean active)
 }
 
 static void
-gst_vdp_video_src_pad_update_caps (GstVdpVideoSrcPad * vdp_pad)
+gst_vdp_video_src_pad_set_device (GstVdpVideoSrcPad * vdp_pad,
+    GstVdpDevice * device)
 {
   GstCaps *caps;
   const GstCaps *templ_caps;
 
+  if (vdp_pad->bpool)
+    g_object_unref (vdp_pad->bpool);
+  if (vdp_pad->device)
+    g_object_unref (vdp_pad->device);
+
+  vdp_pad->device = device;
+  vdp_pad->bpool = gst_vdp_video_buffer_pool_new (device);
+
+  /* update caps */
   if (vdp_pad->caps)
     gst_caps_unref (vdp_pad->caps);
 
-  caps = gst_vdp_video_buffer_get_allowed_caps (vdp_pad->device);
+  caps = gst_vdp_video_buffer_get_allowed_caps (device);
 
   if ((templ_caps = gst_pad_get_pad_template_caps (GST_PAD (vdp_pad)))) {
     vdp_pad->caps = gst_caps_intersect (caps, templ_caps);
@@ -307,10 +286,7 @@ gst_vdp_video_src_pad_set_property (GObject * object, guint prop_id,
 
   switch (prop_id) {
     case PROP_DEVICE:
-      if (vdp_pad->device)
-        g_object_unref (vdp_pad->device);
-      vdp_pad->device = g_value_dup_object (value);
-      gst_vdp_video_src_pad_update_caps (vdp_pad);
+      gst_vdp_video_src_pad_set_device (vdp_pad, g_value_dup_object (value));
       break;
 
     default: