diff options
Diffstat (limited to 'gst')
-rw-r--r-- | gst/colorspace/Makefile.am | 15 | ||||
-rw-r--r-- | gst/colorspace/gstcolorspace.c | 635 | ||||
-rw-r--r-- | gst/colorspace/gstcolorspace.h | 99 | ||||
-rw-r--r-- | gst/colorspace/yuv2rgb.c | 720 | ||||
-rw-r--r-- | gst/colorspace/yuv2rgb.h | 97 |
5 files changed, 1566 insertions, 0 deletions
diff --git a/gst/colorspace/Makefile.am b/gst/colorspace/Makefile.am new file mode 100644 index 00000000..a6d2ca32 --- /dev/null +++ b/gst/colorspace/Makefile.am @@ -0,0 +1,15 @@ + +plugin_LTLIBRARIES = libgstcolorspace.la + +if HAVE_CPU_I386 +ARCHSRCS = yuv2rgb_mmx16.s +else +ARCHSRCS = +endif + +libgstcolorspace_la_SOURCES = gstcolorspace.c yuv2rgb.c yuv2yuv.c +libgstcolorspace_la_CFLAGS = $(GST_CFLAGS) +libgstcolorspace_la_LIBADD = +libgstcolorspace_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS) + +#noinst_HEADERS = yuv2rgb.h yuv2yuv.h diff --git a/gst/colorspace/gstcolorspace.c b/gst/colorspace/gstcolorspace.c new file mode 100644 index 00000000..e923c3a1 --- /dev/null +++ b/gst/colorspace/gstcolorspace.c @@ -0,0 +1,635 @@ +/* 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. + */ + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif +#include <gstcolorspace.h> +#include <gst/gst.h> +#include <gst/video/video.h> + +#include "yuv2rgb.h" + + +static GstColorspaceFormat gst_colorspace_formats[] = { + { GST_STATIC_CAPS (GST_VIDEO_YUV_PAD_TEMPLATE_CAPS("I420")) }, + { GST_STATIC_CAPS (GST_VIDEO_YUV_PAD_TEMPLATE_CAPS("YV12")) }, + { GST_STATIC_CAPS (GST_VIDEO_RGB_PAD_TEMPLATE_CAPS_32) }, + { GST_STATIC_CAPS (GST_VIDEO_RGB_PAD_TEMPLATE_CAPS_24) }, + { GST_STATIC_CAPS (GST_VIDEO_RGB_PAD_TEMPLATE_CAPS_16) }, +}; + +static GstColorspaceConverter gst_colorspace_converters[] = { + { GST_COLORSPACE_I420, GST_COLORSPACE_RGB32, gst_colorspace_I420_to_rgb32 }, + { GST_COLORSPACE_YV12, GST_COLORSPACE_RGB32, gst_colorspace_YV12_to_rgb32 }, + { GST_COLORSPACE_I420, GST_COLORSPACE_RGB24, gst_colorspace_I420_to_rgb24 }, + { GST_COLORSPACE_YV12, GST_COLORSPACE_RGB24, gst_colorspace_YV12_to_rgb24 }, + { GST_COLORSPACE_I420, GST_COLORSPACE_RGB16, gst_colorspace_I420_to_rgb16 }, + { GST_COLORSPACE_YV12, GST_COLORSPACE_RGB16, gst_colorspace_YV12_to_rgb16 }, +}; + +static GstElementDetails colorspace_details = GST_ELEMENT_DETAILS ( + "Colorspace converter", + "Filter/Converter/Video", + "Converts video from one colorspace to another", + "Wim Taymans <wim.taymans@chello.be>" +); + +static GstStaticPadTemplate gst_colorspace_sink_template = +GST_STATIC_PAD_TEMPLATE ( + "sink", + GST_PAD_SINK, + GST_PAD_ALWAYS, + GST_STATIC_CAPS (GST_VIDEO_YUV_PAD_TEMPLATE_CAPS ("{ I420, YV12 }")) +); + +static GstStaticPadTemplate gst_colorspace_src_template = +GST_STATIC_PAD_TEMPLATE ( + "src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_STATIC_CAPS ( + GST_VIDEO_RGB_PAD_TEMPLATE_CAPS_32 "; " + GST_VIDEO_RGB_PAD_TEMPLATE_CAPS_24 "; " + GST_VIDEO_RGB_PAD_TEMPLATE_CAPS_16 + ) +); + +/* Stereo signals and args */ +enum { + /* FILL ME */ + LAST_SIGNAL +}; + +enum { + ARG_0, + ARG_SOURCE, + ARG_DEST, +}; + +static void gst_colorspace_base_init (gpointer g_class); +static void gst_colorspace_class_init (GstColorspaceClass *klass); +static void gst_colorspace_init (GstColorspace *space); + +static void gst_colorspace_set_property (GObject *object, guint prop_id, + const GValue *value, GParamSpec *pspec); +static void gst_colorspace_get_property (GObject *object, guint prop_id, + GValue *value, GParamSpec *pspec); + +static GstPadLinkReturn + gst_colorspace_link (GstPad *pad, const GstCaps *caps); +static void gst_colorspace_chain (GstPad *pad, GstData *_data); +static GstElementStateReturn + gst_colorspace_change_state (GstElement *element); + + +static GstElementClass *parent_class = NULL; +/*static guint gst_colorspace_signals[LAST_SIGNAL] = { 0 }; */ + +#if 0 +static gboolean +colorspace_setup_converter (GstColorspace *space, GstCaps *from_caps, GstCaps *to_caps) +{ + guint32 from_space, to_space; + GstStructure *from_struct; + GstStructure *to_struct; + + g_return_val_if_fail (to_caps != NULL, FALSE); + g_return_val_if_fail (from_caps != NULL, FALSE); + + from_struct = gst_caps_get_structure (from_caps, 0); + to_struct = gst_caps_get_structure (to_caps, 0); + + from_space = GST_MAKE_FOURCC ('R','G','B',' '); + gst_structure_get_fourcc (from_struct, "format", &from_space); + + to_space = GST_MAKE_FOURCC ('R','G','B',' '); + gst_structure_get_fourcc (to_struct, "format", &to_space); + + GST_INFO ("set up converter for " GST_FOURCC_FORMAT + " (%08x) to " GST_FOURCC_FORMAT " (%08x)", + GST_FOURCC_ARGS (from_space), from_space, + GST_FOURCC_ARGS (to_space), to_space); + + switch (from_space) { + case GST_MAKE_FOURCC ('R','G','B',' '): + { + gint from_bpp; + + gst_structure_get_int (from_struct, "bpp", &from_bpp); + + switch (to_space) { + case GST_MAKE_FOURCC ('R','G','B',' '): +#ifdef HAVE_HERMES + { + gint to_bpp; + + gst_structure_get_int (to_struct, "bpp", &to_bpp); + + gst_structure_get_int (from_struct, "red_mask", &space->source.r); + gst_structure_get_int (from_struct, "green_mask", &space->source.g); + gst_structure_get_int (from_struct, "blue_mask", &space->source.b); + space->source.a = 0; + space->srcbpp = space->source.bits = from_bpp; + space->source.indexed = 0; + space->source.has_colorkey = 0; + + GST_INFO ( "source red mask %08x", space->source.r); + GST_INFO ( "source green mask %08x", space->source.g); + GST_INFO ( "source blue mask %08x", space->source.b); + GST_INFO ( "source bpp %08x", space->srcbpp); + + gst_structure_get_int (to_struct, "red_mask", &space->dest.r); + gst_structure_get_int (to_struct, "green_mask", &space->dest.g); + gst_structure_get_int (to_struct, "blue_mask", &space->dest.b); + space->dest.a = 0; + space->destbpp = space->dest.bits = to_bpp; + space->dest.indexed = 0; + space->dest.has_colorkey = 0; + + GST_INFO ( "dest red mask %08x", space->dest.r); + GST_INFO ( "dest green mask %08x", space->dest.g); + GST_INFO ( "dest blue mask %08x", space->dest.b); + GST_INFO ( "dest bpp %08x", space->destbpp); + + if (!Hermes_ConverterRequest (space->h_handle, &space->source, &space->dest)) { + g_warning ("Hermes: could not get converter\n"); + return FALSE; + } + GST_INFO ( "converter set up"); + space->type = GST_COLORSPACE_HERMES; + return TRUE; + } +#else + g_warning ("colorspace: compiled without hermes!"); + return FALSE; +#endif + case GST_MAKE_FOURCC ('Y','V','1','2'): + if (from_bpp == 32) { + space->type = GST_COLORSPACE_RGB32_YV12; + space->destbpp = 12; + return TRUE; + } + case GST_MAKE_FOURCC ('I','4','2','0'): + if (from_bpp == 32) { + space->type = GST_COLORSPACE_RGB32_I420; + space->destbpp = 12; + return TRUE; + } + case GST_MAKE_FOURCC ('Y','U','Y','2'): + GST_INFO ( "colorspace: RGB to YUV with bpp %d not implemented!!", from_bpp); + return FALSE; + } + break; + } + case GST_MAKE_FOURCC ('I','4','2','0'): + switch (to_space) { + case GST_MAKE_FOURCC ('R','G','B',' '): + GST_INFO ( "colorspace: YUV to RGB"); + + gst_structure_get_int (to_struct, "bpp", &space->destbpp); + space->converter = gst_colorspace_yuv2rgb_get_converter (from_caps, to_caps); + space->type = GST_COLORSPACE_YUV_RGB; + return TRUE; + case GST_MAKE_FOURCC ('I','4','2','0'): + space->type = GST_COLORSPACE_NONE; + space->destbpp = 12; + return TRUE; + case GST_MAKE_FOURCC ('Y','V','1','2'): + space->type = GST_COLORSPACE_420_SWAP; + space->destbpp = 12; + return TRUE; + + } + break; + case GST_MAKE_FOURCC ('Y','U','Y','2'): + switch (to_space) { + case GST_MAKE_FOURCC ('I','4','2','0'): + space->type = GST_COLORSPACE_YUY2_I420; + space->destbpp = 12; + return TRUE; + case GST_MAKE_FOURCC ('Y','U','Y','2'): + space->type = GST_COLORSPACE_NONE; + space->destbpp = 16; + return TRUE; + case GST_MAKE_FOURCC ('R','G','B',' '): + GST_INFO ( "colorspace: YUY2 to RGB not implemented!!"); + return FALSE; + } + break; + case GST_MAKE_FOURCC ('Y','V','1','2'): + switch (to_space) { + case GST_MAKE_FOURCC ('R','G','B',' '): + GST_INFO ( "colorspace: YV12 to RGB"); + + gst_structure_get_int (to_struct, "bpp", &space->destbpp); + space->converter = gst_colorspace_yuv2rgb_get_converter (from_caps, to_caps); + space->type = GST_COLORSPACE_YUV_RGB; + return TRUE; + case GST_MAKE_FOURCC ('I','4','2','0'): + space->type = GST_COLORSPACE_420_SWAP; + space->destbpp = 12; + return TRUE; + case GST_MAKE_FOURCC ('Y','V','1','2'): + space->type = GST_COLORSPACE_NONE; + space->destbpp = 12; + return TRUE; + } + break; + } + return FALSE; +} +#endif + +static GstCaps * +gst_colorspace_caps_remove_format_info (GstCaps *caps, const char *media_type) +{ + int i; + GstStructure *structure; + GstCaps *rgbcaps; + + for (i=0; i<gst_caps_get_size (caps); i++) { + structure = gst_caps_get_structure (caps, i); + + gst_structure_set_name (structure, media_type); + 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"); + } + + rgbcaps = gst_caps_simplify (caps); + gst_caps_free (caps); + + return rgbcaps; +} + +static GstCaps* +gst_colorspace_getcaps (GstPad *pad) +{ + GstColorspace *space; + GstPad *otherpad; + GstCaps *othercaps; + GstCaps *caps; + + space = GST_COLORSPACE (gst_pad_get_parent (pad)); + + otherpad = (pad == space->srcpad) ? space->sinkpad : space->srcpad; + + othercaps = gst_pad_get_allowed_caps (otherpad); + + othercaps = gst_colorspace_caps_remove_format_info (othercaps, + (pad == space->srcpad) ? "video/x-raw-rgb" : "video/x-raw-yuv"); + + caps = gst_caps_intersect (othercaps, gst_pad_get_pad_template_caps (pad)); + gst_caps_free (othercaps); + + return caps; +} + +static GstColorSpaceFormatType +gst_colorspace_get_format (const GstCaps *caps) +{ + int i; + + for(i=0; i<G_N_ELEMENTS (gst_colorspace_formats); i++) { + GstCaps *icaps; + GstCaps *fcaps; + + fcaps = gst_caps_copy (gst_static_caps_get ( + &gst_colorspace_formats[i].caps)); + + icaps = gst_caps_intersect (caps, fcaps); + if (!gst_caps_is_empty (icaps)) { + gst_caps_free (icaps); + return i; + } + gst_caps_free (icaps); + } + + g_assert_not_reached(); + return -1; +} + +#define ROUND_UP_2(x) (((x)+1)&~1) +#define ROUND_UP_4(x) (((x)+3)&~3) +#define ROUND_UP_8(x) (((x)+7)&~7) + +static int +gst_colorspace_format_get_size(GstColorSpaceFormatType index, int width, + int height) +{ + int size; + + switch (index) { + case GST_COLORSPACE_I420: + case GST_COLORSPACE_YV12: + size = ROUND_UP_4 (width) * ROUND_UP_2 (height); + size += ROUND_UP_8 (width)/2 * ROUND_UP_2 (height)/2; + size += ROUND_UP_8 (width)/2 * ROUND_UP_2 (height)/2; + return size; + break; + case GST_COLORSPACE_RGB32: + return width*height*4; + break; + case GST_COLORSPACE_RGB24: + return ROUND_UP_4 (width*3) * height; + break; + case GST_COLORSPACE_RGB16: + return ROUND_UP_4 (width*2) * height; + break; + } + + g_assert_not_reached(); + return 0; +} + +static int +gst_colorspace_get_converter (GstColorSpaceFormatType from, + GstColorSpaceFormatType to) +{ + int i; + + for (i=0; i<G_N_ELEMENTS (gst_colorspace_converters); i++) { + GstColorspaceConverter *converter = gst_colorspace_converters + i; + if (from == converter->from && to == converter->to) { + return i; + } + } + g_assert_not_reached(); + return -1; +} + +static GstPadLinkReturn +gst_colorspace_link (GstPad *pad, const GstCaps *caps) +{ + GstColorspace *space; + GstPad *otherpad; + GstStructure *structure; + GstPadLinkReturn link_ret; + int width, height; + double fps; + int format_index; + + space = GST_COLORSPACE (gst_pad_get_parent (pad)); + otherpad = (pad == space->sinkpad) ? space->srcpad : space->sinkpad; + + link_ret = gst_pad_try_set_caps (otherpad, caps); + if (link_ret == GST_PAD_LINK_OK) { + return link_ret; + } + + structure = gst_caps_get_structure (caps, 0); + + format_index = gst_colorspace_get_format (caps); + g_print("format index is %d\n", format_index); + + gst_structure_get_int (structure, "width", &width); + gst_structure_get_int (structure, "height", &height); + gst_structure_get_double (structure, "framerate", &fps); + + GST_INFO ( "size: %dx%d", space->width, space->height); + + if (gst_pad_is_negotiated (otherpad)) { + GstCaps *othercaps; + + othercaps = gst_caps_copy (gst_pad_get_negotiated_caps (otherpad)); + + gst_caps_set_simple (othercaps, + "width", G_TYPE_INT, width, + "height", G_TYPE_INT, height, + "framerate", G_TYPE_DOUBLE, fps, NULL); + + link_ret = gst_pad_try_set_caps (otherpad, othercaps); + if (link_ret != GST_PAD_LINK_OK) { + return link_ret; + } + } + + if (pad == space->srcpad) { + space->src_format_index = format_index; + } else { + space->sink_format_index = format_index; + } + + if (gst_pad_is_negotiated (otherpad)) { + space->converter_index = gst_colorspace_get_converter ( + space->sink_format_index, space->src_format_index); + + g_print("using index %d\n", space->converter_index); + + space->sink_size = gst_colorspace_format_get_size(space->sink_format_index, + width,height); + space->src_size = gst_colorspace_format_get_size(space->src_format_index, + width,height); + space->width = width; + space->height = height; + space->fps = fps; + } + +#if 0 + if (gst_pad_is_negotiated (otherpad)) { + g_warning ("could not get converter\n"); + return GST_PAD_LINK_REFUSED; + } +#endif + + return GST_PAD_LINK_OK; +} + +GType +gst_colorspace_get_type (void) +{ + static GType colorspace_type = 0; + + if (!colorspace_type) { + static const GTypeInfo colorspace_info = { + sizeof(GstColorspaceClass), + gst_colorspace_base_init, + NULL, + (GClassInitFunc)gst_colorspace_class_init, + NULL, + NULL, + sizeof(GstColorspace), + 0, + (GInstanceInitFunc)gst_colorspace_init, + }; + colorspace_type = g_type_register_static(GST_TYPE_ELEMENT, "GstColorspace", &colorspace_info, 0); + } + return colorspace_type; +} + +static void +gst_colorspace_base_init (gpointer g_class) +{ + GstElementClass *element_class = GST_ELEMENT_CLASS (g_class); + + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&gst_colorspace_src_template)); + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&gst_colorspace_sink_template)); + + gst_element_class_set_details (element_class, &colorspace_details); +} + +static void +gst_colorspace_class_init (GstColorspaceClass *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_colorspace_set_property; + gobject_class->get_property = gst_colorspace_get_property; + + gstelement_class->change_state = gst_colorspace_change_state; + + gst_colorspace_table_init(NULL); +} + +static void +gst_colorspace_init (GstColorspace *space) +{ + space->sinkpad = gst_pad_new_from_template ( + gst_static_pad_template_get (&gst_colorspace_sink_template), + "sink"); + gst_pad_set_link_function (space->sinkpad, gst_colorspace_link); + gst_pad_set_getcaps_function (space->sinkpad, gst_colorspace_getcaps); + gst_pad_set_chain_function(space->sinkpad,gst_colorspace_chain); + gst_element_add_pad(GST_ELEMENT(space),space->sinkpad); + + space->srcpad = gst_pad_new_from_template ( + gst_static_pad_template_get (&gst_colorspace_src_template), + "src"); + gst_element_add_pad(GST_ELEMENT(space),space->srcpad); + gst_pad_set_link_function (space->srcpad, gst_colorspace_link); +} + +static void +gst_colorspace_chain (GstPad *pad,GstData *_data) +{ + GstBuffer *buf = GST_BUFFER (_data); + GstColorspace *space; + GstBuffer *outbuf = NULL; + GstColorspaceConverter *converter; + + g_return_if_fail (pad != NULL); + g_return_if_fail (GST_IS_PAD (pad)); + g_return_if_fail (buf != NULL); + + space = GST_COLORSPACE (gst_pad_get_parent (pad)); + + g_return_if_fail (space != NULL); + g_return_if_fail (GST_IS_COLORSPACE (space)); + + if (GST_BUFFER_SIZE (buf) < space->sink_size) { + g_critical ("input size is smaller than expected"); + } + + outbuf = gst_pad_alloc_buffer (space->srcpad, GST_BUFFER_OFFSET_NONE, + space->src_size); + + converter = gst_colorspace_converters + space->converter_index; + converter->convert (space, GST_BUFFER_DATA (outbuf), + GST_BUFFER_DATA (buf)); + + GST_BUFFER_TIMESTAMP (outbuf) = GST_BUFFER_TIMESTAMP (buf); + GST_BUFFER_DURATION (outbuf) = GST_BUFFER_DURATION (buf); + + gst_buffer_unref (buf); + gst_pad_push (space->srcpad, GST_DATA (outbuf)); +} + +static GstElementStateReturn +gst_colorspace_change_state (GstElement *element) +{ + GstColorspace *space; + + space = GST_COLORSPACE (element); + + switch (GST_STATE_TRANSITION (element)) { + case GST_STATE_PAUSED_TO_PLAYING: + break; + case GST_STATE_PLAYING_TO_PAUSED: + break; + case GST_STATE_PAUSED_TO_READY: + break; + } + + return parent_class->change_state (element); +} + +static void +gst_colorspace_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec) +{ + GstColorspace *space; + + /* it's not null if we got it, but it might not be ours */ + g_return_if_fail(GST_IS_COLORSPACE(object)); + space = GST_COLORSPACE(object); + + switch (prop_id) { + default: + break; + } +} + +static void +gst_colorspace_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec) +{ + GstColorspace *space; + + /* it's not null if we got it, but it might not be ours */ + g_return_if_fail(GST_IS_COLORSPACE(object)); + space = GST_COLORSPACE(object); + + switch (prop_id) { + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static gboolean +plugin_init (GstPlugin *plugin) +{ + if (!gst_element_register (plugin, "colorspace", GST_RANK_PRIMARY, + GST_TYPE_COLORSPACE)) + return FALSE; + + return TRUE; +} + +GST_PLUGIN_DEFINE ( + GST_VERSION_MAJOR, + GST_VERSION_MINOR, + "colorspace", + "Hermes colorspace converter", + plugin_init, + VERSION, + "LGPL", + GST_PACKAGE, + GST_ORIGIN) diff --git a/gst/colorspace/gstcolorspace.h b/gst/colorspace/gstcolorspace.h new file mode 100644 index 00000000..22e2f7ec --- /dev/null +++ b/gst/colorspace/gstcolorspace.h @@ -0,0 +1,99 @@ +/* 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. + */ + +#ifndef _GST_COLORSPACE_H_ +#define _GST_COLORSPACE_H_ + +#include <gst/gst.h> + +G_BEGIN_DECLS + +#define GST_TYPE_COLORSPACE \ + (gst_colorspace_get_type()) +#define GST_COLORSPACE(obj) \ + (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_COLORSPACE,GstColorspace)) +#define GST_COLORSPACE_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_ULAW,GstColorspace)) +#define GST_IS_COLORSPACE(obj) \ + (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_COLORSPACE)) +#define GST_IS_COLORSPACE_CLASS(obj) \ + (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_COLORSPACE)) + +typedef struct _GstColorspace GstColorspace; +typedef struct _GstColorspaceClass GstColorspaceClass; + +typedef enum { + GST_COLORSPACE_NONE, + GST_COLORSPACE_HERMES, + GST_COLORSPACE_YUV_RGB, + GST_COLORSPACE_YUY2_I420, + GST_COLORSPACE_RGB32_I420, + GST_COLORSPACE_RGB32_YV12, + GST_COLORSPACE_420_SWAP, +} GstColorSpaceConverterType; + +struct _GstColorspace { + GstElement element; + + GstPad *sinkpad,*srcpad; + + int converter_index; + + int src_format_index; + int sink_format_index; + + int src_size; + int sink_size; + + int src_stride; + int sink_stride; + + gint width, height; + gdouble fps; +}; + +struct _GstColorspaceClass { + GstElementClass parent_class; +}; + +GType gst_colorspace_get_type(void); + +typedef struct _GstColorspaceFormat { + GstStaticCaps caps; + +} GstColorspaceFormat; + +typedef enum { + GST_COLORSPACE_I420, + GST_COLORSPACE_YV12, + GST_COLORSPACE_RGB32, + GST_COLORSPACE_RGB24, + GST_COLORSPACE_RGB16, +} GstColorSpaceFormatType; + +typedef struct _GstColorspaceConverter { + GstColorSpaceFormatType from; + GstColorSpaceFormatType to; + void (*convert) (GstColorspace *colorspace, unsigned char *dest, unsigned char *src); +} GstColorspaceConverter; + +G_END_DECLS + +#endif + diff --git a/gst/colorspace/yuv2rgb.c b/gst/colorspace/yuv2rgb.c new file mode 100644 index 00000000..3627e896 --- /dev/null +++ b/gst/colorspace/yuv2rgb.c @@ -0,0 +1,720 @@ +/* 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. + */ + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif + +#include <math.h> +#include <stdlib.h> + +#include "yuv2rgb.h" + +#undef HAVE_LIBMMX + +#ifdef HAVE_LIBMMX +#include <mmx.h> +#endif + +static int V_r_tab [256]; +static int V_g_tab [256]; +static int U_g_tab [256]; +static int U_b_tab [256]; + +#define CB_BASE 1 +#define CR_BASE (CB_BASE*CB_RANGE) +#define LUM_BASE (CR_BASE*CR_RANGE) + +#define Min(x,y) (((x) < (y)) ? (x) : (y)) +#define Max(x,y) (((x) > (y)) ? (x) : (y)) + +#define GAMMA_CORRECTION(x) ((int)(pow((x) / 255.0, 1.0 / gammaCorrect) * 255.0)) +#define CHROMA_CORRECTION256(x) ((x) >= 128 \ + ? 128 + Min(127, (int)(((x) - 128.0) * chromaCorrect)) \ + : 128 - Min(128, (int)((128.0 - (x)) * chromaCorrect))) +#define CHROMA_CORRECTION128(x) ((x) >= 0 \ + ? Min(127, (int)(((x) * chromaCorrect))) \ + : Max(-128, (int)(((x) * chromaCorrect)))) +#define CHROMA_CORRECTION256D(x) ((x) >= 128 \ + ? 128.0 + Min(127.0, (((x) - 128.0) * chromaCorrect)) \ + : 128.0 - Min(128.0, (((128.0 - (x)) * chromaCorrect)))) +#define CHROMA_CORRECTION128D(x) ((x) >= 0 \ + ? Min(127.0, ((x) * chromaCorrect)) \ + : Max(-128.0, ((x) * chromaCorrect))) + + +void gst_colorspace_I420_to_rgb16 (GstColorspace *space, unsigned char *src, unsigned char *dest); +void gst_colorspace_I420_to_rgb24 (GstColorspace *space, unsigned char *src, unsigned char *dest); +void gst_colorspace_I420_to_rgb32 (GstColorspace *space, unsigned char *src, unsigned char *dest); +#ifdef HAVE_LIBMMX +void gst_colorspace_I420_to_bgr16_mmx (GstColorspace *space, unsigned char *src, unsigned char *dest); +void gst_colorspace_I420_to_bgr32_mmx (GstColorspace *space, unsigned char *src, unsigned char *dest); +#endif + +void gst_colorspace_YV12_to_rgb16 (GstColorspace *space, unsigned char *src, unsigned char *dest); +void gst_colorspace_YV12_to_rgb24 (GstColorspace *space, unsigned char *src, unsigned char *dest); +void gst_colorspace_YV12_to_rgb32 (GstColorspace *space, unsigned char *src, unsigned char *dest); +#ifdef HAVE_LIBMMX +void gst_colorspace_YV12_to_bgr16_mmx (GstColorspace *space, unsigned char *src, unsigned char *dest); +void gst_colorspace_YV12_to_bgr32_mmx (GstColorspace *space, unsigned char *src, unsigned char *dest); +#endif + +static void +gst_colorspace_yuv_to_rgb16(GstColorspace *space, + unsigned char *out, + unsigned char *lum, + unsigned char *cr, + unsigned char *cb, + int cols, int rows); +static void +gst_colorspace_yuv_to_rgb24(GstColorspace *space, + unsigned char *out, + unsigned char *lum, + unsigned char *cr, + unsigned char *cb, + int cols, int rows); +static void +gst_colorspace_yuv_to_rgb32(GstColorspace *space, + unsigned char *out, + unsigned char *lum, + unsigned char *cr, + unsigned char *cb, + int cols, int rows); +#if 0 +static void gst_colorspace_yuv_to_rgb16(GstColorspaceYUVTables *tables, + unsigned char *lum, + unsigned char *cr, + unsigned char *cb, + unsigned char *out, + int cols, int rows); +static void gst_colorspace_yuv_to_rgb24(GstColorspaceYUVTables *tables, + unsigned char *lum, + unsigned char *cr, + unsigned char *cb, + unsigned char *out, + int cols, int rows); +static void gst_colorspace_yuv_to_rgb32(GstColorspaceYUVTables *tables, + unsigned char *lum, + unsigned char *cr, + unsigned char *cb, + unsigned char *out, + int cols, int rows); +#ifdef HAVE_LIBMMX +void gst_colorspace_yuv_to_bgr32_mmx(GstColorspaceYUVTables *tables, + unsigned char *lum, + unsigned char *cr, + unsigned char *cb, + unsigned char *out, + int cols, int rows); +extern void gst_colorspace_yuv_to_bgr16_mmx(GstColorspaceYUVTables *tables, + unsigned char *lum, + unsigned char *cr, + unsigned char *cb, + unsigned char *out, + int cols, int rows); +#endif +#endif + +#define ROUND_UP_2(x) (((x)+1)&~1) +#define ROUND_UP_4(x) (((x)+3)&~3) +#define ROUND_UP_8(x) (((x)+7)&~7) + + +void gst_colorspace_I420_to_rgb32(GstColorspace *space, unsigned char *dest, + unsigned char *src) +{ + unsigned char *src_U; + unsigned char *src_V; + + src_U = src + ROUND_UP_4 (space->width) * ROUND_UP_2 (space->height); + src_V = src_U + ROUND_UP_8 (space->width)/2 * ROUND_UP_2 (space->height)/2; + + gst_colorspace_yuv_to_rgb32(space, + dest, + src, src_U, src_V, + space->width, space->height); +} + +void gst_colorspace_I420_to_rgb24(GstColorspace *space, unsigned char *dest, + unsigned char *src) +{ + unsigned char *src_U; + unsigned char *src_V; + + src_U = src + ROUND_UP_4 (space->width) * ROUND_UP_2 (space->height); + src_V = src_U + ROUND_UP_8 (space->width)/2 * ROUND_UP_2 (space->height)/2; + + gst_colorspace_yuv_to_rgb24(space, + dest, + src, src_U, src_V, + space->width, space->height); +} + +void gst_colorspace_I420_to_rgb16(GstColorspace *space, unsigned char *dest, + unsigned char *src) +{ + unsigned char *src_U; + unsigned char *src_V; + + src_U = src + ROUND_UP_4 (space->width) * ROUND_UP_2 (space->height); + src_V = src_U + ROUND_UP_8 (space->width)/2 * ROUND_UP_2 (space->height)/2; + + gst_colorspace_yuv_to_rgb16(space, + dest, + src, src_U, src_V, + space->width, space->height); +} + +void gst_colorspace_YV12_to_rgb32(GstColorspace *space, unsigned char *dest, + unsigned char *src) +{ + unsigned char *src_U; + unsigned char *src_V; + + src_V = src + ROUND_UP_4 (space->width) * ROUND_UP_2 (space->height); + src_U = src_V + ROUND_UP_8 (space->width)/2 * ROUND_UP_2 (space->height)/2; + + gst_colorspace_yuv_to_rgb32(space, + dest, + src, src_U, src_V, + space->width, space->height); +} + +void gst_colorspace_YV12_to_rgb24(GstColorspace *space, unsigned char *dest, + unsigned char *src) +{ + unsigned char *src_U; + unsigned char *src_V; + + src_V = src + ROUND_UP_4 (space->width) * ROUND_UP_2 (space->height); + src_U = src_V + ROUND_UP_8 (space->width)/2 * ROUND_UP_2 (space->height)/2; + + gst_colorspace_yuv_to_rgb24(space, + dest, + src, src_U, src_V, + space->width, space->height); +} + +void gst_colorspace_YV12_to_rgb16(GstColorspace *space, unsigned char *dest, + unsigned char *src) +{ + unsigned char *src_U; + unsigned char *src_V; + + src_V = src + ROUND_UP_4 (space->width) * ROUND_UP_2 (space->height); + src_U = src_V + ROUND_UP_8 (space->width)/2 * ROUND_UP_2 (space->height)/2; + + gst_colorspace_yuv_to_rgb16(space, + dest, + src, src_U, src_V, + space->width, space->height); +} + +#ifdef HAVE_LIBMMX +void gst_colorspace_I420_to_bgr32_mmx(GstColorspace *space, unsigned char *src, unsigned char *dest) { + int size; + GST_DEBUG ("gst_colorspace_I420_to_rgb32_mmx"); + + size = space->width * space->height; + + gst_colorspace_yuv_to_bgr32_mmx(NULL, + src, /* Y component */ + src+size, /* cr component */ + src+size+(size>>2), /* cb component */ + dest, + space->height, + space->width); + +} + +void gst_colorspace_I420_to_bgr16_mmx(GstColorspace *space, unsigned char *src, unsigned char *dest) { + int size; + GST_DEBUG ("gst_colorspace_I420_to_bgr16_mmx "); + + size = space->width * space->height; + + gst_colorspace_yuv_to_bgr16_mmx(NULL, + src, /* Y component */ + src+size, /* cr component */ + src+size+(size>>2), /* cb component */ + dest, + space->height, + space->width); + GST_DEBUG ("gst_colorspace_I420_to_bgr16_mmx done"); + +} + +void gst_colorspace_YV12_to_bgr32_mmx(GstColorspace *space, unsigned char *src, unsigned char *dest) { + int size; + GST_DEBUG ("gst_colorspace_YV12_to_rgb32_mmx"); + + size = space->width * space->height; + + gst_colorspace_yuv_to_bgr32_mmx(NULL, + src, /* Y component */ + src+size+(size>>2), /* cb component */ + src+size, /* cr component */ + dest, + space->height, + space->width); + +} + +void gst_colorspace_YV12_to_bgr16_mmx(GstColorspace *space, unsigned char *src, unsigned char *dest) { + int size; + GST_DEBUG ("gst_colorspace_YV12_to_bgr16_mmx "); + + size = space->width * space->height; + + gst_colorspace_yuv_to_bgr16_mmx(NULL, + src, /* Y component */ + src+size+(size>>2), /* cb component */ + src+size, /* cr component */ + dest, + space->height, + space->width); + GST_DEBUG ("gst_colorspace_YV12_to_bgr16_mmx done"); + +} +#endif +/* + * How many 1 bits are there in the longword. + * Low performance, do not call often. + */ + +static int +number_of_bits_set(a) +unsigned long a; +{ + if(!a) return 0; + if(a & 1) return 1 + number_of_bits_set(a >> 1); + return(number_of_bits_set(a >> 1)); +} + +/* + * How many 0 bits are there at most significant end of longword. + * Low performance, do not call often. + */ +static int +free_bits_at_top(a) +unsigned long a; +{ + /* assume char is 8 bits */ + if(!a) return sizeof(unsigned long) * 8; + /* assume twos complement */ + if(((long)a) < 0l) return 0; + return 1 + free_bits_at_top ( a << 1); +} + +/* + * How many 0 bits are there at least significant end of longword. + * Low performance, do not call often. + */ +static int +free_bits_at_bottom(a) +unsigned long a; +{ + /* assume char is 8 bits */ + if(!a) return sizeof(unsigned long) * 8; + if(((long)a) & 1l) return 0; + return 1 + free_bits_at_bottom ( a >> 1); +} + +/* + *-------------------------------------------------------------- + * + * InitColor16Dither -- + * + * To get rid of the multiply and other conversions in color + * dither, we use a lookup table. + * + * Results: + * None. + * + * Side effects: + * The lookup tables are initialized. + * + *-------------------------------------------------------------- + */ + +void +gst_colorspace_table_init (GstColorspace *space) +{ + int i; + + for (i=0; i<256; i++) { + V_r_tab[i] = (0.419/0.299) * (i-128); + V_g_tab[i] = -(0.299/0.419) * (i-128); + U_g_tab[i] = -(0.114/0.331) * (i-128); + U_b_tab[i] = (0.587/0.331) * (i-128); + } +#if 0 + int CR, CB, i; + int *L_tab, *Cr_r_tab, *Cr_g_tab, *Cb_g_tab, *Cb_b_tab; + long *r_2_pix_alloc; + long *g_2_pix_alloc; + long *b_2_pix_alloc; + long depth = 32; + long red_mask = 0xff0000; + long green_mask = 0x00ff00; + long blue_mask = 0x0000ff; + + L_tab = space->L_tab = (int *)malloc(256*sizeof(int)); + Cr_r_tab = space->Cr_r_tab = (int *)malloc(256*sizeof(int)); + Cr_g_tab = space->Cr_g_tab = (int *)malloc(256*sizeof(int)); + Cb_g_tab = space->Cb_g_tab = (int *)malloc(256*sizeof(int)); + Cb_b_tab = space->Cb_b_tab = (int *)malloc(256*sizeof(int)); + + r_2_pix_alloc = (long *)malloc(768*sizeof(long)); + g_2_pix_alloc = (long *)malloc(768*sizeof(long)); + b_2_pix_alloc = (long *)malloc(768*sizeof(long)); + + if (L_tab == NULL || + Cr_r_tab == NULL || + Cr_g_tab == NULL || + Cb_g_tab == NULL || + Cb_b_tab == NULL || + r_2_pix_alloc == NULL || + g_2_pix_alloc == NULL || + b_2_pix_alloc == NULL) { + fprintf(stderr, "Could not get enough memory in InitColorDither\n"); + exit(1); + } + + for (i=0; i<256; i++) { + L_tab[i] = i; + /* + if (gammaCorrectFlag) { + L_tab[i] = GAMMA_CORRECTION(i); + } + */ + + CB = CR = i; + /* + if (chromaCorrectFlag) { + CB -= 128; + CB = CHROMA_CORRECTION128(CB); + CR -= 128; + CR = CHROMA_CORRECTION128(CR); + } + else + */ + { + CB -= 128; CR -= 128; + } + Cr_r_tab[i] = (0.419/0.299) * CR; + Cr_g_tab[i] = -(0.299/0.419) * CR; + Cb_g_tab[i] = -(0.114/0.331) * CB; + Cb_b_tab[i] = (0.587/0.331) * CB; + + } + + /* + * Set up entries 0-255 in rgb-to-pixel value tables. + */ + for (i = 0; i < 256; i++) { + r_2_pix_alloc[i + 256] = i >> (8 - number_of_bits_set(red_mask)); + r_2_pix_alloc[i + 256] <<= free_bits_at_bottom(red_mask); + g_2_pix_alloc[i + 256] = i >> (8 - number_of_bits_set(green_mask)); + g_2_pix_alloc[i + 256] <<= free_bits_at_bottom(green_mask); + b_2_pix_alloc[i + 256] = i >> (8 - number_of_bits_set(blue_mask)); + b_2_pix_alloc[i + 256] <<= free_bits_at_bottom(blue_mask); + /* + * If we have 16-bit output depth, then we double the value + * in the top word. This means that we can write out both + * pixels in the pixel doubling mode with one op. It is + * harmless in the normal case as storing a 32-bit value + * through a short pointer will lose the top bits anyway. + * A similar optimisation for Alpha for 64 bit has been + * prepared for, but is not yet implemented. + */ + if(!(depth == 32) && !(depth == 24)) { + + r_2_pix_alloc[i + 256] |= (r_2_pix_alloc[i + 256]) << 16; + g_2_pix_alloc[i + 256] |= (g_2_pix_alloc[i + 256]) << 16; + b_2_pix_alloc[i + 256] |= (b_2_pix_alloc[i + 256]) << 16; + + } +#ifdef SIXTYFOUR_BIT + if(depth == 32) { + + r_2_pix_alloc[i + 256] |= (r_2_pix_alloc[i + 256]) << 32; + g_2_pix_alloc[i + 256] |= (g_2_pix_alloc[i + 256]) << 32; + b_2_pix_alloc[i + 256] |= (b_2_pix_alloc[i + 256]) << 32; + + } +#endif + } + + /* + * Spread out the values we have to the rest of the array so that + * we do not need to check for overflow. + */ + for (i = 0; i < 256; i++) { + r_2_pix_alloc[i] = r_2_pix_alloc[256]; + r_2_pix_alloc[i+ 512] = r_2_pix_alloc[511]; + g_2_pix_alloc[i] = g_2_pix_alloc[256]; + g_2_pix_alloc[i+ 512] = g_2_pix_alloc[511]; + b_2_pix_alloc[i] = b_2_pix_alloc[256]; + b_2_pix_alloc[i+ 512] = b_2_pix_alloc[511]; + } + + space->r_2_pix = r_2_pix_alloc + 256; + space->g_2_pix = g_2_pix_alloc + 256; + space->b_2_pix = b_2_pix_alloc + 256; +#endif +} + +static void +gst_colorspace_yuv_to_rgb32(GstColorspace *space, + unsigned char *dest, + unsigned char *Y, + unsigned char *U, + unsigned char *V, + int width, int height) +{ + int x,y; + int src_rowstride; + int dest_rowstride; + + src_rowstride = ROUND_UP_4 (space->width); + dest_rowstride = width * 4; + for(y=0;y<height;y++){ + for(x=0;x<width;x++){ + dest[x*4 + 0] = 0; + dest[x*3 + 1] = CLAMP (Y[x] + V_r_tab[V[x/2]], 0, 255); + dest[x*3 + 2] = CLAMP (Y[x] + U_g_tab[U[x/2]] + V_g_tab[V[x/2]], 0, 255); + dest[x*3 + 3] = CLAMP (Y[x] + U_b_tab[U[x/2]], 0, 255); + } + Y += src_rowstride; + dest += dest_rowstride; + if (y&1) { + U += src_rowstride/2; + V += src_rowstride/2; + } + } +} + +static void +gst_colorspace_yuv_to_rgb24(GstColorspace *space, + unsigned char *dest, + unsigned char *Y, + unsigned char *U, + unsigned char *V, + int width, int height) +{ + int x,y; + int src_rowstride; + int dest_rowstride; + + src_rowstride = ROUND_UP_4 (space->width); + dest_rowstride = ROUND_UP_4 (width * 3); + for(y=0;y<height;y++){ + for(x=0;x<width;x++){ + dest[x*3 + 0] = CLAMP (Y[x] + V_r_tab[V[x/2]], 0, 255); + dest[x*3 + 1] = CLAMP (Y[x] + U_g_tab[U[x/2]] + V_g_tab[V[x/2]], 0, 255); + dest[x*3 + 2] = CLAMP (Y[x] + U_b_tab[U[x/2]], 0, 255); + } + Y += src_rowstride; + dest += dest_rowstride; + if (y&1) { + U += src_rowstride/2; + V += src_rowstride/2; + } + } +} + +static void +gst_colorspace_yuv_to_rgb16(GstColorspace *space, + unsigned char *dest, + unsigned char *Y, + unsigned char *U, + unsigned char *V, + int width, int height) +{ + int x,y; + int src_rowstride; + int dest_rowstride; + int r, g, b; + + src_rowstride = ROUND_UP_4 (space->width); + dest_rowstride = ROUND_UP_4 (width * 2); + for(y=0;y<height;y++){ + for(x=0;x<width;x++){ + r = CLAMP (Y[x] + V_r_tab[V[x/2]], 0, 255); + g = CLAMP (Y[x] + U_g_tab[U[x/2]] + V_g_tab[V[x/2]], 0, 255); + b = CLAMP (Y[x] + U_b_tab[U[x/2]], 0, 255); + *(unsigned short *)(dest + x*2) = ((r&0xf8)<<8) | ((g&0xfc)<<3) | (b>>3); + } + Y += src_rowstride; + dest += dest_rowstride; + if (y&1) { + U += src_rowstride/2; + V += src_rowstride/2; + } + } +} + +#ifdef HAVE_LIBMMX +static mmx_t MMX_80w = (mmx_t)(long long)0x0080008000800080LL; /*dd 00080 0080h, 000800080h */ + +static mmx_t MMX_00FFw = (mmx_t)(long long)0x00ff00ff00ff00ffLL; /*dd 000FF 00FFh, 000FF00FFh */ +static mmx_t MMX_FF00w = (mmx_t)(long long)0xff00ff00ff00ff00LL; /*dd 000FF 00FFh, 000FF00FFh */ + +static mmx_t MMX32_Vredcoeff = (mmx_t)(long long)0x0059005900590059LL; +static mmx_t MMX32_Ubluecoeff = (mmx_t)(long long)0x0072007200720072LL; +static mmx_t MMX32_Ugrncoeff = (mmx_t)(long long)0xffeaffeaffeaffeaLL; +static mmx_t MMX32_Vgrncoeff = (mmx_t)(long long)0xffd2ffd2ffd2ffd2LL; + +static void +gst_colorspace_yuv_to_bgr32_mmx(tables, lum, cr, cb, out, rows, cols) + GstColorspaceYUVTables *tables; + unsigned char *lum; + unsigned char *cr; + unsigned char *cb; + unsigned char *out; + int cols, rows; + +{ + guint32 *row1 = (guint32 *)out; /* 32 bit target */ + int cols4 = cols>>2; + + int y, x; + + for (y=rows>>1; y; y--) { + for (x=cols4; x; x--) { + + /* create Cr (result in mm1) */ + movd_m2r(*(mmx_t *)cb, mm1); /* 0 0 0 0 v3 v2 v1 v0 */ + pxor_r2r(mm7, mm7); /* 00 00 00 00 00 00 00 00 */ + movd_m2r(*(mmx_t *)lum, mm2); /* 0 0 0 0 l3 l2 l1 l0 */ + punpcklbw_r2r(mm7, mm1); /* 0 v3 0 v2 00 v1 00 v0 */ + punpckldq_r2r(mm1, mm1); /* 00 v1 00 v0 00 v1 00 v0 */ + psubw_m2r(MMX_80w, mm1); /* mm1-128:r1 r1 r0 r0 r1 r1 r0 r0 */ + + /* create Cr_g (result in mm0) */ + movq_r2r(mm1, mm0); /* r1 r1 r0 r0 r1 r1 r0 r0 */ + pmullw_m2r(MMX32_Vgrncoeff, mm0); /* red*-46dec=0.7136*64 */ + pmullw_m2r(MMX32_Vredcoeff, mm1); /* red*89dec=1.4013*64 */ + psraw_i2r(6, mm0); /* red=red/64 */ + psraw_i2r(6, mm1); /* red=red/64 */ + + /* create L1 L2 (result in mm2,mm4) */ + /* L2=lum+cols */ + movq_m2r(*(mmx_t *)(lum+cols),mm3); /* 0 0 0 0 L3 L2 L1 L0 */ + punpckldq_r2r(mm3, mm2); /* L3 L2 L1 L0 l3 l2 l1 l0 */ + movq_r2r(mm2, mm4); /* L3 L2 L1 L0 l3 l2 l1 l0 */ + pand_m2r(MMX_FF00w, mm2); /* L3 0 L1 0 l3 0 l1 0 */ + pand_m2r(MMX_00FFw, mm4); /* 0 L2 0 L0 0 l2 0 l0 */ + psrlw_i2r(8, mm2); /* 0 L3 0 L1 0 l3 0 l1 */ + + /* create R (result in mm6) */ + movq_r2r(mm2, mm5); /* 0 L3 0 L1 0 l3 0 l1 */ + movq_r2r(mm4, mm6); /* 0 L2 0 L0 0 l2 0 l0 */ + paddsw_r2r(mm1, mm5); /* lum1+red:x R3 x R1 x r3 x r1 */ + paddsw_r2r(mm1, mm6); /* lum1+red:x R2 x R0 x r2 x r0 */ + packuswb_r2r(mm5, mm5); /* R3 R1 r3 r1 R3 R1 r3 r1 */ + packuswb_r2r(mm6, mm6); /* R2 R0 r2 r0 R2 R0 r2 r0 */ + pxor_r2r(mm7, mm7); /* 00 00 00 00 00 00 00 00 */ + punpcklbw_r2r(mm5, mm6); /* R3 R2 R1 R0 r3 r2 r1 r0 */ + + /* create Cb (result in mm1) */ + movd_m2r(*(mmx_t *)cr, mm1); /* 0 0 0 0 u3 u2 u1 u0 */ + punpcklbw_r2r(mm7, mm1); /* 0 u3 0 u2 00 u1 00 u0 */ + punpckldq_r2r(mm1, mm1); /* 00 u1 00 u0 00 u1 00 u0 */ + psubw_m2r(MMX_80w, mm1); /* mm1-128:u1 u1 u0 u0 u1 u1 u0 u0 */ + /* create Cb_g (result in mm5) */ + movq_r2r(mm1, mm5); /* u1 u1 u0 u0 u1 u1 u0 u0 */ + pmullw_m2r(MMX32_Ugrncoeff, mm5); /* blue*-109dec=1.7129*64 */ + pmullw_m2r(MMX32_Ubluecoeff, mm1); /* blue*114dec=1.78125*64 */ + psraw_i2r(6, mm5); /* blue=red/64 */ + psraw_i2r(6, mm1); /* blue=blue/64 */ + + /* create G (result in mm7) */ + movq_r2r(mm2, mm3); /* 0 L3 0 L1 0 l3 0 l1 */ + movq_r2r(mm4, mm7); /* 0 L2 0 L0 0 l2 0 l1 */ + paddsw_r2r(mm5, mm3); /* lum1+Cb_g:x G3t x G1t x g3t x g1t */ + paddsw_r2r(mm5, mm7); /* lum1+Cb_g:x G2t x G0t x g2t x g0t */ + paddsw_r2r(mm0, mm3); /* lum1+Cr_g:x G3 x G1 x g3 x g1 */ + paddsw_r2r(mm0, mm7); /* lum1+blue:x G2 x G0 x g2 x g0 */ + packuswb_r2r(mm3, mm3); /* G3 G1 g3 g1 G3 G1 g3 g1 */ + packuswb_r2r(mm7, mm7); /* G2 G0 g2 g0 G2 G0 g2 g0 */ + punpcklbw_r2r(mm3, mm7); /* G3 G2 G1 G0 g3 g2 g1 g0 */ + + /* create B (result in mm5) */ + movq_r2r(mm2, mm3); /* 0 L3 0 L1 0 l3 0 l1 */ + movq_r2r(mm4, mm5); /* 0 L2 0 L0 0 l2 0 l1 */ + paddsw_r2r(mm1, mm3); /* lum1+blue:x B3 x B1 x b3 x b1 */ + paddsw_r2r(mm1, mm5); /* lum1+blue:x B2 x B0 x b2 x b0 */ + packuswb_r2r(mm3, mm3); /* B3 B1 b3 b1 B3 B1 b3 b1 */ + packuswb_r2r(mm5, mm5); /* B2 B0 b2 b0 B2 B0 b2 b0 */ + punpcklbw_r2r(mm3, mm5); /* B3 B2 B1 B0 b3 b2 b1 b0 */ + + /* fill destination row1 (needed are mm6=Rr,mm7=Gg,mm5=Bb) */ + + pxor_r2r(mm2, mm2); /* 0 0 0 0 0 0 0 0 */ + pxor_r2r(mm4, mm4); /* 0 0 0 0 0 0 0 0 */ + movq_r2r(mm6, mm1); /* R3 R2 R1 R0 r3 r2 r1 r0 */ + movq_r2r(mm5, mm3); /* B3 B2 B1 B0 b3 b2 b1 b0 */ + /* process lower lum */ + punpcklbw_r2r(mm4, mm1); /* 0 r3 0 r2 0 r1 0 r0 */ + punpcklbw_r2r(mm4, mm3); /* 0 b3 0 b2 0 b1 0 b0 */ + movq_r2r(mm1, mm2); /* 0 r3 0 r2 0 r1 0 r0 */ + movq_r2r(mm3, mm0); /* 0 b3 0 b2 0 b1 0 b0 */ + punpcklwd_r2r(mm1, mm3); /* 0 r1 0 b1 0 r0 0 b0 */ + punpckhwd_r2r(mm2, mm0); /* 0 r3 0 b3 0 r2 0 b2 */ + + pxor_r2r(mm2, mm2); /* 0 0 0 0 0 0 0 0 */ + movq_r2r(mm7, mm1); /* G3 G2 G1 G0 g3 g2 g1 g0 */ + punpcklbw_r2r(mm1, mm2); /* g3 0 g2 0 g1 0 g0 0 */ + punpcklwd_r2r(mm4, mm2); /* 0 0 g1 0 0 0 g0 0 */ + por_r2r(mm3, mm2); /* 0 r1 g1 b1 0 r0 g0 b0 */ + movq_r2m(mm2, *(mmx_t *)row1); /* wrote out ! row1 */ + + pxor_r2r(mm2, mm2); /* 0 0 0 0 0 0 0 0 */ + punpcklbw_r2r(mm1, mm4); /* g3 0 g2 0 g1 0 g0 0 */ + punpckhwd_r2r(mm2, mm4); /* 0 0 g3 0 0 0 g2 0 */ + por_r2r(mm0, mm4); /* 0 r3 g3 b3 0 r2 g2 b2 */ + movq_r2m(mm4, *(mmx_t *)(row1+2)); /* wrote out ! row1 */ + + /* fill destination row2 (needed are mm6=Rr,mm7=Gg,mm5=Bb) */ + /* this can be done "destructive" */ + pxor_r2r(mm2, mm2); /* 0 0 0 0 0 0 0 0 */ + punpckhbw_r2r(mm2, mm6); /* 0 R3 0 R2 0 R1 0 R0 */ + punpckhbw_r2r(mm1, mm5); /* G3 B3 G2 B2 G1 B1 G0 B0 */ + movq_r2r(mm5, mm1); /* G3 B3 G2 B2 G1 B1 G0 B0 */ + punpcklwd_r2r(mm6, mm1); /* 0 R1 G1 B1 0 R0 G0 B0 */ + movq_r2m(mm1, *(mmx_t *)(row1+cols)); /* wrote out ! row2 */ + punpckhwd_r2r(mm6, mm5); /* 0 R3 G3 B3 0 R2 G2 B2 */ + movq_r2m(mm5, *(mmx_t *)(row1+cols+2)); /* wrote out ! row2 */ + + lum+=4; + cr+=2; + cb+=2; + row1 +=4; + } + lum += cols; + row1 += cols; + } + + emms(); + +} +#endif + diff --git a/gst/colorspace/yuv2rgb.h b/gst/colorspace/yuv2rgb.h new file mode 100644 index 00000000..be173bca --- /dev/null +++ b/gst/colorspace/yuv2rgb.h @@ -0,0 +1,97 @@ +/* + * Copyright (c) 1995 The Regents of the University of California. + * All rights reserved. + * + * Permission to use, copy, modify, and distribute this software and its + * documentation for any purpose, without fee, and without written agreement is + * hereby granted, provided that the above copyright notice and the following + * two paragraphs appear in all copies of this software. + * + * IN NO EVENT SHALL THE UNIVERSITY OF CALIFORNIA BE LIABLE TO ANY PARTY FOR + * DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING OUT + * OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF THE UNIVERSITY OF + * CALIFORNIA HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + * + * THE UNIVERSITY OF CALIFORNIA SPECIFICALLY DISCLAIMS ANY WARRANTIES, + * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY + * AND FITNESS FOR A PARTICULAR PURPOSE. THE SOFTWARE PROVIDED HEREUNDER IS + * ON AN "AS IS" BASIS, AND THE UNIVERSITY OF CALIFORNIA HAS NO OBLIGATION TO + * PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS. + */ + +#ifndef __YUV2RGB_H__ +#define __YUV2RGB_H__ + +#include <gst/gst.h> +#include <gstcolorspace.h> + +G_BEGIN_DECLS + +#if 0 +typedef struct _GstColorspaceYUVTables GstColorspaceYUVTables; + +struct _GstColorspaceYUVTables { + int gammaCorrectFlag; + double gammaCorrect; + int chromaCorrectFlag; + double chromaCorrect; + + int *L_tab, *Cr_r_tab, *Cr_g_tab, *Cb_g_tab, *Cb_b_tab; + + /* + * We define tables that convert a color value between -256 and 512 + * into the R, G and B parts of the pixel. The normal range is 0-255. + **/ + + long *r_2_pix; + long *g_2_pix; + long *b_2_pix; +}; + + +typedef struct _GstColorspaceConverter GstColorspaceConverter; +typedef void (*GstColorspaceConvertFunction) (GstColorspaceConverter *space, guchar *src, guchar *dest); + +struct _GstColorspaceConverter { + guint width; + guint height; + guint insize; + guint outsize; + /* private */ + GstColorspaceYUVTables *color_tables; + GstColorspaceConvertFunction convert; +}; +#endif + +void gst_colorspace_table_init (GstColorspace *space); + +void gst_colorspace_I420_to_rgb32(GstColorspace *space, + unsigned char *src, unsigned char *dest); +void gst_colorspace_I420_to_rgb24(GstColorspace *space, + unsigned char *src, unsigned char *dest); +void gst_colorspace_I420_to_rgb16(GstColorspace *space, + unsigned char *src, unsigned char *dest); +void gst_colorspace_YV12_to_rgb32(GstColorspace *space, + unsigned char *src, unsigned char *dest); +void gst_colorspace_YV12_to_rgb24(GstColorspace *space, + unsigned char *src, unsigned char *dest); +void gst_colorspace_YV12_to_rgb16(GstColorspace *space, + unsigned char *src, unsigned char *dest); + +#if 0 +GstColorspaceYUVTables * gst_colorspace_init_yuv(long depth, + long red_mask, long green_mask, long blue_mask); +#endif + + +#if 0 +GstColorspaceConverter* gst_colorspace_yuv2rgb_get_converter (const GstCaps *from, const GstCaps *to); +#define gst_colorspace_convert(converter, src, dest) \ + (converter)->convert((converter), (src), (dest)) +void gst_colorspace_converter_destroy (GstColorspaceConverter *space); +#endif + +G_END_DECLS + +#endif + |