gstreamer/gst/matroska/matroska-ids.c
Seungha Yang 5dd39d8946 matroskademux: Add parsing Colour element
... and forward colorimetry to downstream. The Colour element describes
various color information (similar to 'colr' box in isobmff).
Note that, due to the comparatively limited syntax for color information
in vpx codecs, the color information in mkv/wemb container level
should be used for sophisticated color handling (e.g., HDR video).

https://bugzilla.gnome.org/show_bug.cgi?id=790023
2017-11-08 11:01:55 +02:00

355 lines
9.4 KiB
C

/* GStreamer Matroska muxer/demuxer
* (C) 2003 Ronald Bultje <rbultje@ronald.bitfreak.net>
* (C) 2006 Tim-Philipp Müller <tim centricular net>
*
* matroska-ids.c: matroska track context utility functions
*
* 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.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "matroska-ids.h"
#include <string.h>
gboolean
gst_matroska_track_init_video_context (GstMatroskaTrackContext ** p_context)
{
GstMatroskaTrackVideoContext *video_context;
g_assert (p_context != NULL && *p_context != NULL);
/* already set up? (track info might come before track type) */
if ((*p_context)->type == GST_MATROSKA_TRACK_TYPE_VIDEO) {
GST_LOG ("video context already set up");
return TRUE;
}
/* it better not have been set up as some other track type ... */
if ((*p_context)->type != 0) {
g_return_val_if_reached (FALSE);
}
video_context = g_renew (GstMatroskaTrackVideoContext, *p_context, 1);
*p_context = (GstMatroskaTrackContext *) video_context;
/* defaults */
(*p_context)->type = GST_MATROSKA_TRACK_TYPE_VIDEO;
video_context->display_width = 0;
video_context->display_height = 0;
video_context->pixel_width = 0;
video_context->pixel_height = 0;
video_context->asr_mode = 0;
video_context->fourcc = 0;
video_context->default_fps = 0.0;
video_context->earliest_time = GST_CLOCK_TIME_NONE;
video_context->multiview_mode = GST_VIDEO_MULTIVIEW_MODE_NONE;
video_context->multiview_flags = GST_VIDEO_MULTIVIEW_FLAGS_NONE;
video_context->colorimetry.range = GST_VIDEO_COLOR_RANGE_UNKNOWN;
video_context->colorimetry.matrix = GST_VIDEO_COLOR_MATRIX_UNKNOWN;
video_context->colorimetry.transfer = GST_VIDEO_TRANSFER_UNKNOWN;
video_context->colorimetry.primaries = GST_VIDEO_COLOR_PRIMARIES_UNKNOWN;
return TRUE;
}
gboolean
gst_matroska_track_init_audio_context (GstMatroskaTrackContext ** p_context)
{
GstMatroskaTrackAudioContext *audio_context;
g_assert (p_context != NULL && *p_context != NULL);
/* already set up? (track info might come before track type) */
if ((*p_context)->type == GST_MATROSKA_TRACK_TYPE_AUDIO)
return TRUE;
/* it better not have been set up as some other track type ... */
if ((*p_context)->type != 0) {
g_return_val_if_reached (FALSE);
}
audio_context = g_renew (GstMatroskaTrackAudioContext, *p_context, 1);
*p_context = (GstMatroskaTrackContext *) audio_context;
/* defaults */
(*p_context)->type = GST_MATROSKA_TRACK_TYPE_AUDIO;
audio_context->channels = 1;
audio_context->samplerate = 8000;
return TRUE;
}
gboolean
gst_matroska_track_init_subtitle_context (GstMatroskaTrackContext ** p_context)
{
GstMatroskaTrackSubtitleContext *subtitle_context;
g_assert (p_context != NULL && *p_context != NULL);
/* already set up? (track info might come before track type) */
if ((*p_context)->type == GST_MATROSKA_TRACK_TYPE_SUBTITLE)
return TRUE;
/* it better not have been set up as some other track type ... */
if ((*p_context)->type != 0) {
g_return_val_if_reached (FALSE);
}
subtitle_context = g_renew (GstMatroskaTrackSubtitleContext, *p_context, 1);
*p_context = (GstMatroskaTrackContext *) subtitle_context;
(*p_context)->type = GST_MATROSKA_TRACK_TYPE_SUBTITLE;
subtitle_context->invalid_utf8 = FALSE;
subtitle_context->seen_markup_tag = FALSE;
return TRUE;
}
void
gst_matroska_register_tags (void)
{
/* TODO: register other custom tags */
}
GstBufferList *
gst_matroska_parse_xiph_stream_headers (gpointer codec_data,
gsize codec_data_size)
{
GstBufferList *list = NULL;
guint8 *p = codec_data;
gint i, offset, num_packets;
guint *length, last;
GST_MEMDUMP ("xiph codec data", codec_data, codec_data_size);
if (codec_data == NULL || codec_data_size == 0)
goto error;
/* start of the stream and vorbis audio or theora video, need to
* send the codec_priv data as first three packets */
num_packets = p[0] + 1;
GST_DEBUG ("%u stream headers, total length=%" G_GSIZE_FORMAT " bytes",
(guint) num_packets, codec_data_size);
length = g_alloca (num_packets * sizeof (guint));
last = 0;
offset = 1;
/* first packets, read length values */
for (i = 0; i < num_packets - 1; i++) {
length[i] = 0;
while (offset < codec_data_size) {
length[i] += p[offset];
if (p[offset++] != 0xff)
break;
}
last += length[i];
}
if (offset + last > codec_data_size)
goto error;
/* last packet is the remaining size */
length[i] = codec_data_size - offset - last;
list = gst_buffer_list_new ();
for (i = 0; i < num_packets; i++) {
GstBuffer *hdr;
GST_DEBUG ("buffer %d: %u bytes", i, (guint) length[i]);
if (offset + length[i] > codec_data_size)
goto error;
hdr = gst_buffer_new_wrapped (g_memdup (p + offset, length[i]), length[i]);
gst_buffer_list_add (list, hdr);
offset += length[i];
}
return list;
/* ERRORS */
error:
{
if (list != NULL)
gst_buffer_list_unref (list);
return NULL;
}
}
GstBufferList *
gst_matroska_parse_speex_stream_headers (gpointer codec_data,
gsize codec_data_size)
{
GstBufferList *list = NULL;
GstBuffer *hdr;
guint8 *pdata = codec_data;
GST_MEMDUMP ("speex codec data", codec_data, codec_data_size);
if (codec_data == NULL || codec_data_size < 80) {
GST_WARNING ("not enough codec priv data for speex headers");
return NULL;
}
if (memcmp (pdata, "Speex ", 8) != 0) {
GST_WARNING ("no Speex marker at start of stream headers");
return NULL;
}
list = gst_buffer_list_new ();
hdr = gst_buffer_new_wrapped (g_memdup (pdata, 80), 80);
gst_buffer_list_add (list, hdr);
if (codec_data_size > 80) {
hdr = gst_buffer_new_wrapped (g_memdup (pdata + 80, codec_data_size - 80),
codec_data_size - 80);
gst_buffer_list_add (list, hdr);
}
return list;
}
GstBufferList *
gst_matroska_parse_opus_stream_headers (gpointer codec_data,
gsize codec_data_size)
{
GstBufferList *list = NULL;
GstBuffer *hdr;
guint8 *pdata = codec_data;
GST_MEMDUMP ("opus codec data", codec_data, codec_data_size);
if (codec_data == NULL || codec_data_size < 19) {
GST_WARNING ("not enough codec priv data for opus headers");
return NULL;
}
if (memcmp (pdata, "OpusHead", 8) != 0) {
GST_WARNING ("no OpusHead marker at start of stream headers");
return NULL;
}
list = gst_buffer_list_new ();
hdr =
gst_buffer_new_wrapped (g_memdup (pdata, codec_data_size),
codec_data_size);
gst_buffer_list_add (list, hdr);
return list;
}
GstBufferList *
gst_matroska_parse_flac_stream_headers (gpointer codec_data,
gsize codec_data_size)
{
GstBufferList *list = NULL;
GstBuffer *hdr;
guint8 *pdata = codec_data;
guint len, off;
GST_MEMDUMP ("flac codec data", codec_data, codec_data_size);
/* need at least 'fLaC' marker + STREAMINFO metadata block */
if (codec_data == NULL || codec_data_size < ((4) + (4 + 34))) {
GST_WARNING ("not enough codec priv data for flac headers");
return NULL;
}
if (memcmp (pdata, "fLaC", 4) != 0) {
GST_WARNING ("no flac marker at start of stream headers");
return NULL;
}
list = gst_buffer_list_new ();
hdr = gst_buffer_new_wrapped (g_memdup (pdata, 4), 4);
gst_buffer_list_add (list, hdr);
/* skip fLaC marker */
off = 4;
while (off < codec_data_size - 3) {
len = GST_READ_UINT8 (pdata + off + 1) << 16;
len |= GST_READ_UINT8 (pdata + off + 2) << 8;
len |= GST_READ_UINT8 (pdata + off + 3);
GST_DEBUG ("header packet: len=%u bytes, flags=0x%02x", len, pdata[off]);
if (off + len > codec_data_size) {
gst_buffer_list_unref (list);
return NULL;
}
hdr = gst_buffer_new_wrapped (g_memdup (pdata + off, len + 4), len + 4);
gst_buffer_list_add (list, hdr);
off += 4 + len;
}
return list;
}
GstClockTime
gst_matroska_track_get_buffer_timestamp (GstMatroskaTrackContext * track,
GstBuffer * buf)
{
if (track->dts_only) {
return GST_BUFFER_DTS_OR_PTS (buf);
} else {
return GST_BUFFER_PTS (buf);
}
}
void
gst_matroska_track_free (GstMatroskaTrackContext * track)
{
g_free (track->codec_id);
g_free (track->codec_name);
g_free (track->name);
g_free (track->language);
g_free (track->codec_priv);
g_free (track->codec_state);
if (track->encodings != NULL) {
int i;
for (i = 0; i < track->encodings->len; ++i) {
GstMatroskaTrackEncoding *enc = &g_array_index (track->encodings,
GstMatroskaTrackEncoding,
i);
g_free (enc->comp_settings);
}
g_array_free (track->encodings, TRUE);
}
if (track->tags)
gst_tag_list_unref (track->tags);
if (track->index_table)
g_array_free (track->index_table, TRUE);
if (track->stream_headers)
gst_buffer_list_unref (track->stream_headers);
g_free (track);
}