gstreamer/gst/smpte/gstsmpte.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.
 */

#include <string.h>
#include <gstsmpte.h>
#include "paint.h"

/* elementfactory information */
static GstElementDetails smpte_details = {
  "SMPTE transitions",
  "Filter/Video",
  "LGPL",
  "Apply the standard SMPTE transitions on video images",
  VERSION,
  "Wim Taymans <wim.taymans@chello.be>",
  "(C) 2002",
};

GST_PAD_TEMPLATE_FACTORY (smpte_src_factory,
  "src",
  GST_PAD_SRC,
  GST_PAD_ALWAYS,
  GST_CAPS_NEW (
   "smpte_src",
   "video/raw",
     "format",   GST_PROPS_FOURCC (GST_STR_FOURCC ("I420"))
  )
)

GST_PAD_TEMPLATE_FACTORY (smpte_sink1_factory,
  "sink1",
  GST_PAD_SINK,
  GST_PAD_ALWAYS,
  GST_CAPS_NEW (
   "smpte_sink1",
   "video/raw",
     "format",   GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")),
     "width",	 GST_PROPS_INT_RANGE (0, 4096),
     "height",	 GST_PROPS_INT_RANGE (0, 4096) 
  )
)

GST_PAD_TEMPLATE_FACTORY (smpte_sink2_factory,
  "sink2",
  GST_PAD_SINK,
  GST_PAD_ALWAYS,
  GST_CAPS_NEW (
   "smpte_sink2",
   "video/raw",
     "format",   GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")),
     "width",	 GST_PROPS_INT_RANGE (0, 4096),
     "height",	 GST_PROPS_INT_RANGE (0, 4096)
  )
)


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

enum {
  ARG_0,
  ARG_TYPE,
};

#define GST_TYPE_SMPTE_TRANSITION_TYPE (gst_smpte_transition_type_get_type())
static GType
gst_smpte_transition_type_get_type (void) 
{
  static GType smpte_transition_type = 0;
  GEnumValue *smpte_transitions;

  if (!smpte_transition_type) {
    const GList *definitions;
    gint i=0;

    definitions = gst_mask_get_definitions ();
    smpte_transitions = g_new0 (GEnumValue, g_list_length ((GList *)definitions)+1);

    while (definitions) {
      GstMaskDefinition *definition = (GstMaskDefinition *) definitions->data;
      definitions = g_list_next (definitions);

      smpte_transitions[i].value = definition->type;
      smpte_transitions[i].value_name = definition->short_name;
      smpte_transitions[i].value_nick = definition->long_name;
      
      i++;
    }

    smpte_transition_type = g_enum_register_static ("GstSMPTETransitionType", smpte_transitions);
  }
  return smpte_transition_type;
}   


static void	gst_smpte_class_init		(GstSMPTEClass *klass);
static void	gst_smpte_init			(GstSMPTE *smpte);

static void	gst_smpte_loop			(GstElement *element);

static void	gst_smpte_set_property		(GObject *object, guint prop_id, 
						 const GValue *value, GParamSpec *pspec);
static void	gst_smpte_get_property		(GObject *object, guint prop_id, 
						 GValue *value, GParamSpec *pspec);

static GstElementClass *parent_class = NULL;
/*static guint gst_smpte_signals[LAST_SIGNAL] = { 0 }; */

static GType
gst_smpte_get_type (void)
{
  static GType smpte_type = 0;

  if (!smpte_type) {
    static const GTypeInfo smpte_info = {
      sizeof(GstSMPTEClass),      
      NULL,
      NULL,
      (GClassInitFunc)gst_smpte_class_init,
      NULL,
      NULL,
      sizeof(GstSMPTE),
      0,
      (GInstanceInitFunc)gst_smpte_init,
    };
    smpte_type = g_type_register_static(GST_TYPE_ELEMENT, "GstSMPTE", &smpte_info, 0);
  }
  return smpte_type;
}

