gstreamer/tests/check/libs/basesrc.c
2019-04-17 23:03:56 +02:00

1057 lines
30 KiB
C

/* GStreamer
*
* some unit tests for GstBaseSrc
*
* Copyright (C) 2006-2017 Tim-Philipp Müller <tim centricular net>
*
* 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., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <gst/gst.h>
#include <gst/check/gstcheck.h>
#include <gst/check/gstconsistencychecker.h>
#include <gst/base/gstbasesrc.h>
static GstPadProbeReturn
eos_event_counter (GstObject * pad, GstPadProbeInfo * info, guint * p_num_eos)
{
GstEvent *event = GST_PAD_PROBE_INFO_EVENT (info);
fail_unless (event != NULL);
fail_unless (GST_IS_EVENT (event));
if (GST_EVENT_TYPE (event) == GST_EVENT_EOS)
*p_num_eos += 1;
return GST_PAD_PROBE_OK;
}
/* basesrc_eos_events_push_live_op:
* - make sure source does send an EOS event when operating in push
* mode and being set to READY explicitly (like one might with
* live sources)
*/
GST_START_TEST (basesrc_eos_events_push_live_op)
{
GstStateChangeReturn state_ret;
GstElement *src, *sink, *pipe;
GstMessage *msg;
GstBus *bus;
GstPad *srcpad;
guint probe, num_eos = 0;
GstStreamConsistency *consistency;
GstEvent *eos_event;
guint32 eos_event_seqnum;
pipe = gst_pipeline_new ("pipeline");
sink = gst_element_factory_make ("fakesink", "sink");
src = gst_element_factory_make ("fakesrc", "src");
g_assert (pipe != NULL);
g_assert (sink != NULL);
g_assert (src != NULL);
fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
fail_unless (gst_element_link (src, sink) == TRUE);
g_object_set (sink, "can-activate-push", TRUE, NULL);
g_object_set (sink, "can-activate-pull", FALSE, NULL);
g_object_set (src, "can-activate-push", TRUE, NULL);
g_object_set (src, "can-activate-pull", FALSE, NULL);
/* set up event probe to count EOS events */
srcpad = gst_element_get_static_pad (src, "src");
fail_unless (srcpad != NULL);
consistency = gst_consistency_checker_new (srcpad);
probe = gst_pad_add_probe (srcpad, GST_PAD_PROBE_TYPE_EVENT_BOTH,
(GstPadProbeCallback) eos_event_counter, &num_eos, NULL);
bus = gst_element_get_bus (pipe);
gst_element_set_state (pipe, GST_STATE_PLAYING);
state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
/* wait a second, then do controlled shutdown */
g_usleep (GST_USECOND * 1);
/* shut down pipeline (should send EOS message) ... */
eos_event = gst_event_new_eos ();
eos_event_seqnum = gst_event_get_seqnum (eos_event);
gst_element_send_event (pipe, eos_event);
/* ... and wait for the EOS message from the sink */
msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
fail_unless (msg != NULL);
fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
/* should be exactly one EOS event */
fail_unless (num_eos == 1);
fail_unless (gst_message_get_seqnum (msg) == eos_event_seqnum);
gst_element_set_state (pipe, GST_STATE_NULL);
gst_element_get_state (pipe, NULL, NULL, -1);
/* make sure source hasn't sent a second one when going PAUSED => READY */
fail_unless (num_eos == 1);
gst_consistency_checker_free (consistency);
gst_pad_remove_probe (srcpad, probe);
gst_object_unref (srcpad);
gst_message_unref (msg);
gst_object_unref (bus);
gst_object_unref (pipe);
}
GST_END_TEST;
/* basesrc_eos_events_push:
* - make sure source only sends one EOS when operating in push-mode,
* reaching the max number of buffers, and is then shut down.
*/
GST_START_TEST (basesrc_eos_events_push)
{
GstStateChangeReturn state_ret;
GstElement *src, *sink, *pipe;
GstMessage *msg;
GstBus *bus;
GstPad *srcpad;
guint probe, num_eos = 0;
GstStreamConsistency *consistency;
pipe = gst_pipeline_new ("pipeline");
sink = gst_element_factory_make ("fakesink", "sink");
src = gst_element_factory_make ("fakesrc", "src");
g_assert (pipe != NULL);
g_assert (sink != NULL);
g_assert (src != NULL);
fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
fail_unless (gst_element_link (src, sink) == TRUE);
g_object_set (sink, "can-activate-push", TRUE, NULL);
g_object_set (sink, "can-activate-pull", FALSE, NULL);
g_object_set (src, "can-activate-push", TRUE, NULL);
g_object_set (src, "can-activate-pull", FALSE, NULL);
g_object_set (src, "num-buffers", 8, NULL);
/* set up event probe to count EOS events */
srcpad = gst_element_get_static_pad (src, "src");
fail_unless (srcpad != NULL);
consistency = gst_consistency_checker_new (srcpad);
probe = gst_pad_add_probe (srcpad, GST_PAD_PROBE_TYPE_EVENT_BOTH,
(GstPadProbeCallback) eos_event_counter, &num_eos, NULL);
bus = gst_element_get_bus (pipe);
gst_element_set_state (pipe, GST_STATE_PLAYING);
state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
fail_unless (msg != NULL);
fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
/* should be exactly one EOS event */
fail_unless (num_eos == 1);
gst_element_set_state (pipe, GST_STATE_NULL);
gst_element_get_state (pipe, NULL, NULL, -1);
/* make sure source hasn't sent a second one when going PAUSED => READY */
fail_unless (num_eos == 1);
gst_consistency_checker_free (consistency);
gst_pad_remove_probe (srcpad, probe);
gst_object_unref (srcpad);
gst_message_unref (msg);
gst_object_unref (bus);
gst_object_unref (pipe);
}
GST_END_TEST;
/* basesrc_eos_events_pull_live_op:
* - make sure source doesn't send an EOS event when operating in
* pull mode and being set to READY explicitly (like one might with
* live sources)
*/
GST_START_TEST (basesrc_eos_events_pull_live_op)
{
GstStateChangeReturn state_ret;
GstElement *src, *sink, *pipe;
GstPad *srcpad;
guint probe, num_eos = 0;
pipe = gst_pipeline_new ("pipeline");
sink = gst_element_factory_make ("fakesink", "sink");
src = gst_element_factory_make ("fakesrc", "src");
g_assert (pipe != NULL);
g_assert (sink != NULL);
g_assert (src != NULL);
fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
fail_unless (gst_element_link (src, sink) == TRUE);
g_object_set (sink, "can-activate-push", FALSE, NULL);
g_object_set (sink, "can-activate-pull", TRUE, NULL);
g_object_set (src, "can-activate-push", FALSE, NULL);
g_object_set (src, "can-activate-pull", TRUE, NULL);
/* set up event probe to count EOS events */
srcpad = gst_element_get_static_pad (src, "src");
fail_unless (srcpad != NULL);
probe = gst_pad_add_probe (srcpad, GST_PAD_PROBE_TYPE_EVENT_BOTH,
(GstPadProbeCallback) eos_event_counter, &num_eos, NULL);
gst_element_set_state (pipe, GST_STATE_PLAYING);
state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
/* wait a second, then do controlled shutdown */
g_usleep (GST_USECOND * 1);
/* shut down source only ... */
gst_element_set_state (src, GST_STATE_NULL);
state_ret = gst_element_get_state (src, NULL, NULL, -1);
fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
fail_unless (gst_element_set_locked_state (src, TRUE) == TRUE);
/* source shouldn't have sent any EOS event in pull mode */
fail_unless (num_eos == 0);
gst_element_set_state (pipe, GST_STATE_NULL);
gst_element_get_state (pipe, NULL, NULL, -1);
/* make sure source hasn't sent an EOS when going PAUSED => READY either */
fail_unless (num_eos == 0);
gst_pad_remove_probe (srcpad, probe);
gst_object_unref (srcpad);
gst_object_unref (pipe);
}
GST_END_TEST;
/* basesrc_eos_events_pull:
* - makes sure source doesn't send EOS event when reaching the max.
* number of buffers configured in pull-mode
* - make sure source doesn't send EOS event either when being shut down
* (PAUSED => READY state change) after EOSing in pull mode
*/
GST_START_TEST (basesrc_eos_events_pull)
{
GstStateChangeReturn state_ret;
GstElement *src, *sink, *pipe;
GstMessage *msg;
GstBus *bus;
GstPad *srcpad;
guint probe, num_eos = 0;
pipe = gst_pipeline_new ("pipeline");
sink = gst_element_factory_make ("fakesink", "sink");
src = gst_element_factory_make ("fakesrc", "src");
g_assert (pipe != NULL);
g_assert (sink != NULL);
g_assert (src != NULL);
fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
fail_unless (gst_element_link (src, sink) == TRUE);
g_object_set (sink, "can-activate-push", FALSE, NULL);
g_object_set (sink, "can-activate-pull", TRUE, NULL);
g_object_set (src, "can-activate-push", FALSE, NULL);
g_object_set (src, "can-activate-pull", TRUE, NULL);
g_object_set (src, "num-buffers", 8, NULL);
/* set up event probe to count EOS events */
srcpad = gst_element_get_static_pad (src, "src");
fail_unless (srcpad != NULL);
probe = gst_pad_add_probe (srcpad, GST_PAD_PROBE_TYPE_EVENT_BOTH,
(GstPadProbeCallback) eos_event_counter, &num_eos, NULL);
bus = gst_element_get_bus (pipe);
gst_element_set_state (pipe, GST_STATE_PLAYING);
state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
fail_unless (msg != NULL);
fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
/* source shouldn't have sent any EOS event in pull mode */
fail_unless (num_eos == 0);
gst_element_set_state (pipe, GST_STATE_NULL);
gst_element_get_state (pipe, NULL, NULL, -1);
/* make sure source hasn't sent an EOS when going PAUSED => READY either */
fail_unless (num_eos == 0);
gst_pad_remove_probe (srcpad, probe);
gst_object_unref (srcpad);
gst_message_unref (msg);
gst_object_unref (bus);
gst_object_unref (pipe);
}
GST_END_TEST;
/* basesrc_eos_events_push_live_eos:
* - make sure the source stops and emits EOS when we send an EOS event to the
* pipeline.
*/
GST_START_TEST (basesrc_eos_events_push_live_eos)
{
GstStateChangeReturn state_ret;
GstElement *src, *sink, *pipe;
GstMessage *msg;
GstBus *bus;
GstPad *srcpad;
guint probe, num_eos = 0;
gboolean res;
pipe = gst_pipeline_new ("pipeline");
sink = gst_element_factory_make ("fakesink", "sink");
src = gst_element_factory_make ("fakesrc", "src");
g_assert (pipe != NULL);
g_assert (sink != NULL);
g_assert (src != NULL);
fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
fail_unless (gst_element_link (src, sink) == TRUE);
g_object_set (sink, "can-activate-push", TRUE, NULL);
g_object_set (sink, "can-activate-pull", FALSE, NULL);
g_object_set (src, "can-activate-push", TRUE, NULL);
g_object_set (src, "can-activate-pull", FALSE, NULL);
/* set up event probe to count EOS events */
srcpad = gst_element_get_static_pad (src, "src");
fail_unless (srcpad != NULL);
probe = gst_pad_add_probe (srcpad, GST_PAD_PROBE_TYPE_EVENT_BOTH,
(GstPadProbeCallback) eos_event_counter, &num_eos, NULL);
bus = gst_element_get_bus (pipe);
gst_element_set_state (pipe, GST_STATE_PLAYING);
state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
/* wait a second, then emit the EOS */
g_usleep (GST_USECOND * 1);
/* shut down source only (should send EOS event) ... */
res = gst_element_send_event (pipe, gst_event_new_eos ());
fail_unless (res == TRUE);
/* ... and wait for the EOS message from the sink */
msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
fail_unless (msg != NULL);
fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
/* should be exactly one EOS event */
fail_unless (num_eos == 1);
gst_element_set_state (pipe, GST_STATE_NULL);
gst_element_get_state (pipe, NULL, NULL, -1);
/* make sure source hasn't sent a second one when going PAUSED => READY */
fail_unless (num_eos == 1);
gst_pad_remove_probe (srcpad, probe);
gst_object_unref (srcpad);
gst_message_unref (msg);
gst_object_unref (bus);
gst_object_unref (pipe);
}
GST_END_TEST;
/* basesrc_eos_events_pull_live_eos:
* - make sure the source stops and emits EOS when we send an EOS event to the
* pipeline.
*/
GST_START_TEST (basesrc_eos_events_pull_live_eos)
{
GstStateChangeReturn state_ret;
GstElement *src, *sink, *pipe;
GstMessage *msg;
GstBus *bus;
GstPad *srcpad;
guint probe, num_eos = 0;
gboolean res;
pipe = gst_pipeline_new ("pipeline");
sink = gst_element_factory_make ("fakesink", "sink");
src = gst_element_factory_make ("fakesrc", "src");
g_assert (pipe != NULL);
g_assert (sink != NULL);
g_assert (src != NULL);
fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
fail_unless (gst_element_link (src, sink) == TRUE);
g_object_set (sink, "can-activate-push", FALSE, NULL);
g_object_set (sink, "can-activate-pull", TRUE, NULL);
g_object_set (src, "can-activate-push", FALSE, NULL);
g_object_set (src, "can-activate-pull", TRUE, NULL);
/* set up event probe to count EOS events */
srcpad = gst_element_get_static_pad (src, "src");
fail_unless (srcpad != NULL);
probe = gst_pad_add_probe (srcpad, GST_PAD_PROBE_TYPE_EVENT_BOTH,
(GstPadProbeCallback) eos_event_counter, &num_eos, NULL);
bus = gst_element_get_bus (pipe);
gst_element_set_state (pipe, GST_STATE_PLAYING);
state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
/* wait a second, then emit the EOS */
g_usleep (GST_USECOND * 1);
/* shut down source only (should send EOS event) ... */
res = gst_element_send_event (pipe, gst_event_new_eos ());
fail_unless (res == TRUE);
/* ... and wait for the EOS message from the sink */
msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
fail_unless (msg != NULL);
fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
/* no EOS in pull mode */
fail_unless (num_eos == 0);
gst_element_set_state (pipe, GST_STATE_NULL);
gst_element_get_state (pipe, NULL, NULL, -1);
/* make sure source hasn't sent a second one when going PAUSED => READY */
fail_unless (num_eos == 0);
gst_pad_remove_probe (srcpad, probe);
gst_object_unref (srcpad);
gst_message_unref (msg);
gst_object_unref (bus);
gst_object_unref (pipe);
}
GST_END_TEST;
static GstPadProbeReturn
segment_event_catcher (GstObject * pad, GstPadProbeInfo * info,
gpointer * user_data)
{
GstEvent *event = GST_PAD_PROBE_INFO_EVENT (info);
GstEvent **last_event = (GstEvent **) user_data;
fail_unless (event != NULL);
fail_unless (GST_IS_EVENT (event));
fail_unless (user_data != NULL);
if (GST_EVENT_TYPE (event) == GST_EVENT_SEGMENT) {
g_mutex_lock (&check_mutex);
fail_unless (*last_event == NULL);
*last_event = gst_event_copy (event);
g_cond_signal (&check_cond);
g_mutex_unlock (&check_mutex);
}
return GST_PAD_PROBE_OK;
}
/* basesrc_seek_events_rate_update:
* - make sure we get expected segment after sending a seek event
*/
GST_START_TEST (basesrc_seek_events_rate_update)
{
GstStateChangeReturn state_ret;
GstElement *src, *sink, *pipe;
GstMessage *msg;
GstBus *bus;
GstPad *probe_pad;
guint probe;
GstEvent *seg_event = NULL;
GstEvent *rate_seek;
gboolean event_ret;
const GstSegment *segment;
pipe = gst_pipeline_new ("pipeline");
sink = gst_element_factory_make ("fakesink", "sink");
src = gst_element_factory_make ("fakesrc", "src");
g_assert (pipe != NULL);
g_assert (sink != NULL);
g_assert (src != NULL);
fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
fail_unless (gst_element_link (src, sink) == TRUE);
bus = gst_element_get_bus (pipe);
/* set up event probe to catch new segment event */
probe_pad = gst_element_get_static_pad (sink, "sink");
fail_unless (probe_pad != NULL);
probe = gst_pad_add_probe (probe_pad, GST_PAD_PROBE_TYPE_EVENT_BOTH,
(GstPadProbeCallback) segment_event_catcher, &seg_event, NULL);
/* prepare the seek */
rate_seek = gst_event_new_seek (0.5, GST_FORMAT_TIME, GST_SEEK_FLAG_NONE,
GST_SEEK_TYPE_NONE, GST_CLOCK_TIME_NONE,
GST_SEEK_TYPE_NONE, GST_CLOCK_TIME_NONE);
GST_INFO ("going to playing");
/* play */
gst_element_set_state (pipe, GST_STATE_PLAYING);
state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
/* wait for the first segment to be posted, and flush it ... */
g_mutex_lock (&check_mutex);
while (seg_event == NULL)
g_cond_wait (&check_cond, &check_mutex);
gst_event_unref (seg_event);
seg_event = NULL;
g_mutex_unlock (&check_mutex);
GST_INFO ("seeking");
/* seek */
event_ret = gst_element_send_event (pipe, rate_seek);
fail_unless (event_ret == TRUE);
/* wait for the updated segment to be posted, posting EOS make the loop
* thread exit before the updated segment is posted ... */
g_mutex_lock (&check_mutex);
while (seg_event == NULL)
g_cond_wait (&check_cond, &check_mutex);
g_mutex_unlock (&check_mutex);
/* shut down pipeline only (should send EOS message) ... */
gst_element_send_event (pipe, gst_event_new_eos ());
/* ... and wait for the EOS message from the sink */
msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
fail_unless (msg != NULL);
fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
gst_element_set_state (pipe, GST_STATE_NULL);
gst_element_get_state (pipe, NULL, NULL, -1);
GST_INFO ("stopped");
/* check that we have go the event */
fail_unless (seg_event != NULL);
gst_event_parse_segment (seg_event, &segment);
fail_unless (segment->rate == 0.5);
gst_pad_remove_probe (probe_pad, probe);
gst_object_unref (probe_pad);
gst_message_unref (msg);
gst_event_unref (seg_event);
gst_object_unref (bus);
gst_object_unref (pipe);
}
GST_END_TEST;
typedef struct
{
gboolean seeked;
gint buffer_count;
GList *events;
} LastBufferSeekData;
static GstPadProbeReturn
seek_on_buffer (GstObject * pad, GstPadProbeInfo * info, gpointer * user_data)
{
LastBufferSeekData *data = (LastBufferSeekData *) user_data;
fail_unless (user_data != NULL);
if (info->type & GST_PAD_PROBE_TYPE_BUFFER) {
data->buffer_count++;
if (!data->seeked) {
fail_unless (gst_pad_push_event (GST_PAD (pad),
gst_event_new_seek (1.0, GST_FORMAT_BYTES, GST_SEEK_FLAG_FLUSH,
GST_SEEK_TYPE_SET, 0, GST_SEEK_TYPE_SET, 1)));
data->seeked = TRUE;
}
} else if (info->type & GST_PAD_PROBE_TYPE_EVENT_DOWNSTREAM) {
data->events = g_list_append (data->events, gst_event_ref (info->data));
} else {
fail ("Should not be reached");
}
return GST_PAD_PROBE_OK;
}
/* basesrc_seek_on_last_buffer:
* - make sure basesrc doesn't go eos if a seek is sent
* after the last buffer push
*
* This is just a test and is a controlled environment.
* For testing purposes sending the seek from the streaming
* thread is ok but doing this in an application might not
* be a good idea.
*/
GST_START_TEST (basesrc_seek_on_last_buffer)
{
GstStateChangeReturn state_ret;
GstElement *src, *sink, *pipe;
GstMessage *msg;
GstBus *bus;
GstPad *probe_pad;
guint probe;
GstEvent *seek;
LastBufferSeekData seek_data;
pipe = gst_pipeline_new ("pipeline");
sink = gst_element_factory_make ("fakesink", "sink");
src = gst_element_factory_make ("fakesrc", "src");
g_assert (pipe != NULL);
g_assert (sink != NULL);
g_assert (src != NULL);
/* use 'sizemax' buffers to avoid receiving empty buffers */
g_object_set (src, "sizetype", 2, NULL);
fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
fail_unless (gst_element_link (src, sink) == TRUE);
bus = gst_element_get_bus (pipe);
/* set up probe to catch the last buffer and send a seek event */
probe_pad = gst_element_get_static_pad (sink, "sink");
fail_unless (probe_pad != NULL);
seek_data.buffer_count = 0;
seek_data.seeked = FALSE;
seek_data.events = NULL;
probe =
gst_pad_add_probe (probe_pad,
GST_PAD_PROBE_TYPE_BUFFER | GST_PAD_PROBE_TYPE_EVENT_DOWNSTREAM,
(GstPadProbeCallback) seek_on_buffer, &seek_data, NULL);
/* prepare the segment so that it has only one buffer */
seek = gst_event_new_seek (1, GST_FORMAT_BYTES, GST_SEEK_FLAG_NONE,
GST_SEEK_TYPE_SET, 0, GST_SEEK_TYPE_SET, 1);
gst_element_set_state (pipe, GST_STATE_READY);
fail_unless (gst_element_send_event (src, seek));
GST_INFO ("going to playing");
/* play */
gst_element_set_state (pipe, GST_STATE_PLAYING);
state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
/* ... and wait for the EOS message from the sink */
msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
fail_unless (msg != NULL);
fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
gst_element_set_state (pipe, GST_STATE_NULL);
gst_element_get_state (pipe, NULL, NULL, -1);
GST_INFO ("stopped");
/* check that we have go the event */
fail_unless (seek_data.buffer_count == 2);
fail_unless (seek_data.seeked);
/* events: stream-start -> segment -> segment -> eos */
fail_unless (g_list_length (seek_data.events) == 4);
{
GstEvent *event;
event = seek_data.events->data;
fail_unless (GST_EVENT_TYPE (event) == GST_EVENT_STREAM_START);
gst_event_unref (event);
seek_data.events = g_list_delete_link (seek_data.events, seek_data.events);
event = seek_data.events->data;
fail_unless (GST_EVENT_TYPE (event) == GST_EVENT_SEGMENT);
gst_event_unref (event);
seek_data.events = g_list_delete_link (seek_data.events, seek_data.events);
event = seek_data.events->data;
fail_unless (GST_EVENT_TYPE (event) == GST_EVENT_SEGMENT);
gst_event_unref (event);
seek_data.events = g_list_delete_link (seek_data.events, seek_data.events);
event = seek_data.events->data;
fail_unless (GST_EVENT_TYPE (event) == GST_EVENT_EOS);
gst_event_unref (event);
seek_data.events = g_list_delete_link (seek_data.events, seek_data.events);
}
gst_pad_remove_probe (probe_pad, probe);
gst_object_unref (probe_pad);
gst_message_unref (msg);
gst_object_unref (bus);
gst_object_unref (pipe);
}
GST_END_TEST;
typedef GstBaseSrc TestSrc;
typedef GstBaseSrcClass TestSrcClass;
static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_STATIC_CAPS_ANY);
static GType test_src_get_type (void);
G_DEFINE_TYPE (TestSrc, test_src, GST_TYPE_BASE_SRC);
static void
test_src_init (TestSrc * src)
{
}
static GstFlowReturn
test_src_create (GstBaseSrc * src, guint64 offset, guint size,
GstBuffer ** p_buf)
{
GstBuffer *buf;
static int num = 0;
fail_if (*p_buf != NULL);
buf = gst_buffer_new ();
GST_BUFFER_OFFSET (buf) = num++;
if (num == 1 || g_random_boolean ()) {
GstBufferList *buflist = gst_buffer_list_new ();
gst_buffer_list_add (buflist, buf);
buf = gst_buffer_new ();
GST_BUFFER_OFFSET (buf) = num++;
gst_buffer_list_add (buflist, buf);
gst_base_src_submit_buffer_list (src, buflist);
} else {
*p_buf = buf;
}
return GST_FLOW_OK;
}
static void
test_src_class_init (TestSrcClass * klass)
{
GstBaseSrcClass *gstbasesrc_class = GST_BASE_SRC_CLASS (klass);
gst_element_class_add_static_pad_template (GST_ELEMENT_CLASS (klass),
&src_template);
gstbasesrc_class->create = test_src_create;
}
static GstPad *mysinkpad;
static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
GST_STATIC_CAPS_ANY);
#define NUM_BUFFERS 100
static gboolean done;
static guint expect_offset;
static GstFlowReturn
chain_____func (GstPad * pad, GstObject * parent, GstBuffer * buf)
{
GST_LOG (" buffer # %3u", (guint) GST_BUFFER_OFFSET (buf));
fail_unless_equals_int (GST_BUFFER_OFFSET (buf), expect_offset);
++expect_offset;
if (GST_BUFFER_OFFSET (buf) > NUM_BUFFERS) {
g_mutex_lock (&check_mutex);
done = TRUE;
g_cond_signal (&check_cond);
g_mutex_unlock (&check_mutex);
}
gst_buffer_unref (buf);
return GST_FLOW_OK;
}
static GstFlowReturn
chainlist_func (GstPad * pad, GstObject * parent, GstBufferList * list)
{
guint i, len;
len = gst_buffer_list_length (list);
GST_DEBUG ("buffer list with %u buffers", len);
for (i = 0; i < len; ++i) {
GstBuffer *buf = gst_buffer_list_get (list, i);
GST_LOG (" buffer # %3u", (guint) GST_BUFFER_OFFSET (buf));
fail_unless_equals_int (GST_BUFFER_OFFSET (buf), expect_offset);
++expect_offset;
}
gst_buffer_list_unref (list);
return GST_FLOW_OK;
}
GST_START_TEST (basesrc_create_bufferlist)
{
GstElement *src;
src = g_object_new (test_src_get_type (), NULL);
mysinkpad = gst_check_setup_sink_pad (src, &sinktemplate);
gst_pad_set_chain_function (mysinkpad, chain_____func);
gst_pad_set_chain_list_function (mysinkpad, chainlist_func);
gst_pad_set_active (mysinkpad, TRUE);
done = FALSE;
expect_offset = 0;
gst_element_set_state (src, GST_STATE_PLAYING);
g_mutex_lock (&check_mutex);
while (!done)
g_cond_wait (&check_cond, &check_mutex);
g_mutex_unlock (&check_mutex);
gst_element_set_state (src, GST_STATE_NULL);
gst_check_teardown_sink_pad (src);
gst_object_unref (src);
}
GST_END_TEST;
typedef struct
{
GstBaseSrc parent;
GstSegment *segment;
gboolean n_output_buffers;
} TimeSrc;
typedef GstBaseSrcClass TimeSrcClass;
static GType time_src_get_type (void);
G_DEFINE_TYPE (TimeSrc, time_src, GST_TYPE_BASE_SRC);
static void
time_src_init (TimeSrc * src)
{
gst_base_src_set_format (GST_BASE_SRC (src), GST_FORMAT_TIME);
gst_base_src_set_automatic_eos (GST_BASE_SRC (src), FALSE);
src->n_output_buffers = 0;
}
/* This test src outputs a compressed format, with a single GOP
* starting at PTS 0.
*
* This means that in reverse playback, we may want to output
* buffers outside the segment bounds, and decide when to EOS
* from the create function.
*/
static GstFlowReturn
time_src_create (GstBaseSrc * bsrc, guint64 offset, guint size,
GstBuffer ** p_buf)
{
TimeSrc *src = (TimeSrc *) bsrc;
if (src->segment->position >= src->segment->stop)
return GST_FLOW_EOS;
*p_buf = gst_buffer_new ();
GST_BUFFER_PTS (*p_buf) = src->segment->position;
GST_BUFFER_DURATION (*p_buf) = GST_SECOND;
src->segment->position += GST_SECOND;
src->n_output_buffers++;
return GST_FLOW_OK;
}
static gboolean
time_src_do_seek (GstBaseSrc * bsrc, GstSegment * segment)
{
TimeSrc *src = (TimeSrc *) bsrc;
fail_unless (segment->format == GST_FORMAT_TIME);
if (src->segment)
gst_segment_free (src->segment);
src->segment = gst_segment_copy (segment);
((TimeSrc *) bsrc)->segment->position = 0;
return TRUE;
}
static gboolean
time_src_is_seekable (GstBaseSrc * bsrc)
{
return TRUE;
}
static void
time_src_class_init (TimeSrcClass * klass)
{
GstBaseSrcClass *gstbasesrc_class = GST_BASE_SRC_CLASS (klass);
gst_element_class_add_static_pad_template (GST_ELEMENT_CLASS (klass),
&src_template);
gstbasesrc_class->create = time_src_create;
gstbasesrc_class->do_seek = time_src_do_seek;
gstbasesrc_class->is_seekable = time_src_is_seekable;
}
GST_START_TEST (basesrc_time_automatic_eos)
{
GstElement *src, *sink;
GstElement *pipe;
GstBus *bus;
src = g_object_new (time_src_get_type (), NULL);
sink = gst_element_factory_make ("fakesink", NULL);
pipe = gst_pipeline_new (NULL);
gst_bin_add (GST_BIN (pipe), src);
gst_bin_add (GST_BIN (pipe), sink);
gst_element_link (src, sink);
gst_element_set_state (pipe, GST_STATE_PAUSED);
gst_element_get_state (pipe, NULL, NULL, GST_CLOCK_TIME_NONE);
gst_element_seek (pipe, -1.0, GST_FORMAT_TIME,
GST_SEEK_FLAG_FLUSH | GST_SEEK_FLAG_ACCURATE, GST_SEEK_TYPE_SET,
GST_SECOND, GST_SEEK_TYPE_SET, 2 * GST_SECOND);
gst_element_set_state (pipe, GST_STATE_PLAYING);
bus = gst_pipeline_get_bus (GST_PIPELINE (pipe));
gst_message_unref (gst_bus_timed_pop_filtered (bus, GST_CLOCK_TIME_NONE,
GST_MESSAGE_EOS));
gst_element_set_state (pipe, GST_STATE_NULL);
/* - One preroll buffer
* - One (1-second) buffer outside the segment, our "keyframe"
* - One buffer inside the segment
*/
fail_unless_equals_int (((TimeSrc *) src)->n_output_buffers, 3);
if (((TimeSrc *) src)->segment)
gst_segment_free (((TimeSrc *) src)->segment);
gst_object_unref (bus);
gst_object_unref (pipe);
}
GST_END_TEST;
static Suite *
gst_basesrc_suite (void)
{
Suite *s = suite_create ("GstBaseSrc");
TCase *tc = tcase_create ("general");
suite_add_tcase (s, tc);
tcase_add_test (tc, basesrc_eos_events_pull);
tcase_add_test (tc, basesrc_eos_events_push);
tcase_add_test (tc, basesrc_eos_events_push_live_op);
tcase_add_test (tc, basesrc_eos_events_pull_live_op);
tcase_add_test (tc, basesrc_eos_events_push_live_eos);
tcase_add_test (tc, basesrc_eos_events_pull_live_eos);
tcase_add_test (tc, basesrc_seek_events_rate_update);
tcase_add_test (tc, basesrc_seek_on_last_buffer);
tcase_add_test (tc, basesrc_create_bufferlist);
tcase_add_test (tc, basesrc_time_automatic_eos);
return s;
}
GST_CHECK_MAIN (gst_basesrc);