gstreamer/tests/check/libs/nalutils.c

120 lines
3.3 KiB
C
Raw Normal View History

/* GStreamer
*
* Copyright (C) 2020 Seungha Yang <seungha@centricular.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>
#include <gst/codecparsers/nalutils.h>
#include <string.h>
GST_START_TEST (test_nal_writer_init)
{
NalWriter nw;
/* init with invalid params */
ASSERT_CRITICAL (nal_writer_init (&nw, 0, TRUE));
ASSERT_CRITICAL (nal_writer_init (&nw, 0, FALSE));
ASSERT_CRITICAL (nal_writer_init (&nw, 5, TRUE));
ASSERT_CRITICAL (nal_writer_init (&nw, 5, FALSE));
nal_writer_init (&nw, 4, FALSE);
nal_writer_reset (&nw);
}
GST_END_TEST;
GST_START_TEST (test_nal_writer_emulation_preventation)
{
NalWriter nw;
gint i, j;
static guint8 patterns[][3] = {
{0x00, 0x00, 0x00},
{0x00, 0x00, 0x01},
{0x00, 0x00, 0x02},
{0x00, 0x00, 0x03},
};
static guint8 expected_rst[][4] = {
{0x00, 0x00, 0x03, 0x00},
{0x00, 0x00, 0x03, 0x01},
{0x00, 0x00, 0x03, 0x02},
{0x00, 0x00, 0x03, 0x03},
};
/* Within the NAL unit, the following three-byte sequences shall not occur
* at any byte-aligned position:
* 0x000000
* 0x000001
* 0x000002
* Within the NAL unit, any four-byte sequence that starts with 0x000003
* other than the following sequences shall not occur
* at any byte-aligned position:
* 0x00000300
* 0x00000301
* 0x00000302
* 0x00000303
*/
for (i = 0; i < G_N_ELEMENTS (patterns); i++) {
GstMemory *mem;
GstMapInfo info;
nal_writer_init (&nw, 4, FALSE);
/* forbidden_zero_bit */
fail_unless (nal_writer_put_bits_uint8 (&nw, 0, 1));
/* nal_ref_idc, just set zero for test */
fail_unless (nal_writer_put_bits_uint8 (&nw, 0, 2));
/* nal_unit_type, unknown h264 nal type */
fail_unless (nal_writer_put_bits_uint8 (&nw, 0x1f, 5));
for (j = 0; j < 3; j++)
fail_unless (nal_writer_put_bits_uint8 (&nw, patterns[i][j], 8));
mem = nal_writer_reset_and_get_memory (&nw);
fail_unless (mem != NULL);
fail_unless (gst_memory_map (mem, &info, GST_MAP_READ));
/* start code prefix 4 + nalu header 1 + written bytes 3 +
* emulation prevention byte 1 */
assert_equals_int (info.size, (4 + 1 + 3 + 1));
fail_if (memcmp (info.data + 5, expected_rst[i], 4));
gst_memory_unmap (mem, &info);
gst_memory_unref (mem);
}
}
GST_END_TEST;
static Suite *
nalutils_suite (void)
{
Suite *s = suite_create ("H264/H264 Nal Utils");
TCase *tc_chain = tcase_create ("general");
suite_add_tcase (s, tc_chain);
tcase_add_test (tc_chain, test_nal_writer_init);
tcase_add_test (tc_chain, test_nal_writer_emulation_preventation);
return s;
}
GST_CHECK_MAIN (nalutils);