/* GStreamer OGM parsing

 * Copyright (C) 2004 Ronald Bultje <rbultje@ronald.bitfreak.net>

 * Copyright (C) 2006 Tim-Philipp Müller <tim centricular net>

 *

 * 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.

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include <string.h>

#include <gst/gst.h>

#include <gst/tag/tag.h>

#include <gst/riff/riff-media.h>

#include <gst/riff/riff-read.h>

GST_DEBUG_CATEGORY_STATIC (gst_ogm_parse_debug);

#define GST_CAT_DEFAULT gst_ogm_parse_debug

#define GST_TYPE_OGM_VIDEO_PARSE (gst_ogm_video_parse_get_type())

#define GST_IS_OGM_VIDEO_PARSE(obj) \

  (G_TYPE_CHECK_INSTANCE_TYPE((obj), GST_TYPE_OGM_VIDEO_PARSE))

#define GST_TYPE_OGM_AUDIO_PARSE (gst_ogm_audio_parse_get_type())

#define GST_IS_OGM_AUDIO_PARSE(obj) \

  (G_TYPE_CHECK_INSTANCE_TYPE((obj), GST_TYPE_OGM_AUDIO_PARSE))

#define GST_TYPE_OGM_TEXT_PARSE (gst_ogm_text_parse_get_type())

#define GST_IS_OGM_TEXT_PARSE(obj) \

  (G_TYPE_CHECK_INSTANCE_TYPE((obj), GST_TYPE_OGM_TEXT_PARSE))

#define GST_TYPE_OGM_PARSE (gst_ogm_parse_get_type())

#define GST_OGM_PARSE(obj) \

  (G_TYPE_CHECK_INSTANCE_CAST((obj), GST_TYPE_OGM_PARSE, GstOgmParse))

#define GST_OGM_PARSE_CLASS(klass) \

  (G_TYPE_CHECK_CLASS_CAST((klass), GST_TYPE_OGM_PARSE, GstOgmParse))

#define GST_IS_OGM_PARSE(obj) \

  (G_TYPE_CHECK_INSTANCE_TYPE((obj), GST_TYPE_OGM_PARSE))

#define GST_IS_OGM_PARSE_CLASS(klass) \

  (G_TYPE_CHECK_CLASS_TYPE((klass), GST_TYPE_OGM_PARSE))

#define GST_OGM_PARSE_GET_CLASS(obj) \

  (G_TYPE_INSTANCE_GET_CLASS ((obj), GST_TYPE_OGM_PARSE, GstOgmParseClass))

static const GstElementDetails gst_ogm_audio_parse_details =

GST_ELEMENT_DETAILS ("OGM audio stream parser",

    "Codec/Decoder/Audio",

    "parse an OGM audio header and stream",

    "Ronald Bultje <rbultje@ronald.bitfreak.net>");

static const GstElementDetails gst_ogm_video_parse_details =

GST_ELEMENT_DETAILS ("OGM video stream parser",

    "Codec/Decoder/Video",

    "parse an OGM video header and stream",

    "Ronald Bultje <rbultje@ronald.bitfreak.net>");

static const GstElementDetails gst_ogm_text_parse_details =

GST_ELEMENT_DETAILS ("OGM text stream parser",

    "Codec/Decoder/Subtitle",

    "parse an OGM text header and stream",

    "Ronald Bultje <rbultje@ronald.bitfreak.net>");

typedef struct _stream_header_video

{

  gint32 width;

  gint32 height;

} stream_header_video;

typedef struct _stream_header_audio

{

  gint16 channels;

  gint16 blockalign;

  gint32 avgbytespersec;

} stream_header_audio;

/* sizeof(stream_header) might differ due to structure packing and

 * alignment differences on some architectures, so not using that */

#define OGM_STREAM_HEADER_SIZE (8+4+4+8+8+4+4+4+8)

typedef struct _stream_header

{

  gchar streamtype[8];

  gchar subtype[4 + 1];

  /* size of the structure */

  gint32 size;

  /* in reference time */

  gint64 time_unit;

  gint64 samples_per_unit;

  /* in media time */

  gint32 default_len;

  gint32 buffersize;

  gint32 bits_per_sample;

  union

  {

    stream_header_video video;

    stream_header_audio audio;

    /* text has no additional data */

  } s;

} stream_header;

typedef struct _GstOgmParse

{

  GstElement element;

  /* pads */

  GstPad *srcpad, *sinkpad;

  GstPadTemplate *srcpadtempl;

  /* we need to cache events that we receive before creating the source pad */

  GList *cached_events;

  /* audio or video */

  stream_header hdr;

  /* expected next granulepos (used for timestamp guessing) */

  guint64 next_granulepos;

} GstOgmParse;

