ext/mpeg2enc/gstmpeg2enc.cc - imx-gst-plugins-bad - Git at Google

 /* GStreamer mpeg2enc (mjpegtools) wrapper
  * (c) 2003 Ronald Bultje <rbultje@ronald.bitfreak.net>
  *
  * gstmpeg2enc.cc: gstreamer wrapping
  *
  * 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.
  */

 #ifdef HAVE_CONFIG_H
 #include "config.h"
 #endif

 #include "gstmpeg2enc.hh"

 /*
  * We can't use fractions in static pad templates, so
  * we do something manual...
  */
 static void
 add_fps (GstCaps *caps)
 {
   GstStructure *structure = gst_caps_get_structure (caps, 0);
   GValue list = { 0 }, fps = { 0 };
   gdouble fpss[] = { 24.0/1.001, 24.0, 25.0,
 		     30.0/1.001, 30.0, 50.0,
 		     60.0/1.001, 60.0, 0 };
   guint n;

   g_value_init (&list, GST_TYPE_LIST);
   g_value_init (&fps, G_TYPE_DOUBLE);
   for (n = 0; fpss[n] != 0; n++) {
     g_value_set_double (&fps, fpss[n]);
     gst_value_list_append_value (&list, &fps);
   }
   gst_structure_set_value (structure, "framerate", &list);
   g_value_unset (&list);
   g_value_unset (&fps);
 }

 static GstPadTemplate *
 sink_templ (void)
 {
   static GstPadTemplate *templ = NULL;

   if (!templ) {
     GstCaps *caps;

     caps = gst_caps_new_simple ("video/x-raw-yuv",
 				"format", GST_TYPE_FOURCC,
 					GST_MAKE_FOURCC ('I','4','2','0'),
 				"width",  GST_TYPE_INT_RANGE, 16, 4096,
 				"height", GST_TYPE_INT_RANGE, 16, 4096,
 				NULL);
     add_fps (caps);

     templ = gst_pad_template_new ("sink", GST_PAD_SINK,
 				  GST_PAD_ALWAYS, caps);
   }

   return templ;
 }

 static GstPadTemplate *
 src_templ (void)
 {
   static GstPadTemplate *templ = NULL;

   if (!templ) {
     GstCaps *caps;

     caps = gst_caps_new_simple ("video/mpeg",
 				"systemstream", G_TYPE_BOOLEAN, FALSE,
 				"mpegversion", GST_TYPE_INT_RANGE, 1, 2,
 				"width",  GST_TYPE_INT_RANGE, 16, 4096,
 				"height", GST_TYPE_INT_RANGE, 16, 4096,
 				NULL);
     add_fps (caps);

     templ = gst_pad_template_new ("src", GST_PAD_SRC,
 				  GST_PAD_ALWAYS, caps);
   }

   return templ;
 }

 static void gst_mpeg2enc_base_init    (GstMpeg2encClass *klass);
 static void gst_mpeg2enc_class_init   (GstMpeg2encClass *klass);
 static void gst_mpeg2enc_init         (GstMpeg2enc  *enc);
 static void gst_mpeg2enc_dispose      (GObject      *object);

 static void gst_mpeg2enc_loop         (GstElement   *element);

 static GstPadLinkReturn
             gst_mpeg2enc_sink_link    (GstPad       *pad,
 				       const GstCaps *caps);

 static GstElementStateReturn
             gst_mpeg2enc_change_state (GstElement  *element);

 static void gst_mpeg2enc_get_property (GObject      *object,
 				       guint         prop_id,
 				       GValue       *value,
 				       GParamSpec   *pspec);
 static void gst_mpeg2enc_set_property (GObject      *object,
 				       guint         prop_id,
 				       const GValue *value,
 				       GParamSpec   *pspec);

 static GstElementClass *parent_class = NULL;

 GType
 gst_mpeg2enc_get_type (void)
 {
   static GType gst_mpeg2enc_type = 0;

   if (!gst_mpeg2enc_type) {
     static const GTypeInfo gst_mpeg2enc_info = {
       sizeof (GstMpeg2encClass),
       (GBaseInitFunc) gst_mpeg2enc_base_init,
       NULL,
       (GClassInitFunc) gst_mpeg2enc_class_init,
       NULL,
       NULL,
       sizeof (GstMpeg2enc),
       0,
       (GInstanceInitFunc) gst_mpeg2enc_init,
     };

     gst_mpeg2enc_type =
 	g_type_register_static (GST_TYPE_ELEMENT,
 				"GstMpeg2enc",
 				&gst_mpeg2enc_info,
 				(GTypeFlags) 0);
   }

   return gst_mpeg2enc_type;
 }

 static void
 gst_mpeg2enc_base_init (GstMpeg2encClass *klass)
 {
   static GstElementDetails gst_mpeg2enc_details = {
     "mpeg2enc video encoder",
     "Codec/Video/Encoder",
     "High-quality MPEG-1/2 video encoder",
     "Andrew Stevens <andrew.stevens@nexgo.de>\n"
     "Ronald Bultje <rbultje@ronald.bitfreak.net>"
   };
   GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

   gst_element_class_add_pad_template (element_class, src_templ ());
   gst_element_class_add_pad_template (element_class, sink_templ ());
   gst_element_class_set_details (element_class,
 				 &gst_mpeg2enc_details);
 }

 static void
 gst_mpeg2enc_class_init (GstMpeg2encClass *klass)
 {
   GObjectClass *object_class = G_OBJECT_CLASS (klass);
   GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

   parent_class = GST_ELEMENT_CLASS (g_type_class_ref (GST_TYPE_ELEMENT));

   /* register arguments */
   mjpeg_default_handler_verbosity (0);
   GstMpeg2EncOptions::initProperties (object_class);

   object_class->set_property = gst_mpeg2enc_set_property;
   object_class->get_property = gst_mpeg2enc_get_property;

   object_class->dispose = gst_mpeg2enc_dispose;

   element_class->change_state = gst_mpeg2enc_change_state;
 }

 static void
 gst_mpeg2enc_dispose (GObject *object)
 {
   GstMpeg2enc *enc = GST_MPEG2ENC (object);

   if (enc->encoder) {
     delete enc->encoder;
     enc->encoder = NULL;
   }
   delete enc->options;
 }

 static void
 gst_mpeg2enc_init (GstMpeg2enc *enc)
 {
   GstElement *element = GST_ELEMENT (enc);
   GstElementClass *klass = GST_ELEMENT_GET_CLASS (element);

   GST_FLAG_SET (element, GST_ELEMENT_EVENT_AWARE);

   enc->sinkpad = gst_pad_new_from_template (
 	gst_element_class_get_pad_template (klass, "sink"), "sink");
   gst_pad_set_link_function (enc->sinkpad, gst_mpeg2enc_sink_link);
   gst_element_add_pad (element, enc->sinkpad);

   enc->srcpad = gst_pad_new_from_template (
 	gst_element_class_get_pad_template (klass, "src"), "src");
   gst_pad_use_explicit_caps (enc->srcpad);
   gst_element_add_pad (element, enc->srcpad);

   enc->options = new GstMpeg2EncOptions ();

   gst_element_set_loop_function (element, gst_mpeg2enc_loop);

   enc->encoder = NULL;
 }

 static void
 gst_mpeg2enc_loop (GstElement *element)
 {
   GstMpeg2enc *enc = GST_MPEG2ENC (element);

   if (!enc->encoder) {
     const GstCaps *caps;
     GstCaps *othercaps;
     GstData *data;

     /* make sure we've had data */
     data = gst_pad_pull (enc->sinkpad);
     gst_pad_set_element_private (enc->sinkpad, data);

     if (!(caps = GST_PAD_CAPS (enc->sinkpad))) {
       GST_ELEMENT_ERROR (element, CORE, NEGOTIATION, (""),
 			 ("format wasn't negotiated before loop function"));
       return;
     }

     /* create new encoder with these settings */
     enc->encoder = new GstMpeg2Encoder (enc->options, enc->sinkpad,
 					caps, enc->srcpad);

     /* and set caps on other side */
     othercaps = enc->encoder->getFormat ();
     if (gst_pad_set_explicit_caps (enc->srcpad, othercaps) <= 0) {
       GST_ELEMENT_ERROR (element, CORE, NEGOTIATION, (""), (""));
       delete enc->encoder;
       enc->encoder = NULL;
       return;
     }
   }

   enc->encoder->encodePicture ();
   gst_pad_event_default (enc->sinkpad, gst_event_new (GST_EVENT_EOS));
 }

 static GstPadLinkReturn
 gst_mpeg2enc_sink_link (GstPad        *pad,
 			const GstCaps *caps)
 {
   GstMpeg2enc *enc = GST_MPEG2ENC (gst_pad_get_parent (pad));

   if (!gst_caps_is_fixed (caps))
     return GST_PAD_LINK_DELAYED;

   if (enc->encoder) {
     delete enc->encoder;
     enc->encoder = NULL;
   }

   return GST_PAD_LINK_OK;
 }

 static void
 gst_mpeg2enc_get_property (GObject    *object,
 			   guint       prop_id,
 			   GValue     *value,
 			   GParamSpec *pspec)
 {
   GST_MPEG2ENC (object)->options->getProperty (prop_id, value);
 }

 static void
 gst_mpeg2enc_set_property (GObject      *object,
 			   guint         prop_id,
 			   const GValue *value,
 			   GParamSpec   *pspec)
 {
   GST_MPEG2ENC (object)->options->setProperty (prop_id, value);
 }

 static GstElementStateReturn
 gst_mpeg2enc_change_state (GstElement  *element)
 {
   GstMpeg2enc *enc = GST_MPEG2ENC (element);

   switch (GST_STATE_TRANSITION (element)) {
     case GST_STATE_PAUSED_TO_READY:
       delete enc->encoder;
       enc->encoder = NULL;
       break;
     default:
       break;
   }

   if (parent_class->change_state)
     return parent_class->change_state (element);

   return GST_STATE_SUCCESS;
 }

 static gboolean
 plugin_init (GstPlugin *plugin)
 {
   return gst_element_register (plugin, "mpeg2enc",
 			       GST_RANK_NONE,
 			       GST_TYPE_MPEG2ENC);
 }

 GST_PLUGIN_DEFINE (
   GST_VERSION_MAJOR,
   GST_VERSION_MINOR,
   "mpeg2enc",
   "High-quality MPEG-1/2 video encoder",
   plugin_init,
   VERSION,
   "GPL",
   GST_PACKAGE,
   GST_ORIGIN
 )
	/* GStreamer mpeg2enc (mjpegtools) wrapper
	* (c) 2003 Ronald Bultje <rbultje@ronald.bitfreak.net>
	*
	* gstmpeg2enc.cc: gstreamer wrapping
	*
	* 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.
	*/

	#ifdef HAVE_CONFIG_H
	#include "config.h"
	#endif

	#include "gstmpeg2enc.hh"

	/*
	* We can't use fractions in static pad templates, so
	* we do something manual...
	*/
	static void
	add_fps (GstCaps *caps)
	{
	GstStructure *structure = gst_caps_get_structure (caps, 0);
	GValue list = { 0 }, fps = { 0 };
	gdouble fpss[] = { 24.0/1.001, 24.0, 25.0,
	30.0/1.001, 30.0, 50.0,
	60.0/1.001, 60.0, 0 };
	guint n;

	g_value_init (&list, GST_TYPE_LIST);
	g_value_init (&fps, G_TYPE_DOUBLE);
	for (n = 0; fpss[n] != 0; n++) {
	g_value_set_double (&fps, fpss[n]);
	gst_value_list_append_value (&list, &fps);
	}
	gst_structure_set_value (structure, "framerate", &list);
	g_value_unset (&list);
	g_value_unset (&fps);
	}

	static GstPadTemplate *
	sink_templ (void)
	{
	static GstPadTemplate *templ = NULL;

	if (!templ) {
	GstCaps *caps;

	caps = gst_caps_new_simple ("video/x-raw-yuv",
	"format", GST_TYPE_FOURCC,
	GST_MAKE_FOURCC ('I','4','2','0'),
	"width", GST_TYPE_INT_RANGE, 16, 4096,
	"height", GST_TYPE_INT_RANGE, 16, 4096,
	NULL);
	add_fps (caps);

	templ = gst_pad_template_new ("sink", GST_PAD_SINK,
	GST_PAD_ALWAYS, caps);
	}

	return templ;
	}

	static GstPadTemplate *
	src_templ (void)
	{
	static GstPadTemplate *templ = NULL;

	if (!templ) {
	GstCaps *caps;

	caps = gst_caps_new_simple ("video/mpeg",
	"systemstream", G_TYPE_BOOLEAN, FALSE,
	"mpegversion", GST_TYPE_INT_RANGE, 1, 2,
	"width", GST_TYPE_INT_RANGE, 16, 4096,
	"height", GST_TYPE_INT_RANGE, 16, 4096,
	NULL);
	add_fps (caps);

	templ = gst_pad_template_new ("src", GST_PAD_SRC,
	GST_PAD_ALWAYS, caps);
	}

	return templ;
	}

	static void gst_mpeg2enc_base_init (GstMpeg2encClass *klass);
	static void gst_mpeg2enc_class_init (GstMpeg2encClass *klass);
	static void gst_mpeg2enc_init (GstMpeg2enc *enc);
	static void gst_mpeg2enc_dispose (GObject *object);

	static void gst_mpeg2enc_loop (GstElement *element);

	static GstPadLinkReturn
	gst_mpeg2enc_sink_link (GstPad *pad,
	const GstCaps *caps);

	static GstElementStateReturn
	gst_mpeg2enc_change_state (GstElement *element);

	static void gst_mpeg2enc_get_property (GObject *object,
	guint prop_id,
	GValue *value,
	GParamSpec *pspec);
	static void gst_mpeg2enc_set_property (GObject *object,
	guint prop_id,
	const GValue *value,
	GParamSpec *pspec);

	static GstElementClass *parent_class = NULL;

	GType
	gst_mpeg2enc_get_type (void)
	{
	static GType gst_mpeg2enc_type = 0;

	if (!gst_mpeg2enc_type) {
	static const GTypeInfo gst_mpeg2enc_info = {
	sizeof (GstMpeg2encClass),
	(GBaseInitFunc) gst_mpeg2enc_base_init,
	NULL,
	(GClassInitFunc) gst_mpeg2enc_class_init,
	NULL,
	NULL,
	sizeof (GstMpeg2enc),
	0,
	(GInstanceInitFunc) gst_mpeg2enc_init,
	};

	gst_mpeg2enc_type =
	g_type_register_static (GST_TYPE_ELEMENT,
	"GstMpeg2enc",
	&gst_mpeg2enc_info,
	(GTypeFlags) 0);
	}

	return gst_mpeg2enc_type;
	}

	static void
	gst_mpeg2enc_base_init (GstMpeg2encClass *klass)
	{
	static GstElementDetails gst_mpeg2enc_details = {
	"mpeg2enc video encoder",
	"Codec/Video/Encoder",
	"High-quality MPEG-1/2 video encoder",
	"Andrew Stevens <andrew.stevens@nexgo.de>\n"
	"Ronald Bultje <rbultje@ronald.bitfreak.net>"
	};
	GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

	gst_element_class_add_pad_template (element_class, src_templ ());
	gst_element_class_add_pad_template (element_class, sink_templ ());
	gst_element_class_set_details (element_class,
	&gst_mpeg2enc_details);
	}

	static void
	gst_mpeg2enc_class_init (GstMpeg2encClass *klass)
	{
	GObjectClass *object_class = G_OBJECT_CLASS (klass);
	GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

	parent_class = GST_ELEMENT_CLASS (g_type_class_ref (GST_TYPE_ELEMENT));

	/* register arguments */
	mjpeg_default_handler_verbosity (0);
	GstMpeg2EncOptions::initProperties (object_class);

	object_class->set_property = gst_mpeg2enc_set_property;
	object_class->get_property = gst_mpeg2enc_get_property;

	object_class->dispose = gst_mpeg2enc_dispose;

	element_class->change_state = gst_mpeg2enc_change_state;
	}

	static void
	gst_mpeg2enc_dispose (GObject *object)
	{
	GstMpeg2enc *enc = GST_MPEG2ENC (object);

	if (enc->encoder) {
	delete enc->encoder;
	enc->encoder = NULL;
	}
	delete enc->options;
	}

	static void
	gst_mpeg2enc_init (GstMpeg2enc *enc)
	{
	GstElement *element = GST_ELEMENT (enc);
	GstElementClass *klass = GST_ELEMENT_GET_CLASS (element);

	GST_FLAG_SET (element, GST_ELEMENT_EVENT_AWARE);

	enc->sinkpad = gst_pad_new_from_template (
	gst_element_class_get_pad_template (klass, "sink"), "sink");
	gst_pad_set_link_function (enc->sinkpad, gst_mpeg2enc_sink_link);
	gst_element_add_pad (element, enc->sinkpad);

	enc->srcpad = gst_pad_new_from_template (
	gst_element_class_get_pad_template (klass, "src"), "src");
	gst_pad_use_explicit_caps (enc->srcpad);
	gst_element_add_pad (element, enc->srcpad);

	enc->options = new GstMpeg2EncOptions ();

	gst_element_set_loop_function (element, gst_mpeg2enc_loop);

	enc->encoder = NULL;
	}

	static void
	gst_mpeg2enc_loop (GstElement *element)
	{
	GstMpeg2enc *enc = GST_MPEG2ENC (element);

	if (!enc->encoder) {
	const GstCaps *caps;
	GstCaps *othercaps;
	GstData *data;

	/* make sure we've had data */
	data = gst_pad_pull (enc->sinkpad);
	gst_pad_set_element_private (enc->sinkpad, data);

	if (!(caps = GST_PAD_CAPS (enc->sinkpad))) {
	GST_ELEMENT_ERROR (element, CORE, NEGOTIATION, (""),
	("format wasn't negotiated before loop function"));
	return;
	}

	/* create new encoder with these settings */
	enc->encoder = new GstMpeg2Encoder (enc->options, enc->sinkpad,
	caps, enc->srcpad);

	/* and set caps on other side */
	othercaps = enc->encoder->getFormat ();
	if (gst_pad_set_explicit_caps (enc->srcpad, othercaps) <= 0) {
	GST_ELEMENT_ERROR (element, CORE, NEGOTIATION, (""), (""));
	delete enc->encoder;
	enc->encoder = NULL;
	return;
	}
	}

	enc->encoder->encodePicture ();
	gst_pad_event_default (enc->sinkpad, gst_event_new (GST_EVENT_EOS));
	}

	static GstPadLinkReturn
	gst_mpeg2enc_sink_link (GstPad *pad,
	const GstCaps *caps)
	{
	GstMpeg2enc *enc = GST_MPEG2ENC (gst_pad_get_parent (pad));

	if (!gst_caps_is_fixed (caps))
	return GST_PAD_LINK_DELAYED;

	if (enc->encoder) {
	delete enc->encoder;
	enc->encoder = NULL;
	}

	return GST_PAD_LINK_OK;
	}

	static void
	gst_mpeg2enc_get_property (GObject *object,
	guint prop_id,
	GValue *value,
	GParamSpec *pspec)
	{
	GST_MPEG2ENC (object)->options->getProperty (prop_id, value);
	}

	static void
	gst_mpeg2enc_set_property (GObject *object,
	guint prop_id,
	const GValue *value,
	GParamSpec *pspec)
	{
	GST_MPEG2ENC (object)->options->setProperty (prop_id, value);
	}

	static GstElementStateReturn
	gst_mpeg2enc_change_state (GstElement *element)
	{
	GstMpeg2enc *enc = GST_MPEG2ENC (element);

	switch (GST_STATE_TRANSITION (element)) {
	case GST_STATE_PAUSED_TO_READY:
	delete enc->encoder;
	enc->encoder = NULL;
	break;
	default:
	break;
	}

	if (parent_class->change_state)
	return parent_class->change_state (element);

	return GST_STATE_SUCCESS;
	}

	static gboolean
	plugin_init (GstPlugin *plugin)
	{
	return gst_element_register (plugin, "mpeg2enc",
	GST_RANK_NONE,
	GST_TYPE_MPEG2ENC);
	}

	GST_PLUGIN_DEFINE (
	GST_VERSION_MAJOR,
	GST_VERSION_MINOR,
	"mpeg2enc",
	"High-quality MPEG-1/2 video encoder",
	plugin_init,
	VERSION,
	"GPL",
	GST_PACKAGE,
	GST_ORIGIN
	)