yair/gst-plugin-linescan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

videolevels: port to 1.0

Browse Source
This commit is contained in:
Joshua M. Doe 2013-05-14 16:16:05 -04:00
parent 6e28e762e0
commit f995ed9987
4 changed files with 162 additions and 275 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gst/videoadjust/gstvideoadjust.c
View File

@ -28,6 +28,6 @@ plugin_init (GstPlugin * plugin)
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR, GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
GST_VERSION_MINOR, GST_VERSION_MINOR,
"videoadjust", videoadjust,
"Filters that apply transform from 16-bit to 8-bit video", "Filters that apply transform from 16-bit to 8-bit video",
plugin_init, VERSION, GST_LICENSE, PACKAGE_NAME, GST_PACKAGE_ORIGIN); plugin_init, VERSION, GST_LICENSE, PACKAGE_NAME, GST_PACKAGE_ORIGIN);

401
gst/videoadjust/gstvideolevels.c
View File

@ -40,8 +40,7 @@
#include "gstvideolevels.h" #include "gstvideolevels.h"
#include <string.h> #include <gst/video/video.h>
#include <math.h>
/* GstVideoLevels signals and args */ /* GstVideoLevels signals and args */
enum enum
@ -71,26 +70,19 @@ static GParamSpec *properties[PROP_LAST];
#define DEFAULT_PROP_AUTO 0 #define DEFAULT_PROP_AUTO 0
#define DEFAULT_PROP_INTERVAL (GST_SECOND / 2) #define DEFAULT_PROP_INTERVAL (GST_SECOND / 2)
static const GstElementDetails videolevels_details =
GST_ELEMENT_DETAILS ("Video videolevels adjustment",
"Filter/Effect/Video",
"Adjusts videolevels on a video stream",
"Joshua Doe <oss@nvl.army.mil");
/* the capabilities of the inputs and outputs */ /* the capabilities of the inputs and outputs */
static GstStaticPadTemplate gst_videolevels_src_template = static GstStaticPadTemplate gst_videolevels_src_template =
GST_STATIC_PAD_TEMPLATE ("sink", GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK, GST_PAD_SINK,
GST_PAD_ALWAYS, GST_PAD_ALWAYS,
GST_STATIC_CAPS (GST_VIDEO_CAPS_GRAY16 ("BIG_ENDIAN") ";" GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("{ GRAY16_LE, GRAY16_BE }"))
GST_VIDEO_CAPS_GRAY16 ("LITTLE_ENDIAN"))
); );
static GstStaticPadTemplate gst_videolevels_sink_template = static GstStaticPadTemplate gst_videolevels_sink_template =
GST_STATIC_PAD_TEMPLATE ("src", GST_STATIC_PAD_TEMPLATE ("src",
GST_PAD_SRC, GST_PAD_SRC,
GST_PAD_ALWAYS, GST_PAD_ALWAYS,
GST_STATIC_CAPS (GST_VIDEO_CAPS_GRAY8) GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("GRAY8"))
); );
#define GST_TYPE_VIDEOLEVELS_AUTO (gst_videolevels_auto_get_type()) #define GST_TYPE_VIDEOLEVELS_AUTO (gst_videolevels_auto_get_type())
@ -121,21 +113,18 @@ static void gst_videolevels_dispose (GObject * object);
/* GstBaseTransform vmethod declarations */ /* GstBaseTransform vmethod declarations */
static GstCaps *gst_videolevels_transform_caps (GstBaseTransform * trans, static GstCaps *gst_videolevels_transform_caps (GstBaseTransform * trans,
GstPadDirection direction, GstCaps * caps); GstPadDirection direction, GstCaps * caps, GstCaps * filter_caps);
static gboolean gst_videolevels_set_caps (GstBaseTransform * base,
GstCaps * incaps, GstCaps * outcaps); /* GstVideoFilter vmethod declarations */
static GstFlowReturn gst_videolevels_transform (GstBaseTransform * base, static gboolean gst_videolevels_set_info (GstVideoFilter * filter,
GstBuffer * inbuf, GstBuffer * outbuf); GstCaps * incaps, GstVideoInfo * in_info, GstCaps * outcaps,
static gboolean gst_videolevels_get_unit_size (GstBaseTransform * base, GstVideoInfo * out_info);
GstCaps * caps, guint * size); static GstFlowReturn gst_videolevels_transform_frame (GstVideoFilter * filter,
static GstFlowReturn gst_videolevels_prepare_output_buffer (GstBaseTransform * GstVideoFrame * in_frame, GstVideoFrame * out_frame);
base, GstBuffer * in_buf, gint size, GstCaps * caps, GstBuffer ** out_buf);
/* GstVideoLevels method declarations */ /* GstVideoLevels method declarations */
static void gst_videolevels_reset (GstVideoLevels * filter); static void gst_videolevels_reset (GstVideoLevels * filter);
static gboolean gst_videolevels_calculate_lut (GstVideoLevels * videolevels); static gboolean gst_videolevels_calculate_lut (GstVideoLevels * videolevels);
static gboolean gst_videolevels_do_levels (GstVideoLevels * videolevels,
gpointer indata, gpointer outdata);
static gboolean gst_videolevels_calculate_histogram (GstVideoLevels * static gboolean gst_videolevels_calculate_histogram (GstVideoLevels *
videolevels, guint16 * data); videolevels, guint16 * data);
static gboolean gst_videolevels_auto_adjust (GstVideoLevels * videolevels, static gboolean gst_videolevels_auto_adjust (GstVideoLevels * videolevels,
@ -144,37 +133,13 @@ static gboolean gst_videolevels_auto_adjust (GstVideoLevels * videolevels,
/* setup debug */ /* setup debug */
GST_DEBUG_CATEGORY_STATIC (videolevels_debug); GST_DEBUG_CATEGORY_STATIC (videolevels_debug);
#define GST_CAT_DEFAULT videolevels_debug #define GST_CAT_DEFAULT videolevels_debug
#define DEBUG_INIT(bla) \
GST_DEBUG_CATEGORY_INIT (videolevels_debug, "videolevels", 0, \
"Video Levels Filter");
GST_BOILERPLATE_FULL (GstVideoLevels, gst_videolevels, GstVideoFilter, G_DEFINE_TYPE (GstVideoLevels, gst_videolevels, GST_TYPE_VIDEO_FILTER);
GST_TYPE_VIDEO_FILTER, DEBUG_INIT);
/************************************************************************/ /************************************************************************/
/* GObject vmethod implementations */ /* GObject vmethod implementations */
/************************************************************************/ /************************************************************************/
/**
* gst_videolevels_base_init:
* @klass: #GstElementClass.
*
*/
static void
gst_videolevels_base_init (gpointer klass)
{
GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
GST_DEBUG ("base init");
gst_element_class_set_details (element_class, &videolevels_details);
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&gst_videolevels_sink_template));
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&gst_videolevels_src_template));
}
/** /**
* gst_videolevels_dispose: * gst_videolevels_dispose:
* @object: #GObject. * @object: #GObject.
@ -190,7 +155,7 @@ gst_videolevels_dispose (GObject * object)
gst_videolevels_reset (videolevels); gst_videolevels_reset (videolevels);
/* chain up to the parent class */ /* chain up to the parent class */
G_OBJECT_CLASS (parent_class)->dispose (object); G_OBJECT_CLASS (gst_videolevels_parent_class)->dispose (object);
} }
/** /**
@ -199,18 +164,25 @@ gst_videolevels_dispose (GObject * object)
* *
*/ */
static void static void
gst_videolevels_class_init (GstVideoLevelsClass * object) gst_videolevels_class_init (GstVideoLevelsClass * klass)
{ {
GObjectClass *obj_class = G_OBJECT_CLASS (object); GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
GstBaseTransformClass *trans_class = GST_BASE_TRANSFORM_CLASS (object); GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);
GstBaseTransformClass *gstbasetransform_class =
GST_BASE_TRANSFORM_CLASS (klass);
GstVideoFilterClass *gstvideofilter_class = GST_VIDEO_FILTER_CLASS (klass);
GST_DEBUG_CATEGORY_INIT (videolevels_debug, "videolevels", 0,
"Video Levels Filter");
GST_DEBUG ("class init"); GST_DEBUG ("class init");
/* Register GObject vmethods */ /* Register GObject vmethods */
obj_class->dispose = GST_DEBUG_FUNCPTR (gst_videolevels_dispose); gobject_class->dispose = GST_DEBUG_FUNCPTR (gst_videolevels_dispose);
obj_class->set_property = GST_DEBUG_FUNCPTR (gst_videolevels_set_property); gobject_class->set_property =
obj_class->get_property = GST_DEBUG_FUNCPTR (gst_videolevels_get_property); GST_DEBUG_FUNCPTR (gst_videolevels_set_property);
gobject_class->get_property =
GST_DEBUG_FUNCPTR (gst_videolevels_get_property);
/* Install GObject properties */ /* Install GObject properties */
properties[PROP_LOWIN] = properties[PROP_LOWIN] =
@ -229,31 +201,39 @@ gst_videolevels_class_init (GstVideoLevelsClass * object)
g_param_spec_double ("upper-output-level", "Upper Output Level", g_param_spec_double ("upper-output-level", "Upper Output Level",
"Upper Output Level", 0.0, 1.0, DEFAULT_PROP_HIGHOUT, "Upper Output Level", 0.0, 1.0, DEFAULT_PROP_HIGHOUT,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS); G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
g_object_class_install_property (obj_class, PROP_LOWIN, g_object_class_install_property (gobject_class, PROP_LOWIN,
properties[PROP_LOWIN]); properties[PROP_LOWIN]);
g_object_class_install_property (obj_class, PROP_HIGHIN, g_object_class_install_property (gobject_class, PROP_HIGHIN,
properties[PROP_HIGHIN]); properties[PROP_HIGHIN]);
g_object_class_install_property (obj_class, PROP_LOWOUT, g_object_class_install_property (gobject_class, PROP_LOWOUT,
properties[PROP_LOWOUT]); properties[PROP_LOWOUT]);
g_object_class_install_property (obj_class, PROP_HIGHOUT, g_object_class_install_property (gobject_class, PROP_HIGHOUT,
properties[PROP_HIGHOUT]); properties[PROP_HIGHOUT]);
g_object_class_install_property (obj_class, PROP_AUTO, g_object_class_install_property (gobject_class, PROP_AUTO,
g_param_spec_enum ("auto", "Auto Adjust", "Auto adjust contrast", g_param_spec_enum ("auto", "Auto Adjust", "Auto adjust contrast",
GST_TYPE_VIDEOLEVELS_AUTO, DEFAULT_PROP_AUTO, G_PARAM_READWRITE)); GST_TYPE_VIDEOLEVELS_AUTO, DEFAULT_PROP_AUTO, G_PARAM_READWRITE));
g_object_class_install_property (obj_class, PROP_INTERVAL, g_object_class_install_property (gobject_class, PROP_INTERVAL,
g_param_spec_uint64 ("interval", "Interval", g_param_spec_uint64 ("interval", "Interval",
"Interval of time between adjustments (in nanoseconds)", 1, "Interval of time between adjustments (in nanoseconds)", 1,
G_MAXUINT64, DEFAULT_PROP_INTERVAL, G_PARAM_READWRITE)); G_MAXUINT64, DEFAULT_PROP_INTERVAL, G_PARAM_READWRITE));
gst_element_class_add_pad_template (gstelement_class,
gst_static_pad_template_get (&gst_videolevels_sink_template));
gst_element_class_add_pad_template (gstelement_class,
gst_static_pad_template_get (&gst_videolevels_src_template));
gst_element_class_set_static_metadata (gstelement_class,
"Video levels adjustment", "Filter/Effect/Video",
"Adjusts videolevels on a video stream",
"Joshua M. Doe <oss@nvl.army.mil>");
/* Register GstBaseTransform vmethods */ /* Register GstBaseTransform vmethods */
trans_class->transform_caps = gstbasetransform_class->transform_caps =
GST_DEBUG_FUNCPTR (gst_videolevels_transform_caps); GST_DEBUG_FUNCPTR (gst_videolevels_transform_caps);
trans_class->set_caps = GST_DEBUG_FUNCPTR (gst_videolevels_set_caps);
trans_class->transform = GST_DEBUG_FUNCPTR (gst_videolevels_transform); gstvideofilter_class->set_info = GST_DEBUG_FUNCPTR (gst_videolevels_set_info);
trans_class->get_unit_size = gstvideofilter_class->transform_frame =
GST_DEBUG_FUNCPTR (gst_videolevels_get_unit_size); GST_DEBUG_FUNCPTR (gst_videolevels_transform_frame);
trans_class->prepare_output_buffer =
GST_DEBUG_FUNCPTR (gst_videolevels_prepare_output_buffer);
} }
/** /**
@ -264,13 +244,11 @@ gst_videolevels_class_init (GstVideoLevelsClass * object)
* Initialize the new element * Initialize the new element
*/ */
static void static void
gst_videolevels_init (GstVideoLevels * videolevels, gst_videolevels_init (GstVideoLevels * videolevels)
GstVideoLevelsClass * g_class)
{ {
GST_DEBUG_OBJECT (videolevels, "init class instance"); GST_DEBUG_OBJECT (videolevels, "init class instance");
gst_videolevels_reset (videolevels); gst_videolevels_reset (videolevels);
} }
/** /**
@ -375,141 +353,67 @@ gst_videolevels_get_property (GObject * object, guint prop_id, GValue * value,
* *
* Returns: #GstCaps allowed on other pad * Returns: #GstCaps allowed on other pad
*/ */
static GstCaps * GstCaps *
gst_videolevels_transform_caps (GstBaseTransform * base, gst_videolevels_transform_caps (GstBaseTransform * trans,
GstPadDirection direction, GstCaps * caps) GstPadDirection direction, GstCaps * caps, GstCaps * filter_caps)
{ {
GstVideoLevels *videolevels; GstVideoLevels *videolevels = GST_VIDEOLEVELS (trans);
GstCaps *newcaps; GstCaps *other_caps;
GstStructure *structure; GstStructure *st;
gint i, n;
videolevels = GST_VIDEOLEVELS (base); videolevels = GST_VIDEOLEVELS (trans);
GST_DEBUG_OBJECT (caps, "transforming caps (from)"); GST_LOG_OBJECT (videolevels, "transforming caps %" GST_PTR_FORMAT, caps);
newcaps = gst_caps_copy (caps); other_caps = gst_caps_new_empty ();
/* finish settings caps of the opposite pad */ n = gst_caps_get_size (caps);
if (direction == GST_PAD_SINK) { for (i = 0; i < n; ++i) {
GST_DEBUG_OBJECT (videolevels, "Pad direction is sink"); st = gst_caps_get_structure (caps, i);
gst_caps_set_simple (newcaps,
"bpp", G_TYPE_INT, 8, "depth", G_TYPE_INT, 8, NULL);
structure = gst_caps_get_structure (newcaps, 0);
gst_structure_remove_field (structure, "endianness");
} else {
GValue endianness = { 0 };
GValue ival = { 0 };
GST_DEBUG_OBJECT (videolevels, "Pad direction is src"); if (gst_caps_is_subset_structure (other_caps, st))
continue;
gst_caps_set_simple (newcaps, st = gst_structure_copy (st);
"bpp", GST_TYPE_INT_RANGE, 1, 16, "depth", G_TYPE_INT, 16, NULL); gst_structure_remove_fields (st, "format", NULL);
structure = gst_caps_get_structure (newcaps, 0);
/* add BIG/LITTLE endianness to caps */ gst_caps_append_structure (other_caps, st);
g_value_init (&ival, G_TYPE_INT);
g_value_init (&endianness, GST_TYPE_LIST);
g_value_set_int (&ival, G_LITTLE_ENDIAN);
gst_value_list_append_value (&endianness, &ival);
g_value_set_int (&ival, G_BIG_ENDIAN);
gst_value_list_append_value (&endianness, &ival);
gst_structure_set_value (structure, "endianness", &endianness);
} }
GST_DEBUG_OBJECT (newcaps, "allowed caps are"); /* finish settings caps of the opposite pad */
//if (direction == GST_PAD_SINK) {
// "GRAY8";
return newcaps;
if (!gst_caps_is_empty (other_caps) && filter_caps) {
GstCaps *tmp = gst_caps_intersect_full (filter_caps, other_caps,
GST_CAPS_INTERSECT_FIRST);
gst_caps_replace (&other_caps, tmp);
gst_caps_unref (tmp);
}
return other_caps;
} }
/**
* gst_videolevels_set_caps:
* base: #GstBaseTransform
* incaps: #GstCaps
* outcaps: #GstCaps
*
* Notification of the actual caps set.
*
* Returns: TRUE on success
*/
static gboolean static gboolean
gst_videolevels_set_caps (GstBaseTransform * base, GstCaps * incaps, gst_videolevels_set_info (GstVideoFilter * filter, GstCaps * incaps,
GstCaps * outcaps) GstVideoInfo * in_info, GstCaps * outcaps, GstVideoInfo * out_info)
{ {
GstVideoLevels *levels = GST_VIDEOLEVELS (base); GstVideoLevels *levels = GST_VIDEOLEVELS (filter);
GstStructure *structure;
gboolean res; gboolean res;
GST_DEBUG_OBJECT (levels, GST_DEBUG_OBJECT (levels,
"set_caps: in %" GST_PTR_FORMAT " out %" GST_PTR_FORMAT, incaps, outcaps); "set_caps: in %" GST_PTR_FORMAT " out %" GST_PTR_FORMAT, incaps, outcaps);
/* retrieve caps info */ memcpy (&levels->info_in, in_info, sizeof (GstVideoInfo));
res = gst_video_format_parse_caps (incaps, &levels->format_in, &levels->width, memcpy (&levels->info_out, out_info, sizeof (GstVideoInfo));
&levels->height);
res &= gst_video_format_parse_caps (outcaps, &levels->format_out, NULL, NULL);
/* FIXME: gst_video_format_get_component_depth is broken in 0.10.36 res = gst_videolevels_calculate_lut (levels);
levels->bpp_in = gst_video_format_get_component_depth (levels->format_in, 0);
levels->bpp_out = gst_video_format_get_component_depth (levels->format_out, 0); */
structure = gst_caps_get_structure (incaps, 0);
res &= gst_structure_get_int (structure, "bpp", &levels->bpp_in);
structure = gst_caps_get_structure (outcaps, 0);
res &= gst_structure_get_int (structure, "bpp", &levels->bpp_out);
levels->framesize = gst_video_format_get_size (levels->format_out,
levels->width, levels->height);
if (!res || levels->framesize <= 0) {
GST_ERROR_OBJECT (levels, "Failed to parse caps");
}
gst_videolevels_calculate_lut (levels);
return res; return res;
} }
/**
* gst_videolevels_get_unit_size:
* @base: #GstBaseTransform
* @caps: #GstCaps
* @size: guint size of unit (one frame for video)
*
* Tells GstBaseTransform the size in bytes of an output frame from the given
* caps.
*
* Returns: TRUE on success
*/
static gboolean
gst_videolevels_get_unit_size (GstBaseTransform * base, GstCaps * caps,
guint * size)
{
if (!gst_video_get_size_from_caps (caps, size)) {
GST_ELEMENT_ERROR (base, CORE, NEGOTIATION,
("Unable to determine frame size from caps"), (NULL));
return FALSE;
}
GST_DEBUG ("Frame size is %d bytes", *size);
return TRUE;
}
static GstFlowReturn
gst_videolevels_prepare_output_buffer (GstBaseTransform * base,
GstBuffer * in_buf, gint size, GstCaps * caps, GstBuffer ** out_buf)
{
GstVideoLevels *levels = GST_VIDEOLEVELS (base);
GstFlowReturn ret = GST_FLOW_OK;
ret = gst_pad_alloc_buffer (base->srcpad, 0, size, caps, out_buf);
if (ret != GST_FLOW_OK) {
GST_DEBUG_OBJECT (levels,
"Couldn't get pad to alloc buffer, creating one directly");
*out_buf = gst_buffer_new_and_alloc (size);
}
return ret;
}
/** /**
* gst_videolevels_transform: * gst_videolevels_transform:
* @base: #GstBaseTransform * @base: #GstBaseTransform
@ -521,50 +425,65 @@ gst_videolevels_prepare_output_buffer (GstBaseTransform * base,
* Returns: GST_FLOW_OK on success * Returns: GST_FLOW_OK on success
*/ */
static GstFlowReturn static GstFlowReturn
gst_videolevels_transform (GstBaseTransform * base, GstBuffer * inbuf, gst_videolevels_transform_frame (GstVideoFilter * filter,
GstBuffer * outbuf) GstVideoFrame * in_frame, GstVideoFrame * out_frame)
{ {
GstVideoLevels *videolevels = GST_VIDEOLEVELS (base); GstVideoLevels *videolevels = GST_VIDEOLEVELS (filter);
gpointer input;
gpointer output;
gboolean ret;
GstClockTimeDiff elapsed; GstClockTimeDiff elapsed;
GstClockTime start = GstClockTime start =
gst_clock_get_time (gst_element_get_clock (GST_ELEMENT (base))); gst_clock_get_time (gst_element_get_clock (GST_ELEMENT (videolevels)));
gint r, c;
gint in_stride, out_stride;
guint8 *in_data, *out_data;
guint8 *lut;
GST_LOG_OBJECT (videolevels, "Performing non-inplace transform"); GST_LOG_OBJECT (videolevels, "Performing non-inplace transform");
input = GST_BUFFER_DATA (inbuf); in_data = GST_VIDEO_FRAME_PLANE_DATA (in_frame, 0);
output = GST_BUFFER_DATA (outbuf); out_data = GST_VIDEO_FRAME_PLANE_DATA (out_frame, 0);
if (videolevels->auto_adjust == 1) { if (videolevels->auto_adjust == 1) {
GST_DEBUG_OBJECT (videolevels, "Auto adjusting levels (once)"); GST_DEBUG_OBJECT (videolevels, "Auto adjusting levels (once)");
gst_videolevels_auto_adjust (videolevels, input); gst_videolevels_auto_adjust (videolevels, in_data);
videolevels->auto_adjust = 0; videolevels->auto_adjust = 0;
g_object_notify (G_OBJECT (videolevels), "auto"); g_object_notify (G_OBJECT (videolevels), "auto");
} else if (videolevels->auto_adjust == 2) { } else if (videolevels->auto_adjust == 2) {
elapsed = elapsed =
GST_CLOCK_DIFF (videolevels->last_auto_timestamp, inbuf->timestamp); GST_CLOCK_DIFF (videolevels->last_auto_timestamp,
GST_BUFFER_TIMESTAMP (in_frame->buffer));
if (videolevels->last_auto_timestamp == GST_CLOCK_TIME_NONE if (videolevels->last_auto_timestamp == GST_CLOCK_TIME_NONE
|| elapsed >= (GstClockTimeDiff) videolevels->interval || elapsed < 0) { || elapsed >= (GstClockTimeDiff) videolevels->interval || elapsed < 0) {
GST_LOG_OBJECT (videolevels, "Auto adjusting levels (%d ns since last)", GST_LOG_OBJECT (videolevels, "Auto adjusting levels (%d ns since last)",
elapsed); elapsed);
gst_videolevels_auto_adjust (videolevels, input); gst_videolevels_auto_adjust (videolevels, in_data);
videolevels->last_auto_timestamp = GST_BUFFER_TIMESTAMP (inbuf); videolevels->last_auto_timestamp =
GST_BUFFER_TIMESTAMP (in_frame->buffer);
} }
} }
ret = gst_videolevels_do_levels (videolevels, input, output); in_stride = GST_VIDEO_FRAME_PLANE_STRIDE (in_frame, 0);
out_stride = GST_VIDEO_FRAME_PLANE_STRIDE (out_frame, 0);
lut = videolevels->lookup_table;
for (r = 0; r < in_frame->info.height; r++) {
guint16 *src = (guint16 *) in_data;
guint8 *dst = out_data;
for (c = 0; c < in_frame->info.width; c++) {
//GST_LOG_OBJECT (videolevels, "Converting pixel (%d, %d), %d->%d", c, r, *src, lut[*src]);
*dst++ = lut[*src++];
}
in_data += in_stride;
out_data += out_stride;
}
GST_LOG_OBJECT (videolevels, "Processing took %" G_GINT64_FORMAT "ms", GST_LOG_OBJECT (videolevels, "Processing took %" G_GINT64_FORMAT "ms",
GST_TIME_AS_MSECONDS (GST_CLOCK_DIFF (start, GST_TIME_AS_MSECONDS (GST_CLOCK_DIFF (start,
gst_clock_get_time (gst_element_get_clock (GST_ELEMENT gst_clock_get_time (gst_element_get_clock (GST_ELEMENT
(videolevels)))))); (videolevels))))));
if (ret)
return GST_FLOW_OK; return GST_FLOW_OK;
else
return GST_FLOW_ERROR;
} }
/************************************************************************/ /************************************************************************/
@ -580,12 +499,8 @@ gst_videolevels_transform (GstBaseTransform * base, GstBuffer * inbuf,
static void static void
gst_videolevels_reset (GstVideoLevels * videolevels) gst_videolevels_reset (GstVideoLevels * videolevels)
{ {
videolevels->width = 0; gst_video_info_init (&videolevels->info_in);
videolevels->height = 0; gst_video_info_init (&videolevels->info_out);
videolevels->format_in = GST_VIDEO_FORMAT_UNKNOWN;
videolevels->format_out = GST_VIDEO_FORMAT_UNKNOWN;
videolevels->bpp_in = 0;
videolevels->bpp_out = 0;
videolevels->lower_input = DEFAULT_PROP_LOWIN; videolevels->lower_input = DEFAULT_PROP_LOWIN;
videolevels->upper_input = DEFAULT_PROP_HIGHIN; videolevels->upper_input = DEFAULT_PROP_HIGHIN;
@ -619,14 +534,13 @@ gst_videolevels_calculate_lut_uint16_to_uint8 (GstVideoLevels * videolevels,
gdouble m; gdouble m;
gdouble b; gdouble b;
guint8 *lut = (guint8 *) videolevels->lookup_table; guint8 *lut = (guint8 *) videolevels->lookup_table;
const guint16 max_in = (1 << videolevels->bpp_in) - 1; const guint16 max_in = (1 << videolevels->info_in.finfo->bits) - 1;
const guint16 low_in = (guint16) (videolevels->lower_input * max_in); const guint16 low_in = (guint16) (videolevels->lower_input * max_in);
const guint16 high_in = (guint16) (videolevels->upper_input * max_in); const guint16 high_in = (guint16) (videolevels->upper_input * max_in);
const guint8 max_out = (1 << videolevels->bpp_out) - 1; const guint8 max_out = (1 << videolevels->info_out.finfo->bits) - 1;
const guint8 low_out = (guint8) (videolevels->lower_output * max_out); const guint8 low_out = (guint8) (videolevels->lower_output * max_out);
const guint8 high_out = (guint8) (videolevels->upper_output * max_out); const guint8 high_out = (guint8) (videolevels->upper_output * max_out);
GST_LOG_OBJECT (videolevels, "Applying linear mapping (%d, %d) -> (%d, %d)", GST_LOG_OBJECT (videolevels, "Applying linear mapping (%d, %d) -> (%d, %d)",
low_in, high_in, low_out, high_out); low_in, high_in, low_out, high_out);
@ -660,43 +574,24 @@ gst_videolevels_calculate_lut (GstVideoLevels * videolevels)
videolevels->lookup_table = g_new (guint8, G_MAXUINT16); videolevels->lookup_table = g_new (guint8, G_MAXUINT16);
} }
if (videolevels->format_in == GST_VIDEO_FORMAT_GRAY16_LE) { switch (videolevels->info_in.finfo->format) {
GST_LOG_OBJECT (videolevels, "Calculating lookup table uint16le -> uint8"); case GST_VIDEO_FORMAT_UNKNOWN:
/* no format set yet, don't do anything */
break;
case GST_VIDEO_FORMAT_GRAY16_LE:
GST_LOG_OBJECT (videolevels,
"Calculating lookup table uint16le -> uint8");
gst_videolevels_calculate_lut_uint16_to_uint8 (videolevels, gst_videolevels_calculate_lut_uint16_to_uint8 (videolevels,
G_LITTLE_ENDIAN); G_LITTLE_ENDIAN);
} else if (videolevels->format_in == GST_VIDEO_FORMAT_GRAY16_BE) { break;
GST_LOG_OBJECT (videolevels, "Calculating lookup table uint16be -> uint8"); case GST_VIDEO_FORMAT_GRAY16_BE:
GST_LOG_OBJECT (videolevels,
"Calculating lookup table uint16be -> uint8");
gst_videolevels_calculate_lut_uint16_to_uint8 (videolevels, G_BIG_ENDIAN); gst_videolevels_calculate_lut_uint16_to_uint8 (videolevels, G_BIG_ENDIAN);
} else break;
return FALSE; default:
g_assert_not_reached ();
return TRUE; }
}
/**
* gst_videolevels_do_levels
* @videolevels: #GstVideoLevels
* @indata: input data
* @outdata: output data
* @size: size of data
*
* Convert frame using previously calculated LUT
*
* Returns: TRUE on success
*/
static gboolean
gst_videolevels_do_levels (GstVideoLevels * videolevels, gpointer indata,
gpointer outdata)
{
guint8 *dst = (guint8 *) outdata;
guint16 *src = (guint16 *) indata;
guint8 *lut = (guint8 *) videolevels->lookup_table;
int i;
GST_LOG_OBJECT (videolevels, "Converting frame using LUT");
for (i = 0; i < videolevels->framesize; i++)
dst[i] = lut[src[i]];
return TRUE; return TRUE;
} }
@ -719,20 +614,18 @@ gst_videolevels_calculate_histogram (GstVideoLevels * videolevels,
gint r; gint r;
gint c; gint c;
gfloat factor; gfloat factor;
gint stride = gst_video_format_get_row_stride (videolevels->format_in, 0, gint stride = GST_VIDEO_INFO_COMP_STRIDE (&videolevels->info_in, 0);
videolevels->width);
gint endianness; gint endianness;
/* TODO: add gst_video_format_get_endianness to video library */ if (videolevels->info_in.finfo->format == GST_VIDEO_FORMAT_GRAY16_BE)
if (videolevels->format_in == GST_VIDEO_FORMAT_GRAY16_BE)
endianness = G_BIG_ENDIAN; endianness = G_BIG_ENDIAN;
else if (videolevels->format_in == GST_VIDEO_FORMAT_GRAY16_LE) else if (videolevels->info_in.finfo->format == GST_VIDEO_FORMAT_GRAY16_LE)
endianness = G_LITTLE_ENDIAN; endianness = G_LITTLE_ENDIAN;
else else
endianness = G_BYTE_ORDER; endianness = G_BYTE_ORDER;
factor = nbins / (gfloat) (1 << videolevels->bpp_in); factor = nbins / (gfloat) (1 << videolevels->info_in.finfo->bits);
if (videolevels->histogram == NULL) { if (videolevels->histogram == NULL) {
GST_DEBUG_OBJECT (videolevels, GST_DEBUG_OBJECT (videolevels,
@ -747,14 +640,14 @@ gst_videolevels_calculate_histogram (GstVideoLevels * videolevels,
GST_LOG_OBJECT (videolevels, "Calculating histogram"); GST_LOG_OBJECT (videolevels, "Calculating histogram");
if (endianness == G_BYTE_ORDER) { if (endianness == G_BYTE_ORDER) {
for (r = 0; r < videolevels->height; r++) { for (r = 0; r < videolevels->info_in.height; r++) {
for (c = 0; c < videolevels->width; c++) { for (c = 0; c < videolevels->info_in.width; c++) {
hist[GINT_CLAMP (data[c + r * stride / 2] * factor, 0, nbins - 1)]++; hist[GINT_CLAMP (data[c + r * stride / 2] * factor, 0, nbins - 1)]++;
} }
} }
} else { } else {
for (r = 0; r < videolevels->height; r++) { for (r = 0; r < videolevels->info_in.height; r++) {
for (c = 0; c < videolevels->width; c++) { for (c = 0; c < videolevels->info_in.width; c++) {
hist[GINT_CLAMP (GUINT16_FROM_BE (data[c + hist[GINT_CLAMP (GUINT16_FROM_BE (data[c +
r * stride / 2]) * factor, 0, nbins - 1)]++; r * stride / 2]) * factor, 0, nbins - 1)]++;
} }
@ -781,11 +674,11 @@ gst_videolevels_auto_adjust (GstVideoLevels * videolevels, guint16 * data)
gint i; gint i;
gint size; gint size;
gdouble min = 0.0; gdouble min = 0.0;
gdouble max = (1 << videolevels->bpp_in) - 1.0; gdouble max = (1 << videolevels->info_in.finfo->bits) - 1.0;
gst_videolevels_calculate_histogram (videolevels, data); gst_videolevels_calculate_histogram (videolevels, data);
size = videolevels->width * videolevels->height; size = videolevels->info_in.width * videolevels->info_in.height;
/* pixels to saturate on low end */ /* pixels to saturate on low end */
npixsat = (guint) (videolevels->lower_pix_sat * size); npixsat = (guint) (videolevels->lower_pix_sat * size);

10
gst/videoadjust/gstvideolevels.h
View File

@ -70,14 +70,8 @@ struct _GstVideoLevels
GstVideoFilter element; GstVideoFilter element;
/* format */ /* format */
gint width; GstVideoInfo info_in;
gint height; GstVideoInfo info_out;
int framesize;
GstVideoFormat format_in;
GstVideoFormat format_out;
gint bpp_in;
gint bpp_out;
/* properties */ /* properties */
gdouble lower_input; gdouble lower_input;

16
vs2010/videoadjust/videoadjust.vcxproj
View File

@ -50,23 +50,23 @@
</ImportGroup> </ImportGroup>
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'"> <ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" /> <Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
<Import Project="$(GSTREAMER_SDK_ROOT_X86)\share\vs\2010\libs\gstreamer-video-0.10.props" /> <Import Project="$(GSTREAMER_1_0_SDK_ROOT_X86)\share\vs\2010\libs\gstreamer-video-1.0.props" />
<Import Project="$(GSTREAMER_SDK_ROOT_X86)\share\vs\2010\msvc\x86.props" /> <Import Project="$(GSTREAMER_1_0_SDK_ROOT_X86)\share\vs\2010\msvc\x86.props" />
</ImportGroup> </ImportGroup>
<ImportGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="PropertySheets"> <ImportGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="PropertySheets">
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" /> <Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
<Import Project="$(GSTREAMER_SDK_ROOT_X86_64)\share\vs\2010\libs\gstreamer-video-0.10.props" /> <Import Project="$(GSTREAMER_1_0_SDK_ROOT_X86_64)\share\vs\2010\libs\gstreamer-video-1.0.props" />
<Import Project="$(GSTREAMER_SDK_ROOT_X86_64)\share\vs\2010\msvc\x86_64.props" /> <Import Project="$(GSTREAMER_1_0_SDK_ROOT_X86_64)\share\vs\2010\msvc\x86_64.props" />
</ImportGroup> </ImportGroup>
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|Win32'"> <ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" /> <Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
<Import Project="$(GSTREAMER_SDK_ROOT_X86)\share\vs\2010\libs\gstreamer-video-0.10.props" /> <Import Project="$(GSTREAMER_1_0_SDK_ROOT_X86)\share\vs\2010\libs\gstreamer-video-1.0.props" />
<Import Project="$(GSTREAMER_SDK_ROOT_X86)\share\vs\2010\msvc\x86.props" /> <Import Project="$(GSTREAMER_1_0_SDK_ROOT_X86)\share\vs\2010\msvc\x86.props" />
</ImportGroup> </ImportGroup>
<ImportGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'" Label="PropertySheets"> <ImportGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'" Label="PropertySheets">
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" /> <Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
<Import Project="$(GSTREAMER_SDK_ROOT_X86_64)\share\vs\2010\libs\gstreamer-video-0.10.props" /> <Import Project="$(GSTREAMER_1_0_SDK_ROOT_X86_64)\share\vs\2010\libs\gstreamer-video-1.0.props" />
<Import Project="$(GSTREAMER_SDK_ROOT_X86_64)\share\vs\2010\msvc\x86_64.props" /> <Import Project="$(GSTREAMER_1_0_SDK_ROOT_X86_64)\share\vs\2010\msvc\x86_64.props" />
</ImportGroup> </ImportGroup>
<PropertyGroup Label="UserMacros" /> <PropertyGroup Label="UserMacros" />
<PropertyGroup /> <PropertyGroup />