diff options
author | Wim Taymans <wim.taymans@gmail.com> | 2002-10-10 21:19:12 +0000 |
---|---|---|
committer | Wim Taymans <wim.taymans@gmail.com> | 2002-10-10 21:19:12 +0000 |
commit | 9b067365c95ee0d987a74acd10a9c9890abab5db (patch) | |
tree | cba71e4d973972f372a4819c8734ac4bd04fe4d8 /gst/smpte/gstsmpte.c | |
parent | 5f4fb7887107189204fa92978400f9d698a4e319 (diff) | |
download | gst-plugins-bad-9b067365c95ee0d987a74acd10a9c9890abab5db.tar.gz gst-plugins-bad-9b067365c95ee0d987a74acd10a9c9890abab5db.tar.bz2 gst-plugins-bad-9b067365c95ee0d987a74acd10a9c9890abab5db.zip |
Added the base framework for the SMPTE transitions and wipes.
Original commit message from CVS:
Added the base framework for the SMPTE transitions and wipes.
Diffstat (limited to 'gst/smpte/gstsmpte.c')
-rw-r--r-- | gst/smpte/gstsmpte.c | 409 |
1 files changed, 409 insertions, 0 deletions
diff --git a/gst/smpte/gstsmpte.c b/gst/smpte/gstsmpte.c new file mode 100644 index 00000000..536be8d4 --- /dev/null +++ b/gst/smpte/gstsmpte.c @@ -0,0 +1,409 @@ +/* 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 +}; + |