/* 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);