ext/gl/gstglsrcbin.c

   1 /*
   2  * GStreamer
   3  * Copyright (C) 2015 Matthew Waters <matthew@centricular.com>
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Library General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Library General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Library General Public
  16  * License along with this library; if not, write to the
  17  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  18  * Boston, MA 02110-1301, USA.
  19  */
  20
  21 #ifdef HAVE_CONFIG_H
  22 #include "config.h"
  23 #endif
  24
  25 #include "gstglsrcbin.h"
  26
  27 GST_DEBUG_CATEGORY (gst_debug_gl_src_bin);
  28 #define GST_CAT_DEFAULT gst_debug_gl_src_bin
  29
  30 static void gst_gl_src_bin_finalize (GObject * object);
  31 static void gst_gl_src_bin_set_property (GObject * object, guint prop_id,
  32     const GValue * value, GParamSpec * param_spec);
  33 static void gst_gl_src_bin_get_property (GObject * object, guint prop_id,
  34     GValue * value, GParamSpec * param_spec);
  35
  36 static GstStateChangeReturn gst_gl_src_bin_change_state (GstElement * element,
  37     GstStateChange transition);
  38
  39 static GstStaticPadTemplate gst_gl_src_bin_template =
  40 GST_STATIC_PAD_TEMPLATE ("src",
  41     GST_PAD_SRC,
  42     GST_PAD_ALWAYS,
  43     GST_STATIC_CAPS ("video/x-raw(ANY)"));
  44
  45 enum
  46 {
  47   PROP_0,
  48   PROP_SRC,
  49 };
  50
  51 enum
  52 {
  53   SIGNAL_0,
  54   SIGNAL_CREATE_ELEMENT,
  55   SIGNAL_LAST,
  56 };
  57
  58 static guint gst_gl_src_bin_signals[SIGNAL_LAST] = { 0, };
  59
  60 #define gst_gl_src_bin_parent_class parent_class
  61 G_DEFINE_TYPE_WITH_CODE (GstGLSrcBin, gst_gl_src_bin,
  62     GST_TYPE_BIN,
  63     GST_DEBUG_CATEGORY_INIT (gst_debug_gl_src_bin, "glsrcbin", 0,
  64         "OpenGL Video Src Bin"));
  65
  66 static void
  67 gst_gl_src_bin_class_init (GstGLSrcBinClass * klass)
  68 {
  69   GObjectClass *gobject_class;
  70   GstElementClass *element_class;
  71
  72   gobject_class = (GObjectClass *) klass;
  73   element_class = GST_ELEMENT_CLASS (klass);
  74
  75   element_class->change_state = gst_gl_src_bin_change_state;
  76
  77   gobject_class->set_property = gst_gl_src_bin_set_property;
  78   gobject_class->get_property = gst_gl_src_bin_get_property;
  79   gobject_class->finalize = gst_gl_src_bin_finalize;
  80
  81   g_object_class_install_property (gobject_class, PROP_SRC,
  82       g_param_spec_object ("src",
  83           "GL src element",
  84           "The GL src chain to use",
  85           GST_TYPE_ELEMENT,
  86           GST_PARAM_MUTABLE_READY | G_PARAM_READWRITE |
  87           G_PARAM_STATIC_STRINGS));
  88
  89   /**
  90    * GstGLSrcBin::create-element:
  91    * @object: the #GstGLSrcBin
  92    *
  93    * Will be emitted when we need the processing element/s that this bin will use
  94    *
  95    * Returns: a new #GstElement
  96    */
  97   gst_gl_src_bin_signals[SIGNAL_CREATE_ELEMENT] =
  98       g_signal_new ("create-element", G_TYPE_FROM_CLASS (klass),
  99       G_SIGNAL_RUN_LAST, 0, NULL, NULL, NULL, GST_TYPE_ELEMENT, 0);
 100
 101   gst_element_class_set_metadata (element_class,
 102       "GL Src Bin", "Src/Video",
 103       "Infrastructure to process GL textures",
 104       "Matthew Waters <matthew@centricular.com>");
 105
 106   gst_element_class_add_static_pad_template (element_class,
 107       &gst_gl_src_bin_template);
 108 }
 109
 110 static void
 111 gst_gl_src_bin_init (GstGLSrcBin * self)
 112 {
 113   gboolean res = TRUE;
 114   GstPad *pad;
 115
 116   self->download = gst_element_factory_make ("gldownload", NULL);
 117   self->convert = gst_element_factory_make ("glcolorconvert", NULL);
 118
 119   res &= gst_bin_add (GST_BIN (self), self->download);
 120   res &= gst_bin_add (GST_BIN (self), self->convert);
 121
 122   res &= gst_element_link_pads (self->convert, "src", self->download, "sink");
 123
 124   pad = gst_element_get_static_pad (self->download, "src");
 125   if (!pad) {
 126     res = FALSE;
 127   } else {
 128     GST_DEBUG_OBJECT (self, "setting target src pad %" GST_PTR_FORMAT, pad);
 129     self->srcpad = gst_ghost_pad_new ("src", pad);
 130     gst_element_add_pad (GST_ELEMENT_CAST (self), self->srcpad);
 131     gst_object_unref (pad);
 132   }
 133
 134   if (!res) {
 135     GST_WARNING_OBJECT (self, "Failed to add/connect the necessary machinery");
 136   }
 137 }
 138
 139 static void
 140 gst_gl_src_bin_finalize (GObject * object)
 141 {
 142   GstGLSrcBin *self = GST_GL_SRC_BIN (object);
 143
 144   if (self->src)
 145     gst_object_unref (self->src);
 146
 147   G_OBJECT_CLASS (parent_class)->finalize (object);
 148 }
 149
 150 static gboolean
 151 _connect_src_element (GstGLSrcBin * self)
 152 {
 153   gboolean res = TRUE;
 154
 155   gst_object_set_name (GST_OBJECT (self->src), "src");
 156   res &= gst_bin_add (GST_BIN (self), self->src);
 157
 158   res &= gst_element_link_pads (self->src, "src", self->convert, "sink");
 159
 160   if (!res)
 161     GST_ERROR_OBJECT (self, "Failed to link src element into the pipeline");
 162
 163   return res;
 164 }
 165
 166 /*
 167  * @src: (transfer floating):
 168  */
 169 static gboolean
 170 gst_gl_src_bin_set_src (GstGLSrcBin * self, GstElement * src)
 171 {
 172   g_return_val_if_fail (GST_IS_ELEMENT (src), FALSE);
 173
 174   if (self->src) {
 175     gst_element_set_locked_state (self->src, TRUE);
 176     gst_bin_remove (GST_BIN (self), self->src);
 177     gst_element_set_state (self->src, GST_STATE_NULL);
 178     gst_object_unref (self->src);
 179     self->src = NULL;
 180   }
 181   self->src = src;
 182
 183   gst_object_ref_sink (src);
 184
 185   if (src && !_connect_src_element (self)) {
 186     gst_object_unref (self->src);
 187     self->src = NULL;
 188     return FALSE;
 189   }
 190
 191   return TRUE;
 192 }
 193
 194 void
 195 gst_gl_src_bin_finish_init_with_element (GstGLSrcBin * self,
 196     GstElement * element)
 197 {
 198   gst_gl_src_bin_set_src (self, element);
 199 }
 200
 201 void
 202 gst_gl_src_bin_finish_init (GstGLSrcBin * self)
 203 {
 204   GstGLSrcBinClass *klass = GST_GL_SRC_BIN_GET_CLASS (self);
 205   GstElement *element = NULL;
 206
 207   if (klass->create_element)
 208     element = klass->create_element ();
 209
 210   if (element)
 211     gst_gl_src_bin_finish_init_with_element (self, element);
 212 }
 213
 214 static void
 215 gst_gl_src_bin_set_property (GObject * object, guint prop_id,
 216     const GValue * value, GParamSpec * pspec)
 217 {
 218   GstGLSrcBin *self = GST_GL_SRC_BIN (object);
 219
 220   switch (prop_id) {
 221     case PROP_SRC:
 222       gst_gl_src_bin_set_src (self, g_value_get_object (value));
 223       break;
 224     default:
 225       if (self->src)
 226         g_object_set_property (G_OBJECT (self->src), pspec->name, value);
 227       break;
 228   }
 229 }
 230
 231 static void
 232 gst_gl_src_bin_get_property (GObject * object, guint prop_id,
 233     GValue * value, GParamSpec * pspec)
 234 {
 235   GstGLSrcBin *self = GST_GL_SRC_BIN (object);
 236
 237   switch (prop_id) {
 238     case PROP_SRC:
 239       g_value_set_object (value, self->src);
 240       break;
 241     default:
 242       if (self->src)
 243         g_object_get_property (G_OBJECT (self->src), pspec->name, value);
 244       break;
 245   }
 246 }
 247
 248 static GstStateChangeReturn
 249 gst_gl_src_bin_change_state (GstElement * element, GstStateChange transition)
 250 {
 251   GstGLSrcBin *self = GST_GL_SRC_BIN (element);
 252   GstGLSrcBinClass *klass = GST_GL_SRC_BIN_GET_CLASS (self);
 253   GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;
 254
 255   GST_DEBUG ("changing state: %s => %s",
 256       gst_element_state_get_name (GST_STATE_TRANSITION_CURRENT (transition)),
 257       gst_element_state_get_name (GST_STATE_TRANSITION_NEXT (transition)));
 258
 259   switch (transition) {
 260     case GST_STATE_CHANGE_NULL_TO_READY:
 261       if (!self->src) {
 262         if (klass->create_element)
 263           self->src = klass->create_element ();
 264
 265         if (!self->src) {
 266           g_signal_emit (element,
 267               gst_gl_src_bin_signals[SIGNAL_CREATE_ELEMENT], 0, &self->src);
 268           if (self->src && g_object_is_floating (self->src))
 269             gst_object_ref_sink (self->src);
 270         }
 271
 272         if (!self->src) {
 273           GST_ERROR_OBJECT (element, "Failed to retrieve element");
 274           return GST_STATE_CHANGE_FAILURE;
 275         }
 276         if (!_connect_src_element (self))
 277           return GST_STATE_CHANGE_FAILURE;
 278       }
 279       break;
 280     default:
 281       break;
 282   }
 283
 284   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
 285   if (ret == GST_STATE_CHANGE_FAILURE)
 286     return ret;
 287
 288   switch (transition) {
 289     default:
 290       break;
 291   }
 292
 293   return ret;
 294 }