typedef struct _GstOgmParseClass

{

  GstElementClass parent_class;

} GstOgmParseClass;

static GstStaticPadTemplate sink_factory_video =

GST_STATIC_PAD_TEMPLATE ("sink", GST_PAD_SINK, GST_PAD_ALWAYS,

    GST_STATIC_CAPS ("application/x-ogm-video"));

static GstStaticPadTemplate sink_factory_audio =

GST_STATIC_PAD_TEMPLATE ("sink", GST_PAD_SINK, GST_PAD_ALWAYS,

    GST_STATIC_CAPS ("application/x-ogm-audio"));

static GstStaticPadTemplate sink_factory_text =

GST_STATIC_PAD_TEMPLATE ("sink", GST_PAD_SINK, GST_PAD_ALWAYS,

    GST_STATIC_CAPS ("application/x-ogm-text"));

static GstPadTemplate *video_src_templ, *audio_src_templ, *text_src_templ;

static GType gst_ogm_audio_parse_get_type (void);

static GType gst_ogm_video_parse_get_type (void);

static GType gst_ogm_text_parse_get_type (void);

static GType gst_ogm_parse_get_type (void);

static void gst_ogm_audio_parse_base_init (GstOgmParseClass * klass);

static void gst_ogm_video_parse_base_init (GstOgmParseClass * klass);

static void gst_ogm_text_parse_base_init (GstOgmParseClass * klass);

static void gst_ogm_parse_class_init (GstOgmParseClass * klass);

static void gst_ogm_parse_init (GstOgmParse * ogm);

static void gst_ogm_video_parse_init (GstOgmParse * ogm);

static void gst_ogm_audio_parse_init (GstOgmParse * ogm);

static void gst_ogm_text_parse_init (GstOgmParse * ogm);

static const GstQueryType *gst_ogm_parse_get_sink_querytypes (GstPad * pad);

static gboolean gst_ogm_parse_sink_event (GstPad * pad, GstEvent * event);

static gboolean gst_ogm_parse_sink_query (GstPad * pad, GstQuery * query);

static gboolean gst_ogm_parse_sink_convert (GstPad * pad, GstFormat src_format,

    gint64 src_value, GstFormat * dest_format, gint64 * dest_value);

static GstFlowReturn gst_ogm_parse_chain (GstPad * pad, GstBuffer * buffer);

static GstStateChangeReturn gst_ogm_parse_change_state (GstElement * element,

    GstStateChange transition);

static GstElementClass *parent_class = NULL;

static GType

gst_ogm_parse_get_type (void)

{

  static GType ogm_parse_type = 0;

  if (!ogm_parse_type) {

    static const GTypeInfo ogm_parse_info = {

      sizeof (GstOgmParseClass),

      NULL,

      NULL,

      (GClassInitFunc) gst_ogm_parse_class_init,

      NULL,

      NULL,

      sizeof (GstOgmParse),

      0,

      (GInstanceInitFunc) gst_ogm_parse_init,

    };

    ogm_parse_type =

        g_type_register_static (GST_TYPE_ELEMENT,

        "GstOgmParse", &ogm_parse_info, 0);

  }

  return ogm_parse_type;

}

static GType

gst_ogm_audio_parse_get_type (void)

{

  static GType ogm_audio_parse_type = 0;

  if (!ogm_audio_parse_type) {

    static const GTypeInfo ogm_audio_parse_info = {

      sizeof (GstOgmParseClass),

      (GBaseInitFunc) gst_ogm_audio_parse_base_init,

      NULL,

      NULL,

      NULL,

      NULL,

      sizeof (GstOgmParse),

      0,

      (GInstanceInitFunc) gst_ogm_audio_parse_init,

    };

    ogm_audio_parse_type =

        g_type_register_static (GST_TYPE_OGM_PARSE,

        "GstOgmAudioParse", &ogm_audio_parse_info, 0);

  }

  return ogm_audio_parse_type;

}

static GType

gst_ogm_video_parse_get_type (void)

{

  static GType ogm_video_parse_type = 0;

  if (!ogm_video_parse_type) {

    static const GTypeInfo ogm_video_parse_info = {

      sizeof (GstOgmParseClass),

      (GBaseInitFunc) gst_ogm_video_parse_base_init,

      NULL,

      NULL,

      NULL,

      NULL,

      sizeof (GstOgmParse),

      0,

      (GInstanceInitFunc) gst_ogm_video_parse_init,

    };

    ogm_video_parse_type =

        g_type_register_static (GST_TYPE_OGM_PARSE,

        "GstOgmVideoParse", &ogm_video_parse_info, 0);

  }

  return ogm_video_parse_type;

}

