/* Gstreamer
 * Copyright (C) <2011> Intel Corporation
 * Copyright (C) <2011> Collabora Ltd.
 * Copyright (C) <2011> Thibault Saunier <thibault.saunier@collabora.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/gsth264parser.h>

static guint8 slice_dpa[] = {
  0x00, 0x00, 0x01, 0x02, 0x00, 0x02, 0x01, 0x03, 0x00,
  0x04, 0x00, 0x05, 0x00, 0x06, 0x00, 0x07, 0x00, 0x09, 0x00, 0x0a, 0x00,
  0x0b, 0x00, 0x0c, 0x00, 0x0d, 0x00, 0x0e, 0x00, 0x0f, 0x00, 0x10, 0x00,
  0x11, 0x00, 0x12, 0x00, 0x13, 0x00, 0x14, 0x00, 0x15, 0x00, 0x16, 0x00,
  0x17, 0x00, 0x18, 0x00, 0x19, 0x00, 0x1a, 0x00, 0x1b, 0x00, 0x1c, 0x00,
  0x1d, 0x00, 0x1e, 0x00, 0x1f, 0x00, 0x20, 0x00, 0x21, 0x00, 0x22, 0x00,
  0x23, 0x00, 0x24, 0x00, 0x25, 0x00, 0x26, 0x00, 0x27, 0x00, 0x28, 0x00,
  0x29, 0x00, 0x2a, 0x00, 0x2b, 0x00, 0x2c, 0x00, 0x2d, 0x00, 0x2e, 0x00,
  0x2f, 0x00, 0x30, 0x00, 0x31, 0x00, 0x32, 0x00, 0x33, 0x00, 0x34, 0x00,
  0x35, 0x00, 0x36, 0x00, 0x37, 0x00, 0x38, 0x00, 0x39, 0x00, 0x3a, 0x00,
  0x3b, 0x00, 0x3c, 0x00, 0x3d, 0x00, 0x3e, 0x00, 0x3f, 0x00, 0x40, 0x00,
  0x41, 0x00, 0x42, 0x00, 0x43, 0x00, 0x44, 0x00, 0x45, 0x00, 0x46, 0x00,
  0x47, 0x00, 0x48, 0x00, 0x49, 0x00, 0x4a, 0x00, 0x4b, 0x00, 0x4c, 0x00,
  0x4d, 0x00, 0x4e, 0x00, 0x4f, 0x00, 0x50, 0x00, 0x51, 0x00, 0x52, 0x00,
  0x53, 0x00, 0x54, 0x00, 0x55, 0x00, 0x56, 0x00, 0x57, 0x00, 0x58, 0x00,
  0x59, 0x00, 0x5a, 0x00, 0x5b, 0x00, 0x5c, 0x00, 0x5d, 0x00, 0x5e, 0x00,
  0x5f, 0x00, 0x60, 0x00, 0x61, 0x01, 0x04, 0x00, 0xc4, 0x00, 0xa6, 0x00,
  0xc5, 0x00, 0xab, 0x00, 0x82, 0x00, 0xc2, 0x00, 0xd8, 0x00, 0xc6, 0x00,
  0xe4, 0x00, 0xbe, 0x00, 0xb0, 0x00, 0xe6, 0x00, 0xb6, 0x00, 0xb7, 0x00,
  0xb4, 0x00, 0xb5, 0x00, 0x87, 0x00, 0xb2, 0x00, 0xb3, 0x00, 0xd9, 0x00,
  0x8c, 0x00, 0xe5, 0x00, 0xbf, 0x00, 0xb1, 0x00, 0xe7, 0x00, 0xbb, 0x00,
  0xa3, 0x00, 0x84, 0x00, 0x85, 0x00, 0xbd, 0x00, 0x96, 0x00, 0xe8, 0x00,
  0x86, 0x00, 0x8e, 0x00, 0x8b, 0x00, 0x9d, 0x00, 0xa9, 0x00, 0x8a, 0x01,
  0x05, 0x00, 0x83, 0x00, 0xf2, 0x00, 0xf3, 0x00, 0x8d, 0x00, 0x97, 0x00,
  0x88, 0x00, 0xde, 0x00, 0xf1, 0x00, 0x9e, 0x00, 0xaa, 0x00, 0xf5, 0x00,
  0xf4, 0x00, 0xf6, 0x00, 0xa2, 0x00, 0xad, 0x00, 0xc9, 0x00, 0xc7, 0x00,
  0xae, 0x00, 0x62, 0x00, 0x63, 0x00, 0x90, 0x00, 0x64, 0x00, 0xcb, 0x00,
  0x65, 0x00, 0xc8, 0x00, 0xca, 0x00, 0xcf, 0x00, 0xcc, 0x00, 0xcd, 0x00,
  0xce, 0x00, 0xe9, 0x00, 0x66, 0x00, 0xd3, 0x00, 0xd0, 0x00, 0xd1, 0x00,
  0xaf, 0x00, 0x67, 0x00, 0x91, 0x00, 0xd6, 0x00, 0xd4, 0x00, 0xd5, 0x00,
  0x68, 0x00, 0xeb, 0x00, 0xed, 0x00, 0x89, 0x00, 0x6a, 0x00, 0x69, 0x00,
  0x6b, 0x00, 0x6d, 0x00, 0x6c, 0x00, 0x6e, 0x00, 0xa0, 0x00, 0x6f, 0x00,
  0x71, 0x00, 0x70, 0x00, 0x72, 0x00, 0x73, 0x00, 0x75, 0x00, 0x74, 0x00,
  0x76, 0x00, 0x77, 0x00, 0xea, 0x00, 0x78, 0x00, 0x7a, 0x00, 0x79, 0x00,
  0x7b, 0x00, 0x7d, 0x00, 0x7c, 0x00, 0xa1, 0x00, 0x7f, 0x00, 0x7e, 0x00,
  0x80, 0x00, 0x81, 0x00, 0xec, 0x00, 0xee, 0x00, 0xba, 0x01, 0x06, 0x00,
  0xef, 0x00, 0xe1, 0x00, 0xe0, 0x00, 0xdc, 0x01, 0x07, 0x01, 0x08, 0x01,
  0x09, 0x01, 0x0a, 0x01, 0x0b, 0x01, 0x0c, 0x00, 0xdb, 0x00, 0xe2, 0x01,
  0x0d, 0x01, 0x0e, 0x01, 0x0f, 0x01, 0x10, 0x01, 0x11, 0x01, 0x12, 0x00,
  0xdf, 0x01, 0x13, 0x01, 0x14, 0x01, 0x15, 0x01, 0x16, 0x01, 0x17, 0x00,
  0xfd, 0x00, 0xff, 0x01, 0x18, 0x01, 0x19, 0x01, 0x1a, 0x01, 0x1b, 0x01,
  0x1c, 0x01, 0x1d, 0x01, 0x1e, 0x01, 0x1f, 0x01, 0x20, 0x01, 0x21, 0x01,
  0x22, 0x01, 0x23, 0x01, 0x24, 0x01, 0x25, 0x01, 0x26, 0x00, 0xfe, 0x01,
  0x00, 0x01, 0x27, 0x01, 0x28, 0x01, 0x29, 0x01, 0x2a, 0x01, 0x2b, 0x01,
  0x2c, 0x01, 0x2d, 0x01, 0x2e, 0x01, 0x2f, 0x01, 0x30, 0x01, 0x31, 0x00,
  0xe3, 0x00, 0xd7, 0x01, 0x32, 0x00, 0xf8, 0x00, 0xf9, 0x01, 0x33, 0x01,
  0x34, 0x01, 0x35, 0x01, 0x36, 0x01, 0x37, 0x01, 0x38, 0x01, 0x39, 0x01,
  0x3a, 0x01, 0x3b, 0x01, 0x3c, 0x01, 0x3d, 0x01, 0x3e, 0x01, 0x3f, 0x01,
  0x40, 0x01, 0x41, 0x01, 0x42, 0x01, 0x43, 0x01, 0x44, 0x01, 0x45, 0x01,
  0x46, 0x01, 0x47, 0x01, 0x48, 0x01, 0x49, 0x01, 0x4a, 0x01, 0x4b, 0x01,
  0x4c, 0x00, 0x08, 0x05, 0x2e, 0x6e, 0x75, 0x6c, 0x6c, 0x0c, 0x76, 0x69,
  0x73, 0x69, 0x62, 0x6c, 0x65, 0x73, 0x70, 0x61, 0x63, 0x65, 0x04, 0x45,
  0x75, 0x72, 0x6f, 0x06, 0x6d, 0x61, 0x63, 0x72, 0x6f, 0x6e, 0x0a, 0x62,
  0x75, 0x6c, 0x6c, 0x65, 0x74, 0x6d, 0x61, 0x74, 0x68, 0x06, 0x53, 0x61,
  0x63, 0x75, 0x74, 0x65, 0x06, 0x54, 0x63, 0x61, 0x72, 0x6f, 0x6e, 0x06,
  0x5a, 0x61, 0x63, 0x75, 0x74, 0x65, 0x06, 0x73, 0x61, 0x63, 0x75, 0x74,
  0x65, 0x06, 0x74, 0x63, 0x61, 0x72, 0x6f, 0x6e, 0x06, 0x7a, 0x61, 0x63,
  0x75, 0x74, 0x65, 0x07, 0x41, 0x6f, 0x67, 0x6f, 0x6e, 0x65, 0x6b, 0x07,
  0x61, 0x6f, 0x67, 0x6f, 0x6e, 0x65, 0x6b, 0x0c, 0x73, 0x63, 0x6f, 0x6d,
  0x6d, 0x61, 0x61, 0x63, 0x63, 0x65, 0x6e, 0x74, 0x0c, 0x53, 0x63, 0x6f,
  0x6d, 0x6d, 0x61, 0x61, 0x63, 0x63, 0x65, 0x6e, 0x74, 0x0a, 0x5a, 0x64,
  0x6f, 0x74, 0x61, 0x63, 0x63, 0x65, 0x6e, 0x74, 0x06, 0x4c, 0x63, 0x61,
  0x72, 0x6f, 0x6e, 0x06, 0x6c, 0x63, 0x61, 0x72, 0x6f, 0x6e, 0x0a, 0x7a,
  0x64, 0x6f, 0x74, 0x61, 0x63, 0x63, 0x65, 0x6e, 0x74, 0x06, 0x52, 0x61,
  0x63, 0x75, 0x74, 0x65, 0x06, 0x41, 0x62, 0x72, 0x65, 0x76, 0x65, 0x06,
  0x4c, 0x61, 0x63, 0x75, 0x74, 0x65, 0x07, 0x45, 0x6f, 0x67, 0x6f, 0x6e,
  0x65, 0x6b, 0x06, 0x45, 0x63, 0x61, 0x72, 0x6f, 0x6e, 0x06, 0x44, 0x63,
  0x61, 0x72, 0x6f, 0x6e, 0x07, 0x44, 0x6d, 0x61, 0x63, 0x72, 0x6f, 0x6e,
  0x06, 0x4e, 0x61, 0x63, 0x75, 0x74, 0x65, 0x06, 0x4e, 0x63, 0x61, 0x72,
  0x6f, 0x6e, 0x0d, 0x4f, 0x68, 0x75, 0x6e, 0x67, 0x61, 0x72, 0x75, 0x6d,
  0x6c, 0x61, 0x75, 0x74, 0x06, 0x52, 0x63, 0x61, 0x72, 0x6f, 0x6e, 0x05,
  0x55, 0x72, 0x69, 0x6e, 0x67, 0x09, 0x6e, 0x75, 0x6e, 0x67, 0x61, 0x64,
  0x65, 0x73, 0x68, 0x0d, 0x55, 0x68, 0x75, 0x6e, 0x67, 0x61, 0x72, 0x75,
  0x6d, 0x6c, 0x61, 0x75, 0x74, 0x0c, 0x54, 0x63, 0x6f, 0x6d, 0x6d, 0x61,
  0x61, 0x63, 0x63, 0x65, 0x6e, 0x74, 0x06, 0x72, 0x61, 0x63, 0x75, 0x74,
  0x65, 0x06, 0x61, 0x62, 0x72, 0x65, 0x76, 0x65, 0x06, 0x6c, 0x61, 0x63,
  0x75, 0x74, 0x65, 0x07, 0x65, 0x6f, 0x67, 0x6f, 0x6e, 0x65, 0x6b, 0x06,
  0x65, 0x63, 0x61, 0x72, 0x6f, 0x6e, 0x06, 0x64, 0x63, 0x61, 0x72, 0x6f,
  0x6e, 0x07, 0x64, 0x6d, 0x61, 0x63, 0x72, 0x6f, 0x6e, 0x06, 0x6e, 0x61,
  0x63, 0x75, 0x74, 0x65, 0x06, 0x6e, 0x63, 0x61, 0x72, 0x6f, 0x6e, 0x0d,
  0x6f, 0x68, 0x75, 0x6e, 0x67, 0x61, 0x72, 0x75, 0x6d, 0x6c, 0x61, 0x75,
  0x74, 0x06, 0x72, 0x63, 0x61, 0x72, 0x6f, 0x6e, 0x05, 0x75, 0x72, 0x69,
  0x6e, 0x67, 0x0d, 0x75, 0x68, 0x75, 0x6e, 0x67, 0x61, 0x72, 0x75, 0x6d,
  0x6c, 0x61, 0x75, 0x74, 0x0c, 0x74, 0x63, 0x6f, 0x6d, 0x6d, 0x61, 0x61,
  0x63, 0x63, 0x65, 0x6e, 0x74, 0x0a, 0x49, 0x64, 0x6f, 0x74, 0x61, 0x63,
  0x63, 0x65, 0x6e, 0x74, 0x0c, 0x52, 0x63, 0x6f, 0x6d, 0x6d, 0x61, 0x61,
  0x63, 0x63, 0x65, 0x6e, 0x74, 0x0c, 0x72, 0x63, 0x6f, 0x6d, 0x6d, 0x61,
  0x61, 0x63, 0x63, 0x65, 0x6e, 0x74, 0x07, 0x49, 0x6f, 0x67, 0x6f, 0x6e,
  0x65, 0x6b, 0x07, 0x41, 0x6d, 0x61, 0x63, 0x72, 0x6f, 0x6e, 0x07, 0x45,
  0x6d, 0x61, 0x63, 0x72, 0x6f, 0x6e, 0x0a, 0x45, 0x64, 0x6f, 0x74, 0x61,
  0x63, 0x63, 0x65, 0x6e, 0x74, 0x0c, 0x47, 0x63, 0x6f, 0x6d, 0x6d, 0x61,
  0x61, 0x63, 0x63, 0x65, 0x6e, 0x74, 0x0c, 0x4b, 0x63, 0x6f, 0x6d, 0x6d,
  0x61, 0x61, 0x63, 0x63, 0x65, 0x6e, 0x74, 0x07, 0x49, 0x6d, 0x61, 0x63,
  0x72, 0x6f, 0x6e, 0x0c, 0x4c, 0x63, 0x6f, 0x6d, 0x6d, 0x61, 0x61, 0x63,
  0x63, 0x65, 0x6e, 0x74, 0x0c, 0x4e, 0x63, 0x6f, 0x6d, 0x6d, 0x61, 0x61,
  0x63, 0x63, 0x65, 0x6e, 0x74, 0x07, 0x4f, 0x6d, 0x61, 0x63, 0x72, 0x6f,
  0x6e, 0x07, 0x55, 0x6f, 0x67, 0x6f, 0x6e, 0x65, 0x6b, 0x07, 0x55, 0x6d,
  0x61, 0x63, 0x72, 0x6f, 0x6e, 0x07, 0x69, 0x6f, 0x67, 0x6f, 0x6e, 0x65,
  0x6b, 0x07, 0x61, 0x6d, 0x61, 0x63, 0x72, 0x6f, 0x6e, 0x07, 0x65, 0x6d,
  0x61, 0x63, 0x72, 0x6f, 0x6e, 0x0a, 0x65, 0x64, 0x6f, 0x74, 0x61, 0x63,
  0x63, 0x65, 0x6e, 0x74, 0x0c, 0x67, 0x63, 0x6f, 0x6d, 0x6d, 0x61, 0x61,
  0x63, 0x63, 0x65, 0x6e, 0x74, 0x0c, 0x6b, 0x63, 0x6f, 0x6d, 0x6d, 0x61,
  0x61, 0x63, 0x63, 0x65, 0x6e, 0x74, 0x07, 0x69, 0x6d, 0x61, 0x63, 0x72,
  0x6f, 0x6e, 0x0c, 0x6c, 0x63, 0x6f, 0x6d, 0x6d, 0x61, 0x61, 0x63, 0x63,
  0x65, 0x6e, 0x74, 0x0c, 0x6e, 0x63, 0x6f, 0x6d, 0x6d, 0x61, 0x61, 0x63,
  0x63, 0x65, 0x6e, 0x74, 0x07, 0x6f, 0x6d, 0x61, 0x63, 0x72, 0x6f, 0x6e,
  0x07, 0x75, 0x6f, 0x67, 0x6f, 0x6e, 0x65, 0x6b, 0x07, 0x75, 0x6d, 0x61,
  0x63, 0x72, 0x6f, 0x6e, 0x00, 0x00, 0x00, 0x00, 0x01, 0x02
};

/* IDR slice, SEQ_END, IDR slice, STREAM_END */
static guint8 slice_eoseq_slice[] = {
  0x00, 0x00, 0x00, 0x01, 0x65, 0x88, 0x84, 0x00,
  0x10, 0xff, 0xfe, 0xf6, 0xf0, 0xfe, 0x05, 0x36,
  0x56, 0x04, 0x50, 0x96, 0x7b, 0x3f, 0x53, 0xe1,
  0x00, 0x00, 0x00, 0x01, 0x0a,
  0x00, 0x00, 0x00, 0x01, 0x65, 0x88, 0x84, 0x00,
  0x10, 0xff, 0xfe, 0xf6, 0xf0, 0xfe, 0x05, 0x36,
  0x56, 0x04, 0x50, 0x96, 0x7b, 0x3f, 0x53, 0xe1,
  0x00, 0x00, 0x00, 0x01, 0x0b
};

