2 * Copyright (C) 2007 Nokia Corporation (contact <stefan.kost@nokia.com>)
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.
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.
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., 51 Franklin St, Fifth Floor,
17 * Boston, MA 02110-1301, USA.
20 * SECTION:element-rndbuffersize
22 * This element pulls buffers with random sizes from the source.
30 #include <gst/base/gstadapter.h>
32 GST_DEBUG_CATEGORY_STATIC (gst_rnd_buffer_size_debug);
33 #define GST_CAT_DEFAULT gst_rnd_buffer_size_debug
35 #define GST_TYPE_RND_BUFFER_SIZE (gst_rnd_buffer_size_get_type())
36 #define GST_RND_BUFFER_SIZE(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_RND_BUFFER_SIZE,GstRndBufferSize))
37 #define GST_RND_BUFFER_SIZE_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_RND_BUFFER_SIZE,GstRndBufferSizeClass))
38 #define GST_IS_RND_BUFFER_SIZE(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_RND_BUFFER_SIZE))
39 #define GST_IS_RND_BUFFER_SIZE_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_RND_BUFFER_SIZE))
41 typedef struct _GstRndBufferSize GstRndBufferSize;
42 typedef struct _GstRndBufferSizeClass GstRndBufferSizeClass;
44 struct _GstRndBufferSize
53 GstPad *sinkpad, *srcpad;
56 gboolean need_newsegment;
61 struct _GstRndBufferSizeClass
63 GstElementClass parent_class;
73 #define DEFAULT_SEED 0
75 #define DEFAULT_MAX (8*1024)
77 static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
82 static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
87 static void gst_rnd_buffer_size_finalize (GObject * object);
88 static void gst_rnd_buffer_size_set_property (GObject * object, guint prop_id,
89 const GValue * value, GParamSpec * pspec);
90 static void gst_rnd_buffer_size_get_property (GObject * object, guint prop_id,
91 GValue * value, GParamSpec * pspec);
93 static gboolean gst_rnd_buffer_size_activate (GstPad * pad, GstObject * parent);
94 static gboolean gst_rnd_buffer_size_activate_mode (GstPad * pad,
95 GstObject * parent, GstPadMode mode, gboolean active);
96 static void gst_rnd_buffer_size_loop (GstRndBufferSize * self);
97 static GstStateChangeReturn gst_rnd_buffer_size_change_state (GstElement *
98 element, GstStateChange transition);
99 static gboolean gst_rnd_buffer_size_src_event (GstPad * pad,
100 GstObject * parent, GstEvent * event);
101 static gboolean gst_rnd_buffer_size_sink_event (GstPad * pad,
102 GstObject * parent, GstEvent * event);
103 static GstFlowReturn gst_rnd_buffer_size_chain (GstPad * pad,
104 GstObject * parent, GstBuffer * buffer);
106 GType gst_rnd_buffer_size_get_type (void);
107 #define gst_rnd_buffer_size_parent_class parent_class
108 G_DEFINE_TYPE (GstRndBufferSize, gst_rnd_buffer_size, GST_TYPE_ELEMENT);
111 gst_rnd_buffer_size_class_init (GstRndBufferSizeClass * klass)
113 GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
114 GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);
116 GST_DEBUG_CATEGORY_INIT (gst_rnd_buffer_size_debug, "rndbuffersize", 0,
117 "rndbuffersize element");
119 gobject_class->set_property = gst_rnd_buffer_size_set_property;
120 gobject_class->get_property = gst_rnd_buffer_size_get_property;
121 gobject_class->finalize = gst_rnd_buffer_size_finalize;
123 gst_element_class_add_pad_template (gstelement_class,
124 gst_static_pad_template_get (&sink_template));
125 gst_element_class_add_pad_template (gstelement_class,
126 gst_static_pad_template_get (&src_template));
128 gst_element_class_set_static_metadata (gstelement_class, "Random buffer size",
129 "Testing", "pull random sized buffers",
130 "Stefan Kost <stefan.kost@nokia.com>");
132 gstelement_class->change_state =
133 GST_DEBUG_FUNCPTR (gst_rnd_buffer_size_change_state);
135 g_object_class_install_property (gobject_class, PROP_SEED,
136 g_param_spec_uint ("seed", "random number seed",
137 "seed for randomness (initialized when going from READY to PAUSED)",
138 0, G_MAXUINT32, DEFAULT_SEED,
139 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
140 g_object_class_install_property (gobject_class, PROP_MINIMUM,
141 g_param_spec_int ("min", "mininum", "mininum buffer size",
142 0, G_MAXINT32, DEFAULT_MIN,
143 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
144 g_object_class_install_property (gobject_class, PROP_MAXIMUM,
145 g_param_spec_int ("max", "maximum", "maximum buffer size",
146 1, G_MAXINT32, DEFAULT_MAX,
147 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
151 gst_rnd_buffer_size_init (GstRndBufferSize * self)
153 self->sinkpad = gst_pad_new_from_static_template (&sink_template, "sink");
154 gst_pad_set_activate_function (self->sinkpad,
155 GST_DEBUG_FUNCPTR (gst_rnd_buffer_size_activate));
156 gst_pad_set_activatemode_function (self->sinkpad,
157 GST_DEBUG_FUNCPTR (gst_rnd_buffer_size_activate_mode));
158 gst_pad_set_event_function (self->sinkpad,
159 GST_DEBUG_FUNCPTR (gst_rnd_buffer_size_sink_event));
160 gst_pad_set_chain_function (self->sinkpad,
161 GST_DEBUG_FUNCPTR (gst_rnd_buffer_size_chain));
162 GST_OBJECT_FLAG_SET (self->sinkpad, GST_PAD_FLAG_PROXY_CAPS);
163 GST_OBJECT_FLAG_SET (self->sinkpad, GST_PAD_FLAG_PROXY_ALLOCATION);
164 GST_OBJECT_FLAG_SET (self->sinkpad, GST_PAD_FLAG_PROXY_SCHEDULING);
165 gst_element_add_pad (GST_ELEMENT (self), self->sinkpad);
167 self->srcpad = gst_pad_new_from_static_template (&src_template, "src");
168 gst_pad_set_event_function (self->srcpad,
169 GST_DEBUG_FUNCPTR (gst_rnd_buffer_size_src_event));
170 GST_OBJECT_FLAG_SET (self->srcpad, GST_PAD_FLAG_PROXY_CAPS);
171 GST_OBJECT_FLAG_SET (self->srcpad, GST_PAD_FLAG_PROXY_ALLOCATION);
172 GST_OBJECT_FLAG_SET (self->srcpad, GST_PAD_FLAG_PROXY_SCHEDULING);
173 gst_element_add_pad (GST_ELEMENT (self), self->srcpad);
178 gst_rnd_buffer_size_finalize (GObject * object)
180 GstRndBufferSize *self = GST_RND_BUFFER_SIZE (object);
183 g_rand_free (self->rand);
187 G_OBJECT_CLASS (parent_class)->finalize (object);
192 gst_rnd_buffer_size_set_property (GObject * object, guint prop_id,
193 const GValue * value, GParamSpec * pspec)
195 GstRndBufferSize *self = GST_RND_BUFFER_SIZE (object);
199 self->seed = g_value_get_uint (value);
202 self->min = g_value_get_int (value);
205 self->max = g_value_get_int (value);
208 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
215 gst_rnd_buffer_size_get_property (GObject * object, guint prop_id,
216 GValue * value, GParamSpec * pspec)
218 GstRndBufferSize *self = GST_RND_BUFFER_SIZE (object);
222 g_value_set_uint (value, self->seed);
225 g_value_set_int (value, self->min);
228 g_value_set_int (value, self->max);
231 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
238 gst_rnd_buffer_size_activate (GstPad * pad, GstObject * parent)
243 query = gst_query_new_scheduling ();
245 if (gst_pad_peer_query (pad, query))
246 pull_mode = gst_query_has_scheduling_mode_with_flags (query,
247 GST_PAD_MODE_PULL, GST_SCHEDULING_FLAG_SEEKABLE);
251 gst_query_unref (query);
254 GST_DEBUG_OBJECT (pad, "activating pull");
255 return gst_pad_activate_mode (pad, GST_PAD_MODE_PULL, TRUE);
257 GST_DEBUG_OBJECT (pad, "activating push");
258 return gst_pad_activate_mode (pad, GST_PAD_MODE_PUSH, TRUE);
264 gst_rnd_buffer_size_activate_mode (GstPad * pad, GstObject * parent,
265 GstPadMode mode, gboolean active)
268 GstRndBufferSize *self = GST_RND_BUFFER_SIZE (parent);
271 case GST_PAD_MODE_PULL:
273 GST_INFO_OBJECT (self, "starting pull");
275 gst_pad_start_task (pad, (GstTaskFunction) gst_rnd_buffer_size_loop,
277 self->need_newsegment = TRUE;
279 GST_INFO_OBJECT (self, "stopping pull");
280 res = gst_pad_stop_task (pad);
283 case GST_PAD_MODE_PUSH:
284 GST_INFO_OBJECT (self, "%sactivating in push mode", (active) ? "" : "de");
295 gst_rnd_buffer_size_src_event (GstPad * pad, GstObject * parent,
298 GstRndBufferSize *self;
299 GstSeekType start_type;
304 if (GST_EVENT_TYPE (event) != GST_EVENT_SEEK) {
305 return gst_pad_event_default (pad, parent, event);
308 self = GST_RND_BUFFER_SIZE (parent);
309 gst_event_parse_seek (event, NULL, &format, &flags, &start_type, &start,
312 if (format != GST_FORMAT_BYTES) {
313 GST_WARNING_OBJECT (pad, "only BYTE format supported");
316 if (start_type != GST_SEEK_TYPE_SET) {
317 GST_WARNING_OBJECT (pad, "only SEEK_TYPE_SET supported");
321 if ((flags & GST_SEEK_FLAG_FLUSH)) {
322 gst_pad_push_event (self->srcpad, gst_event_new_flush_start ());
323 gst_pad_push_event (self->sinkpad, gst_event_new_flush_start ());
325 gst_pad_pause_task (self->sinkpad);
328 GST_PAD_STREAM_LOCK (self->sinkpad);
330 if ((flags & GST_SEEK_FLAG_FLUSH)) {
331 gst_pad_push_event (self->srcpad, gst_event_new_flush_stop (TRUE));
332 gst_pad_push_event (self->sinkpad, gst_event_new_flush_stop (TRUE));
335 GST_INFO_OBJECT (pad, "seeking to offset %" G_GINT64_FORMAT, start);
337 self->offset = start;
338 self->need_newsegment = TRUE;
340 gst_pad_start_task (self->sinkpad, (GstTaskFunction) gst_rnd_buffer_size_loop,
343 GST_PAD_STREAM_UNLOCK (self->sinkpad);
348 gst_rnd_buffer_size_drain_adapter (GstRndBufferSize * self, gboolean eos)
352 guint num_bytes, avail;
356 if (G_UNLIKELY (self->min > self->max))
360 if (self->min != self->max) {
361 num_bytes = g_rand_int_range (self->rand, self->min, self->max);
363 num_bytes = self->min;
366 GST_LOG_OBJECT (self, "pulling %u bytes out of adapter", num_bytes);
368 buf = gst_adapter_take_buffer (self->adapter, num_bytes);
372 GST_LOG_OBJECT (self, "not enough bytes in adapter");
376 avail = gst_adapter_available (self->adapter);
381 if (avail < self->min) {
382 GST_WARNING_OBJECT (self, "discarding %u bytes at end (min=%u)",
384 gst_adapter_clear (self->adapter);
387 buf = gst_adapter_take_buffer (self->adapter, avail);
388 g_assert (buf != NULL);
391 flow = gst_pad_push (self->srcpad, buf);
393 while (flow == GST_FLOW_OK);
400 GST_ELEMENT_ERROR (self, LIBRARY, SETTINGS,
401 ("The minimum buffer size is smaller than the maximum buffer size."),
402 ("buffer sizes: max=%d, min=%d", self->min, self->max));
403 return GST_FLOW_ERROR;
408 gst_rnd_buffer_size_sink_event (GstPad * pad, GstObject * parent,
411 GstRndBufferSize *rnd = GST_RND_BUFFER_SIZE (parent);
413 switch (GST_EVENT_TYPE (event)) {
415 gst_rnd_buffer_size_drain_adapter (rnd, TRUE);
417 case GST_EVENT_FLUSH_STOP:
418 if (rnd->adapter != NULL)
419 gst_adapter_clear (rnd->adapter);
425 return gst_pad_event_default (pad, parent, event);
429 gst_rnd_buffer_size_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
431 GstRndBufferSize *rnd = GST_RND_BUFFER_SIZE (parent);
434 if (rnd->adapter == NULL)
435 rnd->adapter = gst_adapter_new ();
437 gst_adapter_push (rnd->adapter, buf);
439 flow = gst_rnd_buffer_size_drain_adapter (rnd, FALSE);
441 if (flow != GST_FLOW_OK)
442 GST_INFO_OBJECT (rnd, "flow: %s", gst_flow_get_name (flow));
448 gst_rnd_buffer_size_loop (GstRndBufferSize * self)
450 GstBuffer *buf = NULL;
452 guint num_bytes, size;
454 if (G_UNLIKELY (self->min > self->max))
457 if (G_UNLIKELY (self->min != self->max)) {
458 num_bytes = g_rand_int_range (self->rand, self->min, self->max);
460 num_bytes = self->min;
463 GST_LOG_OBJECT (self, "pulling %u bytes at offset %" G_GUINT64_FORMAT,
464 num_bytes, self->offset);
466 ret = gst_pad_pull_range (self->sinkpad, self->offset, num_bytes, &buf);
468 if (ret != GST_FLOW_OK)
471 size = gst_buffer_get_size (buf);
473 if (size < num_bytes) {
474 GST_WARNING_OBJECT (self, "short buffer: %u bytes", size);
477 if (self->need_newsegment) {
480 gst_segment_init (&segment, GST_FORMAT_BYTES);
481 segment.start = self->offset;
482 gst_pad_push_event (self->srcpad, gst_event_new_segment (&segment));
483 self->need_newsegment = FALSE;
486 self->offset += size;
488 ret = gst_pad_push (self->srcpad, buf);
490 if (ret != GST_FLOW_OK)
497 GST_DEBUG_OBJECT (self, "pausing task");
498 gst_pad_pause_task (self->sinkpad);
504 if (ret == GST_FLOW_EOS) {
505 GST_DEBUG_OBJECT (self, "eos");
506 gst_pad_push_event (self->srcpad, gst_event_new_eos ());
508 GST_WARNING_OBJECT (self, "pull_range flow: %s", gst_flow_get_name (ret));
515 GST_DEBUG_OBJECT (self, "push flow: %s", gst_flow_get_name (ret));
516 if (ret == GST_FLOW_EOS) {
517 GST_DEBUG_OBJECT (self, "eos");
518 gst_pad_push_event (self->srcpad, gst_event_new_eos ());
519 } else if (ret < GST_FLOW_EOS || ret == GST_FLOW_NOT_LINKED) {
520 GST_ELEMENT_ERROR (self, STREAM, FAILED,
521 ("Internal data stream error."),
522 ("streaming stopped, reason: %s", gst_flow_get_name (ret)));
529 GST_ELEMENT_ERROR (self, LIBRARY, SETTINGS,
530 ("The minimum buffer size is smaller than the maximum buffer size."),
531 ("buffer sizes: max=%d, min=%d", self->min, self->max));
536 static GstStateChangeReturn
537 gst_rnd_buffer_size_change_state (GstElement * element,
538 GstStateChange transition)
540 GstRndBufferSize *self = GST_RND_BUFFER_SIZE (element);
541 GstStateChangeReturn ret;
543 switch (transition) {
544 case GST_STATE_CHANGE_NULL_TO_READY:
546 case GST_STATE_CHANGE_READY_TO_PAUSED:
549 self->rand = g_rand_new_with_seed (self->seed);
552 case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
558 ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
560 switch (transition) {
561 case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
563 case GST_STATE_CHANGE_PAUSED_TO_READY:
565 g_rand_free (self->rand);
569 case GST_STATE_CHANGE_READY_TO_NULL:
571 g_object_unref (self->adapter);
572 self->adapter = NULL;