diff options
| -rw-r--r-- | ext/lcs/Makefile.am | 10 | ||||
| -rw-r--r-- | ext/lcs/gstcolorspace.c | 532 |
2 files changed, 542 insertions, 0 deletions
diff --git a/ext/lcs/Makefile.am b/ext/lcs/Makefile.am new file mode 100644 index 00000000..8a4c1d6f --- /dev/null +++ b/ext/lcs/Makefile.am @@ -0,0 +1,10 @@ +plugindir = $(libdir)/gst + +plugin_LTLIBRARIES = libgstcolorspace.la + +libgstcolorspace_la_SOURCES = gstcolorspace.c +libgstcolorspace_la_CFLAGS = $(GST_CFLAGS) $(LCS_CFLAGS) +libgstcolorspace_la_LIBADD = $(LCS_LIBS) +libgstcolorspace_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS) + +noinst_HEADERS = diff --git a/ext/lcs/gstcolorspace.c b/ext/lcs/gstcolorspace.c new file mode 100644 index 00000000..8bad1b96 --- /dev/null +++ b/ext/lcs/gstcolorspace.c @@ -0,0 +1,532 @@ +/* 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 <gst/gst.h> +#include <lcs/lcs.h> + +#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_LCS, +} GstColorSpaceConverterType; + +struct _GstColorspace { + GstElement element; + + GstPad *sinkpad, *srcpad; + + LCSConverter *converter; + + GstColorSpaceConverterType type; + gint width, height; + gboolean disabled; + + GstCaps *sinkcaps; + + GstBufferPool *pool; +}; + +struct _GstColorspaceClass { + GstElementClass parent_class; +}; + + +static GstElementDetails colorspace_details = { + "Colorspace converter", + "Filter/Effect", + "Converts video from one colorspace to another", + VERSION, + "Wim Taymans <wim.taymans@chello.be>", + "(C) 2001", +}; + + +/* Stereo signals and args */ +enum { + /* FILL ME */ + LAST_SIGNAL +}; + +enum { + ARG_0, + ARG_SOURCE, + ARG_DEST, +}; + +GST_PADTEMPLATE_FACTORY (colorspace_src_template_factory, + "src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_CAPS_NEW ( + "colorspace_src", + "video/raw", + "format", GST_PROPS_LIST ( + GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")), + GST_PROPS_FOURCC (GST_STR_FOURCC ("YUY2")), + GST_PROPS_FOURCC (GST_STR_FOURCC ("RGB ")) + ), + "width", GST_PROPS_INT_RANGE (0, G_MAXINT), + "height", GST_PROPS_INT_RANGE (0, G_MAXINT) + ) +) + +GST_PADTEMPLATE_FACTORY (colorspace_sink_template_factory, + "sink", + GST_PAD_SINK, + GST_PAD_ALWAYS, + GST_CAPS_NEW ( + "colorspace_sink", + "video/raw", + "format", GST_PROPS_LIST ( + GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")), + GST_PROPS_FOURCC (GST_STR_FOURCC ("YUY2")), + GST_PROPS_FOURCC (GST_STR_FOURCC ("RGB ")) + ), + "width", GST_PROPS_INT_RANGE (0, G_MAXINT), + "height", GST_PROPS_INT_RANGE (0, G_MAXINT) + ) +) + +static GType gst_colorspace_get_type (void); + +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 GstPadConnectReturn + gst_colorspace_sinkconnect (GstPad *pad, GstCaps *caps); +static GstPadConnectReturn + gst_colorspace_srcconnect (GstPad *pad, GstCaps *caps); +static GstPadConnectReturn + gst_colorspace_srcconnect_func (GstPad *pad, GstCaps *caps, gboolean newcaps); +static void gst_colorspace_chain (GstPad *pad, GstBuffer *buf); +static GstElementStateReturn + gst_colorspace_change_state (GstElement *element); + +static GstElementClass *parent_class = NULL; +/*static guint gst_colorspace_signals[LAST_SIGNAL] = { 0 }; */ + +static GstBufferPool* +colorspace_get_bufferpool (GstPad *pad) +{ + GstColorspace *space; + + space = GST_COLORSPACE (gst_pad_get_parent (pad)); + + if (space->type == GST_COLORSPACE_NONE && !space->disabled) + return gst_pad_get_bufferpool (space->srcpad); + else + return NULL; +} + +static const LCSFormat* +colorspace_find_lcs_format (GstCaps *caps) +{ + const LCSFormat *format = NULL; + guint32 space; + + gst_caps_get_fourcc_int (caps, "format", &space); + + if (space == GST_MAKE_FOURCC ('R','G','B',' ')) { + gint mask, bpp; + + gulong red_bits, red_shift; + gulong green_bits, green_shift; + gulong blue_bits, blue_shift; + + gst_caps_get_int (caps, "bpp", &bpp); + gst_caps_get_int (caps, "red_mask", &mask); + lcs_utils_mask_to_shift (mask, &red_bits, &red_shift); + gst_caps_get_int (caps, "green_mask", &mask); + lcs_utils_mask_to_shift (mask, &green_bits, &green_shift); + gst_caps_get_int (caps, "blue_mask", &mask); + lcs_utils_mask_to_shift (mask, &blue_bits, &blue_shift); + + format = lcs_format_new_simple_rgb_packed (red_bits, green_bits, blue_bits, 0, + red_shift, green_shift, blue_shift, 0, bpp); + + g_print ("%lu %lu %lu %lu %lu %lu %s\n", red_bits, green_bits, blue_bits, + red_shift, green_shift, blue_shift, lcs_format_get_layout (format)); + } + else { + gchar fourcc[5]; + + lcs_utils_fourcc_to_string (space, fourcc); + format = lcs_alias_find_format (fourcc); + } + + return format; +} + +static gboolean +colorspace_setup_converter (GstColorspace *space, GstCaps *from_caps, GstCaps *to_caps) +{ + const LCSFormat *from_format = NULL; + const LCSFormat *to_format = NULL; + + g_return_val_if_fail (to_caps != NULL, FALSE); + g_return_val_if_fail (from_caps != NULL, FALSE); + + from_format = colorspace_find_lcs_format (from_caps); + to_format = colorspace_find_lcs_format (to_caps); + + space->converter = lcs_get_converter (from_format, to_format, LCS_FLAG_FAST); + + if (space->converter) { + space->type = GST_COLORSPACE_LCS; + return TRUE; + } + + return FALSE; +} + +static GstCaps* +gst_colorspace_getcaps (GstPad *pad, GstCaps *caps) +{ + GstColorspace *space; + GstCaps *result; + GstCaps *peercaps; + GstCaps *ourcaps; + + space = GST_COLORSPACE (gst_pad_get_parent (pad)); + + /* we can do everything our peer can... */ + peercaps = gst_caps_copy (gst_pad_get_allowed_caps (space->srcpad)); + /* and our own template of course */ + ourcaps = gst_caps_copy (gst_pad_get_padtemplate_caps (pad)); + + /* merge them together, we prefer the peercaps first */ + result = gst_caps_prepend (ourcaps, peercaps); + + return result; +} + +static GstPadConnectReturn +gst_colorspace_sinkconnect (GstPad *pad, GstCaps *caps) +{ + GstColorspace *space; + GstPad *peer; + + space = GST_COLORSPACE (gst_pad_get_parent (pad)); + + if (!GST_CAPS_IS_FIXED (caps)) { + return GST_PAD_CONNECT_DELAYED; + } + + gst_caps_get_int (caps, "width", &space->width); + gst_caps_get_int (caps, "height", &space->height); + + GST_INFO (GST_CAT_PROPERTIES, "size: %dx%d", space->width, space->height); + + space->sinkcaps = caps; + + peer = gst_pad_get_peer (pad); + if (peer) { + if (!gst_colorspace_srcconnect_func (pad, gst_pad_get_allowed_caps (space->srcpad), FALSE)) { + space->sinkcaps = NULL; + return GST_PAD_CONNECT_REFUSED; + } + } + + return GST_PAD_CONNECT_OK; +} + +static GstPadConnectReturn +gst_colorspace_srcconnect (GstPad *pad, GstCaps *caps) +{ + return gst_colorspace_srcconnect_func (pad, caps, TRUE); +} + +static GstPadConnectReturn +gst_colorspace_srcconnect_func (GstPad *pad, GstCaps *caps, gboolean newcaps) +{ + GstColorspace *space; + GstCaps *peercaps; + GstCaps *ourcaps; + + space = GST_COLORSPACE (gst_pad_get_parent (pad)); + + /* we cannot operate if we didn't get src caps */ + ourcaps = space->sinkcaps; + if (!ourcaps) { + if (newcaps) + gst_pad_recalc_allowed_caps (space->sinkpad); + + return GST_PAD_CONNECT_DELAYED; + } + + /* first see if we can do the format natively by filtering the peer caps + * with our incomming caps */ + peercaps = gst_caps_intersect (caps, ourcaps); + if (peercaps) { + /* see if the peer likes it too, it should as the caps say so.. */ + if (gst_pad_try_set_caps (space->srcpad, peercaps)) { + space->type = GST_COLORSPACE_NONE; + space->disabled = FALSE; + return GST_PAD_CONNECT_DONE; + } + } + /* then see what the peer has that matches the size */ + peercaps = gst_caps_intersect (caps, + GST_CAPS_NEW ( + "colorspace_filter", + "video/raw", + "width", GST_PROPS_INT (space->width), + "height", GST_PROPS_INT (space->height) + )); + + /* we are looping over the caps, so we have to get rid of the lists */ + peercaps = gst_caps_normalize (peercaps); + + /* loop over all possibilities and select the first one we can convert and + * is accepted by the peer */ + while (peercaps) { + if (colorspace_setup_converter (space, ourcaps, peercaps)) { + if (gst_pad_try_set_caps (space->srcpad, peercaps)) { + space->disabled = FALSE; + return GST_PAD_CONNECT_DONE; + } + } + peercaps = peercaps->next; + } + + gst_element_error (GST_ELEMENT (space), "could not agree on caps with peer pads"); + /* we disable ourself here */ + space->disabled = TRUE; + + return GST_PAD_CONNECT_REFUSED; +} + +static GType +gst_colorspace_get_type (void) +{ + static GType colorspace_type = 0; + + if (!colorspace_type) { + static const GTypeInfo colorspace_info = { + sizeof(GstColorspaceClass), + NULL, + NULL, + (GClassInitFunc)gst_colorspace_class_init, + NULL, + NULL, + sizeof(GstColorspace), + 0, + (GInstanceInitFunc)gst_colorspace_init, + }; + colorspace_type = g_type_register_static(GST_TYPE_ELEMENT, "GstColorspaceLCS", &colorspace_info, 0); + } + return colorspace_type; +} + +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; +} + +static void +gst_colorspace_init (GstColorspace *space) +{ + space->sinkpad = gst_pad_new_from_template ( + GST_PADTEMPLATE_GET (colorspace_sink_template_factory), "sink"); + gst_pad_set_connect_function (space->sinkpad, gst_colorspace_sinkconnect); + gst_pad_set_getcaps_function (space->sinkpad, gst_colorspace_getcaps); + gst_pad_set_bufferpool_function (space->sinkpad, colorspace_get_bufferpool); + 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_PADTEMPLATE_GET (colorspace_src_template_factory), "src"); + gst_element_add_pad(GST_ELEMENT(space),space->srcpad); + gst_pad_set_connect_function (space->srcpad, gst_colorspace_srcconnect); + + space->pool = NULL; + space->disabled = TRUE; +} + +static void +gst_colorspace_chain (GstPad *pad,GstBuffer *buf) +{ + GstColorspace *space; + GstBuffer *outbuf = NULL; + + 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 (space->disabled) { + gst_buffer_unref (buf); + return; + } + + if (space->type == GST_COLORSPACE_NONE) { + outbuf = buf; + } + else { + if (space->pool) { + outbuf = gst_buffer_new_from_pool (space->pool, 0, 0); + } + + if (!outbuf) { + guint size; + outbuf = gst_buffer_new (); + + lcs_format_buffer_size ( + lcs_convert_get_dest_format (space->converter), + space->width, space->height, &size); + + GST_BUFFER_SIZE (outbuf) = size; + GST_BUFFER_DATA (outbuf) = g_malloc (GST_BUFFER_SIZE (outbuf)); + } + + if (space->type == GST_COLORSPACE_LCS) { + lcs_convert_auto (space->converter, + GST_BUFFER_DATA (buf), + GST_BUFFER_DATA (outbuf), + space->width, space->height); + } + + GST_BUFFER_TIMESTAMP (outbuf) = GST_BUFFER_TIMESTAMP (buf); + + gst_buffer_unref (buf); + } + gst_pad_push (space->srcpad, 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: + space->pool = gst_pad_get_bufferpool (space->srcpad); + break; + case GST_STATE_PLAYING_TO_PAUSED: + space->pool = NULL; + break; + } + + parent_class->change_state (element); + + return GST_STATE_SUCCESS; +} + +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 (GModule *module, GstPlugin *plugin) +{ + GstElementFactory *factory; + + lcs_init (NULL, NULL); + + factory = gst_elementfactory_new ("colorspacelcs", GST_TYPE_COLORSPACE, + &colorspace_details); + g_return_val_if_fail (factory != NULL, FALSE); + + gst_elementfactory_add_padtemplate (factory, + GST_PADTEMPLATE_GET (colorspace_src_template_factory)); + gst_elementfactory_add_padtemplate (factory, + GST_PADTEMPLATE_GET (colorspace_sink_template_factory)); + + gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory)); + + return TRUE; +} + +GstPluginDesc plugin_desc = { + GST_VERSION_MAJOR, + GST_VERSION_MINOR, + "colorspacelcs", + plugin_init +}; + + + + + + + + |