GST_START_TEST (test_h264_parse_slice_dpa)
{
  GstH264ParserResult res;
  GstH264NalUnit nalu;

  GstH264NalParser *parser = gst_h264_nal_parser_new ();

  res = gst_h264_parser_identify_nalu (parser, slice_dpa, 0,
      sizeof (slice_dpa), &nalu);

  assert_equals_int (res, GST_H264_PARSER_OK);
  assert_equals_int (nalu.type, GST_H264_NAL_SLICE_DPA);

  gst_h264_nal_parser_free (parser);
}

GST_END_TEST;

GST_START_TEST (test_h264_parse_slice_eoseq_slice)
{
  GstH264ParserResult res;
  GstH264NalUnit nalu;
  GstH264NalParser *const parser = gst_h264_nal_parser_new ();
  const guint8 *buf = slice_eoseq_slice;
  guint n, buf_size = sizeof (slice_eoseq_slice);

  res = gst_h264_parser_identify_nalu (parser, buf, 0, buf_size, &nalu);

  assert_equals_int (res, GST_H264_PARSER_OK);
  assert_equals_int (nalu.type, GST_H264_NAL_SLICE_IDR);
  assert_equals_int (nalu.size, 20);

  n = nalu.offset + nalu.size;
  buf += n;
  buf_size -= n;

  res = gst_h264_parser_identify_nalu (parser, buf, 0, buf_size, &nalu);

  assert_equals_int (res, GST_H264_PARSER_OK);
  assert_equals_int (nalu.type, GST_H264_NAL_SEQ_END);
  assert_equals_int (nalu.size, 1);

  n = nalu.offset + nalu.size;
  buf += n;
  buf_size -= n;

  res = gst_h264_parser_identify_nalu (parser, buf, 0, buf_size, &nalu);

  assert_equals_int (res, GST_H264_PARSER_OK);
  assert_equals_int (nalu.type, GST_H264_NAL_SLICE_IDR);
  assert_equals_int (nalu.size, 20);

  n = nalu.offset + nalu.size;
  buf += n;
  buf_size -= n;

  res = gst_h264_parser_identify_nalu (parser, buf, 0, buf_size, &nalu);

  assert_equals_int (res, GST_H264_PARSER_OK);
  assert_equals_int (nalu.type, GST_H264_NAL_STREAM_END);
  assert_equals_int (nalu.size, 1);

  gst_h264_nal_parser_free (parser);
}

