/* GStreamer * * unit test for mxfmux * * 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., 51 Franklin St, Fifth Floor, * Boston, MA 02110-1301, 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 ("avenc_mpeg2video"))) { gst_object_unref (factory); return "avenc_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 run_test (const gchar * pipeline_string) { GstElement *pipeline; GstBus *bus; GMainLoop *loop; OnMessageUserData omud = { NULL, }; GstStateChangeReturn ret; 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); 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); 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 ! " "fakesink", mpeg2enc_name); run_test (pipeline); 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 ! " "fakesink " "audiotestsrc num-buffers=250 ! " "audioconvert ! " "audio/x-raw,rate=48000,channels=2 ! " "mux. "); run_test (pipeline); 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 ! " "fakesink"); run_test (pipeline); g_free (pipeline); } GST_END_TEST; GST_START_TEST (test_jpeg2000_alaw) { gchar *pipeline; GstElementFactory *factory = NULL; if ((factory = gst_element_factory_find ("openjpegenc")) == 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 ! " "openjpegenc ! " "mxfmux name=mux ! " "fakesink " "audiotestsrc num-buffers=250 ! " "audioconvert ! " "alawenc ! " "mux. "); run_test (pipeline); g_free (pipeline); } GST_END_TEST; GST_START_TEST (test_dnxhd_mp3) { gchar *pipeline; GstElementFactory *factory = NULL; if ((factory = gst_element_factory_find ("avenc_dnxhd")) == NULL) return; gst_object_unref (factory); if ((factory = gst_element_factory_find ("lamemp3enc")) == NULL) return; gst_object_unref (factory); if ((factory = gst_element_factory_find ("mpegaudioparse")) == 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 ! " "avenc_dnxhd bitrate=36000000 ! " "mxfmux name=mux ! " "fakesink " "audiotestsrc num-buffers=250 ! " "audioconvert ! " "audio/x-raw,channels=2 ! lamemp3enc ! mpegaudioparse ! mux. "); run_test (pipeline); 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 ! " "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); g_free (pipeline); } GST_END_TEST; GST_START_TEST (test_h264_raw_audio) { gchar *pipeline; GstElementFactory *factory = NULL; if ((factory = gst_element_factory_find ("x264enc")) == NULL) return; gst_object_unref (factory); if ((factory = gst_element_factory_find ("h264parse")) == NULL) return; gst_object_unref (factory); pipeline = g_strdup_printf ("videotestsrc num-buffers=250 ! " "video/x-raw,framerate=25/1 ! " "x264enc ! h264parse ! " "mxfmux name=mux ! " "fakesink " "audiotestsrc num-buffers=250 ! " "audioconvert ! " "audio/x-raw,format=S24LE,channels=2 ! mux. "); run_test (pipeline); g_free (pipeline); } GST_END_TEST; static Suite * mxfmux_suite (void) { Suite *s = suite_create ("mxfmux"); 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_h264_raw_audio); tcase_add_test (tc_chain, test_multiple_av_streams); return s; } GST_CHECK_MAIN (mxfmux);