New plugin: h264parse (#340638)

Original commit message from CVS:
Patch by: Michal Benes  <michal dot benes at xeris dot cz>
* configure.ac:
* gst/h264parse/Makefile.am:
* gst/h264parse/gsth264parse.c: (gst_h264_parse_base_init),
(gst_h264_parse_finalize), (gst_h264_parse_class_init),
(gst_h264_parse_init), (gst_h264_parse_chain),
(gst_h264_parse_handle_event), (plugin_init):
* gst/h264parse/gsth264parse.h:
New plugin: h264parse (#340638)

5 files changed, 306 insertions, 0 deletions

diff --git a/ChangeLog b/ChangeLog
index ba7149df..af0dce94 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,16 @@
+2006-05-22  Tim-Philipp Müller  <tim at centricular dot net>
+
+	Patch by: Michal Benes  <michal dot benes at xeris dot cz>
+
+	* configure.ac:
+	* gst/h264parse/Makefile.am:
+	* gst/h264parse/gsth264parse.c: (gst_h264_parse_base_init),
+	(gst_h264_parse_finalize), (gst_h264_parse_class_init),
+	(gst_h264_parse_init), (gst_h264_parse_chain),
+	(gst_h264_parse_handle_event), (plugin_init):
+	* gst/h264parse/gsth264parse.h:
+	  New plugin: h264parse (#340638)
+
 2006-05-22  Thomas Vander Stichele  <thomas at apestaart dot org>
 
 	* gst/gdp/gstgdpdepay.c: (gst_gdp_depay_chain),
diff --git a/configure.ac b/configure.ac
index 592ab421..cfc04a85 100644
--- a/configure.ac
+++ b/configure.ac
@@ -78,6 +78,7 @@ dnl these are all the gst plug-ins, compilable without additional libs
 GST_PLUGINS_ALL="\
   cdxaparse \
   freeze \
+  h264parse \
   modplug \
   spectrum \
   speed \
@@ -703,6 +704,7 @@ gst-plugins-bad.spec
 gst/Makefile
 gst/cdxaparse/Makefile
 gst/freeze/Makefile
+gst/h264parse/Makefile
 gst/modplug/Makefile
 gst/modplug/libmodplug/Makefile
 gst/spectrum/Makefile
diff --git a/gst/h264parse/Makefile.am b/gst/h264parse/Makefile.am
new file mode 100644
index 00000000..40122d4d
--- /dev/null
+++ b/gst/h264parse/Makefile.am
@@ -0,0 +1,11 @@
+plugin_LTLIBRARIES = libgsth264parse.la
+
+libgsth264parse_la_SOURCES = \
+	gsth264parse.c
+
+noinst_HEADERS = \
+	gsth264parse.h
+
+libgsth264parse_la_CFLAGS = $(GST_CFLAGS) $(GST_BASE_CFLAGS)
+libgsth264parse_la_LIBADD = $(GST_LIBS) $(GST_BASE_LIBS)
+libgsth264parse_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS)
diff --git a/gst/h264parse/gsth264parse.c b/gst/h264parse/gsth264parse.c
new file mode 100644
index 00000000..ca1d8ea3
--- /dev/null
+++ b/gst/h264parse/gsth264parse.c
@@ -0,0 +1,216 @@
+/* GStreamer h264 parser
+ * Copyright (C) 2005 Michal Benes <michal.benes@itonis.tv>
+ *
+ * gsth264parse.c:
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Library General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, or (at your option) any later version.
+ *
+ * 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
+ * Library General Public License for more details.
+ *
+ * You should have received a copy of the GNU Library General Public
+ * License along with this library; if not, write to the
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+
+#include <stdlib.h>
+#include <string.h>
+
+#ifdef HAVE_CONFIG_H
+#  include "config.h"
+#endif
+
+#include "gsth264parse.h"
+
+static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
+    GST_PAD_SINK,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS ("video/x-h264"));
+
+static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
+    GST_PAD_SRC,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS ("video/x-h264"));
+
+GST_DEBUG_CATEGORY_STATIC (h264_parse_debug);
+#define GST_CAT_DEFAULT h264_parse_debug
+
+static const GstElementDetails gst_h264_parse_details =
+GST_ELEMENT_DETAILS ("H264Parse",
+    "Codec/Parser",
+    "Parses raw h264 stream",
+    "Michal Benes <michal.benes@itonis.tv>");
+
+GST_BOILERPLATE (GstH264Parse, gst_h264_parse, GstElement, GST_TYPE_ELEMENT);
+
+static void gst_h264_parse_finalize (GObject * object);
+static GstFlowReturn gst_h264_parse_chain (GstPad * pad, GstBuffer * buf);
+static gboolean gst_h264_parse_handle_event (GstPad * pad, GstEvent * event);
+
+static void
+gst_h264_parse_base_init (gpointer g_class)
+{
+  GstElementClass *gstelement_class = GST_ELEMENT_CLASS (g_class);
+
+  gst_element_class_add_pad_template (gstelement_class,
+      gst_static_pad_template_get (&srctemplate));
+  gst_element_class_add_pad_template (gstelement_class,
+      gst_static_pad_template_get (&sinktemplate));
+  gst_element_class_set_details (gstelement_class, &gst_h264_parse_details);
+
+  GST_DEBUG_CATEGORY_INIT (h264_parse_debug, "h264parse", 0, "h264 parser");
+}
+
+static void
+gst_h264_parse_finalize (GObject * object)
+{
+  GstH264Parse *h264parse;
+
+  h264parse = GST_H264PARSE (object);
+
+  G_OBJECT_CLASS (parent_class)->finalize (object);
+}
+
+static void
+gst_h264_parse_class_init (GstH264ParseClass * klass)
+{
+  GObjectClass *gobject_class;
+
+  gobject_class = G_OBJECT_CLASS (klass);
+
+  gobject_class->finalize = GST_DEBUG_FUNCPTR (gst_h264_parse_finalize);
+}
+
+static void
+gst_h264_parse_init (GstH264Parse * h264parse, GstH264ParseClass * g_class)
+{
+  h264parse->sink = gst_pad_new_from_static_template (&sinktemplate, "sink");
+  gst_pad_set_chain_function (h264parse->sink,
+      GST_DEBUG_FUNCPTR (gst_h264_parse_chain));
+  gst_pad_set_event_function (h264parse->sink,
+      GST_DEBUG_FUNCPTR (gst_h264_parse_handle_event));
+  gst_element_add_pad (GST_ELEMENT (h264parse), h264parse->sink);
+
+  h264parse->src = gst_pad_new_from_static_template (&srctemplate, "src");
+  gst_element_add_pad (GST_ELEMENT (h264parse), h264parse->src);
+
+  h264parse->adapter = gst_adapter_new ();
+}
+
+static GstFlowReturn
+gst_h264_parse_chain (GstPad * pad, GstBuffer * buffer)
+{
+  GstFlowReturn res = GST_FLOW_OK;
+  GstH264Parse *h264parse;
+
+  h264parse = GST_H264PARSE (GST_PAD_PARENT (pad));
+
+  if (!GST_PAD_CAPS (h264parse->src)) {
+    GstCaps *caps;
+
+    /* Forward sink caps if possible */
+    caps = GST_PAD_CAPS (h264parse->sink);
+    if (caps)
+      gst_caps_ref (caps);
+
+    /* Set default caps if the sink caps were not negotiated */
+    if (!caps)
+      caps = gst_caps_new_simple ("video/x-h264", NULL);
+
+    /* Set source caps */
+    if (!gst_pad_set_caps (h264parse->src, caps)) {
+      GST_ELEMENT_ERROR (GST_ELEMENT (h264parse),
+          CORE, NEGOTIATION, (NULL), ("failed to set caps"));
+      gst_caps_unref (caps);
+      return GST_FLOW_ERROR;
+    }
+    gst_caps_unref (caps);
+  }
+
+  gst_adapter_push (h264parse->adapter, buffer);
+
+  while (res == GST_FLOW_OK) {
+    gint i;
+    gint next_nalu_pos = -1;
+    guint32 nalu_size;
+    gint avail;
+    const guint8 *data;
+
+    avail = gst_adapter_available (h264parse->adapter);
+    if (avail < 5)
+      break;
+    data = gst_adapter_peek (h264parse->adapter, avail);
+
+    nalu_size = (data[0] << 24)
+        + (data[1] << 16) + (data[2] << 8) + data[3];
+
+    if (nalu_size == 1) {
+      /* Bytestream format */
+      /* Find next NALU header */
+      for (i = 1; i < avail - 4; ++i) {
+        if (data[i + 0] == 0 && data[i + 1] == 0 && data[i + 2] == 0
+            && data[i + 3] == 1) {
+          next_nalu_pos = i;
+          break;
+        }
+      }
+    } else {
+      /* Packetized format */
+      if (nalu_size + 4 <= avail)
+        next_nalu_pos = nalu_size + 4;
+    }
+
+    /* Send packet */
+    if (next_nalu_pos > 0) {
+      GstBuffer *outbuf;
+
+      outbuf = gst_buffer_new_and_alloc (next_nalu_pos);
+      memcpy (GST_BUFFER_DATA (outbuf), data, next_nalu_pos);
+      gst_adapter_flush (h264parse->adapter, next_nalu_pos);
+
+      gst_buffer_set_caps (outbuf, GST_PAD_CAPS (h264parse->src));
+      GST_BUFFER_TIMESTAMP (outbuf) = GST_BUFFER_TIMESTAMP (buffer);
+      res = gst_pad_push (h264parse->src, outbuf);
+    } else {
+      break;                    /* NALU can not be parsed yet */
+    }
+  }
+
+  return res;
+}
+
+
+static gboolean
+gst_h264_parse_handle_event (GstPad * pad, GstEvent * event)
+{
+  GstH264Parse *h264parse;
+  gboolean res;
+
+  h264parse = GST_H264PARSE (gst_pad_get_parent (pad));
+
+  res = gst_pad_event_default (pad, event);
+
+  gst_object_unref (h264parse);
+  return res;
+}
+
+
+static gboolean
+plugin_init (GstPlugin * plugin)
+{
+  return gst_element_register (plugin, "h264parse",
+      GST_RANK_NONE, GST_TYPE_H264PARSE);
+}
+
+GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
+    GST_VERSION_MINOR,
+    "h264parse",
+    "Element parsing raw h264 streams",
+    plugin_init, VERSION, "LGPL", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)
diff --git a/gst/h264parse/gsth264parse.h b/gst/h264parse/gsth264parse.h
new file mode 100644
index 00000000..8f283be9
--- /dev/null
+++ b/gst/h264parse/gsth264parse.h
@@ -0,0 +1,64 @@
+/* GStreamer h264 parser
+ * Copyright (C) 2005 Michal Benes <michal.benes@itonis.tv>
+ *
+ * gsth264parse.h:
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Library General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, or (at your option) any later version.
+ *
+ * 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
+ * Library General Public License for more details.
+ *
+ * You should have received a copy of the GNU Library General Public
+ * License along with this library; if not, write to the
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+
+#ifndef __GST_H264_PARSE_H__
+#define __GST_H264_PARSE_H__
+
+#include <gst/gst.h>
+#include <gst/base/gstadapter.h>
+
+G_BEGIN_DECLS
+
+#define GST_TYPE_H264PARSE \
+  (gst_h264_parse_get_type())
+#define GST_H264PARSE(obj) \
+  (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_H264PARSE,GstH264Parse))
+#define GST_H264PARSE_CLASS(klass) \
+  (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_H264PARSE,GstH264ParseClass))
+#define GST_IS_H264PARSE(obj) \
+  (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_H264PARSE))
+#define GST_IS_H264PARSE_CLASS(klass) \
+  (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_H264PARSE))
+
+typedef struct _GstH264Parse GstH264Parse;
+typedef struct _GstH264ParseClass GstH264ParseClass;
+
+struct _GstH264Parse
+{
+  GstElement element;
+
+  GstPad *sink;
+  GstPad *src;
+
+  GstAdapter *adapter;
+};
+
+struct _GstH264ParseClass
+{
+  GstElementClass parent_class;
+};
+
+GType gst_h264_parse_get_type (void);
+
+G_END_DECLS
+
+#endif /* __GST_H264_PARSE_H__ */