GST_END_TEST;

GST_START_TEST (test_h264_parse_slice_5bytes)
{
  GstH264ParserResult res;
  GstH264NalUnit nalu;
  GstH264NalParser *const parser = gst_h264_nal_parser_new ();
  const guint8 *buf = slice_eoseq_slice;

  res = gst_h264_parser_identify_nalu (parser, buf, 0, 5, &nalu);

  assert_equals_int (res, GST_H264_PARSER_NO_NAL_END);
  assert_equals_int (nalu.type, GST_H264_NAL_SLICE_IDR);
  assert_equals_int (nalu.size, 1);

  gst_h264_nal_parser_free (parser);
}

GST_END_TEST;

GST_START_TEST (test_h264_parse_identify_nalu_avc)
{
  GstH264ParserResult res;
  GstH264NalUnit nalu;
  GstH264NalParser *const parser = gst_h264_nal_parser_new ();
  /* Skip 3 bytes for the start code */
  const gsize nal_size = sizeof (slice_dpa) - 3;
  const gsize buf_size = 4 + nal_size;
  guint8 *buf = g_new (guint8, buf_size);

  memcpy (buf + 4, slice_dpa + 3, nal_size);

  GST_WRITE_UINT16_BE (buf + 2, nal_size);
  res = gst_h264_parser_identify_nalu_avc (parser, buf, 2, buf_size, 2, &nalu);

  assert_equals_int (res, GST_H264_PARSER_OK);
  assert_equals_int (nalu.type, GST_H264_NAL_SLICE_DPA);
  assert_equals_int (nalu.offset, 4);
  assert_equals_int (nalu.size, nal_size);

  GST_WRITE_UINT32_BE (buf, nal_size);
  res = gst_h264_parser_identify_nalu_avc (parser, buf, 0, buf_size, 4, &nalu);

  assert_equals_int (res, GST_H264_PARSER_OK);
  assert_equals_int (nalu.type, GST_H264_NAL_SLICE_DPA);
  assert_equals_int (nalu.offset, 4);
  assert_equals_int (nalu.size, nal_size);

  GST_WRITE_UINT32_BE (buf, G_MAXUINT32);
  res = gst_h264_parser_identify_nalu_avc (parser, buf, 0, buf_size, 4, &nalu);

  assert_equals_int (res, GST_H264_PARSER_BROKEN_DATA);

  GST_WRITE_UINT32_BE (buf, G_MAXUINT32 - 2);
  res = gst_h264_parser_identify_nalu_avc (parser, buf, 0, buf_size, 4, &nalu);

  assert_equals_int (res, GST_H264_PARSER_BROKEN_DATA);

  GST_WRITE_UINT32_BE (buf, G_MAXUINT32 - 3);
  res = gst_h264_parser_identify_nalu_avc (parser, buf, 0, buf_size, 4, &nalu);

  assert_equals_int (res, GST_H264_PARSER_BROKEN_DATA);

  GST_WRITE_UINT32_BE (buf, G_MAXUINT32 - 4);
  res = gst_h264_parser_identify_nalu_avc (parser, buf, 0, buf_size, 4, &nalu);

  assert_equals_int (res, GST_H264_PARSER_NO_NAL_END);

  GST_WRITE_UINT32_BE (buf, G_MAXUINT32 - 6);
  res = gst_h264_parser_identify_nalu_avc (parser, buf, 0, buf_size, 4, &nalu);

  assert_equals_int (res, GST_H264_PARSER_NO_NAL_END);

  g_free (buf);
  gst_h264_nal_parser_free (parser);
}