static void
gst_smpte_class_init (GstSMPTEClass *klass)
{
  GObjectClass *gobject_class;
  GstElementClass *gstelement_class;

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

  parent_class = g_type_class_ref(GST_TYPE_ELEMENT);

  gobject_class->set_property = gst_smpte_set_property;
  gobject_class->get_property = gst_smpte_get_property;

  _gst_mask_init ();

  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_TYPE,
    g_param_spec_enum ("type", "Type", "The type of transition to use",
                       GST_TYPE_SMPTE_TRANSITION_TYPE, 1, G_PARAM_READWRITE));

}

/*                        wht  yel  cya  grn  mag  red  blu  blk   -I    Q */
static int y_colors[] = { 255, 226, 179, 150, 105,  76,  29,  16,  16,   0 };
static int u_colors[] = { 128,   0, 170,  46, 212,  85, 255, 128,   0, 128 };
static int v_colors[] = { 128, 155,   0,  21, 235, 255, 107, 128, 128, 255 };

static void
fill_i420 (guint8 *data, gint width, gint height, gint color)
{
  guint8 *yp = data;
  guint8 *up = data + width * height;
  guint8 *vp = data + width * height + (width * height) / 4;
  
  gst_smpte_paint_rect (yp, width,   0, 0, width,   height,   y_colors[color]);
  gst_smpte_paint_rect (up, width/2, 0, 0, width/2, height/2, u_colors[color]);
  gst_smpte_paint_rect (vp, width/2, 0, 0, width/2, height/2, v_colors[color]);
}

static gboolean
gst_smpte_sinkconnect (GstPad *pad, GstCaps *caps)
{
  GstSMPTE *smpte;

  smpte = GST_SMPTE (gst_pad_get_parent (pad));

  if (!GST_CAPS_IS_FIXED (caps))
    return GST_PAD_CONNECT_DELAYED;

  gst_caps_get_int (caps, "width", &smpte->width);
  gst_caps_get_int (caps, "height", &smpte->height);

  smpte->mask = gst_mask_factory_new (smpte->type, 8, smpte->width, smpte->height);

  /* forward to the next plugin */
  return gst_pad_try_set_caps(smpte->srcpad, gst_caps_copy_1(caps));
}

static void 
gst_smpte_init (GstSMPTE *smpte)
{
  smpte->sinkpad1 = gst_pad_new_from_template (
		  GST_PAD_TEMPLATE_GET (smpte_sink1_factory), "sink1");
  gst_pad_set_connect_function (smpte->sinkpad1, gst_smpte_sinkconnect);
  gst_element_add_pad (GST_ELEMENT (smpte), smpte->sinkpad1);

  smpte->sinkpad2 = gst_pad_new_from_template (
		  GST_PAD_TEMPLATE_GET (smpte_sink2_factory), "sink2");
  gst_pad_set_connect_function (smpte->sinkpad2, gst_smpte_sinkconnect);
  gst_element_add_pad (GST_ELEMENT (smpte), smpte->sinkpad2);

  smpte->srcpad = gst_pad_new_from_template (
		  GST_PAD_TEMPLATE_GET (smpte_src_factory), "src");
  gst_element_add_pad (GST_ELEMENT (smpte), smpte->srcpad);

  gst_element_set_loop_function (GST_ELEMENT (smpte), gst_smpte_loop);

  smpte->width = 320;
  smpte->height = 200;
  smpte->duration = 64;
  smpte->position = 0;
  smpte->type = 1;
  smpte->mask = gst_mask_factory_new (smpte->type, 8, smpte->width, smpte->height);
}

static void
gst_smpte_blend_i420 (guint8 *in1, guint8 *in2, guint8 *out, GstMask *mask,
		      gint width, gint height)
{
  gint i, j;
  guint8 *maskporig, *maskp;
  guint8 *in1u, *in1v, *in2u, *in2v, *outu, *outv; 
  guint8 value;
  gint chromsize = (width * height) >> 2;

  maskp = maskporig = mask->data;
  
  for (i = width * height; i; i--) {
    value = *maskp++;
    *out++ = ((*in2++ * value) + (*in1++ * (255 - value))) >> 8;
  }
  in1u = in1; in1v = in1 + chromsize;
  in2u = in2; in2v = in2 + chromsize;
  outu = out; outv = out + chromsize;

  maskp = maskporig;
  for (i = height/2; i; i--, maskp += width) {
    for (j = width/2; j; j--, maskp += 2) {
      value = *maskp;
      *outu++ = ((*in2u++ * value) + (*in1u++ * (255 - value))) >> 8;
      *outv++ = ((*in2v++ * value) + (*in1v++ * (255 - value))) >> 8;
    }
  }
}

