gst/debugutils/breakmydata.c

   1 /* GStreamer
   2  * Copyright (C) 2004 Benjamin Otte <otte@gnome.org>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19 /**
  20  * SECTION:element-breakmydata
  21  *
  22  * This element modifies the contents of the buffer it is passed randomly
  23  * according to the parameters set.
  24  * It otherwise acts as an identity.
  25  */
  26
  27 #ifdef HAVE_CONFIG_H
  28 #  include "config.h"
  29 #endif
  30
  31 #include <gst/gst.h>
  32 #include <gst/base/gstbasetransform.h>
  33
  34 GST_DEBUG_CATEGORY_STATIC (gst_break_my_data_debug);
  35 #define GST_CAT_DEFAULT gst_break_my_data_debug
  36
  37 #define GST_TYPE_BREAK_MY_DATA \
  38   (gst_break_my_data_get_type())
  39 #define GST_BREAK_MY_DATA(obj) \
  40   (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_BREAK_MY_DATA,GstBreakMyData))
  41 #define GST_BREAK_MY_DATA_CLASS(klass) \
  42   (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_BREAK_MY_DATA,GstBreakMyDataClass))
  43 #define GST_IS_BREAK_MY_DATA(obj) \
  44   (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_BREAK_MY_DATA))
  45 #define GST_IS_BREAK_MY_DATA_CLASS(klass) \
  46   (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_BREAK_MY_DATA))
  47
  48 enum
  49 {
  50   ARG_0,
  51   ARG_SEED,
  52   ARG_SET_TO,
  53   ARG_SKIP,
  54   ARG_PROBABILITY
  55 };
  56
  57 typedef struct _GstBreakMyData GstBreakMyData;
  58 typedef struct _GstBreakMyDataClass GstBreakMyDataClass;
  59
  60 struct _GstBreakMyData
  61 {
  62   GstBaseTransform basetransform;
  63
  64   GRand *rand;
  65   guint skipped;
  66
  67   guint32 seed;
  68   gint set;
  69   guint skip;
  70   gdouble probability;
  71 };
  72
  73 struct _GstBreakMyDataClass
  74 {
  75   GstBaseTransformClass parent_class;
  76 };
  77
  78 static void gst_break_my_data_set_property (GObject * object,
  79     guint prop_id, const GValue * value, GParamSpec * pspec);
  80 static void gst_break_my_data_get_property (GObject * object,
  81     guint prop_id, GValue * value, GParamSpec * pspec);
  82
  83 static GstFlowReturn gst_break_my_data_transform_ip (GstBaseTransform * trans,
  84     GstBuffer * buf);
  85 static gboolean gst_break_my_data_stop (GstBaseTransform * trans);
  86 static gboolean gst_break_my_data_start (GstBaseTransform * trans);
  87
  88 GstStaticPadTemplate bmd_src_template = GST_STATIC_PAD_TEMPLATE ("src",
  89     GST_PAD_SRC,
  90     GST_PAD_ALWAYS,
  91     GST_STATIC_CAPS_ANY);
  92
  93 GstStaticPadTemplate bmd_sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
  94     GST_PAD_SINK,
  95     GST_PAD_ALWAYS,
  96     GST_STATIC_CAPS_ANY);
  97
  98
  99 #define DEBUG_INIT(bla) \
 100   GST_DEBUG_CATEGORY_INIT (gst_break_my_data_debug, "breakmydata", 0, \
 101       "debugging category for breakmydata element");
 102
 103 GType gst_break_my_data_get_type (void);
 104 GST_BOILERPLATE_FULL (GstBreakMyData, gst_break_my_data, GstBaseTransform,
 105     GST_TYPE_BASE_TRANSFORM, DEBUG_INIT);
 106
 107
 108 static void
 109 gst_break_my_data_base_init (gpointer g_class)
 110 {
 111   GstElementClass *gstelement_class = GST_ELEMENT_CLASS (g_class);
 112
 113   gst_element_class_add_pad_template (gstelement_class,
 114       gst_static_pad_template_get (&bmd_sink_template));
 115   gst_element_class_add_pad_template (gstelement_class,
 116       gst_static_pad_template_get (&bmd_src_template));
 117
 118   gst_element_class_set_details_simple (gstelement_class, "Break my data",
 119       "Testing",
 120       "randomly change data in the stream", "Benjamin Otte <otte@gnome>");
 121 }
 122
 123 static void
 124 gst_break_my_data_class_init (GstBreakMyDataClass * klass)
 125 {
 126   GstBaseTransformClass *gstbasetrans_class;
 127   GObjectClass *gobject_class;
 128
 129   gobject_class = G_OBJECT_CLASS (klass);
 130   gstbasetrans_class = GST_BASE_TRANSFORM_CLASS (klass);
 131
 132   gobject_class->set_property = gst_break_my_data_set_property;
 133   gobject_class->get_property = gst_break_my_data_get_property;
 134
 135   g_object_class_install_property (gobject_class, ARG_SEED,
 136       g_param_spec_uint ("seed", "seed",
 137           "seed for randomness (initialized when going from READY to PAUSED)",
 138           0, G_MAXUINT32, 0,
 139           G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
 140   g_object_class_install_property (gobject_class, ARG_SET_TO,
 141       g_param_spec_int ("set-to", "set-to",
 142           "set changed bytes to this value (-1 means random value",
 143           -1, G_MAXUINT8, -1,
 144           G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
 145   g_object_class_install_property (gobject_class, ARG_SKIP,
 146       g_param_spec_uint ("skip", "skip",
 147           "amount of bytes skipped at the beginning of stream",
 148           0, G_MAXUINT, 0,
 149           G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
 150   g_object_class_install_property (gobject_class, ARG_PROBABILITY,
 151       g_param_spec_double ("probability", "probability",
 152           "probability for each byte in the buffer to be changed", 0.0, 1.0,
 153           0.0, G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
 154
 155   gstbasetrans_class->transform_ip =
 156       GST_DEBUG_FUNCPTR (gst_break_my_data_transform_ip);
 157   gstbasetrans_class->start = GST_DEBUG_FUNCPTR (gst_break_my_data_start);
 158   gstbasetrans_class->stop = GST_DEBUG_FUNCPTR (gst_break_my_data_stop);
 159 }
 160
 161 static void
 162 gst_break_my_data_init (GstBreakMyData * bmd, GstBreakMyDataClass * g_class)
 163 {
 164   gst_base_transform_set_in_place (GST_BASE_TRANSFORM (bmd), TRUE);
 165   gst_base_transform_set_passthrough (GST_BASE_TRANSFORM (bmd), TRUE);
 166 }
 167
 168 static void
 169 gst_break_my_data_set_property (GObject * object, guint prop_id,
 170     const GValue * value, GParamSpec * pspec)
 171 {
 172   GstBreakMyData *bmd = GST_BREAK_MY_DATA (object);
 173
 174   GST_OBJECT_LOCK (bmd);
 175
 176   switch (prop_id) {
 177     case ARG_SEED:
 178       bmd->seed = g_value_get_uint (value);
 179       break;
 180     case ARG_SET_TO:
 181       bmd->set = g_value_get_int (value);
 182       break;
 183     case ARG_SKIP:
 184       bmd->skip = g_value_get_uint (value);
 185       break;
 186     case ARG_PROBABILITY:
 187       bmd->probability = g_value_get_double (value);
 188       break;
 189     default:
 190       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 191       break;
 192   }
 193
 194   GST_OBJECT_UNLOCK (bmd);
 195 }
 196
 197 static void
 198 gst_break_my_data_get_property (GObject * object, guint prop_id, GValue * value,
 199     GParamSpec * pspec)
 200 {
 201   GstBreakMyData *bmd = GST_BREAK_MY_DATA (object);
 202
 203   GST_OBJECT_LOCK (bmd);
 204
 205   switch (prop_id) {
 206     case ARG_SEED:
 207       g_value_set_uint (value, bmd->seed);
 208       break;
 209     case ARG_SET_TO:
 210       g_value_set_int (value, bmd->set);
 211       break;
 212     case ARG_SKIP:
 213       g_value_set_uint (value, bmd->skip);
 214       break;
 215     case ARG_PROBABILITY:
 216       g_value_set_double (value, bmd->probability);
 217       break;
 218     default:
 219       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 220       break;
 221   }
 222
 223   GST_OBJECT_UNLOCK (bmd);
 224 }
 225
 226 static GstFlowReturn
 227 gst_break_my_data_transform_ip (GstBaseTransform * trans, GstBuffer * buf)
 228 {
 229   GstBreakMyData *bmd = GST_BREAK_MY_DATA (trans);
 230   guint i, size;
 231
 232   g_return_val_if_fail (gst_buffer_is_writable (buf), GST_FLOW_ERROR);
 233
 234   GST_OBJECT_LOCK (bmd);
 235
 236   if (bmd->skipped < bmd->skip) {
 237     i = bmd->skip - bmd->skipped;
 238   } else {
 239     i = 0;
 240   }
 241
 242   size = GST_BUFFER_SIZE (buf);
 243
 244   GST_LOG_OBJECT (bmd,
 245       "got buffer %p (size %u, timestamp %" G_GUINT64_FORMAT ", offset %"
 246       G_GUINT64_FORMAT "", buf, size, GST_BUFFER_TIMESTAMP (buf),
 247       GST_BUFFER_OFFSET (buf));
 248
 249   for (; i < size; i++) {
 250     if (g_rand_double_range (bmd->rand, 0, 1.0) <= bmd->probability) {
 251       guint8 new;
 252
 253       if (bmd->set < 0) {
 254         new = g_rand_int_range (bmd->rand, 0, 256);
 255       } else {
 256         new = bmd->set;
 257       }
 258       GST_INFO_OBJECT (bmd, "changing byte %u from 0x%02X to 0x%02X", i,
 259           (guint) GST_READ_UINT8 (GST_BUFFER_DATA (buf) + i),
 260           (guint) ((guint8) new));
 261       GST_BUFFER_DATA (buf)[i] = new;
 262     }
 263   }
 264   /* don't overflow */
 265   bmd->skipped += MIN (G_MAXUINT - bmd->skipped, GST_BUFFER_SIZE (buf));
 266
 267   GST_OBJECT_UNLOCK (bmd);
 268
 269   return GST_FLOW_OK;
 270 }
 271
 272 static gboolean
 273 gst_break_my_data_start (GstBaseTransform * trans)
 274 {
 275   GstBreakMyData *bmd = GST_BREAK_MY_DATA (trans);
 276
 277   GST_OBJECT_LOCK (bmd);
 278   bmd->rand = g_rand_new_with_seed (bmd->seed);
 279   bmd->skipped = 0;
 280   GST_OBJECT_UNLOCK (bmd);
 281
 282   return TRUE;
 283 }
 284
 285 static gboolean
 286 gst_break_my_data_stop (GstBaseTransform * trans)
 287 {
 288   GstBreakMyData *bmd = GST_BREAK_MY_DATA (trans);
 289
 290   GST_OBJECT_LOCK (bmd);
 291   g_rand_free (bmd->rand);
 292   bmd->rand = NULL;
 293   GST_OBJECT_UNLOCK (bmd);
 294
 295   return TRUE;
 296 }