mxfmux: Add support for muxing MPEG audio and video

1 files changed, 384 insertions, 1 deletions

diff --git a/gst/mxf/mxfmpeg.c b/gst/mxf/mxfmpeg.c
index 0865577c..68c6ab76 100644
--- a/gst/mxf/mxfmpeg.c
+++ b/gst/mxf/mxfmpeg.c
@@ -33,10 +33,12 @@
 #endif
 
 #include <gst/gst.h>
+#include <gst/video/video.h>
 #include <string.h>
 
 #include "mxfmpeg.h"
 #include "mxfquark.h"
+#include "mxfwrite.h"
 
 #include <gst/base/gstbytereader.h>
 
@@ -243,10 +245,133 @@ mxf_metadata_mpeg_video_descriptor_to_structure (MXFMetadataBase * m)
   return ret;
 }
 
+static GList *
+mxf_metadata_mpeg_video_descriptor_write_tags (MXFMetadataBase * m,
+    MXFPrimerPack * primer)
+{
+  MXFMetadataMPEGVideoDescriptor *self = MXF_METADATA_MPEG_VIDEO_DESCRIPTOR (m);
+  GList *ret =
+      MXF_METADATA_BASE_CLASS
+      (mxf_metadata_mpeg_video_descriptor_parent_class)->write_tags (m, primer);
+  MXFLocalTag *t;
+
+  if (self->single_sequence != -1) {
+    t = g_slice_new0 (MXFLocalTag);
+    memcpy (&t->key, &_single_sequence_ul, 16);
+    t->size = 1;
+    t->data = g_slice_alloc (t->size);
+    t->g_slice = TRUE;
+    GST_WRITE_UINT8 (t->data, (self->single_sequence) ? 1 : 0);
+    mxf_primer_pack_add_mapping (primer, 0, &t->key);
+    ret = g_list_prepend (ret, t);
+  }
+
+  if (self->const_b_frames) {
+    t = g_slice_new0 (MXFLocalTag);
+    memcpy (&t->key, &_constant_b_frames_ul, 16);
+    t->size = 1;
+    t->data = g_slice_alloc (t->size);
+    t->g_slice = TRUE;
+    GST_WRITE_UINT8 (t->data, (self->const_b_frames) ? 1 : 0);
+    mxf_primer_pack_add_mapping (primer, 0, &t->key);
+    ret = g_list_prepend (ret, t);
+  }
+
+  if (self->coded_content_type) {
+    t = g_slice_new0 (MXFLocalTag);
+    memcpy (&t->key, &_coded_content_type_ul, 16);
+    t->size = 1;
+    t->data = g_slice_alloc (t->size);
+    t->g_slice = TRUE;
+    GST_WRITE_UINT8 (t->data, self->coded_content_type);
+    mxf_primer_pack_add_mapping (primer, 0, &t->key);
+    ret = g_list_prepend (ret, t);
+  }
+
+  if (self->low_delay) {
+    t = g_slice_new0 (MXFLocalTag);
+    memcpy (&t->key, &_low_delay_ul, 16);
+    t->size = 1;
+    t->data = g_slice_alloc (t->size);
+    t->g_slice = TRUE;
+    GST_WRITE_UINT8 (t->data, (self->low_delay) ? 1 : 0);
+    mxf_primer_pack_add_mapping (primer, 0, &t->key);
+    ret = g_list_prepend (ret, t);
+  }
+
+  if (self->closed_gop) {
+    t = g_slice_new0 (MXFLocalTag);
+    memcpy (&t->key, &_closed_gop_ul, 16);
+    t->size = 1;
+    t->data = g_slice_alloc (t->size);
+    t->g_slice = TRUE;
+    GST_WRITE_UINT8 (t->data, (self->closed_gop) ? 1 : 0);
+    mxf_primer_pack_add_mapping (primer, 0, &t->key);
+    ret = g_list_prepend (ret, t);
+  }
+
+  if (self->identical_gop) {
+    t = g_slice_new0 (MXFLocalTag);
+    memcpy (&t->key, &_identical_gop_ul, 16);
+    t->size = 1;
+    t->data = g_slice_alloc (t->size);
+    t->g_slice = TRUE;
+    GST_WRITE_UINT8 (t->data, (self->identical_gop) ? 1 : 0);
+    mxf_primer_pack_add_mapping (primer, 0, &t->key);
+    ret = g_list_prepend (ret, t);
+  }
+
+  if (self->max_gop) {
+    t = g_slice_new0 (MXFLocalTag);
+    memcpy (&t->key, &_identical_gop_ul, 16);
+    t->size = 2;
+    t->data = g_slice_alloc (t->size);
+    t->g_slice = TRUE;
+    GST_WRITE_UINT16_BE (t->data, self->max_gop);
+    mxf_primer_pack_add_mapping (primer, 0, &t->key);
+    ret = g_list_prepend (ret, t);
+  }
+
+  if (self->b_picture_count) {
+    t = g_slice_new0 (MXFLocalTag);
+    memcpy (&t->key, &_b_picture_count_ul, 16);
+    t->size = 2;
+    t->data = g_slice_alloc (t->size);
+    t->g_slice = TRUE;
+    GST_WRITE_UINT16_BE (t->data, self->b_picture_count);
+    mxf_primer_pack_add_mapping (primer, 0, &t->key);
+    ret = g_list_prepend (ret, t);
+  }
+
+  if (self->bitrate) {
+    t = g_slice_new0 (MXFLocalTag);
+    memcpy (&t->key, &_bitrate_ul, 16);
+    t->size = 4;
+    t->data = g_slice_alloc (t->size);
+    t->g_slice = TRUE;
+    GST_WRITE_UINT32_BE (t->data, self->bitrate);
+    mxf_primer_pack_add_mapping (primer, 0, &t->key);
+    ret = g_list_prepend (ret, t);
+  }
+
+  if (self->profile_and_level) {
+    t = g_slice_new0 (MXFLocalTag);
+    memcpy (&t->key, &_profile_and_level_ul, 16);
+    t->size = 1;
+    t->data = g_slice_alloc (t->size);
+    t->g_slice = TRUE;
+    GST_WRITE_UINT8 (t->data, self->profile_and_level);
+    mxf_primer_pack_add_mapping (primer, 0, &t->key);
+    ret = g_list_prepend (ret, t);
+  }
+
+  return ret;
+}
+
 static void
 mxf_metadata_mpeg_video_descriptor_init (MXFMetadataMPEGVideoDescriptor * self)
 {
-
+  self->single_sequence = -1;
 }
 
 static void
