gstreamer/gst/avi/gstavidecoder.c

/* GStreamer
 * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
 *
 * 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.
 */


/*#define GST_DEBUG_ENABLED */
#include <string.h>

#include "gstavidecoder.h"



/* elementfactory information */
static GstElementDetails gst_avi_decoder_details = {
  ".avi decoder",
  "Decoder/Video",
  "Decodes a .avi file into audio and video",
  VERSION,
  "Erik Walthinsen <omega@cse.ogi.edu>\n" "Wim Taymans <wim.taymans@tvd.be>",
  "(C) 1999",
};

static GstCaps *avi_typefind (GstBuffer * buf, gpointer private);

/* typefactory for 'avi' */
static GstTypeDefinition avidefinition = {
  "avidecoder_video/avi",
  "video/avi",
  ".avi",
  avi_typefind,
};

/* AviDecoder signals and args */
enum
{
  /* FILL ME */
  LAST_SIGNAL
};

enum
{
  ARG_0,
  ARG_BITRATE,
  ARG_MEDIA_TIME,
  ARG_CURRENT_TIME,
  /* FILL ME */
};

GST_PADTEMPLATE_FACTORY (sink_templ,
    "sink",
    GST_PAD_SINK,
    GST_PAD_ALWAYS,
    GST_CAPS_NEW ("avidecoder_sink",
	"video/avi", "RIFF", GST_PROPS_STRING ("AVI")
    )
    )

    GST_PADTEMPLATE_FACTORY (src_video_templ,
    "video_src",
    GST_PAD_SRC,
    GST_PAD_ALWAYS,
    GST_CAPS_NEW ("wincodec_src",
	"video/raw",
	"format", GST_PROPS_LIST (GST_PROPS_FOURCC (GST_MAKE_FOURCC ('Y', 'U', 'Y',
		    '2')), GST_PROPS_FOURCC (GST_MAKE_FOURCC ('I', '4', '2', '0')),
	    GST_PROPS_FOURCC (GST_MAKE_FOURCC ('R', 'G', 'B', ' '))
	), "width", GST_PROPS_INT_RANGE (16, 4096), "height",
	GST_PROPS_INT_RANGE (16, 4096)
    )
    )

    GST_PADTEMPLATE_FACTORY (src_audio_templ,
    "audio_src",
    GST_PAD_SRC,
    GST_PAD_ALWAYS,
    GST_CAPS_NEW ("src_audio",
	"audio/raw",
	"format", GST_PROPS_STRING ("int"),
	"law", GST_PROPS_INT (0),
	"endianness", GST_PROPS_INT (G_BYTE_ORDER),
	"signed", GST_PROPS_LIST (GST_PROPS_BOOLEAN (TRUE),
	    GST_PROPS_BOOLEAN (FALSE)
	), "width", GST_PROPS_LIST (GST_PROPS_INT (8), GST_PROPS_INT (16)
	), "depth", GST_PROPS_LIST (GST_PROPS_INT (8), GST_PROPS_INT (16)
	),
	"rate", GST_PROPS_INT_RANGE (11025, 48000),
	"channels", GST_PROPS_INT_RANGE (1, 2)
    )
    )

     static void gst_avi_decoder_class_init (GstAviDecoderClass * klass);
     static void gst_avi_decoder_init (GstAviDecoder * avi_decoder);

     static void gst_avi_decoder_get_property (GObject * object, guint prop_id,
    GValue * value, GParamSpec * pspec);



     static GstElementClass *parent_class = NULL;

/*static guint gst_avi_decoder_signals[LAST_SIGNAL] = { 0 }; */

GType
gst_avi_decoder_get_type (void)
{
  static GType avi_decoder_type = 0;

  if (!avi_decoder_type) {
    static const GTypeInfo avi_decoder_info = {
      sizeof (GstAviDecoderClass),
      NULL,
      NULL,
      (GClassInitFunc) gst_avi_decoder_class_init,
      NULL,
      NULL,
      sizeof (GstAviDecoder),
      0,
      (GInstanceInitFunc) gst_avi_decoder_init,
    };
    avi_decoder_type =
	g_type_register_static (GST_TYPE_BIN, "GstAviDecoder",
	&avi_decoder_info, 0);
  }
  return avi_decoder_type;
}

