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