diff options
Diffstat (limited to 'gst/mxf/mxfmpeg.c')
| -rw-r--r-- | gst/mxf/mxfmpeg.c | 318 |
1 files changed, 318 insertions, 0 deletions
diff --git a/gst/mxf/mxfmpeg.c b/gst/mxf/mxfmpeg.c new file mode 100644 index 00000000..e46d5388 --- /dev/null +++ b/gst/mxf/mxfmpeg.c @@ -0,0 +1,318 @@ +/* GStreamer + * Copyright (C) 2008 Sebastian Dröge <sebastian.droege@collabora.co.uk> + * + * 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., 59 Temple Place - Suite 330, + * Boston, MA 02111-1307, USA. + */ + +/* Implementation of SMPTE 381M - Mapping MPEG streams into the MXF + * Generic Container + */ + +/* TODO: + * - Handle PES streams + */ + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif + +#include <gst/gst.h> +#include <string.h> + +#include "mxfmpeg.h" + +GST_DEBUG_CATEGORY_EXTERN (mxf_debug); +#define GST_CAT_DEFAULT mxf_debug + +/* SMPTE 381M 8.1 - ULs of local tags */ +static const guint8 _single_sequence_ul[] = { + 0x06, 0x0e, 0x2b, 0x34, 0x01, 0x01, 0x01, 0x05, 0x04, 0x01, 0x06, 0x02, 0x01, + 0x02, 0x00, 0x00 +}; + +static const guint8 _constant_b_frames_ul[] = { + 0x06, 0x0e, 0x2b, 0x34, 0x01, 0x01, 0x01, 0x05, 0x04, 0x01, 0x06, 0x02, 0x01, + 0x03, 0x00, 0x00 +}; + +static const guint8 _coded_content_type_ul[] = { + 0x06, 0x0e, 0x2b, 0x34, 0x01, 0x01, 0x01, 0x05, 0x04, 0x01, 0x06, 0x02, 0x01, + 0x04, 0x00, 0x00 +}; + +static const guint8 _low_delay_ul[] = { + 0x06, 0x0e, 0x2b, 0x34, 0x01, 0x01, 0x01, 0x05, 0x04, 0x01, 0x06, 0x02, 0x01, + 0x05, 0x00, 0x00 +}; + +static const guint8 _closed_gop_ul[] = { + 0x06, 0x0e, 0x2b, 0x34, 0x01, 0x01, 0x01, 0x05, 0x04, 0x01, 0x06, 0x02, 0x01, + 0x06, 0x00, 0x00 +}; + +static const guint8 _identical_gop_ul[] = { + 0x06, 0x0e, 0x2b, 0x34, 0x01, 0x01, 0x01, 0x05, 0x04, 0x01, 0x06, 0x02, 0x01, + 0x07, 0x00, 0x00 +}; + +static const guint8 _max_gop_ul[] = { + 0x06, 0x0e, 0x2b, 0x34, 0x01, 0x01, 0x01, 0x05, 0x04, 0x01, 0x06, 0x02, 0x01, + 0x08, 0x00, 0x00 +}; + +static const guint8 _b_picture_count_ul[] = { + 0x06, 0x0e, 0x2b, 0x34, 0x01, 0x01, 0x01, 0x05, 0x04, 0x01, 0x06, 0x02, 0x01, + 0x09, 0x00, 0x00 +}; + +static const guint8 _bitrate_ul[] = { + 0x06, 0x0e, 0x2b, 0x34, 0x01, 0x01, 0x01, 0x05, 0x04, 0x01, 0x06, 0x02, 0x01, + 0x0b, 0x00, 0x00 +}; + +static const guint8 _profile_and_level_ul[] = { + 0x06, 0x0e, 0x2b, 0x34, 0x01, 0x01, 0x01, 0x05, 0x04, 0x01, 0x06, 0x02, 0x01, + 0x0a, 0x00, 0x00 +}; + +gboolean +mxf_metadata_mpeg_video_descriptor_parse (const MXFUL * key, + MXFMetadataMPEGVideoDescriptor * descriptor, + const MXFPrimerPack * primer, guint16 type, const guint8 * data, gsize size) +{ + guint16 tag, tag_size; + const guint8 *tag_data; + + g_return_val_if_fail (data != NULL, FALSE); + + memset (descriptor, 0, sizeof (MXFMetadataMPEGVideoDescriptor)); + + if (!mxf_metadata_cdci_picture_essence_descriptor_parse (key, + (MXFMetadataCDCIPictureEssenceDescriptor *) descriptor, primer, type, + data, size)) + goto error; + + while (mxf_local_tag_parse (data, size, &tag, &tag_size, &tag_data)) { + MXFUL *tag_ul = NULL; + + if (tag_size == 0 || tag == 0x0000) + goto next; + + if (!(tag_ul = + (MXFUL *) g_hash_table_lookup (primer->mappings, + GUINT_TO_POINTER (((guint) tag))))) + goto next; + + if (memcmp (tag_ul, &_single_sequence_ul, 16) == 0) { + GST_WRITE_UINT16_BE (data, 0x0000); + if (tag_size != 1) + goto error; + descriptor->single_sequence = GST_READ_UINT8 (tag_data); + } else if (memcmp (tag_ul, &_constant_b_frames_ul, 16) == 0) { + GST_WRITE_UINT16_BE (data, 0x0000); + if (tag_size != 1) + goto error; + descriptor->const_b_frames = GST_READ_UINT8 (tag_data); + } else if (memcmp (tag_ul, &_coded_content_type_ul, 16) == 0) { + GST_WRITE_UINT16_BE (data, 0x0000); + if (tag_size != 1) + goto error; + descriptor->coded_content_type = GST_READ_UINT8 (tag_data); + } else if (memcmp (tag_ul, &_low_delay_ul, 16) == 0) { + GST_WRITE_UINT16_BE (data, 0x0000); + if (tag_size != 1) + goto error; + descriptor->low_delay = GST_READ_UINT8 (tag_data); + } else if (memcmp (tag_ul, &_closed_gop_ul, 16) == 0) { + GST_WRITE_UINT16_BE (data, 0x0000); + if (tag_size != 1) + goto error; + descriptor->closed_gop = GST_READ_UINT8 (tag_data); + } else if (memcmp (tag_ul, &_identical_gop_ul, 16) == 0) { + GST_WRITE_UINT16_BE (data, 0x0000); + if (tag_size != 1) + goto error; + descriptor->identical_gop = GST_READ_UINT8 (tag_data); + } else if (memcmp (tag_ul, &_max_gop_ul, 16) == 0) { + GST_WRITE_UINT16_BE (data, 0x0000); + if (tag_size != 2) + goto error; + descriptor->max_gop = GST_READ_UINT16_BE (tag_data); + } else if (memcmp (tag_ul, &_b_picture_count_ul, 16) == 0) { + GST_WRITE_UINT16_BE (data, 0x0000); + if (tag_size != 2) + goto error; + descriptor->b_picture_count = GST_READ_UINT16_BE (tag_data); + } else if (memcmp (tag_ul, &_bitrate_ul, 16) == 0) { + GST_WRITE_UINT16_BE (data, 0x0000); + if (tag_size != 4) + goto error; + descriptor->bitrate = GST_READ_UINT32_BE (tag_data); + } else if (memcmp (tag_ul, &_profile_and_level_ul, 16) == 0) { + GST_WRITE_UINT16_BE (data, 0x0000); + if (tag_size != 1) + goto error; + descriptor->profile_and_level = GST_READ_UINT8 (tag_data); + } else { + if (type != MXF_METADATA_MPEG_VIDEO_DESCRIPTOR) + goto next; + GST_WRITE_UINT16_BE (data, 0x0000); + if (!gst_metadata_add_custom_tag (primer, tag, tag_data, tag_size, + &((MXFMetadataGenericDescriptor *) descriptor)->other_tags)) + goto error; + } + next: + data += 4 + tag_size; + size -= 4 + tag_size; + } + + GST_DEBUG ("Parsed mpeg video descriptors:"); + GST_DEBUG (" single sequence = %s", + (descriptor->single_sequence) ? "yes" : "no"); + GST_DEBUG (" constant b frames = %s", + (descriptor->single_sequence) ? "yes" : "no"); + GST_DEBUG (" coded content type = %u", descriptor->coded_content_type); + GST_DEBUG (" low delay = %s", (descriptor->low_delay) ? "yes" : "no"); + GST_DEBUG (" closed gop = %s", (descriptor->closed_gop) ? "yes" : "no"); + GST_DEBUG (" identical gop = %s", + (descriptor->identical_gop) ? "yes" : "no"); + GST_DEBUG (" max gop = %u", descriptor->max_gop); + GST_DEBUG (" b picture count = %u", descriptor->b_picture_count); + GST_DEBUG (" bitrate = %u", descriptor->bitrate); + GST_DEBUG (" profile & level = %u", descriptor->profile_and_level); + + return TRUE; + +error: + GST_ERROR ("Invalid mpeg video descriptor"); + mxf_metadata_mpeg_video_descriptor_reset (descriptor); + + return FALSE; +} + +void mxf_metadata_mpeg_video_descriptor_reset + (MXFMetadataMPEGVideoDescriptor * descriptor) +{ + g_return_if_fail (descriptor != NULL); + + mxf_metadata_cdci_picture_essence_descriptor_reset ( + (MXFMetadataCDCIPictureEssenceDescriptor *) descriptor); + + memset (descriptor, 0, sizeof (MXFMetadataMPEGVideoDescriptor)); +} + +gboolean +mxf_is_mpeg_video_essence_track (const MXFMetadataTrack * track) +{ + guint i; + + g_return_val_if_fail (track != NULL, FALSE); + + if (track->descriptor == NULL) + return FALSE; + + for (i = 0; i < track->n_descriptor; i++) { + MXFMetadataFileDescriptor *d = track->descriptor[i]; + MXFUL *key = &d->essence_container; + /* SMPTE 381M 7 */ + if (mxf_is_generic_container_essence_container_label (key) && + key->u[12] == 0x02 && + (key->u[13] == 0x04 || + key->u[13] == 0x07 || key->u[13] == 0x08 || key->u[13] == 0x09)) + return TRUE; + } + + return FALSE; +} + +static GstFlowReturn +mxf_mpeg_video_handle_essence_element (const MXFUL * key, GstBuffer * buffer, + GstCaps * caps, MXFMetadataGenericPackage * package, + MXFMetadataTrack * track, MXFMetadataStructuralComponent * component, + gpointer mapping_data, GstBuffer ** outbuf) +{ + *outbuf = buffer; + + /* SMPTE 381M 6.1 */ + if (key->u[12] != 0x15 || (key->u[14] != 0x05 && key->u[14] != 0x06 + && key->u[14] != 0x07)) { + GST_ERROR ("Invalid MPEG video essence element"); + return GST_FLOW_ERROR; + } + + return GST_FLOW_OK; +} + + +GstCaps * +mxf_mpeg_video_create_caps (MXFMetadataGenericPackage * package, + MXFMetadataTrack * track, GstTagList ** tags, + MXFEssenceElementHandler * handler, gpointer * mapping_data) +{ + MXFMetadataMPEGVideoDescriptor *d = NULL; + MXFMetadataFileDescriptor *f = NULL; + guint i; + + g_return_val_if_fail (package != NULL, NULL); + g_return_val_if_fail (track != NULL, NULL); + + if (track->descriptor == NULL) { + GST_ERROR ("No descriptor found for this track"); + return NULL; + } + + for (i = 0; i < track->n_descriptor; i++) { + if (((MXFMetadataGenericDescriptor *) track->descriptor[i])->type == + MXF_METADATA_MPEG_VIDEO_DESCRIPTOR) { + d = (MXFMetadataMPEGVideoDescriptor *) track->descriptor[i]; + f = track->descriptor[i]; + break; + } else if (((MXFMetadataGenericDescriptor *) track->descriptor[i])->type == + MXF_METADATA_CDCI_PICTURE_ESSENCE_DESCRIPTOR + || ((MXFMetadataGenericDescriptor *) track->descriptor[i])->type == + MXF_METADATA_GENERIC_PICTURE_ESSENCE_DESCRIPTOR + || ((MXFMetadataGenericDescriptor *) track->descriptor[i])->type == + MXF_METADATA_FILE_DESCRIPTOR) { + f = track->descriptor[i]; + } + } + + if (!f) { + GST_ERROR ("No descriptor found for this track"); + return NULL; + } + + *handler = mxf_mpeg_video_handle_essence_element; + /* SMPTE 381M 7 */ + if (f->essence_container.u[13] == 0x04) { + /* FIXME: get mpeg version somehow */ + GST_DEBUG ("Found MPEG ES stream"); + return gst_caps_new_simple ("video/mpeg", "mpegversion", G_TYPE_INT, 1, + "systemstream", G_TYPE_BOOLEAN, FALSE, NULL); + } else if (f->essence_container.u[13] == 0x07) { + GST_ERROR ("MPEG PES streams not supported yet"); + return NULL; + } else if (f->essence_container.u[13] == 0x08) { + /* FIXME: get mpeg version somehow */ + GST_DEBUG ("Found MPEG PS stream"); + return gst_caps_new_simple ("video/mpeg", "mpegversion", G_TYPE_INT, 1, + "systemstream", G_TYPE_BOOLEAN, TRUE, NULL); + } else if (f->essence_container.u[13] == 0x09) { + GST_DEBUG ("Found MPEG TS stream"); + return gst_caps_new_simple ("video/mpegts", NULL); + } + return NULL; +} |