static GType

gst_ogm_text_parse_get_type (void)

{

  static GType ogm_text_parse_type = 0;

  if (!ogm_text_parse_type) {

    static const GTypeInfo ogm_text_parse_info = {

      sizeof (GstOgmParseClass),

      (GBaseInitFunc) gst_ogm_text_parse_base_init,

      NULL,

      NULL,

      NULL,

      NULL,

      sizeof (GstOgmParse),

      0,

      (GInstanceInitFunc) gst_ogm_text_parse_init,

    };

    ogm_text_parse_type =

        g_type_register_static (GST_TYPE_OGM_PARSE,

        "GstOgmTextParse", &ogm_text_parse_info, 0);

  }

  return ogm_text_parse_type;

}

static void

gst_ogm_audio_parse_base_init (GstOgmParseClass * klass)

{

  GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

  GstCaps *caps = gst_riff_create_audio_template_caps ();

  gst_element_class_set_details (element_class, &gst_ogm_audio_parse_details);

  gst_element_class_add_pad_template (element_class,

      gst_static_pad_template_get (&sink_factory_audio));

  audio_src_templ = gst_pad_template_new ("src",

      GST_PAD_SRC, GST_PAD_SOMETIMES, caps);

  gst_element_class_add_pad_template (element_class, audio_src_templ);

}

static void

gst_ogm_video_parse_base_init (GstOgmParseClass * klass)

{

  GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

  GstCaps *caps = gst_riff_create_video_template_caps ();

  gst_element_class_set_details (element_class, &gst_ogm_video_parse_details);

  gst_element_class_add_pad_template (element_class,

      gst_static_pad_template_get (&sink_factory_video));

  video_src_templ = gst_pad_template_new ("src",

      GST_PAD_SRC, GST_PAD_SOMETIMES, caps);

  gst_element_class_add_pad_template (element_class, video_src_templ);

}

static void

gst_ogm_text_parse_base_init (GstOgmParseClass * klass)

{

  GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

  GstCaps *caps = gst_caps_new_simple ("text/plain", NULL, NULL);

  gst_element_class_set_details (element_class, &gst_ogm_text_parse_details);

  gst_element_class_add_pad_template (element_class,

      gst_static_pad_template_get (&sink_factory_text));

  text_src_templ = gst_pad_template_new ("src",

      GST_PAD_SRC, GST_PAD_SOMETIMES, caps);

  gst_element_class_add_pad_template (element_class, text_src_templ);

}

static void

gst_ogm_parse_class_init (GstOgmParseClass * klass)

{

  GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);

  parent_class = g_type_class_peek_parent (klass);

  gstelement_class->change_state =

      GST_DEBUG_FUNCPTR (gst_ogm_parse_change_state);

}

static void

gst_ogm_parse_init (GstOgmParse * ogm)

{

  memset (&ogm->hdr, 0, sizeof (ogm->hdr));

  ogm->next_granulepos = 0;

  ogm->srcpad = NULL;

  ogm->cached_events = NULL;

}

static void

gst_ogm_audio_parse_init (GstOgmParse * ogm)

{

  ogm->sinkpad = gst_pad_new_from_static_template (&sink_factory_audio, "sink");

  gst_pad_set_query_function (ogm->sinkpad,

      GST_DEBUG_FUNCPTR (gst_ogm_parse_sink_query));

  gst_pad_set_chain_function (ogm->sinkpad,

      GST_DEBUG_FUNCPTR (gst_ogm_parse_chain));

  gst_pad_set_event_function (ogm->sinkpad,

      GST_DEBUG_FUNCPTR (gst_ogm_parse_sink_event));

  gst_element_add_pad (GST_ELEMENT (ogm), ogm->sinkpad);

  ogm->srcpad = NULL;

  ogm->srcpadtempl = audio_src_templ;

}

static void

gst_ogm_video_parse_init (GstOgmParse * ogm)

{

  ogm->sinkpad = gst_pad_new_from_static_template (&sink_factory_video, "sink");

  gst_pad_set_query_function (ogm->sinkpad,

      GST_DEBUG_FUNCPTR (gst_ogm_parse_sink_query));

  gst_pad_set_chain_function (ogm->sinkpad,

      GST_DEBUG_FUNCPTR (gst_ogm_parse_chain));

  gst_pad_set_event_function (ogm->sinkpad,

      GST_DEBUG_FUNCPTR (gst_ogm_parse_sink_event));

  gst_element_add_pad (GST_ELEMENT (ogm), ogm->sinkpad);

  ogm->srcpad = NULL;

  ogm->srcpadtempl = video_src_templ;

}