GST_END_TEST;

static guint8 nalu_sps_with_vui[] = {
  0x00, 0x00, 0x00, 0x01, 0x67, 0x64, 0x00, 0x28,
  0xac, 0xd9, 0x40, 0x78, 0x04, 0x4f, 0xde, 0x03,
  0xd2, 0x02, 0x02, 0x02, 0x80, 0x00, 0x01, 0xf4,
  0x80, 0x00, 0x75, 0x30, 0x4f, 0x8b, 0x16, 0xcb
};

static guint8 nalu_sei_pic_timing[] = {
  0x00, 0x00, 0x00, 0x01, 0x06, 0x01, 0x01, 0x32, 0x80
};

static guint8 nalu_chained_sei[] = {
  0x00, 0x00, 0x01, 0x06, 0x01, 0x02, 0x32, 0x80,
  0x06, 0x01, 0xc4, 0x80
};

/* Content light level information SEI message */
static guint8 h264_sei_cll[] = {
  0x00, 0x00, 0x00, 0x01, 0x06, 0x90, 0x04, 0x03, 0xe8, 0x01, 0x90, 0x80
};

/* Mastering display colour volume information SEI message */
static guint8 h264_sei_mdcv[] = {
  0x00, 0x00, 0x00, 0x01, 0x06, 0x89, 0x18, 0x84,
  0xd0, 0x3e, 0x80, 0x33, 0x90, 0x86, 0xc4, 0x1d,
  0x4c, 0x0b, 0xb8, 0x3d, 0x13, 0x40, 0x42, 0x00,
  0x98, 0x96, 0x80, 0x00, 0x00, 0x03, 0x00, 0x01,
  0x80
};