static void
gst_avi_decoder_class_init (GstAviDecoderClass * klass)
{
  GObjectClass *gobject_class;
  GstElementClass *gstelement_class;

  gobject_class = (GObjectClass *) klass;
  gstelement_class = (GstElementClass *) klass;

  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_BITRATE, g_param_spec_long ("bitrate", "bitrate", "bitrate", G_MINLONG, G_MAXLONG, 0, G_PARAM_READABLE));	/* CHECKME */
  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_MEDIA_TIME, g_param_spec_long ("media_time", "media_time", "media_time", G_MINLONG, G_MAXLONG, 0, G_PARAM_READABLE));	/* CHECKME */
  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_CURRENT_TIME, g_param_spec_long ("current_time", "current_time", "current_time", G_MINLONG, G_MAXLONG, 0, G_PARAM_READABLE));	/* CHECKME */

  parent_class = g_type_class_ref (GST_TYPE_BIN);

  gobject_class->get_property = gst_avi_decoder_get_property;
}

static void
gst_avi_decoder_new_pad (GstElement * element, GstPad * pad,
    GstAviDecoder * avi_decoder)
{
  GstCaps *caps;
  GstCaps *targetcaps = NULL;
  const gchar *format;
  gboolean type_found;
  GstElement *type;
  GstElement *new_element = NULL;
  gchar *padname = NULL;
  gchar *gpadname = NULL;

#define AVI_TYPE_VIDEO  1
#define AVI_TYPE_AUDIO  2
  gint media_type = 0;

  GST_DEBUG (0, "avidecoder: new pad for element \"%s\"",
      gst_element_get_name (element));

  caps = gst_pad_get_caps (pad);
  format = gst_caps_get_string (caps, "format");

  if (!strcmp (format, "strf_vids")) {
    targetcaps =
	gst_padtemplate_get_caps (GST_PADTEMPLATE_GET (src_video_templ));
    media_type = AVI_TYPE_VIDEO;
    gpadname = g_strdup_printf ("video_%02d", avi_decoder->video_count++);
  } else if (!strcmp (format, "strf_auds")) {
    targetcaps =
	gst_padtemplate_get_caps (GST_PADTEMPLATE_GET (src_audio_templ));
    media_type = AVI_TYPE_AUDIO;
    gpadname = g_strdup_printf ("audio_%02d", avi_decoder->audio_count++);
  } else if (!strcmp (format, "strf_iavs")) {
    targetcaps =
	gst_padtemplate_get_caps (GST_PADTEMPLATE_GET (src_video_templ));
    media_type = AVI_TYPE_VIDEO;
    gpadname = g_strdup_printf ("video_%02d", avi_decoder->video_count++);
  } else {
    g_assert_not_reached ();
  }

  gst_element_set_state (GST_ELEMENT (avi_decoder), GST_STATE_PAUSED);

  type = gst_elementfactory_make ("avitypes",
      g_strdup_printf ("typeconvert%d", avi_decoder->count));

  /* brin the element to the READY state so it can do our caps negotiation */
  gst_element_set_state (type, GST_STATE_READY);

  gst_pad_connect (pad, gst_element_get_pad (type, "sink"));
  type_found = gst_util_get_bool_arg (G_OBJECT (type), "type_found");

  if (type_found) {

    gst_bin_add (GST_BIN (avi_decoder), type);

    pad = gst_element_get_pad (type, "src");
    caps = gst_pad_get_caps (pad);

    if (gst_caps_is_always_compatible (caps, targetcaps)) {
      gst_element_add_ghost_pad (GST_ELEMENT (avi_decoder),
	  gst_element_get_pad (type, "src"), gpadname);

      avi_decoder->count++;
      goto done;
    }
#ifndef GST_DISABLE_AUTOPLUG
    else {
      GstAutoplug *autoplug;

      autoplug = gst_autoplugfactory_make ("static");

      new_element = gst_autoplug_to_caps (autoplug, caps, targetcaps, NULL);

      padname = "src_00";
    }
#endif /* GST_DISABLE_AUTOPLUG */
  }

  if (!new_element && (media_type == AVI_TYPE_VIDEO)) {
    padname = "src";
  } else if (!new_element && (media_type == AVI_TYPE_AUDIO)) {
    /*FIXME */
    padname = "src";
  }

  if (new_element) {
    gst_pad_connect (pad, gst_element_get_pad (new_element, "sink"));
    gst_element_set_name (new_element, g_strdup_printf ("element%d",
	    avi_decoder->count));
    gst_bin_add (GST_BIN (avi_decoder), new_element);

    gst_element_add_ghost_pad (GST_ELEMENT (avi_decoder),
	gst_element_get_pad (new_element, padname), gpadname);

    avi_decoder->count++;
  } else {
    g_warning ("avidecoder: could not autoplug\n");
  }

done:
  gst_element_set_state (GST_ELEMENT (avi_decoder), GST_STATE_PLAYING);
}

