diff -r 71e347f905f2 -r 4a7fac7dd34a gst_plugins_good/gst/alpha/gstalphacolor.c --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/gst_plugins_good/gst/alpha/gstalphacolor.c Fri Apr 16 15:15:52 2010 +0300 @@ -0,0 +1,279 @@ +/* GStreamer + * Copyright (C) <1999> Erik Walthinsen + * + * 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. + */ + +/** + * SECTION:element-alphacolor + * + * The alphacolor element does memory-efficient (in-place) colourspace + * conversion from RGBA to AYUV, preserving the alpha channel. + */ + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif + +#include "gstalphacolor.h" + +#include +#include + +#include + +GST_DEBUG_CATEGORY_STATIC (alpha_color_debug); +#define GST_CAT_DEFAULT alpha_color_debug + +/* elementfactory information */ +static const GstElementDetails gst_alpha_color_details = +GST_ELEMENT_DETAILS ("Alpha color filter", + "Filter/Effect/Video", + "RGBA to AYUV colorspace conversion preserving the alpha channel", + "Wim Taymans "); + +static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink", + GST_PAD_SINK, + GST_PAD_ALWAYS, + GST_STATIC_CAPS (GST_VIDEO_CAPS_RGBA ";" GST_VIDEO_CAPS_BGRA) + ); + +static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("AYUV")) + ); + +GST_BOILERPLATE (GstAlphaColor, gst_alpha_color, GstBaseTransform, + GST_TYPE_BASE_TRANSFORM); + +static GstCaps *gst_alpha_color_transform_caps (GstBaseTransform * btrans, + GstPadDirection direction, GstCaps * caps); +static gboolean gst_alpha_color_set_caps (GstBaseTransform * btrans, + GstCaps * incaps, GstCaps * outcaps); +static GstFlowReturn gst_alpha_color_transform_ip (GstBaseTransform * btrans, + GstBuffer * inbuf); + +static void +gst_alpha_color_base_init (gpointer g_class) +{ + GstElementClass *element_class = GST_ELEMENT_CLASS (g_class); + + gst_element_class_set_details (element_class, &gst_alpha_color_details); + + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&sink_template)); + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&src_template)); +} + +static void +gst_alpha_color_class_init (GstAlphaColorClass * klass) +{ + GstBaseTransformClass *gstbasetransform_class; + + gstbasetransform_class = (GstBaseTransformClass *) klass; + + gstbasetransform_class->transform_caps = + GST_DEBUG_FUNCPTR (gst_alpha_color_transform_caps); + gstbasetransform_class->set_caps = + GST_DEBUG_FUNCPTR (gst_alpha_color_set_caps); + gstbasetransform_class->transform_ip = + GST_DEBUG_FUNCPTR (gst_alpha_color_transform_ip); + + GST_DEBUG_CATEGORY_INIT (alpha_color_debug, "alphacolor", 0, + "RGB->YUV colorspace conversion preserving the alpha channels"); +} + +static void +gst_alpha_color_init (GstAlphaColor * alpha, GstAlphaColorClass * g_class) +{ + GstBaseTransform *btrans = NULL; + + btrans = GST_BASE_TRANSFORM (alpha); + + btrans->always_in_place = TRUE; +} + +static GstCaps * +gst_alpha_color_transform_caps (GstBaseTransform * btrans, + GstPadDirection direction, GstCaps * caps) +{ + const GstCaps *tmpl_caps = NULL; + GstCaps *result = NULL, *local_caps = NULL; + guint i; + + local_caps = gst_caps_copy (caps); + + for (i = 0; i < gst_caps_get_size (local_caps); i++) { + GstStructure *structure = gst_caps_get_structure (local_caps, i); + + /* Throw away the structure name and set it to transformed format */ + if (direction == GST_PAD_SINK) { + gst_structure_set_name (structure, "video/x-raw-yuv"); + } else if (direction == GST_PAD_SRC) { + gst_structure_set_name (structure, "video/x-raw-rgb"); + } + /* Remove any specific parameter from the structure */ + gst_structure_remove_field (structure, "format"); + gst_structure_remove_field (structure, "endianness"); + gst_structure_remove_field (structure, "depth"); + gst_structure_remove_field (structure, "bpp"); + gst_structure_remove_field (structure, "red_mask"); + gst_structure_remove_field (structure, "green_mask"); + gst_structure_remove_field (structure, "blue_mask"); + gst_structure_remove_field (structure, "alpha_mask"); + } + + /* Get the appropriate template */ + if (direction == GST_PAD_SINK) { + tmpl_caps = gst_static_pad_template_get_caps (&src_template); + } else if (direction == GST_PAD_SRC) { + tmpl_caps = gst_static_pad_template_get_caps (&sink_template); + } + + /* Intersect with our template caps */ + result = gst_caps_intersect (local_caps, tmpl_caps); + + gst_caps_unref (local_caps); + gst_caps_do_simplify (result); + + GST_LOG ("transformed %" GST_PTR_FORMAT " to %" GST_PTR_FORMAT, caps, result); + + return result; +} + +static gboolean +gst_alpha_color_set_caps (GstBaseTransform * btrans, GstCaps * incaps, + GstCaps * outcaps) +{ + GstAlphaColor *alpha; + GstStructure *structure; + gboolean ret; + const GValue *fps; + gint red_mask, alpha_mask; + gint w, h, depth, bpp; + + alpha = GST_ALPHA_COLOR (btrans); + structure = gst_caps_get_structure (incaps, 0); + + ret = gst_structure_get_int (structure, "width", &w); + ret &= gst_structure_get_int (structure, "height", &h); + fps = gst_structure_get_value (structure, "framerate"); + ret &= (fps != NULL && GST_VALUE_HOLDS_FRACTION (fps)); + ret &= gst_structure_get_int (structure, "red_mask", &red_mask); + + /* make sure these are really full RGBA caps */ + ret &= gst_structure_get_int (structure, "alpha_mask", &alpha_mask); + ret &= gst_structure_get_int (structure, "depth", &depth); + ret &= gst_structure_get_int (structure, "bpp", &bpp); + + if (!ret || alpha_mask == 0 || red_mask == 0 || depth != 32 || bpp != 32) { + GST_DEBUG_OBJECT (alpha, "incomplete or non-RGBA input caps!"); + return FALSE; + } + + alpha->in_width = w; + alpha->in_height = h; + alpha->in_rgba = TRUE; +#if (G_BYTE_ORDER == G_BIG_ENDIAN) + if (red_mask != 0x000000ff) +#else + if (red_mask != 0xff000000) +#endif + alpha->in_rgba = FALSE; + + return TRUE; +} + +static void +transform_rgb (guint8 * data, gint size) +{ + guint8 y, u, v; + + while (size > 0) { + y = data[0] * 0.299 + data[1] * 0.587 + data[2] * 0.114 + 0; + u = data[0] * -0.169 + data[1] * -0.332 + data[2] * 0.500 + 128; + v = data[0] * 0.500 + data[1] * -0.419 + data[2] * -0.0813 + 128; + + data[0] = data[3]; + data[1] = y; + data[2] = u; + data[3] = v; + + data += 4; + size -= 4; + } +} + +static void +transform_bgr (guint8 * data, gint size) +{ + guint8 y, u, v; + + while (size > 0) { + y = data[2] * 0.299 + data[1] * 0.587 + data[0] * 0.114 + 0; + u = data[2] * -0.169 + data[1] * -0.332 + data[0] * 0.500 + 128; + v = data[2] * 0.500 + data[1] * -0.419 + data[0] * -0.0813 + 128; + + data[0] = data[3]; + data[1] = y; + data[2] = u; + data[3] = v; + + data += 4; + size -= 4; + } +} + +static GstFlowReturn +gst_alpha_color_transform_ip (GstBaseTransform * btrans, GstBuffer * inbuf) +{ + GstFlowReturn ret = GST_FLOW_OK; + GstAlphaColor *alpha; + + alpha = GST_ALPHA_COLOR (btrans); + + /* Transform in place */ + if (alpha->in_rgba) + transform_rgb (GST_BUFFER_DATA (inbuf), GST_BUFFER_SIZE (inbuf)); + else + transform_bgr (GST_BUFFER_DATA (inbuf), GST_BUFFER_SIZE (inbuf)); + + return ret; +} + +static gboolean +plugin_init (GstPlugin * plugin) +{ + return gst_element_register (plugin, "alphacolor", GST_RANK_NONE, + GST_TYPE_ALPHA_COLOR); +} + +GST_PLUGIN_DEFINE (GST_VERSION_MAJOR, + GST_VERSION_MINOR, + "alphacolor", + "RGBA to AYUV colorspace conversion preserving the alpha channel", + plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN) + + +#ifdef __SYMBIAN32__ +EXPORT_C GstPluginDesc* _GST_PLUGIN_DESC() +{ + return &gst_plugin_desc; +} + +#endif