static void

gst_ogm_text_parse_init (GstOgmParse * ogm)

{

  ogm->sinkpad = gst_pad_new_from_static_template (&sink_factory_text, "sink");

  gst_pad_set_query_type_function (ogm->sinkpad,

      gst_ogm_parse_get_sink_querytypes);

  gst_pad_set_query_function (ogm->sinkpad,

      GST_DEBUG_FUNCPTR (gst_ogm_parse_sink_query));

  gst_pad_set_chain_function (ogm->sinkpad,

      GST_DEBUG_FUNCPTR (gst_ogm_parse_chain));

  gst_pad_set_event_function (ogm->sinkpad,

      GST_DEBUG_FUNCPTR (gst_ogm_parse_sink_event));

  gst_element_add_pad (GST_ELEMENT (ogm), ogm->sinkpad);

  ogm->srcpad = NULL;

  ogm->srcpadtempl = text_src_templ;

}

static const GstQueryType *

gst_ogm_parse_get_sink_querytypes (GstPad * pad)

{

  static const GstQueryType types[] = {

    GST_QUERY_POSITION,

    0

  };

  return types;

}

static gboolean

gst_ogm_parse_sink_convert (GstPad * pad,

    GstFormat src_format, gint64 src_value,

    GstFormat * dest_format, gint64 * dest_value)

{

  gboolean res = FALSE;

  GstOgmParse *ogm = GST_OGM_PARSE (gst_pad_get_parent (pad));

  switch (src_format) {

    case GST_FORMAT_DEFAULT:

      switch (*dest_format) {

        case GST_FORMAT_TIME:

          switch (ogm->hdr.streamtype[0]) {

            case 'a':

              *dest_value = GST_SECOND * src_value / ogm->hdr.samples_per_unit;

              res = TRUE;

              break;

            case 'v':

            case 't':

              *dest_value = (GST_SECOND / 10000000) *

                  ogm->hdr.time_unit * src_value;

              res = TRUE;

              break;

            default:

              break;

          }

          break;

        default:

          break;

      }

      break;

    case GST_FORMAT_TIME:

      switch (*dest_format) {

        case GST_FORMAT_DEFAULT:

          switch (ogm->hdr.streamtype[0]) {

            case 'a':

              *dest_value = ogm->hdr.samples_per_unit * src_value / GST_SECOND;

              res = TRUE;

              break;

            case 'v':

            case 't':

              *dest_value = src_value /

                  ((GST_SECOND / 10000000) * ogm->hdr.time_unit);

              res = TRUE;

              break;

            default:

              break;

          }

          break;

        default:

          break;

      }

      break;

    default:

      break;

  }

  gst_object_unref (ogm);

  return res;

}

static gboolean

gst_ogm_parse_sink_query (GstPad * pad, GstQuery * query)

{

  GstOgmParse *ogm = GST_OGM_PARSE (gst_pad_get_parent (pad));

  GstFormat format;

  gboolean res = FALSE;

  switch (GST_QUERY_TYPE (query)) {

    case GST_QUERY_POSITION:

    {

      gint64 val;

      gst_query_parse_position (query, &format, NULL);

      if (format != GST_FORMAT_DEFAULT && format != GST_FORMAT_TIME)

        break;

      if ((res = gst_ogm_parse_sink_convert (pad,

                  GST_FORMAT_DEFAULT, ogm->next_granulepos, &format, &val))) {

        /* don't know the total length here.. */

        gst_query_set_position (query, format, val);

      }

      break;

    }

    case GST_QUERY_CONVERT:

    {

      GstFormat src_fmt, dest_fmt;

      gint64 src_val, dest_val;

      /* peel off input */

      gst_query_parse_convert (query, &src_fmt, &src_val, &dest_fmt, &dest_val);

      if ((res = gst_ogm_parse_sink_convert (pad, src_fmt, src_val,

                  &dest_fmt, &dest_val))) {

        gst_query_set_convert (query, src_fmt, src_val, dest_fmt, dest_val);

      }

      break;

    }

    default:

      res = gst_pad_query_default (pad, query);

      break;

  }

  gst_object_unref (ogm);

  return res;

}

static GstFlowReturn

gst_ogm_parse_stream_header (GstOgmParse * ogm, const guint8 * data, guint size)

{

  GstCaps *caps = NULL;

  /* stream header */

  if (size < OGM_STREAM_HEADER_SIZE)

    goto buffer_too_small;

  if (!memcmp (data, "video\000\000\000", 8)) {