omxh264videodec: Add h.264 video decoder

omx/Makefile.am

@@ -4,6 +4,7 @@ libgstomx_la_SOURCES = \
 	gstomx.c \
 	gstomxvideodec.c \
 	gstomxmpeg4videodec.c \
+	gstomxh264videodec.c \
 	gstbasevideocodec.c \
 	gstbasevideodecoder.c \
 	gstbasevideoencoder.c \
@@ -12,6 +13,7 @@ libgstomx_la_SOURCES = \
 noinst_HEADERS = \
 	gstomxvideodec.h \
 	gstomxmpeg4videodec.h \
+	gstomxh264videodec.h \
 	gstbasevideocodec.h \
 	gstbasevideodecoder.h \
 	gstbasevideoencoder.h \

omx/gstomx.c

@@ -27,6 +27,7 @@
 
 #include "gstomx.h"
 #include "gstomxmpeg4videodec.h"
+#include "gstomxh264videodec.h"
 
 GST_DEBUG_CATEGORY (gstomx_debug);
 #define GST_CAT_DEFAULT gstomx_debug
@@ -1443,6 +1444,10 @@ plugin_init (GstPlugin * plugin)
       gst_element_register (plugin, "omxmpeg4videodec", GST_RANK_PRIMARY,
       GST_TYPE_OMX_MPEG4_VIDEO_DEC);
 
+  ret |=
+      gst_element_register (plugin, "omxh264videodec", GST_RANK_PRIMARY,
+      GST_TYPE_OMX_H264_VIDEO_DEC);
+
   return ret;
 }
 

omx/gstomxh264videodec.c

@@ -0,0 +1,137 @@
+/*
+ * Copyright (C) 2011, Hewlett-Packard Development Company, L.P.
+ *   Author: Sebastian Dröge <sebastian.droege@collabora.co.uk>, Collabora Ltd.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation
+ * version 2.1 of the License.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
+ *
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <gst/gst.h>
+
+#include "gstomxh264videodec.h"
+
+GST_DEBUG_CATEGORY_STATIC (gst_omx_h264_video_dec_debug_category);
+#define GST_CAT_DEFAULT gst_omx_h264_video_dec_debug_category
+
+/* prototypes */
+static void gst_omx_h264_video_dec_finalize (GObject * object);
+static gboolean gst_omx_h264_video_dec_is_format_change (GstOMXVideoDec * dec,
+    GstOMXPort * port, GstVideoState * state);
+static gboolean gst_omx_h264_video_dec_set_format (GstOMXVideoDec * dec,
+    GstOMXPort * port, GstVideoState * state);
+
+enum
+{
+  PROP_0
+};
+
+/* class initialization */
+
+#define DEBUG_INIT(bla) \
+  GST_DEBUG_CATEGORY_INIT (gst_omx_h264_video_dec_debug_category, "omxh264videodec", 0, \
+      "debug category for gst-omx video decoder base class");
+
+GST_BOILERPLATE_FULL (GstOMXH264VideoDec, gst_omx_h264_video_dec,
+    GstOMXVideoDec, GST_TYPE_OMX_VIDEO_DEC, DEBUG_INIT);
+
+static GstStaticPadTemplate gst_omx_h264_video_dec_sink_template =
+GST_STATIC_PAD_TEMPLATE ("sink",
+    GST_PAD_SINK,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS ("video/x-h264, " "parsed=(boolean) true")
+    );
+
+static GstStaticPadTemplate gst_omx_h264_video_dec_src_template =
+GST_STATIC_PAD_TEMPLATE ("src",
+    GST_PAD_SRC,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("I420"))
+    );
+
+static void
+gst_omx_h264_video_dec_base_init (gpointer g_class)
+{
+  GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
+
+  gst_element_class_add_pad_template (element_class,
+      gst_static_pad_template_get (&gst_omx_h264_video_dec_src_template));
+  gst_element_class_add_pad_template (element_class,
+      gst_static_pad_template_get (&gst_omx_h264_video_dec_sink_template));
+
+  gst_element_class_set_details_simple (element_class,
+      "OpenMAX H264 Video Decoder",
+      "Codec/Decoder/Video",
+      "Decode H264 video streams",
+      "Sebastian Dröge <sebastian.droege@collabora.co.uk>");
+}
+
+static void
+gst_omx_h264_video_dec_class_init (GstOMXH264VideoDecClass * klass)
+{
+  GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
+  GstOMXVideoDecClass *videodec_class = GST_OMX_VIDEO_DEC_CLASS (klass);
+
+  gobject_class->finalize = gst_omx_h264_video_dec_finalize;
+
+  /* TODO: Make this configurable */
+  videodec_class->core_name = "/usr/local/lib/libomxil-bellagio.so.0";
+  videodec_class->component_name = "OMX.st.video_decoder.avc";
+  videodec_class->in_port_index = 0;
+  videodec_class->out_port_index = 1;
+
+  videodec_class->is_format_change =
+      GST_DEBUG_FUNCPTR (gst_omx_h264_video_dec_is_format_change);
+  videodec_class->set_format =
+      GST_DEBUG_FUNCPTR (gst_omx_h264_video_dec_set_format);
+}
+
+static void
+gst_omx_h264_video_dec_init (GstOMXH264VideoDec * self,
+    GstOMXH264VideoDecClass * klass)
+{
+}
+
+static void
+gst_omx_h264_video_dec_finalize (GObject * object)
+{
+  /* GstOMXH264VideoDec *self = GST_OMX_H264_VIDEO_DEC (object); */
+
+  G_OBJECT_CLASS (parent_class)->finalize (object);
+}
+
+static gboolean
+gst_omx_h264_video_dec_is_format_change (GstOMXVideoDec * dec,
+    GstOMXPort * port, GstVideoState * state)
+{
+  return FALSE;
+}
+
+static gboolean
+gst_omx_h264_video_dec_set_format (GstOMXVideoDec * dec, GstOMXPort * port,
+    GstVideoState * state)
+{
+  gboolean ret;
+  OMX_PARAM_PORTDEFINITIONTYPE port_def;
+
+  gst_omx_port_get_port_definition (port, &port_def);
+  port_def.format.video.eCompressionFormat = OMX_VIDEO_CodingAVC;
+  ret = gst_omx_port_update_port_definition (port, &port_def);
+
+  return ret;
+}