/* closed caption data */
static guint8 h264_sei_user_data_registered[] = {
  0x00, 0x00, 0x00, 0x01, 0x06, 0x04, 0x47, 0xb5, 0x00, 0x31, 0x47, 0x41,
  0x39, 0x34, 0x03, 0xd4,
  0xff, 0xfc, 0x80, 0x80, 0xfd, 0x80, 0x80, 0xfa, 0x00, 0x00, 0xfa, 0x00,
  0x00, 0xfa, 0x00, 0x00,
  0xfa, 0x00, 0x00, 0xfa, 0x00, 0x00, 0xfa, 0x00, 0x00, 0xfa, 0x00, 0x00,
  0xfa, 0x00, 0x00, 0xfa,
  0x00, 0x00, 0xfa, 0x00, 0x00, 0xfa, 0x00, 0x00, 0xfa, 0x00, 0x00, 0xfa,
  0x00, 0x00, 0xfa, 0x00,
  0x00, 0xfa, 0x00, 0x00, 0xfa, 0x00, 0x00, 0xfa, 0x00, 0x00, 0xfa, 0x00,
  0x00, 0xff, 0x80
};

/* frame packing, side-by-side */
static guint8 h264_sei_frame_packing[] = {
  0x00, 0x00, 0x00, 0x01, 0x06, 0x2d, 0x07, 0x81, 0x81, 0x00, 0x00, 0x03,
  0x00, 0x01, 0x20, 0x80
};

