gstreamer/testsuite/bytestream/gstbstest.c

/* GStreamer
 * Copyright (C) 1999,2000 Erik Walthinsen <omega@cse.ogi.edu>
 *                    2000 Wim Taymans <wtay@chello.be>
 *
 * gstbstest.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 <gst/gst.h>
#include <libs/bytestream/gstbytestream.h>

#define GST_TYPE_BSTEST  		(gst_bstest_get_type())
#define GST_BSTEST(obj)  		(G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_BSTEST,GstBsTest))
#define GST_BSTEST_CLASS(klass) 	(G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_BSTEST,GstBsTestClass))
#define GST_IS_BSTEST(obj)  		(G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_BSTEST))
#define GST_IS_BSTEST_CLASS(obj)  	(G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_BSTEST))

typedef struct _GstBsTest GstBsTest;
typedef struct _GstBsTestClass GstBsTestClass;

typedef enum
{
  BSTEST_SIZETYPE_FIXED = 1,
  BSTEST_SIZETYPE_RANDOM
}
GstBsTestSizeType;

typedef enum
{
  BSTEST_ACCESSTYPE_READ = 1,
  BSTEST_ACCESSTYPE_PEEK_READ,
  BSTEST_ACCESSTYPE_PEEK_READRAND,
}
GstBsTestAccessType;

struct _GstBsTest
{
  GstElement element;

  GstPad *sinkpad;
  GstPad *srcpad;

  GstByteStream *bs;
  GstBsTestSizeType sizetype;
  GstBsTestAccessType accesstype;
  guint sizemin;
  guint sizemax;
  gint count;
  gboolean silent;
};

struct _GstBsTestClass
{
  GstElementClass parent_class;
};

GType gst_bstest_get_type (void);


GstElementDetails gst_bstest_details = {
  "ByteStreamTest",
  "Filter",
  "Test for the GstByteStream code",
  VERSION,
  "Erik Walthinsen <omega@temple-baptist.com>," "Wim Taymans <wim.taymans@chello.be>",
  "(C) 2001",
};


/* BsTest signals and args */
enum
{
  /* FILL ME */
  LAST_SIGNAL
};

enum
{
  ARG_0,
  ARG_SIZETYPE,
  ARG_SIZEMIN,
  ARG_SIZEMAX,
  ARG_ACCESSTYPE,
  ARG_COUNT,
  ARG_SILENT,
};


static void 	gst_bstest_class_init 		(GstBsTestClass * klass);
static void 	gst_bstest_init 		(GstBsTest * bstest);

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

static GstElementStateReturn 	gst_bstest_change_state 	(GstElement *element);
static void 			gst_bstest_loop 		(GstElement * element);

static GstElementClass *parent_class = NULL;

// static guint gst_bstest_signals[LAST_SIGNAL] = { 0 };

#define GST_TYPE_BSTEST_SIZETYPE (gst_bstest_sizetype_get_type())
static GType
gst_bstest_sizetype_get_type (void)
{
  static GType bstest_sizetype_type = 0;
  static GEnumValue bstest_sizetype[] = {
    {BSTEST_SIZETYPE_FIXED,  "1", "Fixed size buffers (sizemax sized)"},
    {BSTEST_SIZETYPE_RANDOM, "2", "Random sized buffers (sizemin <= size <= sizemax)"},
    {0, NULL, NULL},
  };

  if (!bstest_sizetype_type) {
    bstest_sizetype_type = g_enum_register_static ("GstBsTestSizeType", bstest_sizetype);
  }
  return bstest_sizetype_type;
}

#define GST_TYPE_BSTEST_ACCESSTYPE (gst_bstest_accesstype_get_type())
static GType
gst_bstest_accesstype_get_type (void)
{
  static GType bstest_accesstype_type = 0;
  static GEnumValue bstest_accesstype[] = {
    {BSTEST_ACCESSTYPE_READ,           "1", "Read data"},
    {BSTEST_ACCESSTYPE_PEEK_READ,      "2", "Peek data, read data (same size)"},
    {BSTEST_ACCESSTYPE_PEEK_READRAND,  "3", "Peek data, read data (different size)"},
    {0, NULL, NULL},
  };

  if (!bstest_accesstype_type) {
    bstest_accesstype_type = g_enum_register_static ("GstBsTestAccessType", bstest_accesstype);
  }
  return bstest_accesstype_type;
}
GType
gst_bstest_get_type (void)
{
  static GType bstest_type = 0;

  if (!bstest_type) {
    static const GTypeInfo bstest_info = {
      sizeof (GstBsTestClass), NULL,
      NULL,
      (GClassInitFunc) gst_bstest_class_init,
      NULL,
      NULL,
      sizeof (GstBsTest),
      0,
      (GInstanceInitFunc) gst_bstest_init,
    };

    bstest_type = g_type_register_static (GST_TYPE_ELEMENT, "BSTest", &bstest_info, 0);
  }
  return bstest_type;
}

