/* GStreamer * Copyright (C) 2019 Seungha Yang <seungha.yang@navercorp.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> static gboolean bus_cb (GstBus * bus, GstMessage * message, gpointer data) { GMainLoop *loop = (GMainLoop *) data; switch (GST_MESSAGE_TYPE (message)) { case GST_MESSAGE_ERROR:{ GError *err = NULL; gchar *debug = NULL; gst_message_parse_error (message, &err, &debug); GST_ERROR ("Error: %s : %s", err->message, debug); g_error_free (err); g_free (debug); fail_if (TRUE, "failed"); g_main_loop_quit (loop); } break; case GST_MESSAGE_EOS: g_main_loop_quit (loop); break; default: break; } return TRUE; } static void run_convert_pipelne (const gchar * in_format, const gchar * out_format) { GstBus *bus; GMainLoop *loop = g_main_loop_new (NULL, FALSE); gchar *pipeline_str = g_strdup_printf ("videotestsrc num-buffers=1 is-live=true ! " "video/x-raw,format=%s,framerate=3/1 ! cudaupload ! " "cudaconvert ! cudadownload ! video/x-raw,format=%s ! " "videoconvert ! autovideosink", in_format, out_format); GstElement *pipeline; pipeline = gst_parse_launch (pipeline_str, NULL); fail_unless (pipeline != NULL); g_free (pipeline_str); bus = gst_pipeline_get_bus (GST_PIPELINE (pipeline)); gst_bus_add_watch (bus, (GstBusFunc) bus_cb, loop); gst_element_set_state (pipeline, GST_STATE_PLAYING); g_main_loop_run (loop); gst_element_set_state (pipeline, GST_STATE_NULL); gst_bus_remove_watch (bus); gst_object_unref (bus); gst_object_unref (pipeline); g_main_loop_unref (loop); } GST_START_TEST (test_convert_yuv_yuv) { const gchar *format_list[] = { "I420", "YV12", "NV12", "NV21", "P010_10LE", "I420_10LE", "Y444", "Y444_16LE", }; gint i, j; for (i = 0; i < G_N_ELEMENTS (format_list); i++) { for (j = 0; j < G_N_ELEMENTS (format_list); j++) { if (i == j) continue; GST_DEBUG ("run conversion %s to %s", format_list[i], format_list[j]); run_convert_pipelne (format_list[i], format_list[j]); } } } GST_END_TEST; GST_START_TEST (test_convert_yuv_rgb) { const gchar *in_format_list[] = { "I420", "YV12", "NV12", "NV21", "P010_10LE", "I420_10LE", "Y444", "Y444_16LE", }; const gchar *out_format_list[] = { "BGRA", "RGBA", "RGBx", "BGRx", "ARGB", "ABGR", "RGB", "BGR", "BGR10A2_LE", "RGB10A2_LE", }; gint i, j; for (i = 0; i < G_N_ELEMENTS (in_format_list); i++) { for (j = 0; j < G_N_ELEMENTS (out_format_list); j++) { GST_DEBUG ("run conversion %s to %s", in_format_list[i], out_format_list[j]); run_convert_pipelne (in_format_list[i], out_format_list[j]); } } } GST_END_TEST; GST_START_TEST (test_convert_rgb_yuv) { const gchar *in_format_list[] = { "BGRA", "RGBA", "RGBx", "BGRx", "ARGB", "ABGR", "RGB", "BGR", "BGR10A2_LE", "RGB10A2_LE", }; const gchar *out_format_list[] = { "I420", "YV12", "NV12", "NV21", "P010_10LE", "I420_10LE", "Y444", "Y444_16LE", }; gint i, j; for (i = 0; i < G_N_ELEMENTS (in_format_list); i++) { for (j = 0; j < G_N_ELEMENTS (out_format_list); j++) { GST_DEBUG ("run conversion %s to %s", in_format_list[i], out_format_list[j]); run_convert_pipelne (in_format_list[i], out_format_list[j]); } } } GST_END_TEST; GST_START_TEST (test_convert_rgb_rgb) { const gchar *format_list[] = { "BGRA", "RGBA", "RGBx", "BGRx", "ARGB", "ABGR", "RGB", "BGR", "BGR10A2_LE", "RGB10A2_LE", }; gint i, j; for (i = 0; i < G_N_ELEMENTS (format_list); i++) { for (j = 0; j < G_N_ELEMENTS (format_list); j++) { if (i == j) continue; GST_DEBUG ("run conversion %s to %s", format_list[i], format_list[j]); run_convert_pipelne (format_list[i], format_list[j]); } } } GST_END_TEST; static gboolean check_cuda_convert_available (void) { gboolean ret = TRUE; GstElement *upload; upload = gst_element_factory_make ("cudaconvert", NULL); if (!upload) { GST_WARNING ("cudaconvert is not available, possibly driver load failure"); return FALSE; } gst_object_unref (upload); return ret; } static Suite * cudaconvert_suite (void) { Suite *s; TCase *tc_chain; /* HACK: cuda device init/deinit with fork seems to problematic */ g_setenv ("CK_FORK", "no", TRUE); s = suite_create ("cudaconvert"); tc_chain = tcase_create ("general"); suite_add_tcase (s, tc_chain); if (!check_cuda_convert_available ()) { GST_DEBUG ("Skip cudaconvert test since cannot open device"); goto end; } tcase_add_test (tc_chain, test_convert_yuv_yuv); tcase_add_test (tc_chain, test_convert_yuv_rgb); tcase_add_test (tc_chain, test_convert_rgb_yuv); tcase_add_test (tc_chain, test_convert_rgb_rgb); end: return s; } GST_CHECK_MAIN (cudaconvert);