GST_START_TEST (test_h264_parse_invalid_sei)
{
  GstH264ParserResult res;
  GstH264NalUnit nalu;
  GstH264NalParser *const parser = gst_h264_nal_parser_new ();
  const guint8 *buf = nalu_sps_with_vui;
  GArray *seis = NULL;
  GstH264SEIMessage *sei;

  /* First try parsing the SEI, which will fail because there's no SPS yet */
  res =
      gst_h264_parser_identify_nalu (parser, nalu_sei_pic_timing, 0,
      sizeof (nalu_sei_pic_timing), &nalu);
  assert_equals_int (res, GST_H264_PARSER_NO_NAL_END);
  assert_equals_int (nalu.type, GST_H264_NAL_SEI);

  res = gst_h264_parser_parse_sei (parser, &nalu, &seis);
  assert_equals_int (res, GST_H264_PARSER_BROKEN_LINK);
  g_array_free (seis, TRUE);

  /* Inject SPS */
  res =
      gst_h264_parser_identify_nalu (parser, buf, 0, sizeof (nalu_sps_with_vui),
      &nalu);
  assert_equals_int (res, GST_H264_PARSER_NO_NAL_END);
  assert_equals_int (nalu.type, GST_H264_NAL_SPS);
  assert_equals_int (nalu.size, 28);

  res = gst_h264_parser_parse_nal (parser, &nalu);
  assert_equals_int (res, GST_H264_PARSER_OK);

  /* Parse the SEI again */
  res =
      gst_h264_parser_identify_nalu (parser, nalu_sei_pic_timing, 0,
      sizeof (nalu_sei_pic_timing), &nalu);
  assert_equals_int (res, GST_H264_PARSER_NO_NAL_END);
  assert_equals_int (nalu.type, GST_H264_NAL_SEI);

  res = gst_h264_parser_parse_sei (parser, &nalu, &seis);
  assert_equals_int (res, GST_H264_PARSER_OK);
  fail_if (seis == NULL);
  assert_equals_int (seis->len, 1);
  g_array_free (seis, TRUE);

  /* Parse NALU with 2 chained SEI */
  res =
      gst_h264_parser_identify_nalu (parser, nalu_chained_sei, 0,
      sizeof (nalu_chained_sei), &nalu);
  assert_equals_int (res, GST_H264_PARSER_NO_NAL_END);
  assert_equals_int (nalu.type, GST_H264_NAL_SEI);

  res = gst_h264_parser_parse_sei (parser, &nalu, &seis);
  assert_equals_int (res, GST_H264_PARSER_OK);
  fail_if (seis == NULL);
  assert_equals_int (seis->len, 2);
  sei = &g_array_index (seis, GstH264SEIMessage, 0);
  assert_equals_int (sei->payloadType, GST_H264_SEI_PIC_TIMING);

  sei = &g_array_index (seis, GstH264SEIMessage, 1);
  assert_equals_int (sei->payloadType, GST_H264_SEI_RECOVERY_POINT);

  g_array_free (seis, TRUE);

  gst_h264_nal_parser_free (parser);
}

GST_END_TEST;

typedef gboolean (*SEICheckFunc) (gconstpointer a, gconstpointer b);

static gboolean
check_sei_user_data_registered (const GstH264RegisteredUserData * a,
    const GstH264RegisteredUserData * b)
{
  if (a->country_code != b->country_code)
    return FALSE;

  if ((a->country_code == 0xff) &&
      (a->country_code_extension != b->country_code_extension))
    return FALSE;

  if (a->size != b->size)
    return FALSE;

  return !memcmp (a->data, b->data, a->size);
}

static gboolean
check_sei_frame_packing (const GstH264FramePacking * a,
    const GstH264FramePacking * b)
{
  if ((a->frame_packing_id != b->frame_packing_id) ||
      (a->frame_packing_cancel_flag != b->frame_packing_cancel_flag))
    return FALSE;

  if (!a->frame_packing_cancel_flag) {
    if ((a->frame_packing_type != b->frame_packing_type) ||
        (a->quincunx_sampling_flag != b->quincunx_sampling_flag) ||
        (a->content_interpretation_type != b->content_interpretation_type) ||
        (a->spatial_flipping_flag != b->spatial_flipping_flag) ||
        (a->frame0_flipped_flag != b->frame0_flipped_flag) ||
        (a->field_views_flag != b->field_views_flag) ||
        (a->current_frame_is_frame0_flag != b->current_frame_is_frame0_flag) ||
        (a->frame0_self_contained_flag != b->frame0_self_contained_flag) ||
        (a->frame1_self_contained_flag != b->frame1_self_contained_flag))
      return FALSE;

    if (!a->quincunx_sampling_flag &&
        a->frame_packing_type != GST_H264_FRAME_PACKING_TEMPORAL_INTERLEAVING) {
      if ((a->frame0_grid_position_x != b->frame0_grid_position_x) ||
          (a->frame0_grid_position_y != b->frame0_grid_position_y) ||
          (a->frame1_grid_position_x != b->frame1_grid_position_x) ||
          (a->frame1_grid_position_y != b->frame1_grid_position_y))
        return FALSE;
    }

    if (a->frame_packing_repetition_period !=
        b->frame_packing_repetition_period)
      return FALSE;
  }

  return TRUE;
}