static void
gst_bstest_class_init (GstBsTestClass * klass)
{
  GObjectClass *gobject_class;
  GstElementClass *gstelement_class;

  gobject_class = (GObjectClass *) klass;
  gstelement_class = (GstElementClass*)klass;

  parent_class = g_type_class_ref (GST_TYPE_ELEMENT);

  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_SIZETYPE,
				   g_param_spec_enum ("sizetype", "sizetype", "sizetype",
						      GST_TYPE_BSTEST_SIZETYPE,
						      BSTEST_SIZETYPE_FIXED, G_PARAM_READWRITE));
  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_SIZEMIN,
				   g_param_spec_int ("sizemin", "sizemin", "sizemin", 0, G_MAXINT,
						     0, G_PARAM_READWRITE));
  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_SIZEMAX,
				   g_param_spec_int ("sizemax", "sizemax", "sizemax", 0, G_MAXINT,
						     384, G_PARAM_READWRITE));
  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_ACCESSTYPE,
				   g_param_spec_enum ("accesstype", "accesstype", "accesstype",
						      GST_TYPE_BSTEST_ACCESSTYPE,
						      BSTEST_ACCESSTYPE_READ, G_PARAM_READWRITE));
  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_COUNT,
				   g_param_spec_uint ("count", "count", "count",
						      0, G_MAXUINT, 0, G_PARAM_READWRITE));
  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_SILENT,
		        	   g_param_spec_boolean ("silent", "silent", "silent",
				                          FALSE, G_PARAM_READWRITE));

  gobject_class->set_property = GST_DEBUG_FUNCPTR (gst_bstest_set_property);
  gobject_class->get_property = GST_DEBUG_FUNCPTR (gst_bstest_get_property);

  gstelement_class->change_state = gst_bstest_change_state;

}

static GstPadNegotiateReturn
gst_bstest_negotiate_src (GstPad * pad, GstCaps ** caps, gpointer * data)
{
  GstBsTest *bstest;

  bstest = GST_BSTEST (gst_pad_get_parent (pad));

  return gst_pad_negotiate_proxy (pad, bstest->sinkpad, caps);
}

static GstPadNegotiateReturn
gst_bstest_negotiate_sink (GstPad * pad, GstCaps ** caps, gpointer * data)
{
  GstBsTest *bstest;

  bstest = GST_BSTEST (gst_pad_get_parent (pad));

  return gst_pad_negotiate_proxy (pad, bstest->srcpad, caps);
}

static void
gst_bstest_init (GstBsTest * bstest)
{
  bstest->sinkpad = gst_pad_new ("sink", GST_PAD_SINK);
  gst_element_add_pad (GST_ELEMENT (bstest), bstest->sinkpad);
  gst_pad_set_negotiate_function (bstest->sinkpad, gst_bstest_negotiate_sink);

  bstest->srcpad = gst_pad_new ("src", GST_PAD_SRC);
  gst_element_add_pad (GST_ELEMENT (bstest), bstest->srcpad);
  gst_pad_set_negotiate_function (bstest->srcpad, gst_bstest_negotiate_src);

  gst_element_set_loop_function (GST_ELEMENT (bstest), gst_bstest_loop);

  bstest->sizetype = BSTEST_SIZETYPE_FIXED;
  bstest->sizemin = 0;
  bstest->sizemax = 384;
  bstest->accesstype = BSTEST_ACCESSTYPE_READ;
  bstest->count = 5;
  bstest->silent = FALSE;
  bstest->bs = NULL;
}

static guint
gst_bstest_get_size (GstBsTest *bstest)
{
  switch (bstest->sizetype) {
    case BSTEST_SIZETYPE_FIXED:
      return bstest->sizemax;
      break;
    case BSTEST_SIZETYPE_RANDOM:
      return bstest->sizemin + (guint8)(((gfloat)bstest->sizemax)*rand()/(RAND_MAX + (gfloat)bstest->sizemin));
      break;
  }
}