static void
gst_avi_decoder_init (GstAviDecoder * avi_decoder)
{
  avi_decoder->demuxer = gst_elementfactory_make ("avidemux", "demux");

  if (avi_decoder->demuxer) {
    gst_bin_add (GST_BIN (avi_decoder), avi_decoder->demuxer);

    gst_element_add_ghost_pad (GST_ELEMENT (avi_decoder),
	gst_element_get_pad (avi_decoder->demuxer, "sink"), "sink");

    g_signal_connect (G_OBJECT (avi_decoder->demuxer), "new_pad",
	G_CALLBACK (gst_avi_decoder_new_pad), avi_decoder);
  } else {
    g_warning ("wow!, no avi demuxer found. help me\n");
  }

  avi_decoder->count = 0;
  avi_decoder->audio_count = 0;
  avi_decoder->video_count = 0;
}

static GstCaps *
avi_typefind (GstBuffer * buf, gpointer private)
{
  gchar *data = GST_BUFFER_DATA (buf);
  GstCaps *new;

  GST_DEBUG (0, "avi_decoder: typefind");
  if (strncmp (&data[0], "RIFF", 4))
    return NULL;
  if (strncmp (&data[8], "AVI ", 4))
    return NULL;

  new = GST_CAPS_NEW ("avi_typefind",
      "video/avi", "RIFF", GST_PROPS_STRING ("AVI"));

  return new;
}

static void
gst_avi_decoder_get_property (GObject * object, guint prop_id, GValue * value,
    GParamSpec * pspec)
{
  GstAviDecoder *src;

  g_return_if_fail (GST_IS_AVI_DECODER (object));

  src = GST_AVI_DECODER (object);

  switch (prop_id) {
    case ARG_BITRATE:
      break;
    case ARG_MEDIA_TIME:
      g_value_set_long (value, gst_util_get_long_arg (G_OBJECT (src->demuxer),
	      "media_time"));
      break;
    case ARG_CURRENT_TIME:
      g_value_set_long (value, gst_util_get_long_arg (G_OBJECT (src->demuxer),
	      "current_time"));
      break;
    default:
      break;
  }
}


static gboolean
plugin_init (GModule * module, GstPlugin * plugin)
{
  GstElementFactory *factory;
  GstTypeFactory *type;

  /* create an elementfactory for the avi_decoder element */
  factory = gst_elementfactory_new ("avidecoder", GST_TYPE_AVI_DECODER,
      &gst_avi_decoder_details);
  g_return_val_if_fail (factory != NULL, FALSE);

  gst_elementfactory_add_padtemplate (factory,
      GST_PADTEMPLATE_GET (src_audio_templ));
  gst_elementfactory_add_padtemplate (factory,
      GST_PADTEMPLATE_GET (src_video_templ));
  gst_elementfactory_add_padtemplate (factory,
      GST_PADTEMPLATE_GET (sink_templ));

  type = gst_typefactory_new (&avidefinition);
  gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (type));

  gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));

  return TRUE;
}

GstPluginDesc plugin_desc = {
  GST_VERSION_MAJOR,
  GST_VERSION_MINOR,
  "avidecoder",
  plugin_init
};