From 8cdc3e5e8c6df139b3667649da3466368f8131ea Mon Sep 17 00:00:00 2001 From: Michal Benes Date: Mon, 22 May 2006 16:08:34 +0000 Subject: [PATCH] New plugin: h264parse (#340638) Original commit message from CVS: Patch by: Michal Benes * 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) --- ChangeLog | 13 +++ configure.ac | 2 + gst/h264parse/Makefile.am | 11 ++ gst/h264parse/gsth264parse.c | 216 +++++++++++++++++++++++++++++++++++ gst/h264parse/gsth264parse.h | 64 +++++++++++ 5 files changed, 306 insertions(+) create mode 100644 gst/h264parse/Makefile.am create mode 100644 gst/h264parse/gsth264parse.c create mode 100644 gst/h264parse/gsth264parse.h diff --git a/ChangeLog b/ChangeLog index ba7149dfe8..af0dce94c5 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,16 @@ +2006-05-22 Tim-Philipp Müller + + Patch by: Michal Benes + + * 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 * gst/gdp/gstgdpdepay.c: (gst_gdp_depay_chain), diff --git a/configure.ac b/configure.ac index 592ab4211a..cfc04a8582 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 0000000000..40122d4d8b --- /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 0000000000..ca1d8ea33e --- /dev/null +++ b/gst/h264parse/gsth264parse.c @@ -0,0 +1,216 @@ +/* GStreamer h264 parser + * Copyright (C) 2005 Michal Benes + * + * 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 +#include + +#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 "); + +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 0000000000..8f283be9f3 --- /dev/null +++ b/gst/h264parse/gsth264parse.h @@ -0,0 +1,64 @@ +/* GStreamer h264 parser + * Copyright (C) 2005 Michal Benes + * + * 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 +#include + +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__ */