gstreamer/tests/check/pipelines/mxf.c

282 lines
7.3 KiB
C

/* GStreamer
*
* unit test for mxfmux ! mxfdemux pipelines
*
* Copyright (C) <2009> Sebastian Dröge <sebastian.droege@collabora.co.uk>
*
* 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 <gst/check/gstcheck.h>
#include <string.h>
static const gchar *
get_mpeg2enc_element_name (void)
{
GstElementFactory *factory = NULL;
if ((factory = gst_element_factory_find ("mpeg2enc"))) {
gst_object_unref (factory);
return "mpeg2enc";
} else if ((factory = gst_element_factory_find ("ffenc_mpeg2video"))) {
gst_object_unref (factory);
return "ffenc_mpeg2video";
} else {
return NULL;
}
}
typedef struct
{
GMainLoop *loop;
gboolean eos;
} OnMessageUserData;
static void
on_message_cb (GstBus * bus, GstMessage * message, gpointer user_data)
{
OnMessageUserData *d = user_data;
switch (GST_MESSAGE_TYPE (message)) {
case GST_MESSAGE_ERROR:
case GST_MESSAGE_WARNING:
g_assert_not_reached ();
break;
case GST_MESSAGE_EOS:
g_main_loop_quit (d->loop);
d->eos = TRUE;
break;
default:
break;
}
}
static void
on_pad_added (GstElement * element, GstPad * pad, gpointer user_data)
{
gint *n_pads = user_data;
*n_pads = *n_pads + 1;
}
static void
run_test (const gchar * pipeline_string, gint n_pads_expected)
{
GstElement *pipeline;
GstBus *bus;
GMainLoop *loop;
OnMessageUserData omud = { NULL, };
GstStateChangeReturn ret;
GstElement *demux;
gint n_pads = 0;
GST_DEBUG ("Testing pipeline '%s'", pipeline_string);
pipeline = gst_parse_launch (pipeline_string, NULL);
fail_unless (pipeline != NULL);
g_object_set (G_OBJECT (pipeline), "async-handling", TRUE, NULL);
demux = gst_bin_get_by_name (GST_BIN (pipeline), "demux");
fail_unless (demux != NULL);
g_signal_connect (demux, "pad-added", (GCallback) on_pad_added, &n_pads);
gst_object_unref (demux);
loop = g_main_loop_new (NULL, FALSE);
bus = gst_element_get_bus (pipeline);
fail_unless (bus != NULL);
gst_bus_add_signal_watch (bus);
omud.loop = loop;
omud.eos = FALSE;
g_signal_connect (bus, "message", (GCallback) on_message_cb, &omud);
gst_object_unref (bus);
ret = gst_element_set_state (pipeline, GST_STATE_PLAYING);
fail_unless (ret == GST_STATE_CHANGE_SUCCESS
|| ret == GST_STATE_CHANGE_ASYNC);
g_main_loop_run (loop);
fail_unless (gst_element_set_state (pipeline,
GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS);
fail_unless (omud.eos == TRUE);
fail_unless_equals_int (n_pads, n_pads_expected);
gst_object_unref (pipeline);
g_main_loop_unref (loop);
}
GST_START_TEST (test_mpeg2)
{
const gchar *mpeg2enc_name = get_mpeg2enc_element_name ();
gchar *pipeline;
if (!mpeg2enc_name)
return;
pipeline = g_strdup_printf ("videotestsrc num-buffers=250 ! "
"video/x-raw-yuv,framerate=25/1 ! "
"%s ! " "mxfmux name=mux ! "
"mxfdemux name=demux ! " "fakesink", mpeg2enc_name);
run_test (pipeline, 1);
g_free (pipeline);
}
GST_END_TEST;
GST_START_TEST (test_raw_video_raw_audio)
{
gchar *pipeline;
pipeline = g_strdup_printf ("videotestsrc num-buffers=250 ! "
"video/x-raw-yuv,format=(GstFourcc)v308,width=1920,height=1080,framerate=25/1 ! "
"mxfmux name=mux ! "
"mxfdemux name=demux ! "
"fakesink "
"audiotestsrc num-buffers=250 ! "
"audioconvert ! " "audio/x-raw-int,rate=48000,channels=2 ! " "mux. ");
run_test (pipeline, 2);
g_free (pipeline);
}
GST_END_TEST;
GST_START_TEST (test_raw_video_stride_transform)
{
gchar *pipeline;
pipeline = g_strdup_printf ("videotestsrc num-buffers=250 ! "
"video/x-raw-yuv,format=(GstFourcc)v308,width=1001,height=501,framerate=25/1 ! "
"mxfmux name=mux ! " "mxfdemux name=demux ! " "fakesink");
run_test (pipeline, 1);
g_free (pipeline);
}
GST_END_TEST;
GST_START_TEST (test_jpeg2000_alaw)
{
gchar *pipeline;
GstElementFactory *factory = NULL;
if ((factory = gst_element_factory_find ("jp2kenc")) == NULL)
return;
gst_object_unref (factory);
if ((factory = gst_element_factory_find ("alawenc")) == NULL)
return;
gst_object_unref (factory);
pipeline = g_strdup_printf ("videotestsrc num-buffers=250 ! "
"video/x-raw-yuv,framerate=25/1 ! "
"jp2kenc ! "
"mxfmux name=mux ! "
"mxfdemux name=demux ! "
"fakesink "
"audiotestsrc num-buffers=250 ! " "audioconvert ! " "alawenc ! " "mux. ");
run_test (pipeline, 2);
g_free (pipeline);
}
GST_END_TEST;
GST_START_TEST (test_dnxhd_mp3)
{
gchar *pipeline;
GstElementFactory *factory = NULL;
if ((factory = gst_element_factory_find ("ffenc_dnxhd")) == NULL)
return;
gst_object_unref (factory);
if ((factory = gst_element_factory_find ("lame")) == NULL)
return;
gst_object_unref (factory);
if ((factory = gst_element_factory_find ("mp3parse")) == NULL)
return;
gst_object_unref (factory);
pipeline = g_strdup_printf ("videotestsrc num-buffers=250 ! "
"video/x-raw-yuv,format=(GstFourcc)Y42B,width=1920,height=1080,framerate=25/1 ! "
"ffenc_dnxhd bitrate=36000000 ! "
"mxfmux name=mux ! "
"mxfdemux name=demux ! "
"fakesink "
"audiotestsrc num-buffers=250 ! "
"audioconvert ! "
"audio/x-raw-int,channels=2 ! " "lame ! " "mp3parse ! " "mux. ");
run_test (pipeline, 2);
g_free (pipeline);
}
GST_END_TEST;
GST_START_TEST (test_multiple_av_streams)
{
gchar *pipeline;
pipeline = g_strdup_printf ("videotestsrc num-buffers=250 ! "
"video/x-raw-yuv,format=(GstFourcc)v308,width=1920,height=1080,framerate=25/1 ! "
"mxfmux name=mux ! "
"mxfdemux name=demux ! "
"fakesink "
"audiotestsrc num-buffers=250 ! "
"audioconvert ! "
"audio/x-raw-int,rate=48000,channels=2 ! "
"mux. "
"videotestsrc num-buffers=100 ! "
"video/x-raw-yuv,format=(GstFourcc)v308,width=1920,height=1080,framerate=25/1 ! "
"mux. "
"audiotestsrc num-buffers=100 ! "
"audioconvert ! "
"audio/x-raw-int,rate=48000,channels=2 ! "
"mux. "
"audiotestsrc num-buffers=250 ! "
"audioconvert ! " "audio/x-raw-int,rate=48000,channels=2 ! " "mux. ");
run_test (pipeline, 5);
g_free (pipeline);
}
GST_END_TEST;
static Suite *
mxf_suite (void)
{
Suite *s = suite_create ("mxf");
TCase *tc_chain = tcase_create ("general");
suite_add_tcase (s, tc_chain);
tcase_set_timeout (tc_chain, 180);
tcase_add_test (tc_chain, test_mpeg2);
tcase_add_test (tc_chain, test_raw_video_raw_audio);
tcase_add_test (tc_chain, test_raw_video_stride_transform);
tcase_add_test (tc_chain, test_jpeg2000_alaw);
tcase_add_test (tc_chain, test_dnxhd_mp3);
tcase_add_test (tc_chain, test_multiple_av_streams);
return s;
}
GST_CHECK_MAIN (mxf);