omx/gstomxh264videodec.h

@@ -0,0 +1,60 @@
+/*
+ * Copyright (C) 2011, Hewlett-Packard Development Company, L.P.
+ *   Author: Sebastian Dröge <sebastian.droege@collabora.co.uk>, Collabora Ltd.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation
+ * version 2.1 of the License.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
+ *
+ */
+
+#ifndef __GST_OMX_H264_VIDEO_DEC_H__
+#define __GST_OMX_H264_VIDEO_DEC_H__
+
+#include <gst/gst.h>
+#include "gstomxvideodec.h"
+
+G_BEGIN_DECLS
+
+#define GST_TYPE_OMX_H264_VIDEO_DEC \
+  (gst_omx_h264_video_dec_get_type())
+#define GST_OMX_H264_VIDEO_DEC(obj) \
+  (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_OMX_H264_VIDEO_DEC,GstOMXH264VideoDec))
+#define GST_OMX_H264_VIDEO_DEC_CLASS(klass) \
+  (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_OMX_H264_VIDEO_DEC,GstOMXH264VideoDecClass))
+#define GST_OMX_H264_VIDEO_DEC_GET_CLASS(obj) \
+  (G_TYPE_INSTANCE_GET_CLASS((obj),GST_TYPE_OMX_H264_VIDEO_DEC,GstOMXH264VideoDecClass))
+#define GST_IS_OMX_H264_VIDEO_DEC(obj) \
+  (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_OMX_H264_VIDEO_DEC))
+#define GST_IS_OMX_H264_VIDEO_DEC_CLASS(obj) \
+  (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_OMX_H264_VIDEO_DEC))
+
+typedef struct _GstOMXH264VideoDec GstOMXH264VideoDec;
+typedef struct _GstOMXH264VideoDecClass GstOMXH264VideoDecClass;
+
+struct _GstOMXH264VideoDec
+{
+  GstOMXVideoDec parent;
+};
+
+struct _GstOMXH264VideoDecClass
+{
+  GstOMXVideoDecClass parent_class;
+};
+
+GType gst_omx_h264_video_dec_get_type (void);
+
+G_END_DECLS
+
+#endif /* __GST_OMX_H264_VIDEO_DEC_H__ */
+