2003-07-02 06:31:33 +00:00
|
|
|
/* GStreamer
|
|
|
|
* Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
|
|
|
|
* Copyright (C) <2003> David Schleef <ds@schleef.org>
|
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
|
|
/*#define DEBUG_ENABLED */
|
|
|
|
#include <gstvideobalance.h>
|
2003-07-02 06:49:37 +00:00
|
|
|
#include <string.h>
|
|
|
|
#include <math.h>
|
2003-07-02 06:31:33 +00:00
|
|
|
|
|
|
|
|
|
|
|
/* elementfactory information */
|
|
|
|
static GstElementDetails videobalance_details = {
|
2003-07-02 06:49:37 +00:00
|
|
|
"Video Balance control",
|
2003-07-02 06:31:33 +00:00
|
|
|
"Filter/Video",
|
2003-07-02 06:49:37 +00:00
|
|
|
"Adjusts brightness, contrast, hue, saturation on a video stream",
|
2003-11-02 22:54:47 +00:00
|
|
|
"David Schleef <ds@schleef.org>"
|
2003-07-02 06:31:33 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
/* GstVideobalance signals and args */
|
|
|
|
enum {
|
|
|
|
/* FILL ME */
|
|
|
|
LAST_SIGNAL
|
|
|
|
};
|
|
|
|
|
|
|
|
enum {
|
|
|
|
ARG_0,
|
2003-07-02 06:49:37 +00:00
|
|
|
ARG_CONTRAST,
|
|
|
|
ARG_BRIGHTNESS,
|
|
|
|
ARG_HUE,
|
|
|
|
ARG_SATURATION,
|
2003-07-02 06:31:33 +00:00
|
|
|
/* FILL ME */
|
|
|
|
};
|
|
|
|
|
|
|
|
static void gst_videobalance_class_init (GstVideobalanceClass *klass);
|
2003-11-02 22:54:47 +00:00
|
|
|
static void gst_videobalance_base_init (GstVideobalanceClass *klass);
|
2003-07-02 06:31:33 +00:00
|
|
|
static void gst_videobalance_init (GstVideobalance *videobalance);
|
|
|
|
|
|
|
|
static void gst_videobalance_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec);
|
|
|
|
static void gst_videobalance_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec);
|
|
|
|
|
|
|
|
static void gst_videobalance_planar411(GstVideofilter *videofilter, void *dest, void *src);
|
|
|
|
static void gst_videobalance_setup(GstVideofilter *videofilter);
|
|
|
|
|
|
|
|
static GstVideobalanceClass *this_class = NULL;
|
|
|
|
static GstVideofilterClass *parent_class = NULL;
|
|
|
|
static GstElementClass *element_class = NULL;
|
|
|
|
|
|
|
|
GType
|
|
|
|
gst_videobalance_get_type (void)
|
|
|
|
{
|
|
|
|
static GType videobalance_type = 0;
|
|
|
|
|
|
|
|
if (!videobalance_type) {
|
|
|
|
static const GTypeInfo videobalance_info = {
|
2003-11-02 22:54:47 +00:00
|
|
|
sizeof(GstVideobalanceClass),
|
|
|
|
(GBaseInitFunc)gst_videobalance_base_init,
|
2003-07-02 06:31:33 +00:00
|
|
|
NULL,
|
|
|
|
(GClassInitFunc)gst_videobalance_class_init,
|
|
|
|
NULL,
|
|
|
|
NULL,
|
|
|
|
sizeof(GstVideobalance),
|
|
|
|
0,
|
|
|
|
(GInstanceInitFunc)gst_videobalance_init,
|
|
|
|
};
|
|
|
|
videobalance_type = g_type_register_static(GST_TYPE_VIDEOFILTER, "GstVideobalance", &videobalance_info, 0);
|
|
|
|
}
|
|
|
|
return videobalance_type;
|
|
|
|
}
|
|
|
|
|
|
|
|
static GstVideofilterFormat gst_videobalance_formats[] = {
|
|
|
|
{ "I420", 12, gst_videobalance_planar411, },
|
|
|
|
};
|
|
|
|
|
2003-11-02 22:54:47 +00:00
|
|
|
static GstPadTemplate *gst_videobalance_sink_template_factory (void);
|
|
|
|
static GstPadTemplate *gst_videobalance_src_template_factory (void);
|
|
|
|
|
|
|
|
static void
|
|
|
|
gst_videobalance_base_init (GstVideobalanceClass *klass)
|
|
|
|
{
|
|
|
|
GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
|
|
|
|
|
|
|
|
gst_element_class_add_pad_template (element_class,
|
|
|
|
GST_PAD_TEMPLATE_GET (gst_videobalance_sink_template_factory));
|
|
|
|
gst_element_class_add_pad_template (element_class,
|
|
|
|
GST_PAD_TEMPLATE_GET (gst_videobalance_src_template_factory));
|
|
|
|
gst_element_class_set_details (element_class, &videobalance_details);
|
|
|
|
}
|
|
|
|
|
2003-07-02 06:31:33 +00:00
|
|
|
static void
|
|
|
|
gst_videobalance_class_init (GstVideobalanceClass *klass)
|
|
|
|
{
|
|
|
|
GObjectClass *gobject_class;
|
|
|
|
GstElementClass *gstelement_class;
|
|
|
|
GstVideofilterClass *gstvideofilter_class;
|
|
|
|
int i;
|
|
|
|
|
|
|
|
gobject_class = (GObjectClass*)klass;
|
|
|
|
gstelement_class = (GstElementClass*)klass;
|
|
|
|
gstvideofilter_class = (GstVideofilterClass *)klass;
|
|
|
|
|
2003-07-02 06:49:37 +00:00
|
|
|
g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_CONTRAST,
|
|
|
|
g_param_spec_double("contrast","Contrast","contrast",
|
|
|
|
0, 2, 1, G_PARAM_READWRITE));
|
|
|
|
g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_BRIGHTNESS,
|
|
|
|
g_param_spec_double("brightness","Brightness","brightness",
|
|
|
|
-1, 1, 0, G_PARAM_READWRITE));
|
|
|
|
g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_HUE,
|
|
|
|
g_param_spec_double("hue","Hue","hue",
|
|
|
|
-1, 1, 0, G_PARAM_READWRITE));
|
|
|
|
g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_SATURATION,
|
|
|
|
g_param_spec_double("saturation","Saturation","saturation",
|
|
|
|
0, 2, 1, G_PARAM_READWRITE));
|
2003-07-02 06:31:33 +00:00
|
|
|
|
|
|
|
this_class = klass;
|
|
|
|
parent_class = g_type_class_ref(GST_TYPE_VIDEOFILTER);
|
|
|
|
element_class = g_type_class_ref(GST_TYPE_ELEMENT);
|
|
|
|
|
|
|
|
gobject_class->set_property = gst_videobalance_set_property;
|
|
|
|
gobject_class->get_property = gst_videobalance_get_property;
|
|
|
|
|
|
|
|
gstvideofilter_class->setup = gst_videobalance_setup;
|
|
|
|
|
|
|
|
for(i=0;i<G_N_ELEMENTS(gst_videobalance_formats);i++){
|
2003-07-02 08:10:35 +00:00
|
|
|
gst_videofilter_class_add_format(gstvideofilter_class,
|
|
|
|
gst_videobalance_formats + i);
|
2003-07-02 06:31:33 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static GstCaps *gst_videobalance_get_capslist(void)
|
|
|
|
{
|
|
|
|
GstVideofilterClass *klass;
|
|
|
|
|
|
|
|
klass = g_type_class_ref(GST_TYPE_VIDEOFILTER);
|
|
|
|
|
|
|
|
return gst_videofilter_class_get_capslist(klass);
|
|
|
|
}
|
|
|
|
|
|
|
|
static GstPadTemplate *
|
|
|
|
gst_videobalance_src_template_factory(void)
|
|
|
|
{
|
|
|
|
static GstPadTemplate *templ = NULL;
|
|
|
|
|
|
|
|
if(!templ){
|
2003-07-14 08:32:38 +00:00
|
|
|
GstCaps *caps = GST_CAPS_NEW("src","video/x-raw-yuv",
|
2003-07-02 06:31:33 +00:00
|
|
|
"width", GST_PROPS_INT_RANGE (0, G_MAXINT),
|
2003-07-14 08:32:38 +00:00
|
|
|
"height", GST_PROPS_INT_RANGE (0, G_MAXINT),
|
|
|
|
"framerate", GST_PROPS_FLOAT_RANGE (0, G_MAXFLOAT));
|
2003-07-02 06:31:33 +00:00
|
|
|
|
|
|
|
caps = gst_caps_intersect(caps, gst_videobalance_get_capslist ());
|
|
|
|
|
|
|
|
templ = GST_PAD_TEMPLATE_NEW("src", GST_PAD_SRC, GST_PAD_ALWAYS, caps);
|
|
|
|
}
|
|
|
|
return templ;
|
|
|
|
}
|
|
|
|
|
|
|
|
static GstPadTemplate *
|
|
|
|
gst_videobalance_sink_template_factory(void)
|
|
|
|
{
|
|
|
|
static GstPadTemplate *templ = NULL;
|
|
|
|
|
|
|
|
if(!templ){
|
2003-07-14 08:32:38 +00:00
|
|
|
GstCaps *caps = GST_CAPS_NEW("src","video/x-raw-yuv",
|
2003-07-02 06:31:33 +00:00
|
|
|
"width", GST_PROPS_INT_RANGE (0, G_MAXINT),
|
2003-07-14 08:32:38 +00:00
|
|
|
"height", GST_PROPS_INT_RANGE (0, G_MAXINT),
|
|
|
|
"framerate", GST_PROPS_FLOAT_RANGE (0, G_MAXFLOAT));
|
2003-07-02 06:31:33 +00:00
|
|
|
|
|
|
|
caps = gst_caps_intersect(caps, gst_videobalance_get_capslist ());
|
|
|
|
|
|
|
|
templ = GST_PAD_TEMPLATE_NEW("src", GST_PAD_SINK, GST_PAD_ALWAYS, caps);
|
|
|
|
}
|
|
|
|
return templ;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
gst_videobalance_init (GstVideobalance *videobalance)
|
|
|
|
{
|
|
|
|
GstVideofilter *videofilter;
|
|
|
|
|
|
|
|
GST_DEBUG("gst_videobalance_init");
|
|
|
|
|
|
|
|
videofilter = GST_VIDEOFILTER(videobalance);
|
|
|
|
|
|
|
|
videofilter->sinkpad = gst_pad_new_from_template (
|
|
|
|
GST_PAD_TEMPLATE_GET (gst_videobalance_sink_template_factory),
|
|
|
|
"sink");
|
|
|
|
|
|
|
|
videofilter->srcpad = gst_pad_new_from_template (
|
|
|
|
GST_PAD_TEMPLATE_GET (gst_videobalance_src_template_factory),
|
|
|
|
"src");
|
|
|
|
|
2003-07-02 06:49:37 +00:00
|
|
|
videobalance->contrast = 1;
|
|
|
|
videobalance->brightness = 0;
|
|
|
|
videobalance->saturation = 1;
|
|
|
|
videobalance->hue = 0;
|
|
|
|
|
2003-07-02 06:31:33 +00:00
|
|
|
gst_videofilter_postinit(GST_VIDEOFILTER(videobalance));
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
gst_videobalance_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
|
|
|
|
{
|
|
|
|
GstVideobalance *src;
|
|
|
|
|
|
|
|
/* it's not null if we got it, but it might not be ours */
|
|
|
|
g_return_if_fail(GST_IS_VIDEOBALANCE(object));
|
|
|
|
src = GST_VIDEOBALANCE(object);
|
|
|
|
|
|
|
|
GST_DEBUG("gst_videobalance_set_property");
|
|
|
|
switch (prop_id) {
|
2003-07-02 06:49:37 +00:00
|
|
|
case ARG_CONTRAST:
|
|
|
|
src->contrast = g_value_get_double (value);
|
|
|
|
break;
|
|
|
|
case ARG_BRIGHTNESS:
|
|
|
|
src->brightness = g_value_get_double (value);
|
|
|
|
break;
|
|
|
|
case ARG_HUE:
|
|
|
|
src->hue = g_value_get_double (value);
|
|
|
|
break;
|
|
|
|
case ARG_SATURATION:
|
|
|
|
src->saturation = g_value_get_double (value);
|
2003-07-02 06:31:33 +00:00
|
|
|
break;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
gst_videobalance_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
|
|
|
|
{
|
|
|
|
GstVideobalance *src;
|
|
|
|
|
|
|
|
/* it's not null if we got it, but it might not be ours */
|
|
|
|
g_return_if_fail(GST_IS_VIDEOBALANCE(object));
|
|
|
|
src = GST_VIDEOBALANCE(object);
|
|
|
|
|
|
|
|
switch (prop_id) {
|
2003-07-02 06:49:37 +00:00
|
|
|
case ARG_CONTRAST:
|
|
|
|
g_value_set_double (value, src->contrast);
|
|
|
|
break;
|
|
|
|
case ARG_BRIGHTNESS:
|
|
|
|
g_value_set_double (value, src->brightness);
|
|
|
|
break;
|
|
|
|
case ARG_HUE:
|
|
|
|
g_value_set_double (value, src->hue);
|
|
|
|
break;
|
|
|
|
case ARG_SATURATION:
|
|
|
|
g_value_set_double (value, src->saturation);
|
2003-07-02 06:31:33 +00:00
|
|
|
break;
|
|
|
|
default:
|
|
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2003-11-02 22:54:47 +00:00
|
|
|
static gboolean plugin_init (GstPlugin *plugin)
|
2003-07-02 06:31:33 +00:00
|
|
|
{
|
|
|
|
if(!gst_library_load("gstvideofilter"))
|
|
|
|
return FALSE;
|
|
|
|
|
2003-11-02 22:54:47 +00:00
|
|
|
return gst_element_register(plugin, "videobalance",
|
|
|
|
GST_RANK_NONE, GST_TYPE_VIDEOBALANCE);
|
2003-07-02 06:31:33 +00:00
|
|
|
}
|
|
|
|
|
2003-11-02 22:54:47 +00:00
|
|
|
GST_PLUGIN_DEFINE (
|
2003-07-02 06:31:33 +00:00
|
|
|
GST_VERSION_MAJOR,
|
|
|
|
GST_VERSION_MINOR,
|
|
|
|
"videobalance",
|
2003-11-02 22:54:47 +00:00
|
|
|
"Changes hue, saturation, brightness etc. on video images",
|
|
|
|
plugin_init,
|
|
|
|
VERSION,
|
|
|
|
"LGPL",
|
|
|
|
GST_COPYRIGHT,
|
|
|
|
GST_PACKAGE,
|
|
|
|
GST_ORIGIN
|
|
|
|
)
|
2003-07-02 06:31:33 +00:00
|
|
|
|
|
|
|
static void gst_videobalance_setup(GstVideofilter *videofilter)
|
|
|
|
{
|
|
|
|
GstVideobalance *videobalance;
|
|
|
|
|
|
|
|
g_return_if_fail(GST_IS_VIDEOBALANCE(videofilter));
|
|
|
|
videobalance = GST_VIDEOBALANCE(videofilter);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
static void gst_videobalance_planar411(GstVideofilter *videofilter,
|
|
|
|
void *dest, void *src)
|
|
|
|
{
|
|
|
|
GstVideobalance *videobalance;
|
2003-07-02 06:49:37 +00:00
|
|
|
int width;
|
|
|
|
int height;
|
|
|
|
int x,y;
|
2003-07-02 06:31:33 +00:00
|
|
|
|
|
|
|
g_return_if_fail(GST_IS_VIDEOBALANCE(videofilter));
|
|
|
|
videobalance = GST_VIDEOBALANCE(videofilter);
|
|
|
|
|
2003-07-02 06:49:37 +00:00
|
|
|
width = videofilter->from_width;
|
|
|
|
height = videofilter->from_height;
|
|
|
|
|
|
|
|
{
|
|
|
|
double Y;
|
|
|
|
double contrast;
|
|
|
|
double brightness;
|
|
|
|
guint8 *cdest = dest;
|
|
|
|
guint8 *csrc = src;
|
|
|
|
|
|
|
|
contrast = videobalance->contrast;
|
|
|
|
brightness = videobalance->brightness;
|
|
|
|
|
|
|
|
for(y=0;y<height;y++){
|
|
|
|
for(x=0;x<width;x++){
|
|
|
|
Y = csrc[y*width + x];
|
|
|
|
Y = 16 + ((Y-16) * contrast + brightness*255);
|
|
|
|
if(Y<0)Y=0;
|
|
|
|
if(Y>255)Y=255;
|
|
|
|
cdest[y*width + x] = rint(Y);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
{
|
|
|
|
double u, v;
|
|
|
|
double u1, v1;
|
|
|
|
double saturation;
|
|
|
|
double hue_cos, hue_sin;
|
|
|
|
guint8 *usrc, *vsrc;
|
|
|
|
guint8 *udest, *vdest;
|
|
|
|
|
|
|
|
saturation = videobalance->saturation;
|
|
|
|
|
|
|
|
usrc = src + width*height;
|
|
|
|
udest = dest + width*height;
|
|
|
|
vsrc = src + width*height + (width/2)*(height/2);
|
|
|
|
vdest = dest + width*height + (width/2)*(height/2);
|
|
|
|
|
|
|
|
/* FIXME this is a bogus transformation for hue, but you get
|
|
|
|
* the idea */
|
|
|
|
hue_cos = cos(M_PI*videobalance->hue);
|
|
|
|
hue_sin = sin(M_PI*videobalance->hue);
|
|
|
|
|
|
|
|
for(y=0;y<height/2;y++){
|
|
|
|
for(x=0;x<width/2;x++){
|
|
|
|
u1 = usrc[y*(width/2) + x] - 128;
|
|
|
|
v1 = vsrc[y*(width/2) + x] - 128;
|
|
|
|
u = 128 + (( u1 * hue_cos + v1 * hue_sin) * saturation);
|
|
|
|
v = 128 + ((-u1 * hue_sin + v1 * hue_cos) * saturation);
|
|
|
|
if(u<0)u=0;
|
|
|
|
if(u>255)u=255;
|
|
|
|
if(v<0)v=0;
|
|
|
|
if(v>255)v=255;
|
|
|
|
udest[y*(width/2) + x] = rint(u);
|
|
|
|
vdest[y*(width/2) + x] = rint(v);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2003-07-02 06:31:33 +00:00
|
|
|
}
|
|
|
|
|