static void
gst_smpte_loop (GstElement *element)
{
  GstSMPTE *smpte;
  GstBuffer *outbuf;
  GstClockTime ts;
  GstBuffer *in1 = NULL, *in2 = NULL;

  smpte = GST_SMPTE (element);

  ts = smpte->position * GST_SECOND;

  if (GST_PAD_IS_USABLE (smpte->sinkpad1)) {
    in1 = gst_pad_pull (smpte->sinkpad1);
    ts = GST_BUFFER_TIMESTAMP (in1);
  }
  else {
    in1 = gst_buffer_new_and_alloc (smpte->width * smpte->height * 3);
    fill_i420 (GST_BUFFER_DATA (in1), smpte->width, smpte->height, 7);
  }
    
  if (GST_PAD_IS_USABLE (smpte->sinkpad2)) {
    in2 = gst_pad_pull (smpte->sinkpad2);
    ts = GST_BUFFER_TIMESTAMP (in2);
  }
  else {
    in2 = gst_buffer_new_and_alloc (smpte->width * smpte->height * 3);
    fill_i420 (GST_BUFFER_DATA (in2), smpte->width, smpte->height, 0);
  }

  if (smpte->position < smpte->duration) { 
    outbuf = gst_buffer_new_and_alloc (smpte->width * smpte->height * 3);

    if (!GST_PAD_CAPS (smpte->srcpad)) {
      if (!gst_pad_try_set_caps (smpte->srcpad,
			    GST_CAPS_NEW (
				    "smpte_srccaps",
				    "video/raw",
				      "format",   GST_PROPS_FOURCC (GST_MAKE_FOURCC ('I','4','2','0')),
				      "width",    GST_PROPS_INT (smpte->width),
				      "height",   GST_PROPS_INT (smpte->height)
				    )))
      {
        gst_element_error (element, "cannot set caps");
        return;
      }
    }

    gst_mask_update (smpte->mask, smpte->position, smpte->duration);

    gst_smpte_blend_i420 (GST_BUFFER_DATA (in1), GST_BUFFER_DATA (in2), GST_BUFFER_DATA (outbuf),
	                  smpte->mask, smpte->width, smpte->height);
  }
  else {
    outbuf = in2;
    gst_buffer_ref (in2);
  }

  smpte->position++;

  if (in1)
    gst_buffer_unref (in1);
  if (in2)
    gst_buffer_unref (in2);

  GST_BUFFER_TIMESTAMP (outbuf) = ts;
  gst_pad_push (smpte->srcpad, outbuf);
}

static void
gst_smpte_set_property (GObject *object, guint prop_id, 
		        const GValue *value, GParamSpec *pspec)
{
  GstSMPTE *smpte;

  smpte = GST_SMPTE(object);

  switch (prop_id) {
    case ARG_TYPE:
    {
      GstMask *newmask;
      gint type = g_value_get_enum (value);

      newmask = gst_mask_factory_new (type, 8, smpte->width, smpte->height);
      if (newmask) {
	if (smpte->mask) {
          gst_mask_destroy (smpte->mask);
	}
	smpte->mask = newmask;
	smpte->type = type;
      }
      break;
    }
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      break;
  }
}

static void
gst_smpte_get_property (GObject *object, guint prop_id, 
		        GValue *value, GParamSpec *pspec)
{
  GstSMPTE *smpte;

  smpte = GST_SMPTE(object);

  switch (prop_id) {
    case ARG_TYPE:
      if (smpte->mask) {
	g_value_set_enum (value, smpte->mask->type);
      }
      break;
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      break;
  }
}


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

  factory = gst_element_factory_new("smpte",GST_TYPE_SMPTE,
                                   &smpte_details);
  g_return_val_if_fail(factory != NULL, FALSE);

  gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET (smpte_sink1_factory));
  gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET (smpte_sink2_factory));
  gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET (smpte_src_factory));

  gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));

  return TRUE;
}

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