static void
gst_bstest_loop (GstElement * element)
{
  GstBsTest *bstest;
  GstBuffer *buf = NULL;
  gint i;

  g_return_if_fail (element != NULL);
  g_return_if_fail (GST_IS_BSTEST (element));

  bstest = GST_BSTEST (element);

/* THIS IS THE BUFFER BASED ONE */
  do {
    for (i = 0; i < bstest->count; i++) {
      guint size;

      size = gst_bstest_get_size (bstest);
      if (size == 0) {
	buf = gst_buffer_new ();
      }
      else {

	switch (bstest->accesstype) {
          case BSTEST_ACCESSTYPE_PEEK_READ:
            if (!bstest->silent) g_print ("bstest: ***** peek %d bytes\n", size);
            gst_bytestream_peek_bytes (bstest->bs, size);
	    // fall though
          case BSTEST_ACCESSTYPE_READ:
            if (!bstest->silent) g_print ("bstest: ***** read %d bytes\n", size);
            buf = gst_bytestream_read (bstest->bs, size);
	    break;
          case BSTEST_ACCESSTYPE_PEEK_READRAND:
            if (!bstest->silent) g_print ("bstest: ***** peek %d bytes\n", size);
            gst_bytestream_peek_bytes (bstest->bs, size);
            size = gst_bstest_get_size (bstest);
            if (!bstest->silent) g_print ("bstest: ***** read %d bytes\n", size);
            if (size == 0) {
	      buf = gst_buffer_new ();
            }
	    else {
              buf = gst_bytestream_read (bstest->bs, size);
	    }
	    break;
	}
      }
      if (!buf)
	g_print ("BUFFER IS BOGUS\n");
      else 
        gst_pad_push (bstest->srcpad, buf);
    }
  } while (!GST_ELEMENT_IS_COTHREAD_STOPPING (element));
}

static void
gst_bstest_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec)
{
  GstBsTest *bstest;

  /* it's not null if we got it, but it might not be ours */
  g_return_if_fail (GST_IS_BSTEST (object));

  bstest = GST_BSTEST (object);

  switch (prop_id) {
    case ARG_SIZETYPE:
      bstest->sizetype = g_value_get_int (value);
      break;
    case ARG_SIZEMIN:
      bstest->sizemin = g_value_get_int (value);
      break;
    case ARG_SIZEMAX:
      bstest->sizemax = g_value_get_int (value);
      break;
    case ARG_ACCESSTYPE:
      bstest->accesstype = g_value_get_int (value);
      break;
    case ARG_COUNT:
      bstest->count = g_value_get_uint (value);
      break;
    case ARG_SILENT:
      bstest->silent = g_value_get_boolean (value);
      break;
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      break;
  }
}

static void
gst_bstest_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec)
{
  GstBsTest *bstest;

  /* it's not null if we got it, but it might not be ours */
  g_return_if_fail (GST_IS_BSTEST (object));

  bstest = GST_BSTEST (object);

  switch (prop_id) {
    case ARG_SIZETYPE:
      g_value_set_int (value, bstest->sizetype);
      break;
    case ARG_SIZEMIN:
      g_value_set_int (value, bstest->sizemin);
      break;
    case ARG_SIZEMAX:
      g_value_set_int (value, bstest->sizemax);
      break;
    case ARG_ACCESSTYPE:
      g_value_set_int (value, bstest->accesstype);
      break;
    case ARG_COUNT:
      g_value_set_uint (value, bstest->count);
      break;
    case ARG_SILENT:
      g_value_set_boolean (value, bstest->silent);
      break;
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      break;
  }
}

static GstElementStateReturn
gst_bstest_change_state (GstElement *element)
{
  GstBsTest *bstest;

  g_return_val_if_fail (GST_IS_BSTEST (element), GST_STATE_FAILURE);

  bstest = GST_BSTEST (element);

  if (GST_STATE_PENDING (element) == GST_STATE_NULL) {
    if (bstest->bs) {
      gst_bytestream_destroy (bstest->bs);
      bstest->bs = NULL;
    }
  }
  else {
    if (!bstest->bs) {
      bstest->bs = gst_bytestream_new (bstest->sinkpad);
    }
  }

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

  return GST_STATE_SUCCESS;
}

static gboolean
plugin_init (GModule * module, GstPlugin * plugin)
{
  GstElementFactory *factory;

  // we need gstbytestream
  if (!gst_library_load ("gstbytestream")) {
    g_print ("can't load bytestream\n");
    return FALSE;
  }

  /* We need to create an ElementFactory for each element we provide.
   * This consists of the name of the element, the GType identifier,
   * and a pointer to the details structure at the top of the file.
   */
  factory = gst_elementfactory_new ("bstest", GST_TYPE_BSTEST, &gst_bstest_details);
  g_return_val_if_fail (factory != NULL, FALSE);

  /* The very last thing is to register the elementfactory with the plugin. */
  gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));

  return TRUE;
}

GST_PLUGIN_DESC (GST_VERSION_MAJOR, GST_VERSION_MINOR, "bstest", plugin_init);