/* GStreamer * * unit test for x264enc * * Copyright (C) <2008> Mark Nauwelaerts * * 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 #include /* For ease of programming we use globals to keep refs for our floating * src and sink pads we create; otherwise we always have to do get_pad, * get_peer, and then remove references in every test function */ static GstPad *mysrcpad, *mysinkpad; #define VIDEO_CAPS_STRING "video/x-raw, " \ "format = (string) { I420, Y42B, Y444 }, " \ "width = (int) 384, " \ "height = (int) 288, " \ "framerate = (fraction) 25/1" #define MPEG_CAPS_STRING "video/x-h264, " \ "width = (int) 384, " \ "height = (int) 288, " \ "framerate = (fraction) 25/1" static GstElement * setup_x264enc (const gchar * profile, const gchar * stream_format, const gchar * input_format) { GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink", GST_PAD_SINK, GST_PAD_ALWAYS, GST_STATIC_CAPS (MPEG_CAPS_STRING)); GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS, GST_STATIC_CAPS (VIDEO_CAPS_STRING)); GstElement *x264enc; gchar *caps_str; GstCaps *caps; GST_DEBUG ("setup_x264enc"); caps_str = g_strdup_printf ("%s, profile = (string) %s, " "stream-format = (string) %s", MPEG_CAPS_STRING, profile, stream_format); sinktemplate.static_caps.string = caps_str; x264enc = gst_check_setup_element ("x264enc"); mysrcpad = gst_check_setup_src_pad (x264enc, &srctemplate); mysinkpad = gst_check_setup_sink_pad (x264enc, &sinktemplate); gst_pad_set_active (mysrcpad, TRUE); gst_pad_set_active (mysinkpad, TRUE); caps = gst_caps_from_string (VIDEO_CAPS_STRING); gst_caps_set_simple (caps, "format", G_TYPE_STRING, input_format, NULL); gst_check_setup_events (mysrcpad, x264enc, caps, GST_FORMAT_TIME); gst_caps_unref (caps); g_free (caps_str); return x264enc; } static void cleanup_x264enc (GstElement * x264enc) { GST_DEBUG ("cleanup_x264enc"); gst_element_set_state (x264enc, GST_STATE_NULL); gst_pad_set_active (mysrcpad, FALSE); gst_pad_set_active (mysinkpad, FALSE); gst_check_teardown_src_pad (x264enc); gst_check_teardown_sink_pad (x264enc); gst_check_teardown_element (x264enc); } static void check_caps (GstCaps * caps, const gchar * profile, gint profile_id) { GstStructure *s; const GValue *sf, *avcc, *pf; const gchar *stream_format; const gchar *caps_profile; fail_unless (caps != NULL); GST_INFO ("caps %" GST_PTR_FORMAT, caps); s = gst_caps_get_structure (caps, 0); fail_unless (s != NULL); fail_if (!gst_structure_has_name (s, "video/x-h264")); sf = gst_structure_get_value (s, "stream-format"); fail_unless (sf != NULL); fail_unless (G_VALUE_HOLDS_STRING (sf)); stream_format = g_value_get_string (sf); fail_unless (stream_format != NULL); if (strcmp (stream_format, "avc") == 0) { GstMapInfo map; GstBuffer *buf; avcc = gst_structure_get_value (s, "codec_data"); fail_unless (avcc != NULL); fail_unless (GST_VALUE_HOLDS_BUFFER (avcc)); buf = gst_value_get_buffer (avcc); fail_unless (buf != NULL); gst_buffer_map (buf, &map, GST_MAP_READ); fail_unless_equals_int (map.data[0], 1); fail_unless (map.data[1] == profile_id); gst_buffer_unmap (buf, &map); } else if (strcmp (stream_format, "byte-stream") == 0) { fail_if (gst_structure_get_value (s, "codec_data") != NULL); } else { fail_if (TRUE, "unexpected stream-format in caps: %s", stream_format); } pf = gst_structure_get_value (s, "profile"); fail_unless (pf != NULL); fail_unless (G_VALUE_HOLDS_STRING (pf)); caps_profile = g_value_get_string (pf); fail_unless (caps_profile != NULL); fail_unless (!strcmp (caps_profile, profile)); } static void test_video_profile (const gchar * profile, gint profile_id, const gchar * input_format) { GstElement *x264enc; GstBuffer *inbuffer, *outbuffer; GstCaps *caps; GstMapInfo map; guint8 *data; gsize size; int i, num_buffers; x264enc = setup_x264enc (profile, "avc", input_format); fail_unless (gst_element_set_state (x264enc, GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS, "could not set to playing"); /* corresponds to I420 buffer for the size mentioned in the caps */ if (!strcmp (input_format, "I420")) inbuffer = gst_buffer_new_and_alloc (384 * 288 * 3 / 2); else if (!strcmp (input_format, "Y42B")) inbuffer = gst_buffer_new_and_alloc (384 * 288 * 2); else if (!strcmp (input_format, "Y444")) inbuffer = gst_buffer_new_and_alloc (384 * 288 * 3); /* makes valgrind's memcheck happier */ gst_buffer_memset (inbuffer, 0, 0, -1); GST_BUFFER_TIMESTAMP (inbuffer) = 0; ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1); fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK); /* send eos to have all flushed if needed */ fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_eos ()) == TRUE); num_buffers = g_list_length (buffers); fail_unless (num_buffers == 1); /* check output caps */ { GstCaps *outcaps; outcaps = gst_pad_get_current_caps (mysinkpad); check_caps (outcaps, profile, profile_id); gst_caps_unref (outcaps); } /* clean up buffers */ for (i = 0; i < num_buffers; ++i) { outbuffer = GST_BUFFER (buffers->data); fail_if (outbuffer == NULL); switch (i) { case 0: { gint nsize, npos, j, type, next_type; GstMapInfo map; const guint8 *data; gsize size; gst_buffer_map (outbuffer, &map, GST_MAP_READ); data = map.data; size = map.size; npos = 0; j = 0; /* need SPS first */ next_type = 7; /* loop through NALs */ while (npos < size) { fail_unless (size - npos >= 4); nsize = GST_READ_UINT32_BE (data + npos); fail_unless (nsize > 0); fail_unless (npos + 4 + nsize <= size); type = data[npos + 4] & 0x1F; /* check the first NALs, disregard AU (9), SEI (6) */ if (type != 9 && type != 6) { fail_unless (type == next_type); switch (type) { case 7: /* SPS */ next_type = 8; break; case 8: /* PPS */ next_type = 5; break; default: break; } j++; } npos += nsize + 4; } gst_buffer_unmap (outbuffer, &map); /* should have reached the exact end */ fail_unless (npos == size); break; } default: break; } buffers = g_list_remove (buffers, outbuffer); ASSERT_BUFFER_REFCOUNT (outbuffer, "outbuffer", 1); gst_buffer_unref (outbuffer); outbuffer = NULL; } cleanup_x264enc (x264enc); g_list_free (buffers); buffers = NULL; } GST_START_TEST (test_video_baseline) { test_video_profile ("constrained-baseline", 0x42, "I420"); } GST_END_TEST; GST_START_TEST (test_video_main) { test_video_profile ("main", 0x4d, "I420"); } GST_END_TEST; GST_START_TEST (test_video_high) { test_video_profile ("high", 0x64, "I420"); } GST_END_TEST; GST_START_TEST (test_video_high422) { test_video_profile ("high-4:2:2", 0x7A, "Y42B"); } GST_END_TEST; GST_START_TEST (test_video_high444) { test_video_profile ("high-4:4:4", 0xF4, "Y444"); } GST_END_TEST; Suite * x264enc_suite (void) { Suite *s = suite_create ("x264enc"); TCase *tc_chain = tcase_create ("general"); suite_add_tcase (s, tc_chain); tcase_add_test (tc_chain, test_video_baseline); tcase_add_test (tc_chain, test_video_main); tcase_add_test (tc_chain, test_video_high); tcase_add_test (tc_chain, test_video_high422); tcase_add_test (tc_chain, test_video_high444); return s; } int main (int argc, char **argv) { int nf; Suite *s = x264enc_suite (); SRunner *sr = srunner_create (s); gst_check_init (&argc, &argv); srunner_run_all (sr, CK_NORMAL); nf = srunner_ntests_failed (sr); srunner_free (sr); return nf; }