/* GStreamer * Copyright (C) 2008-2009 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., 51 Franklin St, Fifth Floor, * Boston, MA 02110-1301, USA. */ /* Implementation of SMPTE 422M - Mapping JPEG2000 codestreams into the MXF * Generic Container */ /* TODO: * - parse the jpeg2000 sub-descriptor, see SMPTE 422M 7.2 * - Add support for XYZ colorspace */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include <gst/gst.h> #include <gst/video/video.h> #include <string.h> #include "mxfjpeg2000.h" #include "mxfessence.h" GST_DEBUG_CATEGORY_EXTERN (mxf_debug); #define GST_CAT_DEFAULT mxf_debug static gboolean mxf_is_jpeg2000_essence_track (const MXFMetadataTimelineTrack * track) { guint i; g_return_val_if_fail (track != NULL, FALSE); if (track->parent.descriptor == NULL) return FALSE; for (i = 0; i < track->parent.n_descriptor; i++) { MXFMetadataFileDescriptor *d = track->parent.descriptor[i]; MXFUL *key; if (!d) continue; key = &d->essence_container; /* SMPTE 422M 5.4 */ if (mxf_is_generic_container_essence_container_label (key) && key->u[12] == 0x02 && key->u[13] == 0x0c && (key->u[14] == 0x01 || key->u[14] == 0x02)) return TRUE; } return FALSE; } static GstFlowReturn mxf_jpeg2000_handle_essence_element (const MXFUL * key, GstBuffer * buffer, GstCaps * caps, MXFMetadataTimelineTrack * track, gpointer mapping_data, GstBuffer ** outbuf) { *outbuf = buffer; /* SMPTE 422M 5.1 */ if (key->u[12] != 0x15 || (key->u[14] != 0x08 && key->u[14] != 0x09)) { GST_ERROR ("Invalid JPEG2000 essence element"); return GST_FLOW_ERROR; } return GST_FLOW_OK; } static MXFEssenceWrapping mxf_jpeg2000_get_track_wrapping (const MXFMetadataTimelineTrack * track) { guint i; g_return_val_if_fail (track != NULL, MXF_ESSENCE_WRAPPING_CUSTOM_WRAPPING); if (track->parent.descriptor == NULL) { GST_ERROR ("No descriptor found for this track"); return MXF_ESSENCE_WRAPPING_CUSTOM_WRAPPING; } for (i = 0; i < track->parent.n_descriptor; i++) { if (!track->parent.descriptor[i]) continue; if (!MXF_IS_METADATA_GENERIC_PICTURE_ESSENCE_DESCRIPTOR (track-> parent.descriptor[i]) && !(MXF_IS_METADATA_FILE_DESCRIPTOR (track->parent.descriptor[i]) && !MXF_IS_METADATA_MULTIPLE_DESCRIPTOR (track-> parent.descriptor[i]))) continue; switch (track->parent.descriptor[i]->essence_container.u[14]) { case 0x01: return MXF_ESSENCE_WRAPPING_FRAME_WRAPPING; break; case 0x02: return MXF_ESSENCE_WRAPPING_CLIP_WRAPPING; break; default: return MXF_ESSENCE_WRAPPING_CUSTOM_WRAPPING; break; } } return MXF_ESSENCE_WRAPPING_CUSTOM_WRAPPING; } static GstCaps * mxf_jpeg2000_create_caps (MXFMetadataTimelineTrack * track, GstTagList ** tags, gboolean * intra_only, MXFEssenceElementHandleFunc * handler, gpointer * mapping_data) { MXFMetadataFileDescriptor *f = NULL; MXFMetadataGenericPictureEssenceDescriptor *p = NULL; guint i; GstCaps *caps = NULL; const gchar *colorspace; g_return_val_if_fail (track != NULL, NULL); if (track->parent.descriptor == NULL) { GST_ERROR ("No descriptor found for this track"); return NULL; } for (i = 0; i < track->parent.n_descriptor; i++) { if (!track->parent.descriptor[i]) continue; if (MXF_IS_METADATA_GENERIC_PICTURE_ESSENCE_DESCRIPTOR (track-> parent.descriptor[i])) { p = (MXFMetadataGenericPictureEssenceDescriptor *) track->parent. descriptor[i]; f = track->parent.descriptor[i]; break; } else if (MXF_IS_METADATA_FILE_DESCRIPTOR (track->parent.descriptor[i]) && !MXF_IS_METADATA_MULTIPLE_DESCRIPTOR (track->parent.descriptor[i])) { f = track->parent.descriptor[i]; } } if (!f) { GST_ERROR ("No descriptor found for this track"); return NULL; } colorspace = "sRGB"; if (p && MXF_IS_METADATA_CDCI_PICTURE_ESSENCE_DESCRIPTOR (p)) { colorspace = "sYUV"; } else if (p && MXF_IS_METADATA_RGBA_PICTURE_ESSENCE_DESCRIPTOR (p)) { MXFMetadataRGBAPictureEssenceDescriptor *r = (MXFMetadataRGBAPictureEssenceDescriptor *) p; gboolean rgb = TRUE; gboolean xyz = TRUE; gboolean yuv = TRUE; guint i; if (r->pixel_layout) { for (i = 0; i < r->n_pixel_layout; i++) { guint8 c = r->pixel_layout[2 * i]; switch (c) { case 'R': case 'r': case 'G': case 'g': case 'B': case 'b': xyz = yuv = FALSE; break; case 'Y': case 'y': rgb = FALSE; break; case 'U': case 'u': case 'V': case 'v': xyz = rgb = FALSE; break; case 'X': case 'x': case 'Z': case 'z': rgb = yuv = FALSE; break; default: break; } } if (rgb) { colorspace = "sRGB"; } else if (yuv) { colorspace = "sYUV"; } else if (xyz) { GST_ERROR ("JPEG2000 with XYZ colorspace not supported yet"); return NULL; } } } *handler = mxf_jpeg2000_handle_essence_element; /* TODO: What about other field values? */ caps = gst_caps_new_simple ("image/x-jpc", "colorspace", G_TYPE_STRING, colorspace, NULL); if (p) { mxf_metadata_generic_picture_essence_descriptor_set_caps (p, caps); } else { GST_WARNING ("Only a generic file descriptor found"); } if (!*tags) *tags = gst_tag_list_new_empty (); gst_tag_list_add (*tags, GST_TAG_MERGE_APPEND, GST_TAG_VIDEO_CODEC, "JPEG 2000", NULL); *intra_only = TRUE; return caps; } static const MXFEssenceElementHandler mxf_jpeg2000_essence_element_handler = { mxf_is_jpeg2000_essence_track, mxf_jpeg2000_get_track_wrapping, mxf_jpeg2000_create_caps }; static GstFlowReturn mxf_jpeg2000_write_func (GstBuffer * buffer, gpointer mapping_data, GstAdapter * adapter, GstBuffer ** outbuf, gboolean flush) { *outbuf = buffer; return GST_FLOW_OK; } static const guint8 jpeg2000_essence_container_ul[] = { 0x06, 0x0e, 0x2b, 0x34, 0x04, 0x01, 0x01, 0x07, 0x0d, 0x01, 0x03, 0x01, 0x02, 0x0c, 0x01, 0x00 }; static const guint jpeg2000_picture_essence_coding[] = { 0x06, 0x0E, 0x2B, 0x34, 0x04, 0x01, 0x01, 0x07, 0x04, 0x01, 0x02, 0x02, 0x03, 0x01, 0x01, 0x00 }; static MXFMetadataFileDescriptor * mxf_jpeg2000_get_descriptor (GstPadTemplate * tmpl, GstCaps * caps, MXFEssenceElementWriteFunc * handler, gpointer * mapping_data) { MXFMetadataRGBAPictureEssenceDescriptor *ret; GstStructure *s; const gchar *colorspace; s = gst_caps_get_structure (caps, 0); if (strcmp (gst_structure_get_name (s), "image/x-jpc") != 0 || !gst_structure_get_string (s, "colorspace")) { GST_ERROR ("Invalid caps %" GST_PTR_FORMAT, caps); return NULL; } colorspace = gst_structure_get_string (s, "colorspace"); ret = (MXFMetadataRGBAPictureEssenceDescriptor *) g_object_new (MXF_TYPE_METADATA_RGBA_PICTURE_ESSENCE_DESCRIPTOR, NULL); memcpy (&ret->parent.parent.essence_container, &jpeg2000_essence_container_ul, 16); memcpy (&ret->parent.picture_essence_coding, &jpeg2000_picture_essence_coding, 16); if (g_str_equal (colorspace, "sRGB")) { ret->n_pixel_layout = 3; ret->pixel_layout = g_new0 (guint8, 6); ret->pixel_layout[0] = 'R'; ret->pixel_layout[1] = 8; ret->pixel_layout[2] = 'G'; ret->pixel_layout[3] = 8; ret->pixel_layout[4] = 'B'; ret->pixel_layout[5] = 8; } else if (g_str_equal (colorspace, "sYUV")) { ret->n_pixel_layout = 3; ret->pixel_layout = g_new0 (guint8, 6); ret->pixel_layout[0] = 'Y'; ret->pixel_layout[1] = 8; ret->pixel_layout[2] = 'U'; ret->pixel_layout[3] = 8; ret->pixel_layout[4] = 'V'; ret->pixel_layout[5] = 8; } else { g_assert_not_reached (); } if (!mxf_metadata_generic_picture_essence_descriptor_from_caps (&ret->parent, caps)) { g_object_unref (ret); return NULL; } *handler = mxf_jpeg2000_write_func; return (MXFMetadataFileDescriptor *) ret; } static void mxf_jpeg2000_update_descriptor (MXFMetadataFileDescriptor * d, GstCaps * caps, gpointer mapping_data, GstBuffer * buf) { return; } static void mxf_jpeg2000_get_edit_rate (MXFMetadataFileDescriptor * a, GstCaps * caps, gpointer mapping_data, GstBuffer * buf, MXFMetadataSourcePackage * package, MXFMetadataTimelineTrack * track, MXFFraction * edit_rate) { edit_rate->n = a->sample_rate.n; edit_rate->d = a->sample_rate.d; } static guint32 mxf_jpeg2000_get_track_number_template (MXFMetadataFileDescriptor * a, GstCaps * caps, gpointer mapping_data) { return (0x15 << 24) | (0x08 << 8); } static MXFEssenceElementWriter mxf_jpeg2000_essence_element_writer = { mxf_jpeg2000_get_descriptor, mxf_jpeg2000_update_descriptor, mxf_jpeg2000_get_edit_rate, mxf_jpeg2000_get_track_number_template, NULL, {{0,}} }; void mxf_jpeg2000_init (void) { mxf_essence_element_handler_register (&mxf_jpeg2000_essence_element_handler); mxf_jpeg2000_essence_element_writer.pad_template = gst_pad_template_new ("jpeg2000_video_sink_%u", GST_PAD_SINK, GST_PAD_REQUEST, gst_caps_from_string ("image/x-jpc, width = " GST_VIDEO_SIZE_RANGE ", height = " GST_VIDEO_SIZE_RANGE ", framerate = " GST_VIDEO_FPS_RANGE ", colorspace = (string) { \"sRGB\", \"sYUV\" }")); memcpy (&mxf_jpeg2000_essence_element_writer.data_definition, mxf_metadata_track_identifier_get (MXF_METADATA_TRACK_PICTURE_ESSENCE), 16); mxf_essence_element_writer_register (&mxf_jpeg2000_essence_element_writer); }