From fb9248b8ea57dd9db4ccc02f86e2af372c7d105e Mon Sep 17 00:00:00 2001 From: Thijs Vermeir Date: Wed, 15 Oct 2014 16:52:09 +0200 Subject: [PATCH] tests/x265enc: add simple unit test --- tests/check/Makefile.am | 7 ++ tests/check/elements/x265enc.c | 165 +++++++++++++++++++++++++++++++++ 2 files changed, 172 insertions(+) create mode 100644 tests/check/elements/x265enc.c diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am index f6990549f3..cad8148efd 100644 --- a/tests/check/Makefile.am +++ b/tests/check/Makefile.am @@ -106,6 +106,12 @@ else check_schro= endif +if USE_X265 +check_x265enc=elements/x265enc +else +check_x265enc= +endif + if USE_TIMIDITY check_timidity=elements/timidity else @@ -260,6 +266,7 @@ check_PROGRAMS = \ $(check_uvch264) \ libs/vc1parser \ $(check_schro) \ + $(check_x265enc) \ elements/viewfinderbin \ $(check_zbar) \ $(check_orc) \ diff --git a/tests/check/elements/x265enc.c b/tests/check/elements/x265enc.c new file mode 100644 index 0000000000..38186f1c32 --- /dev/null +++ b/tests/check/elements/x265enc.c @@ -0,0 +1,165 @@ +/* GStreamer + * + * Copyright (c) 2010 Sebastian Dröge + * Copyright (c) 2010 David Schleef + * Copyright (c) 2014 Thijs Vermeir + * + * 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 + +static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink", + GST_PAD_SINK, + GST_PAD_ALWAYS, + GST_STATIC_CAPS ("video/x-h265, " + "width = (int) [1, MAX], " + "height = (int) [1, MAX], " "framerate = (fraction) [0, MAX]")); + +static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_STATIC_CAPS ("video/x-raw, " + "format = (string) I420, " + "width = (int) [1, MAX], " + "height = (int) [1, MAX], " "framerate = (fraction) [0, MAX]")); + +static GstPad *sinkpad, *srcpad; + +static GstElement * +setup_x265enc (const gchar * src_caps_str) +{ + GstElement *x265enc; + GstCaps *srccaps = NULL; + GstBus *bus; + + if (src_caps_str) { + srccaps = gst_caps_from_string (src_caps_str); + fail_unless (srccaps != NULL); + } + + x265enc = gst_check_setup_element ("x265enc"); + fail_unless (x265enc != NULL); + srcpad = gst_check_setup_src_pad (x265enc, &srctemplate); + sinkpad = gst_check_setup_sink_pad (x265enc, &sinktemplate); + gst_pad_set_active (srcpad, TRUE); + gst_pad_set_active (sinkpad, TRUE); + + gst_check_setup_events (srcpad, x265enc, srccaps, GST_FORMAT_TIME); + + bus = gst_bus_new (); + gst_element_set_bus (x265enc, bus); + + fail_unless (gst_element_set_state (x265enc, + GST_STATE_PLAYING) != GST_STATE_CHANGE_FAILURE, + "could not set to playing"); + + if (srccaps) + gst_caps_unref (srccaps); + + buffers = NULL; + return x265enc; +} + +static void +cleanup_x265enc (GstElement * x265enc) +{ + GstBus *bus; + + /* Free parsed buffers */ + gst_check_drop_buffers (); + + bus = GST_ELEMENT_BUS (x265enc); + gst_bus_set_flushing (bus, TRUE); + gst_object_unref (bus); + + gst_pad_set_active (srcpad, FALSE); + gst_pad_set_active (sinkpad, FALSE); + gst_check_teardown_src_pad (x265enc); + gst_check_teardown_sink_pad (x265enc); + gst_check_teardown_element (x265enc); +} + +GST_START_TEST (test_encode_simple) +{ + GstElement *x265enc; + GstBuffer *buffer; + gint i; + GList *l; + GstCaps *outcaps, *sinkcaps; + GstSegment seg; + + x265enc = + setup_x265enc + ("video/x-raw,format=(string)I420,width=(int)320,height=(int)240,framerate=(fraction)25/1"); + + gst_segment_init (&seg, GST_FORMAT_TIME); + seg.stop = gst_util_uint64_scale (10, GST_SECOND, 25); + + fail_unless (gst_pad_push_event (srcpad, gst_event_new_segment (&seg))); + + buffer = gst_buffer_new_allocate (NULL, 320 * 240 + 2 * 160 * 120, NULL); + gst_buffer_memset (buffer, 0, 0, -1); + + for (i = 0; i < 10; i++) { + GST_BUFFER_TIMESTAMP (buffer) = gst_util_uint64_scale (i, GST_SECOND, 25); + GST_BUFFER_DURATION (buffer) = gst_util_uint64_scale (1, GST_SECOND, 25); + fail_unless (gst_pad_push (srcpad, gst_buffer_ref (buffer)) == GST_FLOW_OK); + } + + gst_buffer_unref (buffer); + + fail_unless (gst_pad_push_event (srcpad, gst_event_new_eos ())); + + /* All buffers must be there now */ + fail_unless_equals_int (g_list_length (buffers), 10); + + outcaps = + gst_caps_from_string + ("video/x-h265,width=(int)320,height=(int)240,framerate=(fraction)25/1"); + + for (l = buffers, i = 0; l; l = l->next, i++) { + buffer = l->data; + + fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), + gst_util_uint64_scale (1, GST_SECOND, 25)); + + sinkcaps = gst_pad_get_current_caps (sinkpad); + fail_unless (gst_caps_can_intersect (sinkcaps, outcaps)); + gst_caps_unref (sinkcaps); + } + + gst_caps_unref (outcaps); + + cleanup_x265enc (x265enc); +} + +GST_END_TEST; + +static Suite * +x265enc_suite (void) +{ + Suite *s = suite_create ("x265enc"); + TCase *tc_chain = tcase_create ("general"); + + suite_add_tcase (s, tc_chain); + + tcase_add_test (tc_chain, test_encode_simple); + + return s; +} + +GST_CHECK_MAIN (x265enc);