static gboolean
check_sei_mdcv (const GstH264MasteringDisplayColourVolume * a,
    const GstH264MasteringDisplayColourVolume * b)
{
  gint i;
  for (i = 0; i < 3; i++) {
    if (a->display_primaries_x[i] != b->display_primaries_x[i] ||
        a->display_primaries_y[i] != b->display_primaries_y[i])
      return FALSE;
  }

  return (a->white_point_x == b->white_point_x) &&
      (a->white_point_y == b->white_point_y) &&
      (a->max_display_mastering_luminance == b->max_display_mastering_luminance)
      && (a->min_display_mastering_luminance ==
      b->min_display_mastering_luminance);
}

static gboolean
check_sei_cll (const GstH264ContentLightLevel * a,
    const GstH264ContentLightLevel * b)
{
  return (a->max_content_light_level == b->max_content_light_level) &&
      (a->max_pic_average_light_level == b->max_pic_average_light_level);
}

static gboolean
check_sei_pic_timing (const GstH264PicTiming * a, const GstH264PicTiming * b)
{
  if (a->CpbDpbDelaysPresentFlag != b->CpbDpbDelaysPresentFlag)
    return FALSE;

  if (a->CpbDpbDelaysPresentFlag) {
    if (a->cpb_removal_delay != b->cpb_removal_delay ||
        a->cpb_removal_delay_length_minus1 != b->cpb_removal_delay_length_minus1
        || a->dpb_output_delay != b->dpb_output_delay
        || a->dpb_output_delay_length_minus1 !=
        b->dpb_output_delay_length_minus1)
      return FALSE;
  }

  if (a->pic_struct_present_flag != b->pic_struct_present_flag)
    return FALSE;

  if (a->pic_struct_present_flag) {
    const guint8 num_clock_ts_table[9] = {
      1, 1, 1, 2, 2, 3, 3, 2, 3
    };
    guint8 num_clock_num_ts;
    guint i;

    if (a->pic_struct != b->pic_struct)
      return FALSE;

    if (a->time_offset_length != b->time_offset_length)
      return FALSE;

    num_clock_num_ts = num_clock_ts_table[a->pic_struct];

    for (i = 0; i < num_clock_num_ts; i++) {
      if (a->clock_timestamp_flag[i] != b->clock_timestamp_flag[i])
        return FALSE;

      if (a->clock_timestamp_flag[i]) {
        const GstH264ClockTimestamp *ta = &a->clock_timestamp[i];
        const GstH264ClockTimestamp *tb = &b->clock_timestamp[i];

        if (ta->ct_type != tb->ct_type ||
            ta->nuit_field_based_flag != tb->nuit_field_based_flag ||
            ta->counting_type != tb->counting_type ||
            ta->discontinuity_flag != tb->discontinuity_flag ||
            ta->cnt_dropped_flag != tb->cnt_dropped_flag ||
            ta->n_frames != tb->n_frames)
          return FALSE;

        if (ta->full_timestamp_flag) {
          if (ta->seconds_value != tb->seconds_value ||
              ta->minutes_value != tb->minutes_value ||
              ta->hours_value != tb->hours_value)
            return FALSE;
        } else {
          if (ta->seconds_flag != tb->seconds_flag)
            return FALSE;

          if (ta->seconds_flag) {
            if (ta->seconds_value != tb->seconds_value ||
                ta->minutes_flag != tb->minutes_flag)
              return FALSE;

            if (ta->minutes_flag) {
              if (ta->minutes_value != tb->minutes_value ||
                  ta->hours_flag != tb->hours_flag)
                return FALSE;

              if (ta->hours_flag) {
                if (ta->hours_value != tb->hours_value)
                  return FALSE;
              }
            }
          }
        }

        if (ta->time_offset != tb->time_offset)
          return FALSE;
      }
    }
  }

  return TRUE;
}

