gst/debug/breakmydata.c - imx-gst-plugins-good - Git at Google

 /* GStreamer
  * Copyright (C) 2004 Benjamin Otte <otte@gnome.org>
  *
  * 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 <gst/gst.h>
 #include <gst/base/gstbasetransform.h>

 GST_DEBUG_CATEGORY_STATIC (gst_break_my_data_debug);
 #define GST_CAT_DEFAULT gst_break_my_data_debug

 /* This plugin modifies the contents of the buffer it is passed randomly
  * according to the parameters set.
  * It otherwise acts as an identity.
  */

 #define GST_TYPE_BREAK_MY_DATA \
   (gst_break_my_data_get_type())
 #define GST_BREAK_MY_DATA(obj) \
   (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_BREAK_MY_DATA,GstBreakMyData))
 #define GST_BREAK_MY_DATA_CLASS(klass) \
   (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_BREAK_MY_DATA,GstBreakMyDataClass))
 #define GST_IS_BREAK_MY_DATA(obj) \
   (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_BREAK_MY_DATA))
 #define GST_IS_BREAK_MY_DATA_CLASS(klass) \
   (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_BREAK_MY_DATA))

 enum
 {
   ARG_0,
   ARG_SEED,
   ARG_SET_TO,
   ARG_SKIP,
   ARG_PROBABILITY
 };

 typedef struct _GstBreakMyData GstBreakMyData;
 typedef struct _GstBreakMyDataClass GstBreakMyDataClass;

 struct _GstBreakMyData
 {
   GstBaseTransform basetransform;

   GRand *rand;
   guint skipped;

   guint32 seed;
   gint set;
   guint skip;
   gdouble probability;
 };

 struct _GstBreakMyDataClass
 {
   GstBaseTransformClass parent_class;
 };

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

 static GstFlowReturn gst_break_my_data_transform_ip (GstBaseTransform * trans,
     GstBuffer * buf);
 static gboolean gst_break_my_data_stop (GstBaseTransform * trans);
 static gboolean gst_break_my_data_start (GstBaseTransform * trans);

 static const GstElementDetails details = GST_ELEMENT_DETAILS ("Break my data",
     "Testing",
     "randomly change data in the stream",
     "Benjamin Otte <otte@gnome>");

 GstStaticPadTemplate bmd_src_template = GST_STATIC_PAD_TEMPLATE ("src",
     GST_PAD_SRC,
     GST_PAD_ALWAYS,
     GST_STATIC_CAPS_ANY);

 GstStaticPadTemplate bmd_sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
     GST_PAD_SINK,
     GST_PAD_ALWAYS,
     GST_STATIC_CAPS_ANY);

 GST_BOILERPLATE (GstBreakMyData, gst_break_my_data, GstBaseTransform,
     GST_TYPE_BASE_TRANSFORM);

 static void
 gst_break_my_data_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 (&bmd_sink_template));
   gst_element_class_add_pad_template (gstelement_class,
       gst_static_pad_template_get (&bmd_src_template));

   gst_element_class_set_details (gstelement_class, &details);
 }

 static void
 gst_break_my_data_class_init (GstBreakMyDataClass * klass)
 {
   GstBaseTransformClass *gstbasetrans_class;
   GObjectClass *gobject_class;

   gobject_class = G_OBJECT_CLASS (klass);
   gstbasetrans_class = GST_BASE_TRANSFORM_CLASS (klass);

   gobject_class->set_property =
       GST_DEBUG_FUNCPTR (gst_break_my_data_set_property);
   gobject_class->get_property =
       GST_DEBUG_FUNCPTR (gst_break_my_data_get_property);

   g_object_class_install_property (gobject_class, ARG_SEED,
       g_param_spec_uint ("seed", "seed",
           "seed for randomness (initialized when going from READY to PAUSED)",
           0, 0xFFFFFFFF, 0, G_PARAM_READWRITE | G_PARAM_CONSTRUCT));
   g_object_class_install_property (gobject_class, ARG_SET_TO,
       g_param_spec_int ("set-to", "set-to",
           "set changed bytes to this value (-1 means random value",
           -1, 255, -1, G_PARAM_READWRITE | G_PARAM_CONSTRUCT));
   g_object_class_install_property (gobject_class, ARG_SKIP,
       g_param_spec_uint ("skip", "skip",
           "amount of bytes skipped at the beginning of stream",
           0, G_MAXUINT, 0, G_PARAM_READWRITE | G_PARAM_CONSTRUCT));
   g_object_class_install_property (gobject_class, ARG_PROBABILITY,
       g_param_spec_double ("probability", "probability",
           "probability for each byte in the buffer to be changed", 0.0, 1.0,
           0.0, G_PARAM_READWRITE | G_PARAM_CONSTRUCT));

   gstbasetrans_class->transform_ip =
       GST_DEBUG_FUNCPTR (gst_break_my_data_transform_ip);
   gstbasetrans_class->start = GST_DEBUG_FUNCPTR (gst_break_my_data_start);
   gstbasetrans_class->stop = GST_DEBUG_FUNCPTR (gst_break_my_data_stop);
 }

 static void
 gst_break_my_data_init (GstBreakMyData * bmd, GstBreakMyDataClass * g_class)
 {
   gst_base_transform_set_in_place (GST_BASE_TRANSFORM (bmd), TRUE);
   gst_base_transform_set_passthrough (GST_BASE_TRANSFORM (bmd), TRUE);
 }

 static void
 gst_break_my_data_set_property (GObject * object, guint prop_id,
     const GValue * value, GParamSpec * pspec)
 {
   GstBreakMyData *bmd = GST_BREAK_MY_DATA (object);

   GST_OBJECT_LOCK (bmd);

   switch (prop_id) {
     case ARG_SEED:
       bmd->seed = g_value_get_uint (value);
       break;
     case ARG_SET_TO:
       bmd->set = g_value_get_int (value);
       break;
     case ARG_SKIP:
       bmd->skip = g_value_get_uint (value);
       break;
     case ARG_PROBABILITY:
       bmd->probability = g_value_get_double (value);
       break;
     default:
       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
       break;
   }

   GST_OBJECT_UNLOCK (bmd);
 }

 static void
 gst_break_my_data_get_property (GObject * object, guint prop_id, GValue * value,
     GParamSpec * pspec)
 {
   GstBreakMyData *bmd = GST_BREAK_MY_DATA (object);

   GST_OBJECT_LOCK (bmd);

   switch (prop_id) {
     case ARG_SEED:
       g_value_set_uint (value, bmd->seed);
       break;
     case ARG_SET_TO:
       g_value_set_int (value, bmd->set);
       break;
     case ARG_SKIP:
       g_value_set_uint (value, bmd->skip);
       break;
     case ARG_PROBABILITY:
       g_value_set_double (value, bmd->probability);
       break;
     default:
       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
       break;
   }

   GST_OBJECT_UNLOCK (bmd);
 }

 static GstFlowReturn
 gst_break_my_data_transform_ip (GstBaseTransform * trans, GstBuffer * buf)
 {
   GstBreakMyData *bmd = GST_BREAK_MY_DATA (trans);
   guint i, size;

   g_return_val_if_fail (gst_buffer_is_writable (buf), GST_FLOW_ERROR);

   GST_OBJECT_LOCK (bmd);

   if (bmd->skipped < bmd->skip) {
     i = bmd->skip - bmd->skipped;
   } else {
     i = 0;
   }

   size = GST_BUFFER_SIZE (buf);

   GST_LOG_OBJECT (bmd,
       "got buffer %p (size %u, timestamp %" G_GUINT64_FORMAT ", offset %"
       G_GUINT64_FORMAT "", buf, size, GST_BUFFER_TIMESTAMP (buf),
       GST_BUFFER_OFFSET (buf));

   for (; i < size; i++) {
     if (g_rand_double_range (bmd->rand, 0, 1.0) <= bmd->probability) {
       guint8 new;

       if (bmd->set < 0) {
         new = g_rand_int_range (bmd->rand, 0, 256);
       } else {
         new = bmd->set;
       }
       GST_INFO_OBJECT (bmd, "changing byte %u from 0x%02X to 0x%02X", i,
           (guint) GST_READ_UINT8 (GST_BUFFER_DATA (buf) + i),
           (guint) ((guint8) new));
       GST_BUFFER_DATA (buf)[i] = new;
     }
   }
   /* don't overflow */
   bmd->skipped += MIN (G_MAXUINT - bmd->skipped, GST_BUFFER_SIZE (buf));

   GST_OBJECT_UNLOCK (bmd);

   return GST_FLOW_OK;
 }

 static gboolean
 gst_break_my_data_start (GstBaseTransform * trans)
 {
   GstBreakMyData *bmd = GST_BREAK_MY_DATA (trans);

   GST_OBJECT_LOCK (bmd);
   bmd->rand = g_rand_new_with_seed (bmd->seed);
   bmd->skipped = 0;
   GST_OBJECT_UNLOCK (bmd);

   return TRUE;
 }

 static gboolean
 gst_break_my_data_stop (GstBaseTransform * trans)
 {
   GstBreakMyData *bmd = GST_BREAK_MY_DATA (trans);

   GST_OBJECT_LOCK (bmd);
   g_rand_free (bmd->rand);
   bmd->rand = NULL;
   GST_OBJECT_UNLOCK (bmd);

   return TRUE;
 }

 gboolean
 gst_break_my_data_plugin_init (GstPlugin * plugin, GstPluginClass * g_class)
 {
   if (!gst_element_register (plugin, "breakmydata", GST_RANK_NONE,
           GST_TYPE_BREAK_MY_DATA))
     return FALSE;

   GST_DEBUG_CATEGORY_INIT (gst_break_my_data_debug, "breakmydata", 0,
       "debugging category for breakmydata element");

   return TRUE;
 }
	/* GStreamer
	* Copyright (C) 2004 Benjamin Otte <otte@gnome.org>
	*
	* 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 <gst/gst.h>
	#include <gst/base/gstbasetransform.h>

	GST_DEBUG_CATEGORY_STATIC (gst_break_my_data_debug);
	#define GST_CAT_DEFAULT gst_break_my_data_debug

	/* This plugin modifies the contents of the buffer it is passed randomly
	* according to the parameters set.
	* It otherwise acts as an identity.
	*/

	#define GST_TYPE_BREAK_MY_DATA \
	(gst_break_my_data_get_type())
	#define GST_BREAK_MY_DATA(obj) \
	(G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_BREAK_MY_DATA,GstBreakMyData))
	#define GST_BREAK_MY_DATA_CLASS(klass) \
	(G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_BREAK_MY_DATA,GstBreakMyDataClass))
	#define GST_IS_BREAK_MY_DATA(obj) \
	(G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_BREAK_MY_DATA))
	#define GST_IS_BREAK_MY_DATA_CLASS(klass) \
	(G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_BREAK_MY_DATA))

	enum
	{
	ARG_0,
	ARG_SEED,
	ARG_SET_TO,
	ARG_SKIP,
	ARG_PROBABILITY
	};

	typedef struct _GstBreakMyData GstBreakMyData;
	typedef struct _GstBreakMyDataClass GstBreakMyDataClass;

	struct _GstBreakMyData
	{
	GstBaseTransform basetransform;

	GRand *rand;
	guint skipped;

	guint32 seed;
	gint set;
	guint skip;
	gdouble probability;
	};

	struct _GstBreakMyDataClass
	{
	GstBaseTransformClass parent_class;
	};

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

	static GstFlowReturn gst_break_my_data_transform_ip (GstBaseTransform * trans,
	GstBuffer * buf);
	static gboolean gst_break_my_data_stop (GstBaseTransform * trans);
	static gboolean gst_break_my_data_start (GstBaseTransform * trans);

	static const GstElementDetails details = GST_ELEMENT_DETAILS ("Break my data",
	"Testing",
	"randomly change data in the stream",
	"Benjamin Otte <otte@gnome>");

	GstStaticPadTemplate bmd_src_template = GST_STATIC_PAD_TEMPLATE ("src",
	GST_PAD_SRC,
	GST_PAD_ALWAYS,
	GST_STATIC_CAPS_ANY);

	GstStaticPadTemplate bmd_sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
	GST_PAD_SINK,
	GST_PAD_ALWAYS,
	GST_STATIC_CAPS_ANY);

	GST_BOILERPLATE (GstBreakMyData, gst_break_my_data, GstBaseTransform,
	GST_TYPE_BASE_TRANSFORM);

	static void
	gst_break_my_data_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 (&bmd_sink_template));
	gst_element_class_add_pad_template (gstelement_class,
	gst_static_pad_template_get (&bmd_src_template));

	gst_element_class_set_details (gstelement_class, &details);
	}

	static void
	gst_break_my_data_class_init (GstBreakMyDataClass * klass)
	{
	GstBaseTransformClass *gstbasetrans_class;
	GObjectClass *gobject_class;

	gobject_class = G_OBJECT_CLASS (klass);
	gstbasetrans_class = GST_BASE_TRANSFORM_CLASS (klass);

	gobject_class->set_property =
	GST_DEBUG_FUNCPTR (gst_break_my_data_set_property);
	gobject_class->get_property =
	GST_DEBUG_FUNCPTR (gst_break_my_data_get_property);

	g_object_class_install_property (gobject_class, ARG_SEED,
	g_param_spec_uint ("seed", "seed",
	"seed for randomness (initialized when going from READY to PAUSED)",
	0, 0xFFFFFFFF, 0, G_PARAM_READWRITE \| G_PARAM_CONSTRUCT));
	g_object_class_install_property (gobject_class, ARG_SET_TO,
	g_param_spec_int ("set-to", "set-to",
	"set changed bytes to this value (-1 means random value",
	-1, 255, -1, G_PARAM_READWRITE \| G_PARAM_CONSTRUCT));
	g_object_class_install_property (gobject_class, ARG_SKIP,
	g_param_spec_uint ("skip", "skip",
	"amount of bytes skipped at the beginning of stream",
	0, G_MAXUINT, 0, G_PARAM_READWRITE \| G_PARAM_CONSTRUCT));
	g_object_class_install_property (gobject_class, ARG_PROBABILITY,
	g_param_spec_double ("probability", "probability",
	"probability for each byte in the buffer to be changed", 0.0, 1.0,
	0.0, G_PARAM_READWRITE \| G_PARAM_CONSTRUCT));

	gstbasetrans_class->transform_ip =
	GST_DEBUG_FUNCPTR (gst_break_my_data_transform_ip);
	gstbasetrans_class->start = GST_DEBUG_FUNCPTR (gst_break_my_data_start);
	gstbasetrans_class->stop = GST_DEBUG_FUNCPTR (gst_break_my_data_stop);
	}

	static void
	gst_break_my_data_init (GstBreakMyData * bmd, GstBreakMyDataClass * g_class)
	{
	gst_base_transform_set_in_place (GST_BASE_TRANSFORM (bmd), TRUE);
	gst_base_transform_set_passthrough (GST_BASE_TRANSFORM (bmd), TRUE);
	}

	static void
	gst_break_my_data_set_property (GObject * object, guint prop_id,
	const GValue * value, GParamSpec * pspec)
	{
	GstBreakMyData *bmd = GST_BREAK_MY_DATA (object);

	GST_OBJECT_LOCK (bmd);

	switch (prop_id) {
	case ARG_SEED:
	bmd->seed = g_value_get_uint (value);
	break;
	case ARG_SET_TO:
	bmd->set = g_value_get_int (value);
	break;
	case ARG_SKIP:
	bmd->skip = g_value_get_uint (value);
	break;
	case ARG_PROBABILITY:
	bmd->probability = g_value_get_double (value);
	break;
	default:
	G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
	break;
	}

	GST_OBJECT_UNLOCK (bmd);
	}

	static void
	gst_break_my_data_get_property (GObject * object, guint prop_id, GValue * value,
	GParamSpec * pspec)
	{
	GstBreakMyData *bmd = GST_BREAK_MY_DATA (object);

	GST_OBJECT_LOCK (bmd);

	switch (prop_id) {
	case ARG_SEED:
	g_value_set_uint (value, bmd->seed);
	break;
	case ARG_SET_TO:
	g_value_set_int (value, bmd->set);
	break;
	case ARG_SKIP:
	g_value_set_uint (value, bmd->skip);
	break;
	case ARG_PROBABILITY:
	g_value_set_double (value, bmd->probability);
	break;
	default:
	G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
	break;
	}

	GST_OBJECT_UNLOCK (bmd);
	}

	static GstFlowReturn
	gst_break_my_data_transform_ip (GstBaseTransform * trans, GstBuffer * buf)
	{
	GstBreakMyData *bmd = GST_BREAK_MY_DATA (trans);
	guint i, size;

	g_return_val_if_fail (gst_buffer_is_writable (buf), GST_FLOW_ERROR);

	GST_OBJECT_LOCK (bmd);

	if (bmd->skipped < bmd->skip) {
	i = bmd->skip - bmd->skipped;
	} else {
	i = 0;
	}

	size = GST_BUFFER_SIZE (buf);

	GST_LOG_OBJECT (bmd,
	"got buffer %p (size %u, timestamp %" G_GUINT64_FORMAT ", offset %"
	G_GUINT64_FORMAT "", buf, size, GST_BUFFER_TIMESTAMP (buf),
	GST_BUFFER_OFFSET (buf));

	for (; i < size; i++) {
	if (g_rand_double_range (bmd->rand, 0, 1.0) <= bmd->probability) {
	guint8 new;

	if (bmd->set < 0) {
	new = g_rand_int_range (bmd->rand, 0, 256);
	} else {
	new = bmd->set;
	}
	GST_INFO_OBJECT (bmd, "changing byte %u from 0x%02X to 0x%02X", i,
	(guint) GST_READ_UINT8 (GST_BUFFER_DATA (buf) + i),
	(guint) ((guint8) new));
	GST_BUFFER_DATA (buf)[i] = new;
	}
	}
	/* don't overflow */
	bmd->skipped += MIN (G_MAXUINT - bmd->skipped, GST_BUFFER_SIZE (buf));

	GST_OBJECT_UNLOCK (bmd);

	return GST_FLOW_OK;
	}

	static gboolean
	gst_break_my_data_start (GstBaseTransform * trans)
	{
	GstBreakMyData *bmd = GST_BREAK_MY_DATA (trans);

	GST_OBJECT_LOCK (bmd);
	bmd->rand = g_rand_new_with_seed (bmd->seed);
	bmd->skipped = 0;
	GST_OBJECT_UNLOCK (bmd);

	return TRUE;
	}

	static gboolean
	gst_break_my_data_stop (GstBaseTransform * trans)
	{
	GstBreakMyData *bmd = GST_BREAK_MY_DATA (trans);

	GST_OBJECT_LOCK (bmd);
	g_rand_free (bmd->rand);
	bmd->rand = NULL;
	GST_OBJECT_UNLOCK (bmd);

	return TRUE;
	}

	gboolean
	gst_break_my_data_plugin_init (GstPlugin * plugin, GstPluginClass * g_class)
	{
	if (!gst_element_register (plugin, "breakmydata", GST_RANK_NONE,
	GST_TYPE_BREAK_MY_DATA))
	return FALSE;

	GST_DEBUG_CATEGORY_INIT (gst_break_my_data_debug, "breakmydata", 0,
	"debugging category for breakmydata element");

	return TRUE;
	}