@@ -261,6 +386,8 @@ static void
   metadata_base_class->name_quark = MXF_QUARK (MPEG_VIDEO_DESCRIPTOR);
   metadata_base_class->to_structure =
       mxf_metadata_mpeg_video_descriptor_to_structure;
+  metadata_base_class->write_tags =
+      mxf_metadata_mpeg_video_descriptor_write_tags;
 
   metadata_class->type = 0x0151;
 }
@@ -736,9 +863,265 @@ static const MXFEssenceElementHandler mxf_mpeg_essence_element_handler = {
   mxf_mpeg_create_caps
 };
 
+static GstFlowReturn
+mxf_mpeg_write_func (GstBuffer * buffer, GstCaps * caps, gpointer mapping_data,
+    GstAdapter * adapter, GstBuffer ** outbuf, gboolean flush)
+{
+  *outbuf = buffer;
+  return GST_FLOW_OK;
+}
+
+static const guint8 mpeg_essence_container_ul[] = {
+  0x06, 0x0e, 0x2b, 0x34, 0x04, 0x01, 0x01, 0x02,
+  0x0d, 0x01, 0x03, 0x01, 0x02, 0x04, 0x01, 0x01
+};
+
+static MXFMetadataFileDescriptor *
+mxf_mpeg_audio_get_descriptor (GstPadTemplate * tmpl, GstCaps * caps,
+    MXFEssenceElementWriteFunc * handler, gpointer * mapping_data)
+{
+  MXFMetadataGenericSoundEssenceDescriptor *ret;
+  GstStructure *s;
+
+  ret = (MXFMetadataGenericSoundEssenceDescriptor *)
+      gst_mini_object_new (MXF_TYPE_METADATA_GENERIC_SOUND_ESSENCE_DESCRIPTOR);
+
+  s = gst_caps_get_structure (caps, 0);
+  if (strcmp (gst_structure_get_name (s), "audio/mpeg") == 0) {
+    gint mpegversion;
+
+    if (!gst_structure_get_int (s, "mpegversion", &mpegversion))
+      return NULL;
+
+    if (mpegversion == 1) {
+      gint layer = 0;
+      gint mpegaudioversion = 0;
+
+      gst_structure_get_int (s, "layer", &layer);
+      gst_structure_get_int (s, "mpegaudioversion", &mpegaudioversion);
+
+      if (mpegaudioversion == 1 && layer == 1)
+        memcpy (&ret->sound_essence_compression,
+            &sound_essence_compression_mpeg1_layer1, 16);
+      else if (mpegaudioversion == 1 && layer == 2)
+        memcpy (&ret->sound_essence_compression,
+            &sound_essence_compression_mpeg1_layer12, 16);
+      else if (mpegaudioversion == 2 && layer == 1)
+        memcpy (&ret->sound_essence_compression,
+            &sound_essence_compression_mpeg2_layer1, 16);
+
+      /* Otherwise all 0x00, must be some kind of mpeg1 audio */
+    } else if (mpegversion == 2) {
+      memcpy (&ret->sound_essence_compression, &sound_essence_compression_aac,
+          16);
+    }
+  } else if (strcmp (gst_structure_get_name (s), "audio/x-ac3") == 0) {
+    memcpy (&ret->sound_essence_compression, &sound_essence_compression_ac3,
+        16);
+  } else {
+    g_assert_not_reached ();
+  }
+
+
+  memcpy (&ret->parent.essence_container, &mpeg_essence_container_ul, 16);
+
+  mxf_metadata_generic_sound_essence_descriptor_from_caps (ret, caps);
+  *handler = mxf_mpeg_write_func;
+
+  return (MXFMetadataFileDescriptor *) ret;
+}
+
+static void
+mxf_mpeg_audio_update_descriptor (MXFMetadataFileDescriptor * d, GstCaps * caps,
+    gpointer mapping_data, GstBuffer * buf)
+{
+  return;
+}
+
+static guint64
+gst_greatest_common_divisor (guint64 a, guint64 b)
+{
+  while (b != 0) {
+    guint temp = a;
+
+    a = b;
+    b = temp % b;
+  }
+
+  return a;
+}
+
+static void
+mxf_mpeg_audio_get_edit_rate (MXFMetadataFileDescriptor * a, GstCaps * caps,
+    gpointer mapping_data, GstBuffer * buf, MXFMetadataSourcePackage * package,
+    MXFMetadataTimelineTrack * track, MXFFraction * edit_rate)
+{
+  GstClockTime duration = GST_BUFFER_DURATION (buf);
+  guint64 gcd;
+
+  g_assert (duration != -1);
+
+  /* FIXME: Get the real edit rate for the format */
+  gcd = gst_greatest_common_divisor (GST_SECOND, duration);
+
+  edit_rate->n = GST_SECOND / gcd;
+  edit_rate->d = duration / gcd;
+}
+
+static guint32
+mxf_mpeg_audio_get_track_number_template (MXFMetadataFileDescriptor * a,
+    GstCaps * caps, gpointer mapping_data)
+{
+  return (0x16 << 24) | (0x05 << 8);
+}
+
+static MXFEssenceElementWriter mxf_mpeg_audio_essence_element_writer = {
+  mxf_mpeg_audio_get_descriptor,
+  mxf_mpeg_audio_update_descriptor,
+  mxf_mpeg_audio_get_edit_rate,
+  mxf_mpeg_audio_get_track_number_template,
+  NULL,
+  {{0,}}
+};
+
+#define MPEG_AUDIO_CAPS \
+      "audio/mpeg, " \
+      "mpegversion = (int) 1, " \
+      "layer = (int) [ 1, 3 ], " \
+      "rate = (int) [ 8000, 48000 ], " \
+      "channels = (int) [ 1, 2 ], " \
+      "parsed = (boolean) TRUE; " \
+      "audio/x-ac3, " \
+      "rate = (int) [ 4000, 96000 ], " \
+      "channels = (int) [ 1, 6 ]; " \
+      "audio/mpeg, " \
+      "mpegversion = (int) 2, " \
+      "rate = (int) [ 8000, 96000 ], " \
+      "channels = (int) [ 1, 8 ]"
+
+
+static const guint8 mpeg_video_picture_essence_compression_ul[] = {
+  0x06, 0x0e, 0x2b, 0x34, 0x04, 0x01, 0x01, 0x00,
+  0x04, 0x01, 0x02, 0x02, 0x01, 0x00, 0x00, 0x00
+};
+
+static MXFMetadataFileDescriptor *
+mxf_mpeg_video_get_descriptor (GstPadTemplate * tmpl, GstCaps * caps,
+    MXFEssenceElementWriteFunc * handler, gpointer * mapping_data)
+{
+  MXFMetadataMPEGVideoDescriptor *ret;
+  GstStructure *s;
+
+  ret = (MXFMetadataMPEGVideoDescriptor *)
+      gst_mini_object_new (MXF_TYPE_METADATA_MPEG_VIDEO_DESCRIPTOR);
+
+  s = gst_caps_get_structure (caps, 0);
+
+  memcpy (&ret->parent.parent.parent.essence_container,
+      &mpeg_essence_container_ul, 16);
+  memcpy (&ret->parent.parent.picture_essence_coding,
+      &mpeg_video_picture_essence_compression_ul, 16);
+  if (strcmp (gst_structure_get_name (s), "video/mpeg") == 0) {
+    gint mpegversion;
+
+    if (!gst_structure_get_int (s, "mpegversion", &mpegversion))
+      return NULL;
+
+    if (mpegversion == 1) {
+      ret->parent.parent.picture_essence_coding.u[13] = 0x10;
+    } else if (mpegversion == 2) {
+      ret->parent.parent.picture_essence_coding.u[13] = 0x01;
+    } else {
+      const GValue *v;
+      const GstBuffer *codec_data;
+
+      ret->parent.parent.picture_essence_coding.u[13] = 0x20;
+      if ((v = gst_structure_get_value (s, "codec_data"))) {
+        MXFLocalTag *t = g_slice_new0 (MXFLocalTag);
+        codec_data = gst_value_get_buffer (v);
+        t->size = GST_BUFFER_SIZE (codec_data);
+        t->data = g_memdup (GST_BUFFER_DATA (codec_data), t->size);
+        memcpy (&t->key, &sony_mpeg4_extradata, 16);
+        mxf_local_tag_insert (t, &MXF_METADATA_BASE (ret)->other_tags);
+      }
+    }
+  } else if (strcmp (gst_structure_get_name (s), "video/x-h264") == 0) {
+    ret->parent.parent.picture_essence_coding.u[13] = 0x30;
+  } else {
+    g_assert_not_reached ();
+  }
+
+
+  mxf_metadata_generic_picture_essence_descriptor_from_caps (&ret->
+      parent.parent, caps);
+  *handler = mxf_mpeg_write_func;
+
+  return (MXFMetadataFileDescriptor *) ret;
+}
+
+static void
+mxf_mpeg_video_update_descriptor (MXFMetadataFileDescriptor * d, GstCaps * caps,
+    gpointer mapping_data, GstBuffer * buf)
+{
+  return;
+}
+
+static void
+mxf_mpeg_video_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_mpeg_video_get_track_number_template (MXFMetadataFileDescriptor * a,
+    GstCaps * caps, gpointer mapping_data)
+{
+  return (0x15 << 24) | (0x05 << 8);
+}
+
+static MXFEssenceElementWriter mxf_mpeg_video_essence_element_writer = {
+  mxf_mpeg_video_get_descriptor,
+  mxf_mpeg_video_update_descriptor,
+  mxf_mpeg_video_get_edit_rate,
+  mxf_mpeg_video_get_track_number_template,
+  NULL,
+  {{0,}}
+};
+
+#define MPEG_VIDEO_CAPS \
+"video/mpeg, " \
+"mpegversion = (int) { 1, 2, 4 }, " \
+"systemstream = (boolean) FALSE, " \
+"width = " GST_VIDEO_SIZE_RANGE ", " \
+"height = " GST_VIDEO_SIZE_RANGE ", " \
+"framerate = " GST_VIDEO_FPS_RANGE "; " \
+"video/x-h264, " \
+"width = " GST_VIDEO_SIZE_RANGE ", " \
+"height = " GST_VIDEO_SIZE_RANGE ", " \
+"framerate = " GST_VIDEO_FPS_RANGE
+
 void
 mxf_mpeg_init (void)
 {
   mxf_metadata_register (MXF_TYPE_METADATA_MPEG_VIDEO_DESCRIPTOR);
   mxf_essence_element_handler_register (&mxf_mpeg_essence_element_handler);
+
+  mxf_mpeg_audio_essence_element_writer.pad_template =
+      gst_pad_template_new ("mpeg_audio_sink_%u", GST_PAD_SINK, GST_PAD_REQUEST,
+      gst_caps_from_string (MPEG_AUDIO_CAPS));
+  memcpy (&mxf_mpeg_audio_essence_element_writer.data_definition,
+      mxf_metadata_track_identifier_get (MXF_METADATA_TRACK_SOUND_ESSENCE), 16);
+  mxf_essence_element_writer_register (&mxf_mpeg_audio_essence_element_writer);
+
+  mxf_mpeg_video_essence_element_writer.pad_template =
+      gst_pad_template_new ("mpeg_video_sink_%u", GST_PAD_SINK, GST_PAD_REQUEST,
+      gst_caps_from_string (MPEG_VIDEO_CAPS));
+  memcpy (&mxf_mpeg_video_essence_element_writer.data_definition,
+      mxf_metadata_track_identifier_get (MXF_METADATA_TRACK_PICTURE_ESSENCE),
+      16);
+  mxf_essence_element_writer_register (&mxf_mpeg_video_essence_element_writer);
+
 }