mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-13 12:51:16 +00:00
a1e8109aa8
Add an API to create raw SEI nal unit. This would be useful in case an user want to create SEI nal data and inject the SEI nal data into bitstream.
576 lines
23 KiB
C
576 lines
23 KiB
C
/* 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;
|
|
|
|
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, 0x01, 0x06, 0x01, 0x02, 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);
|
|
}
|
|
|
|
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},
|
|
/* *INDENT-ON* */
|
|
};
|
|
|
|
parser = gst_h264_nal_parser_new ();
|
|
|
|
/* 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_invalid_sei);
|
|
tcase_add_test (tc_chain, test_h264_create_sei);
|
|
|
|
return s;
|
|
}
|
|
|
|
GST_CHECK_MAIN (h264parser);
|