/* GStreamer * Copyright (C) 2010 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 #include static gboolean bus_handler (GstBus * bus, GstMessage * message, gpointer data) { GMainLoop *loop = (GMainLoop *) data; switch (message->type) { case GST_MESSAGE_EOS: g_main_loop_quit (loop); break; case GST_MESSAGE_WARNING: case GST_MESSAGE_ERROR:{ GError *gerror; gchar *debug; if (message->type == GST_MESSAGE_WARNING) gst_message_parse_warning (message, &gerror, &debug); else gst_message_parse_error (message, &gerror, &debug); g_error ("error from %s: %s (%s)\n", GST_ELEMENT_NAME (GST_MESSAGE_SRC (message)), gerror->message, GST_STR_NULL (debug)); gst_message_unref (message); g_error_free (gerror); g_free (debug); g_main_loop_quit (loop); break; } default: break; } return TRUE; } static GstElement * setup_imagefreeze (const GstCaps * caps1, const GstCaps * caps2, GCallback sink_handoff, gpointer sink_handoff_data) { GstElement *pipeline; GstElement *videotestsrc, *capsfilter1, *imagefreeze, *capsfilter2, *fakesink; pipeline = gst_pipeline_new ("pipeline"); fail_unless (pipeline != NULL); videotestsrc = gst_element_factory_make ("videotestsrc", NULL); fail_unless (videotestsrc != NULL); g_object_set (videotestsrc, "num-buffers", 1, NULL); capsfilter1 = gst_element_factory_make ("capsfilter", NULL); fail_unless (capsfilter1 != NULL); g_object_set (capsfilter1, "caps", caps1, NULL); imagefreeze = gst_element_factory_make ("imagefreeze", NULL); fail_unless (imagefreeze != NULL); capsfilter2 = gst_element_factory_make ("capsfilter", NULL); fail_unless (capsfilter2 != NULL); g_object_set (capsfilter2, "caps", caps2, NULL); fakesink = gst_element_factory_make ("fakesink", NULL); fail_unless (fakesink != NULL); g_object_set (fakesink, "signal-handoffs", TRUE, "async", FALSE, NULL); g_signal_connect (fakesink, "handoff", sink_handoff, sink_handoff_data); gst_bin_add_many (GST_BIN (pipeline), videotestsrc, capsfilter1, imagefreeze, capsfilter2, fakesink, NULL); fail_unless (gst_element_link_pads (videotestsrc, "src", capsfilter1, "sink")); fail_unless (gst_element_link_pads (capsfilter1, "src", imagefreeze, "sink")); fail_unless (gst_element_link_pads (imagefreeze, "src", capsfilter2, "sink")); fail_unless (gst_element_link_pads (capsfilter2, "src", fakesink, "sink")); return pipeline; } static void sink_handoff_cb_0_1 (GstElement * object, GstBuffer * buffer, GstPad * pad, gpointer user_data) { guint *n_buffers = (guint *) user_data; fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer), 0); fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), GST_CLOCK_TIME_NONE); fail_unless_equals_uint64 (GST_BUFFER_OFFSET (buffer), 0); fail_unless_equals_uint64 (GST_BUFFER_OFFSET_END (buffer), 1); *n_buffers = *n_buffers + 1; } GST_START_TEST (test_imagefreeze_0_1) { GstElement *pipeline; GstCaps *caps1, *caps2; GstBus *bus; GMainLoop *loop; guint n_buffers = 0; caps1 = gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1); caps2 = gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 0, 1, 1, 1); pipeline = setup_imagefreeze (caps1, caps2, G_CALLBACK (sink_handoff_cb_0_1), &n_buffers); loop = g_main_loop_new (NULL, TRUE); fail_unless (loop != NULL); bus = gst_element_get_bus (pipeline); fail_unless (bus != NULL); gst_bus_add_watch (bus, bus_handler, loop); gst_object_unref (bus); fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING), GST_STATE_CHANGE_SUCCESS); g_main_loop_run (loop); fail_unless_equals_int (n_buffers, 1); gst_element_set_state (pipeline, GST_STATE_NULL); g_object_unref (pipeline); g_main_loop_unref (loop); gst_caps_unref (caps1); gst_caps_unref (caps2); } GST_END_TEST; static void sink_handoff_cb_25_1_0ms_400ms (GstElement * object, GstBuffer * buffer, GstPad * pad, gpointer user_data) { guint *n_buffers = (guint *) user_data; fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer), *n_buffers * 40 * GST_MSECOND); fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 40 * GST_MSECOND); fail_unless_equals_uint64 (GST_BUFFER_OFFSET (buffer), *n_buffers); fail_unless_equals_uint64 (GST_BUFFER_OFFSET_END (buffer), *n_buffers + 1); *n_buffers = *n_buffers + 1; } GST_START_TEST (test_imagefreeze_25_1_0ms_400ms) { GstElement *pipeline; GstCaps *caps1, *caps2; GstBus *bus; GMainLoop *loop; guint n_buffers = 0; caps1 = gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1); caps2 = gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1); pipeline = setup_imagefreeze (caps1, caps2, G_CALLBACK (sink_handoff_cb_25_1_0ms_400ms), &n_buffers); loop = g_main_loop_new (NULL, TRUE); fail_unless (loop != NULL); bus = gst_element_get_bus (pipeline); fail_unless (bus != NULL); gst_bus_add_watch (bus, bus_handler, loop); gst_object_unref (bus); fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING), GST_STATE_CHANGE_SUCCESS); fail_unless (gst_element_seek (pipeline, 1.0, GST_FORMAT_TIME, GST_SEEK_FLAG_FLUSH, GST_SEEK_TYPE_SET, 0, GST_SEEK_TYPE_SET, 400 * GST_MSECOND)); fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING), GST_STATE_CHANGE_SUCCESS); g_main_loop_run (loop); fail_unless_equals_int (n_buffers, 10); gst_element_set_state (pipeline, GST_STATE_NULL); g_object_unref (pipeline); g_main_loop_unref (loop); gst_caps_unref (caps1); gst_caps_unref (caps2); } GST_END_TEST; static void sink_handoff_cb_25_1_200ms_400ms (GstElement * object, GstBuffer * buffer, GstPad * pad, gpointer user_data) { guint *n_buffers = (guint *) user_data; fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer), 200 * GST_MSECOND + *n_buffers * 40 * GST_MSECOND); fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 40 * GST_MSECOND); fail_unless_equals_uint64 (GST_BUFFER_OFFSET (buffer), 5 + *n_buffers); fail_unless_equals_uint64 (GST_BUFFER_OFFSET_END (buffer), 5 + *n_buffers + 1); *n_buffers = *n_buffers + 1; } GST_START_TEST (test_imagefreeze_25_1_200ms_400ms) { GstElement *pipeline; GstCaps *caps1, *caps2; GstBus *bus; GMainLoop *loop; guint n_buffers = 0; caps1 = gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1); caps2 = gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1); pipeline = setup_imagefreeze (caps1, caps2, G_CALLBACK (sink_handoff_cb_25_1_200ms_400ms), &n_buffers); loop = g_main_loop_new (NULL, TRUE); fail_unless (loop != NULL); bus = gst_element_get_bus (pipeline); fail_unless (bus != NULL); gst_bus_add_watch (bus, bus_handler, loop); gst_object_unref (bus); fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING), GST_STATE_CHANGE_SUCCESS); fail_unless (gst_element_seek (pipeline, 1.0, GST_FORMAT_TIME, GST_SEEK_FLAG_FLUSH, GST_SEEK_TYPE_SET, 200 * GST_MSECOND, GST_SEEK_TYPE_SET, 400 * GST_MSECOND)); fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING), GST_STATE_CHANGE_SUCCESS); g_main_loop_run (loop); fail_unless_equals_int (n_buffers, 5); gst_element_set_state (pipeline, GST_STATE_NULL); g_object_unref (pipeline); g_main_loop_unref (loop); gst_caps_unref (caps1); gst_caps_unref (caps2); } GST_END_TEST; static void sink_handoff_cb_25_1_400ms_0ms (GstElement * object, GstBuffer * buffer, GstPad * pad, gpointer user_data) { guint *n_buffers = (guint *) user_data; fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer), 400 * GST_MSECOND - (*n_buffers + 1) * 40 * GST_MSECOND); fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 40 * GST_MSECOND); fail_unless_equals_uint64 (GST_BUFFER_OFFSET (buffer), 10 - (*n_buffers + 1)); fail_unless_equals_uint64 (GST_BUFFER_OFFSET_END (buffer), 10 - (*n_buffers + 1) + 1); *n_buffers = *n_buffers + 1; } GST_START_TEST (test_imagefreeze_25_1_400ms_0ms) { GstElement *pipeline; GstCaps *caps1, *caps2; GstBus *bus; GMainLoop *loop; guint n_buffers = 0; caps1 = gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1); caps2 = gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1); pipeline = setup_imagefreeze (caps1, caps2, G_CALLBACK (sink_handoff_cb_25_1_400ms_0ms), &n_buffers); loop = g_main_loop_new (NULL, TRUE); fail_unless (loop != NULL); bus = gst_element_get_bus (pipeline); fail_unless (bus != NULL); gst_bus_add_watch (bus, bus_handler, loop); gst_object_unref (bus); fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING), GST_STATE_CHANGE_SUCCESS); fail_unless (gst_element_seek (pipeline, -1.0, GST_FORMAT_TIME, GST_SEEK_FLAG_FLUSH, GST_SEEK_TYPE_SET, 0, GST_SEEK_TYPE_SET, 400 * GST_MSECOND)); fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING), GST_STATE_CHANGE_SUCCESS); g_main_loop_run (loop); fail_unless_equals_int (n_buffers, 10); gst_element_set_state (pipeline, GST_STATE_NULL); g_object_unref (pipeline); g_main_loop_unref (loop); gst_caps_unref (caps1); gst_caps_unref (caps2); } GST_END_TEST; static void sink_handoff_cb_25_1_220ms_380ms (GstElement * object, GstBuffer * buffer, GstPad * pad, gpointer user_data) { guint *n_buffers = (guint *) user_data; if (*n_buffers == 0) { fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer), 220 * GST_MSECOND); fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 20 * GST_MSECOND); } else if (*n_buffers == 4) { fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer), 360 * GST_MSECOND); fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 20 * GST_MSECOND); } else { fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer), 200 * GST_MSECOND + *n_buffers * 40 * GST_MSECOND); fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 40 * GST_MSECOND); } fail_unless_equals_uint64 (GST_BUFFER_OFFSET (buffer), 5 + *n_buffers); fail_unless_equals_uint64 (GST_BUFFER_OFFSET_END (buffer), 5 + *n_buffers + 1); *n_buffers = *n_buffers + 1; } GST_START_TEST (test_imagefreeze_25_1_220ms_380ms) { GstElement *pipeline; GstCaps *caps1, *caps2; GstBus *bus; GMainLoop *loop; guint n_buffers = 0; caps1 = gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1); caps2 = gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1); pipeline = setup_imagefreeze (caps1, caps2, G_CALLBACK (sink_handoff_cb_25_1_220ms_380ms), &n_buffers); loop = g_main_loop_new (NULL, TRUE); fail_unless (loop != NULL); bus = gst_element_get_bus (pipeline); fail_unless (bus != NULL); gst_bus_add_watch (bus, bus_handler, loop); gst_object_unref (bus); fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING), GST_STATE_CHANGE_SUCCESS); fail_unless (gst_element_seek (pipeline, 1.0, GST_FORMAT_TIME, GST_SEEK_FLAG_FLUSH, GST_SEEK_TYPE_SET, 220 * GST_MSECOND, GST_SEEK_TYPE_SET, 380 * GST_MSECOND)); fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING), GST_STATE_CHANGE_SUCCESS); g_main_loop_run (loop); fail_unless_equals_int (n_buffers, 5); gst_element_set_state (pipeline, GST_STATE_NULL); g_object_unref (pipeline); g_main_loop_unref (loop); gst_caps_unref (caps1); gst_caps_unref (caps2); } GST_END_TEST; static Suite * imagefreeze_suite (void) { Suite *s = suite_create ("imagefreeze"); TCase *tc_chain = tcase_create ("linear"); /* time out after 120s, not the default 3 */ tcase_set_timeout (tc_chain, 120); suite_add_tcase (s, tc_chain); tcase_add_test (tc_chain, test_imagefreeze_0_1); tcase_add_test (tc_chain, test_imagefreeze_25_1_0ms_400ms); tcase_add_test (tc_chain, test_imagefreeze_25_1_200ms_400ms); tcase_add_test (tc_chain, test_imagefreeze_25_1_400ms_0ms); tcase_add_test (tc_chain, test_imagefreeze_25_1_220ms_380ms); return s; } GST_CHECK_MAIN (imagefreeze);