/* GStreamer * * unit test for mxfmux ! mxfdemux pipelines * * Copyright (C) <2009> Sebastian Dröge * * 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 #include 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); 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_bus_remove_signal_watch (bus); gst_object_unref (bus); } 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,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,format=(string)v308,width=1920,height=1080,framerate=25/1 ! " "mxfmux name=mux ! " "mxfdemux name=demux ! " "fakesink " "audiotestsrc num-buffers=250 ! " "audioconvert ! " "audio/x-raw,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,format=(string)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,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,format=(string)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,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,format=(string)v308,width=1920,height=1080,framerate=25/1 ! " "mxfmux name=mux ! " "mxfdemux name=demux ! " "fakesink " "audiotestsrc num-buffers=250 ! " "audioconvert ! " "audio/x-raw,rate=48000,channels=2 ! " "mux. " "videotestsrc num-buffers=100 ! " "video/x-raw,format=(string)v308,width=1920,height=1080,framerate=25/1 ! " "mux. " "audiotestsrc num-buffers=100 ! " "audioconvert ! " "audio/x-raw,rate=48000,channels=2 ! " "mux. " "audiotestsrc num-buffers=250 ! " "audioconvert ! " "audio/x-raw,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);