mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-08 13:18:52 +00:00
396 lines
13 KiB
C
396 lines
13 KiB
C
/* GStreamer
|
|
*
|
|
* unit test for vorbisenc
|
|
*
|
|
* Copyright (C) 2006 Andy Wingo <wingo at pobox.com>
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#include <gst/check/gstcheck.h>
|
|
#include <gst/check/gstbufferstraw.h>
|
|
|
|
#ifndef GST_DISABLE_PARSE
|
|
|
|
#define TIMESTAMP_OFFSET G_GINT64_CONSTANT(3249870963)
|
|
|
|
static void
|
|
check_buffer_timestamp (GstBuffer * buffer, GstClockTime timestamp)
|
|
{
|
|
fail_unless (GST_BUFFER_TIMESTAMP (buffer) == timestamp,
|
|
"expected timestamp %" GST_TIME_FORMAT
|
|
", but got timestamp %" GST_TIME_FORMAT,
|
|
GST_TIME_ARGS (timestamp), GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buffer)));
|
|
}
|
|
|
|
static void
|
|
check_buffer_duration (GstBuffer * buffer, GstClockTime duration)
|
|
{
|
|
fail_unless (GST_BUFFER_DURATION (buffer) == duration,
|
|
"expected duration %" GST_TIME_FORMAT
|
|
", but got duration %" GST_TIME_FORMAT,
|
|
GST_TIME_ARGS (duration), GST_TIME_ARGS (GST_BUFFER_DURATION (buffer)));
|
|
}
|
|
|
|
static void
|
|
check_buffer_granulepos (GstBuffer * buffer, gint64 granulepos)
|
|
{
|
|
GstClockTime clocktime;
|
|
|
|
fail_unless (GST_BUFFER_OFFSET_END (buffer) == granulepos,
|
|
"expected granulepos %" G_GUINT64_FORMAT
|
|
", but got granulepos %" G_GUINT64_FORMAT,
|
|
granulepos, GST_BUFFER_OFFSET_END (buffer));
|
|
|
|
/* contrary to what we record as TIMESTAMP, we can use OFFSET to check
|
|
* the granulepos correctly here */
|
|
clocktime = gst_util_uint64_scale (GST_BUFFER_OFFSET_END (buffer), 44100,
|
|
GST_SECOND);
|
|
|
|
fail_unless (clocktime == GST_BUFFER_OFFSET (buffer),
|
|
"expected OFFSET set to clocktime %" GST_TIME_FORMAT
|
|
", but got %" GST_TIME_FORMAT,
|
|
GST_TIME_ARGS (clocktime), GST_TIME_ARGS (GST_BUFFER_OFFSET (buffer)));
|
|
}
|
|
|
|
/* this check is here to check that the granulepos we derive from the timestamp
|
|
is about correct. This is "about correct" because you can't precisely go from
|
|
timestamp to granulepos due to the downward-rounding characteristics of
|
|
gst_util_uint64_scale, so you check if granulepos is equal to the number, or
|
|
the number plus one. */
|
|
static void
|
|
check_buffer_granulepos_from_endtime (GstBuffer * buffer, GstClockTime endtime)
|
|
{
|
|
gint64 granulepos, expected;
|
|
|
|
granulepos = GST_BUFFER_OFFSET_END (buffer);
|
|
expected = gst_util_uint64_scale (endtime, 44100, GST_SECOND);
|
|
|
|
fail_unless (granulepos == expected || granulepos == expected + 1,
|
|
"expected granulepos %" G_GUINT64_FORMAT
|
|
" or %" G_GUINT64_FORMAT
|
|
", but got granulepos %" G_GUINT64_FORMAT,
|
|
expected, expected + 1, granulepos);
|
|
}
|
|
|
|
GST_START_TEST (test_granulepos_offset)
|
|
{
|
|
GstElement *bin;
|
|
GstPad *pad;
|
|
gchar *pipe_str;
|
|
GstBuffer *buffer;
|
|
GError *error = NULL;
|
|
|
|
pipe_str = g_strdup_printf ("audiotestsrc timestamp-offset=%" G_GUINT64_FORMAT
|
|
" ! audio/x-raw,rate=44100"
|
|
" ! audioconvert ! vorbisenc ! fakesink name=sink", TIMESTAMP_OFFSET);
|
|
|
|
bin = gst_parse_launch (pipe_str, &error);
|
|
fail_unless (bin != NULL, "Error parsing pipeline: %s",
|
|
error ? error->message : "(invalid error)");
|
|
g_free (pipe_str);
|
|
|
|
/* get the pad */
|
|
{
|
|
GstElement *sink = gst_bin_get_by_name (GST_BIN (bin), "sink");
|
|
|
|
fail_unless (sink != NULL, "Could not get fakesink out of bin");
|
|
pad = gst_element_get_static_pad (sink, "sink");
|
|
fail_unless (pad != NULL, "Could not get pad out of fakesink");
|
|
gst_object_unref (sink);
|
|
}
|
|
|
|
gst_buffer_straw_start_pipeline (bin, pad);
|
|
|
|
/* header packets should have timestamp == NONE, granulepos 0 */
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
GST_DEBUG ("Got buffer in test");
|
|
check_buffer_timestamp (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_duration (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_granulepos (buffer, 0);
|
|
gst_buffer_unref (buffer);
|
|
GST_DEBUG ("Unreffed buffer in test");
|
|
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
check_buffer_timestamp (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_duration (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_granulepos (buffer, 0);
|
|
gst_buffer_unref (buffer);
|
|
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
check_buffer_timestamp (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_duration (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_granulepos (buffer, 0);
|
|
gst_buffer_unref (buffer);
|
|
|
|
{
|
|
GstClockTime next_timestamp;
|
|
gint64 last_granulepos = 0;
|
|
|
|
/* first buffer should have timestamp of TIMESTAMP_OFFSET, granulepos to
|
|
* match the timestamp of the end of the last sample in the output buffer.
|
|
* Note that one cannot go timestamp->granulepos->timestamp and get the same
|
|
* value due to loss of precision with granulepos. vorbisenc does take care
|
|
* to timestamp correctly based on the offset of the input data however, so
|
|
* it does do sub-granulepos timestamping. */
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
last_granulepos = GST_BUFFER_OFFSET_END (buffer);
|
|
check_buffer_timestamp (buffer, TIMESTAMP_OFFSET);
|
|
/* don't really have a good way of checking duration... */
|
|
check_buffer_granulepos_from_endtime (buffer,
|
|
TIMESTAMP_OFFSET + GST_BUFFER_DURATION (buffer));
|
|
|
|
next_timestamp = TIMESTAMP_OFFSET + GST_BUFFER_DURATION (buffer);
|
|
|
|
gst_buffer_unref (buffer);
|
|
|
|
/* check continuity with the next buffer */
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
check_buffer_timestamp (buffer, next_timestamp);
|
|
check_buffer_duration (buffer,
|
|
gst_util_uint64_scale (GST_BUFFER_OFFSET_END (buffer), GST_SECOND,
|
|
44100)
|
|
- gst_util_uint64_scale (last_granulepos, GST_SECOND, 44100));
|
|
check_buffer_granulepos_from_endtime (buffer,
|
|
next_timestamp + GST_BUFFER_DURATION (buffer));
|
|
|
|
gst_buffer_unref (buffer);
|
|
}
|
|
|
|
gst_buffer_straw_stop_pipeline (bin, pad);
|
|
|
|
gst_object_unref (pad);
|
|
gst_object_unref (bin);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
GST_START_TEST (test_timestamps)
|
|
{
|
|
GstElement *bin;
|
|
GstPad *pad;
|
|
gchar *pipe_str;
|
|
GstBuffer *buffer;
|
|
GError *error = NULL;
|
|
|
|
pipe_str = g_strdup_printf ("audiotestsrc"
|
|
" ! audio/x-raw,rate=44100 ! audioconvert ! vorbisenc "
|
|
" ! fakesink name=sink");
|
|
|
|
bin = gst_parse_launch (pipe_str, &error);
|
|
fail_unless (bin != NULL, "Error parsing pipeline: %s",
|
|
error ? error->message : "(invalid error)");
|
|
g_free (pipe_str);
|
|
|
|
/* get the pad */
|
|
{
|
|
GstElement *sink = gst_bin_get_by_name (GST_BIN (bin), "sink");
|
|
|
|
fail_unless (sink != NULL, "Could not get fakesink out of bin");
|
|
pad = gst_element_get_static_pad (sink, "sink");
|
|
fail_unless (pad != NULL, "Could not get pad out of fakesink");
|
|
gst_object_unref (sink);
|
|
}
|
|
|
|
gst_buffer_straw_start_pipeline (bin, pad);
|
|
|
|
/* check header packets */
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
check_buffer_timestamp (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_duration (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_granulepos (buffer, 0);
|
|
gst_buffer_unref (buffer);
|
|
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
check_buffer_timestamp (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_duration (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_granulepos (buffer, 0);
|
|
gst_buffer_unref (buffer);
|
|
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
check_buffer_timestamp (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_duration (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_granulepos (buffer, 0);
|
|
gst_buffer_unref (buffer);
|
|
|
|
{
|
|
GstClockTime next_timestamp;
|
|
gint64 last_granulepos;
|
|
|
|
/* first buffer has timestamp 0 */
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
last_granulepos = GST_BUFFER_OFFSET_END (buffer);
|
|
check_buffer_timestamp (buffer, 0);
|
|
/* don't really have a good way of checking duration... */
|
|
check_buffer_granulepos_from_endtime (buffer, GST_BUFFER_DURATION (buffer));
|
|
|
|
next_timestamp = GST_BUFFER_DURATION (buffer);
|
|
|
|
gst_buffer_unref (buffer);
|
|
|
|
/* check continuity with the next buffer */
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
check_buffer_timestamp (buffer, next_timestamp);
|
|
check_buffer_duration (buffer,
|
|
gst_util_uint64_scale (GST_BUFFER_OFFSET_END (buffer), GST_SECOND,
|
|
44100)
|
|
- gst_util_uint64_scale (last_granulepos, GST_SECOND, 44100));
|
|
check_buffer_granulepos_from_endtime (buffer,
|
|
next_timestamp + GST_BUFFER_DURATION (buffer));
|
|
|
|
gst_buffer_unref (buffer);
|
|
}
|
|
|
|
gst_buffer_straw_stop_pipeline (bin, pad);
|
|
|
|
gst_object_unref (pad);
|
|
gst_object_unref (bin);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
static GstPadProbeReturn
|
|
drop_second_data_buffer (GstPad * droppad, GstPadProbeInfo * info,
|
|
gpointer unused)
|
|
{
|
|
GstBuffer *buffer = GST_PAD_PROBE_INFO_BUFFER (info);
|
|
GstPadProbeReturn res = GST_PAD_PROBE_OK;
|
|
|
|
if (GST_BUFFER_OFFSET (buffer) == 4096)
|
|
res = GST_PAD_PROBE_DROP;
|
|
|
|
GST_DEBUG ("dropping %d", res);
|
|
|
|
return res;
|
|
}
|
|
|
|
GST_START_TEST (test_discontinuity)
|
|
{
|
|
GstElement *bin;
|
|
GstPad *pad, *droppad;
|
|
gchar *pipe_str;
|
|
GstBuffer *buffer;
|
|
GError *error = NULL;
|
|
guint drop_id;
|
|
|
|
/* make audioencoder act sufficiently pedantic */
|
|
pipe_str = g_strdup_printf ("audiotestsrc samplesperbuffer=1024"
|
|
" ! audio/x-raw,rate=44100 ! audioconvert "
|
|
" ! vorbisenc tolerance=10000000 name=enc ! fakesink name=sink");
|
|
|
|
bin = gst_parse_launch (pipe_str, &error);
|
|
fail_unless (bin != NULL, "Error parsing pipeline: %s",
|
|
error ? error->message : "(invalid error)");
|
|
g_free (pipe_str);
|
|
|
|
/* the plan: same as test_timestamps, but dropping a buffer and seeing if
|
|
vorbisenc correctly notes the discontinuity */
|
|
|
|
/* get the pad to use to drop buffers */
|
|
{
|
|
GstElement *sink = gst_bin_get_by_name (GST_BIN (bin), "enc");
|
|
|
|
fail_unless (sink != NULL, "Could not get vorbisenc out of bin");
|
|
droppad = gst_element_get_static_pad (sink, "sink");
|
|
fail_unless (droppad != NULL, "Could not get pad out of vorbisenc");
|
|
gst_object_unref (sink);
|
|
}
|
|
|
|
/* get the pad */
|
|
{
|
|
GstElement *sink = gst_bin_get_by_name (GST_BIN (bin), "sink");
|
|
|
|
fail_unless (sink != NULL, "Could not get fakesink out of bin");
|
|
pad = gst_element_get_static_pad (sink, "sink");
|
|
fail_unless (pad != NULL, "Could not get pad out of fakesink");
|
|
gst_object_unref (sink);
|
|
}
|
|
|
|
drop_id = gst_pad_add_probe (droppad, GST_PAD_PROBE_TYPE_BUFFER,
|
|
(GstPadProbeCallback) drop_second_data_buffer, NULL, NULL);
|
|
gst_buffer_straw_start_pipeline (bin, pad);
|
|
|
|
/* check header packets */
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
check_buffer_timestamp (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_duration (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_granulepos (buffer, 0);
|
|
gst_buffer_unref (buffer);
|
|
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
check_buffer_timestamp (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_duration (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_granulepos (buffer, 0);
|
|
gst_buffer_unref (buffer);
|
|
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
check_buffer_timestamp (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_duration (buffer, GST_CLOCK_TIME_NONE);
|
|
check_buffer_granulepos (buffer, 0);
|
|
gst_buffer_unref (buffer);
|
|
|
|
{
|
|
GstClockTime next_timestamp = 0;
|
|
gint64 last_granulepos = 0, granulepos;
|
|
gint i;
|
|
|
|
for (i = 0; i < 10; i++) {
|
|
buffer = gst_buffer_straw_get_buffer (bin, pad);
|
|
granulepos = GST_BUFFER_OFFSET_END (buffer);
|
|
/* discont is either at start, or following gap */
|
|
if (GST_BUFFER_IS_DISCONT (buffer)) {
|
|
if (next_timestamp) {
|
|
fail_unless (granulepos - last_granulepos > 1024,
|
|
"expected discont of at least 1024 samples");
|
|
next_timestamp = GST_BUFFER_TIMESTAMP (buffer);
|
|
}
|
|
}
|
|
check_buffer_timestamp (buffer, next_timestamp);
|
|
next_timestamp += GST_BUFFER_DURATION (buffer);
|
|
last_granulepos = granulepos;
|
|
gst_buffer_unref (buffer);
|
|
}
|
|
}
|
|
|
|
gst_buffer_straw_stop_pipeline (bin, pad);
|
|
gst_pad_remove_probe (droppad, drop_id);
|
|
|
|
gst_object_unref (droppad);
|
|
gst_object_unref (pad);
|
|
gst_object_unref (bin);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
#endif /* #ifndef GST_DISABLE_PARSE */
|
|
|
|
static Suite *
|
|
vorbisenc_suite (void)
|
|
{
|
|
Suite *s = suite_create ("vorbisenc");
|
|
TCase *tc_chain = tcase_create ("general");
|
|
|
|
suite_add_tcase (s, tc_chain);
|
|
#ifndef GST_DISABLE_PARSE
|
|
tcase_add_test (tc_chain, test_granulepos_offset);
|
|
tcase_add_test (tc_chain, test_timestamps);
|
|
tcase_add_test (tc_chain, test_discontinuity);
|
|
#endif
|
|
|
|
return s;
|
|
}
|
|
|
|
GST_CHECK_MAIN (vorbisenc);
|