GST_START_TEST (test_h264_create_sei)
{
  GstH264NalParser *parser;
  GstH264ParserResult parse_ret;
  GstH264NalUnit nalu;
  GArray *msg_array = NULL;
  GstMemory *mem;
  gint i;
  GstMapInfo info;
  struct
  {
    guint8 *raw_data;
    guint len;
    GstH264SEIPayloadType type;
    GstH264SEIMessage parsed_message;
    SEICheckFunc check_func;
  } test_list[] = {
    /* *INDENT-OFF* */
    {h264_sei_user_data_registered, G_N_ELEMENTS (h264_sei_user_data_registered),
        GST_H264_SEI_REGISTERED_USER_DATA, {0,},
        (SEICheckFunc) check_sei_user_data_registered},
    {h264_sei_frame_packing, G_N_ELEMENTS (h264_sei_frame_packing),
        GST_H264_SEI_FRAME_PACKING, {0,},
        (SEICheckFunc) check_sei_frame_packing},
    {h264_sei_mdcv, G_N_ELEMENTS (h264_sei_mdcv),
        GST_H264_SEI_MASTERING_DISPLAY_COLOUR_VOLUME, {0,},
        (SEICheckFunc) check_sei_mdcv},
    {h264_sei_cll, G_N_ELEMENTS (h264_sei_cll),
        GST_H264_SEI_CONTENT_LIGHT_LEVEL, {0,},
        (SEICheckFunc) check_sei_cll},
    {nalu_sei_pic_timing, G_N_ELEMENTS (nalu_sei_pic_timing),
        GST_H264_SEI_PIC_TIMING, {0,},
        (SEICheckFunc) check_sei_pic_timing},
    /* *INDENT-ON* */
  };

  parser = gst_h264_nal_parser_new ();

  /* inject SPS for picture timing sei */
  parse_ret =
      gst_h264_parser_identify_nalu_unchecked (parser, nalu_sps_with_vui, 0,
      sizeof (nalu_sps_with_vui), &nalu);
  assert_equals_int (parse_ret, GST_H264_PARSER_OK);
  assert_equals_int (nalu.type, GST_H264_NAL_SPS);
  assert_equals_int (nalu.size, 28);

  parse_ret = gst_h264_parser_parse_nal (parser, &nalu);
  assert_equals_int (parse_ret, GST_H264_PARSER_OK);

  /* test single sei message per sei nal unit */
  for (i = 0; i < G_N_ELEMENTS (test_list); i++) {
    gsize nal_size;

    parse_ret = gst_h264_parser_identify_nalu_unchecked (parser,
        test_list[i].raw_data, 0, test_list[i].len, &nalu);
    assert_equals_int (parse_ret, GST_H264_PARSER_OK);
    assert_equals_int (nalu.type, GST_H264_NAL_SEI);

    parse_ret = gst_h264_parser_parse_sei (parser, &nalu, &msg_array);
    assert_equals_int (parse_ret, GST_H264_PARSER_OK);
    assert_equals_int (msg_array->len, 1);

    /* test bytestream */
    mem = gst_h264_create_sei_memory (4, msg_array);
    fail_unless (mem != NULL);
    fail_unless (gst_memory_map (mem, &info, GST_MAP_READ));
    GST_MEMDUMP ("created sei nal", info.data, info.size);
    GST_MEMDUMP ("original sei nal", test_list[i].raw_data, test_list[i].len);
    assert_equals_int (info.size, test_list[i].len);
    fail_if (memcmp (info.data, test_list[i].raw_data, test_list[i].len));
    gst_memory_unmap (mem, &info);
    gst_memory_unref (mem);

    /* test packetized */
    mem = gst_h264_create_sei_memory_avc (4, msg_array);
    fail_unless (mem != NULL);
    fail_unless (gst_memory_map (mem, &info, GST_MAP_READ));
    assert_equals_int (info.size, test_list[i].len);
    fail_if (memcmp (info.data + 4, test_list[i].raw_data + 4,
            test_list[i].len - 4));
    nal_size = GST_READ_UINT32_BE (info.data);
    assert_equals_int (nal_size, info.size - 4);
    gst_memory_unmap (mem, &info);
    gst_memory_unref (mem);

    /* store parsed SEI for following tests */
    test_list[i].parsed_message =
        g_array_index (msg_array, GstH264SEIMessage, 0);
    if (test_list[i].type == GST_H264_SEI_REGISTERED_USER_DATA) {
      GstH264RegisteredUserData *dst_rud =
          &test_list[i].parsed_message.payload.registered_user_data;
      const GstH264SEIMessage *src_msg =
          &g_array_index (msg_array, GstH264SEIMessage, 0);
      const GstH264RegisteredUserData *src_rud =
          &src_msg->payload.registered_user_data;

      dst_rud->data = g_malloc (src_rud->size);
      memcpy ((guint8 *) dst_rud->data, src_rud->data, src_rud->size);
    }
    g_array_unref (msg_array);
  }

  /* test multiple SEI messages in a nal unit */
  msg_array = g_array_new (FALSE, FALSE, sizeof (GstH264SEIMessage));
  for (i = 0; i < G_N_ELEMENTS (test_list); i++)
    g_array_append_val (msg_array, test_list[i].parsed_message);

  mem = gst_h264_create_sei_memory (4, msg_array);
  fail_unless (mem != NULL);
  g_array_unref (msg_array);

  /* parse sei message from buffer */
  fail_unless (gst_memory_map (mem, &info, GST_MAP_READ));
  parse_ret = gst_h264_parser_identify_nalu_unchecked (parser,
      info.data, 0, info.size, &nalu);
  assert_equals_int (parse_ret, GST_H264_PARSER_OK);
  assert_equals_int (nalu.type, GST_H264_NAL_SEI);
  parse_ret = gst_h264_parser_parse_sei (parser, &nalu, &msg_array);
  gst_memory_unmap (mem, &info);
  gst_memory_unref (mem);

  assert_equals_int (parse_ret, GST_H264_PARSER_OK);
  assert_equals_int (msg_array->len, G_N_ELEMENTS (test_list));
  for (i = 0; i < msg_array->len; i++) {
    GstH264SEIMessage *msg = &g_array_index (msg_array, GstH264SEIMessage, i);

    assert_equals_int (msg->payloadType, test_list[i].type);
    fail_unless (test_list[i].check_func (&msg->payload,
            &test_list[i].parsed_message.payload));
  }

  /* clean up */
  for (i = 0; i < G_N_ELEMENTS (test_list); i++)
    gst_h264_sei_clear (&test_list[i].parsed_message);

  g_array_unref (msg_array);
  gst_h264_nal_parser_free (parser);
}

GST_END_TEST;

static Suite *
h264parser_suite (void)
{
  Suite *s = suite_create ("H264 Parser library");

  TCase *tc_chain = tcase_create ("general");

  suite_add_tcase (s, tc_chain);
  tcase_add_test (tc_chain, test_h264_parse_slice_dpa);
  tcase_add_test (tc_chain, test_h264_parse_slice_eoseq_slice);
  tcase_add_test (tc_chain, test_h264_parse_slice_5bytes);
  tcase_add_test (tc_chain, test_h264_parse_identify_nalu_avc);
  tcase_add_test (tc_chain, test_h264_parse_invalid_sei);
  tcase_add_test (tc_chain, test_h264_create_sei);

  return s;
}

GST_CHECK_MAIN (h264parser);
