mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2024-11-03 16:09:39 +00:00
Code Issues 1 Projects Releases Wiki Activity
gstreamer/gst/videobox/gstvideobox.c
Sebastian Dröge cd6b4214a6 videobox: Add const qualifier to the source frame data
2010-03-26 13:28:53 +01:00

1293 lines
39 KiB
C
Raw Blame History

/* GStreamer
* Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
* Copyright (C) <2010> Sebastian Dröge <sebastian.droege@collabora.co.uk>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
/**
* SECTION:element-videobox
* @see_also: #GstVideoCrop
*
* This plugin crops or enlarges the image. It takes 4 values as input, a
* top, bottom, left and right offset. Positive values will crop that much
* pixels from the respective border of the image, negative values will add
* that much pixels. When pixels are added, you can specify their color.
* Some predefined colors are usable with an enum property.
*
* The plugin is alpha channel aware and will try to negotiate with a format
* that supports alpha channels first. When alpha channel is active two
* other properties, alpha and border_alpha can be used to set the alpha
* values of the inner picture and the border respectively. an alpha value of
* 0.0 means total transparency, 1.0 is opaque.
*
* The videobox plugin has many uses such as doing a mosaic of pictures,
* letterboxing video, cutting out pieces of video, picture in picture, etc..
*
* Setting autocrop to true changes the behavior of the plugin so that
* caps determine crop properties rather than the other way around: given
* input and output dimensions, the crop values are selected so that the
* smaller frame is effectively centered in the larger frame. This
* involves either cropping or padding.
*
* If you use autocrop there is little point in setting the other
* properties manually because they will be overriden if the caps change,
* but nothing stops you from doing so.
*
* Sample pipeline:
* |[
* gst-launch videotestsrc ! videobox autocrop=true ! \
* "video/x-raw-yuv, width=600, height=400" ! ffmpegcolorspace ! ximagesink
* ]|
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "gstvideobox.h"
#include <math.h>
#include <liboil/liboil.h>
#include <string.h>
#include <gst/controller/gstcontroller.h>
GST_DEBUG_CATEGORY_STATIC (videobox_debug);
#define GST_CAT_DEFAULT videobox_debug
#define DEFAULT_LEFT 0
#define DEFAULT_RIGHT 0
#define DEFAULT_TOP 0
#define DEFAULT_BOTTOM 0
#define DEFAULT_FILL_TYPE VIDEO_BOX_FILL_BLACK
#define DEFAULT_ALPHA 1.0
#define DEFAULT_BORDER_ALPHA 1.0
enum
{
PROP_0,
PROP_LEFT,
PROP_RIGHT,
PROP_TOP,
PROP_BOTTOM,
PROP_FILL_TYPE,
PROP_ALPHA,
PROP_BORDER_ALPHA,
PROP_AUTOCROP
/* FILL ME */
};
static GstStaticPadTemplate gst_video_box_src_template =
GST_STATIC_PAD_TEMPLATE ("src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("AYUV") ";"
GST_VIDEO_CAPS_YUV ("I420"))
);
static GstStaticPadTemplate gst_video_box_sink_template =
GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("AYUV") ";"
GST_VIDEO_CAPS_YUV ("I420"))
);
GST_BOILERPLATE (GstVideoBox, gst_video_box, GstBaseTransform,
GST_TYPE_BASE_TRANSFORM);
static void gst_video_box_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec);
static void gst_video_box_get_property (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec);
static gboolean video_box_recalc_transform (GstVideoBox * video_box);
static GstCaps *gst_video_box_transform_caps (GstBaseTransform * trans,
GstPadDirection direction, GstCaps * from);
static gboolean gst_video_box_set_caps (GstBaseTransform * trans,
GstCaps * in, GstCaps * out);
static gboolean gst_video_box_get_unit_size (GstBaseTransform * trans,
GstCaps * caps, guint * size);
static GstFlowReturn gst_video_box_transform (GstBaseTransform * trans,
GstBuffer * in, GstBuffer * out);
#define GST_TYPE_VIDEO_BOX_FILL (gst_video_box_fill_get_type())
static GType
gst_video_box_fill_get_type (void)
{
static GType video_box_fill_type = 0;
static const GEnumValue video_box_fill[] = {
{VIDEO_BOX_FILL_BLACK, "Black", "black"},
{VIDEO_BOX_FILL_GREEN, "Colorkey green", "green"},
{VIDEO_BOX_FILL_BLUE, "Colorkey blue", "blue"},
{0, NULL, NULL},
};
if (!video_box_fill_type) {
video_box_fill_type =
g_enum_register_static ("GstVideoBoxFill", video_box_fill);
}
return video_box_fill_type;
}
static void
gst_video_box_base_init (gpointer g_class)
{
GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
gst_element_class_set_details_simple (element_class, "Video box filter",
"Filter/Effect/Video",
"Resizes a video by adding borders or cropping",
"Wim Taymans <wim@fluendo.com>");
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&gst_video_box_sink_template));
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&gst_video_box_src_template));
}
static void
gst_video_box_finalize (GObject * object)
{
GstVideoBox *video_box = GST_VIDEO_BOX (object);
if (video_box->mutex) {
g_mutex_free (video_box->mutex);
video_box->mutex = NULL;
}
G_OBJECT_CLASS (parent_class)->finalize (object);
}
static void
gst_video_box_class_init (GstVideoBoxClass * klass)
{
GObjectClass *gobject_class = (GObjectClass *) klass;
GstBaseTransformClass *trans_class = (GstBaseTransformClass *) klass;
gobject_class->set_property = gst_video_box_set_property;
gobject_class->get_property = gst_video_box_get_property;
gobject_class->finalize = gst_video_box_finalize;
g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_FILL_TYPE,
g_param_spec_enum ("fill", "Fill", "How to fill the borders",
GST_TYPE_VIDEO_BOX_FILL, DEFAULT_FILL_TYPE,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS | GST_PARAM_CONTROLLABLE));
g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_LEFT,
g_param_spec_int ("left", "Left",
"Pixels to box at left (<0 = add a border)", G_MININT, G_MAXINT,
DEFAULT_LEFT,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS | GST_PARAM_CONTROLLABLE));
g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_RIGHT,
g_param_spec_int ("right", "Right",
"Pixels to box at right (<0 = add a border)", G_MININT, G_MAXINT,
DEFAULT_RIGHT,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS | GST_PARAM_CONTROLLABLE));
g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_TOP,
g_param_spec_int ("top", "Top",
"Pixels to box at top (<0 = add a border)", G_MININT, G_MAXINT,
DEFAULT_TOP,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS | GST_PARAM_CONTROLLABLE));
g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_BOTTOM,
g_param_spec_int ("bottom", "Bottom",
"Pixels to box at bottom (<0 = add a border)", G_MININT, G_MAXINT,
DEFAULT_BOTTOM,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS | GST_PARAM_CONTROLLABLE));
g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_ALPHA,
g_param_spec_double ("alpha", "Alpha", "Alpha value picture", 0.0, 1.0,
DEFAULT_ALPHA,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS | GST_PARAM_CONTROLLABLE));
g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_BORDER_ALPHA,
g_param_spec_double ("border-alpha", "Border Alpha",
"Alpha value of the border", 0.0, 1.0, DEFAULT_BORDER_ALPHA,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS | GST_PARAM_CONTROLLABLE));
/**
* GstVideoBox:autocrop
*
* If set to %TRUE videobox will automatically crop/pad the input
* video to be centered in the output.
*
* Since: 0.10.16
**/
g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_AUTOCROP,
g_param_spec_boolean ("autocrop", "Auto crop",
"Auto crop", FALSE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
trans_class->transform = GST_DEBUG_FUNCPTR (gst_video_box_transform);
trans_class->transform_caps =
GST_DEBUG_FUNCPTR (gst_video_box_transform_caps);
trans_class->set_caps = GST_DEBUG_FUNCPTR (gst_video_box_set_caps);
trans_class->get_unit_size = GST_DEBUG_FUNCPTR (gst_video_box_get_unit_size);
}
static void
gst_video_box_init (GstVideoBox * video_box, GstVideoBoxClass * g_class)
{
video_box->box_right = DEFAULT_RIGHT;
video_box->box_left = DEFAULT_LEFT;
video_box->box_top = DEFAULT_TOP;
video_box->box_bottom = DEFAULT_BOTTOM;
video_box->crop_right = 0;
video_box->crop_left = 0;
video_box->crop_top = 0;
video_box->crop_bottom = 0;
video_box->fill_type = DEFAULT_FILL_TYPE;
video_box->alpha = DEFAULT_ALPHA;
video_box->border_alpha = DEFAULT_BORDER_ALPHA;
video_box->autocrop = FALSE;
video_box->mutex = g_mutex_new ();
}
static void
gst_video_box_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec)
{
GstVideoBox *video_box = GST_VIDEO_BOX (object);
g_mutex_lock (video_box->mutex);
switch (prop_id) {
case PROP_LEFT:
video_box->box_left = g_value_get_int (value);
if (video_box->box_left < 0) {
video_box->border_left = -video_box->box_left;
video_box->crop_left = 0;
} else {
video_box->border_left = 0;
video_box->crop_left = video_box->box_left;
}
break;
case PROP_RIGHT:
video_box->box_right = g_value_get_int (value);
if (video_box->box_right < 0) {
video_box->border_right = -video_box->box_right;
video_box->crop_right = 0;
} else {
video_box->border_right = 0;
video_box->crop_right = video_box->box_right;
}
break;
case PROP_TOP:
video_box->box_top = g_value_get_int (value);
if (video_box->box_top < 0) {
video_box->border_top = -video_box->box_top;
video_box->crop_top = 0;
} else {
video_box->border_top = 0;
video_box->crop_top = video_box->box_top;
}
break;
case PROP_BOTTOM:
video_box->box_bottom = g_value_get_int (value);
if (video_box->box_bottom < 0) {
video_box->border_bottom = -video_box->box_bottom;
video_box->crop_bottom = 0;
} else {
video_box->border_bottom = 0;
video_box->crop_bottom = video_box->box_bottom;
}
break;
case PROP_FILL_TYPE:
video_box->fill_type = g_value_get_enum (value);
break;
case PROP_ALPHA:
video_box->alpha = g_value_get_double (value);
break;
case PROP_BORDER_ALPHA:
video_box->border_alpha = g_value_get_double (value);
break;
case PROP_AUTOCROP:
video_box->autocrop = g_value_get_boolean (value);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
video_box_recalc_transform (video_box);
GST_DEBUG_OBJECT (video_box, "Calling reconfigure");
gst_base_transform_reconfigure (GST_BASE_TRANSFORM_CAST (video_box));
g_mutex_unlock (video_box->mutex);
}
static void
gst_video_box_autocrop (GstVideoBox * video_box)
{
gint crop_w = video_box->in_width - video_box->out_width;
gint crop_h = video_box->in_height - video_box->out_height;
video_box->box_left = crop_w / 2;
if (video_box->box_left < 0) {
video_box->border_left = -video_box->box_left;
video_box->crop_left = 0;
} else {
video_box->border_left = 0;
video_box->crop_left = video_box->box_left;
}
/* Round down/up for odd width differences */
if (crop_w < 0)
crop_w -= 1;
else
crop_w += 1;
video_box->box_right = crop_w / 2;
if (video_box->box_right < 0) {
video_box->border_right = -video_box->box_right;
video_box->crop_right = 0;
} else {
video_box->border_right = 0;
video_box->crop_right = video_box->box_right;
}
video_box->box_top = crop_h / 2;
if (video_box->box_top < 0) {
video_box->border_top = -video_box->box_top;
video_box->crop_top = 0;
} else {
video_box->border_top = 0;
video_box->crop_top = video_box->box_top;
}
/* Round down/up for odd height differences */
if (crop_h < 0)
crop_h -= 1;
else
crop_h += 1;
video_box->box_bottom = crop_h / 2;
if (video_box->box_bottom < 0) {
video_box->border_bottom = -video_box->box_bottom;
video_box->crop_bottom = 0;
} else {
video_box->border_bottom = 0;
video_box->crop_bottom = video_box->box_bottom;
}
}
static void
gst_video_box_get_property (GObject * object, guint prop_id, GValue * value,
GParamSpec * pspec)
{
GstVideoBox *video_box = GST_VIDEO_BOX (object);
switch (prop_id) {
case PROP_LEFT:
g_value_set_int (value, video_box->box_left);
break;
case PROP_RIGHT:
g_value_set_int (value, video_box->box_right);
break;
case PROP_TOP:
g_value_set_int (value, video_box->box_top);
break;
case PROP_BOTTOM:
g_value_set_int (value, video_box->box_bottom);
break;
case PROP_FILL_TYPE:
g_value_set_enum (value, video_box->fill_type);
break;
case PROP_ALPHA:
g_value_set_double (value, video_box->alpha);
break;
case PROP_BORDER_ALPHA:
g_value_set_double (value, video_box->border_alpha);
break;
case PROP_AUTOCROP:
g_value_set_boolean (value, video_box->autocrop);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
static GstCaps *
gst_video_box_transform_caps (GstBaseTransform * trans,
GstPadDirection direction, GstCaps * from)
{
GstVideoBox *video_box = GST_VIDEO_BOX (trans);
GstCaps *to, *ret;
const GstCaps *templ;
GstStructure *structure;
GstPad *other;
gint width, height;
to = gst_caps_copy (from);
structure = gst_caps_get_structure (to, 0);
/* get rid of format */
gst_structure_remove_field (structure, "format");
/* otherwise caps nego will fail: */
if (video_box->autocrop) {
gst_structure_remove_field (structure, "width");
gst_structure_remove_field (structure, "height");
} else {
/* calculate width and height */
if (gst_structure_get_int (structure, "width", &width)) {
if (direction == GST_PAD_SINK) {
width -= video_box->box_left;
width -= video_box->box_right;
} else {
width += video_box->box_left;
width += video_box->box_right;
}
if (width <= 0)
width = 1;
GST_DEBUG_OBJECT (trans, "New caps width: %d", width);
gst_structure_set (structure, "width", G_TYPE_INT, width, NULL);
}
if (gst_structure_get_int (structure, "height", &height)) {
if (direction == GST_PAD_SINK) {
height -= video_box->box_top;
height -= video_box->box_bottom;
} else {
height += video_box->box_top;
height += video_box->box_bottom;
}
if (height <= 0)
height = 1;
GST_DEBUG_OBJECT (trans, "New caps height: %d", height);
gst_structure_set (structure, "height", G_TYPE_INT, height, NULL);
}
}
/* filter against set allowed caps on the pad */
other = (direction == GST_PAD_SINK) ? trans->srcpad : trans->sinkpad;
templ = gst_pad_get_pad_template_caps (other);
ret = gst_caps_intersect (to, templ);
gst_caps_unref (to);
GST_DEBUG_OBJECT (video_box, "direction %d, transformed %" GST_PTR_FORMAT
" to %" GST_PTR_FORMAT, direction, from, ret);
return ret;
}
static gboolean
video_box_recalc_transform (GstVideoBox * video_box)
{
gboolean res = TRUE;
/* if we have the same format in and out and we don't need to perform any
* cropping at all, we can just operate in passthrough mode */
if (video_box->in_format == video_box->out_format &&
video_box->box_left == 0 && video_box->box_right == 0 &&
video_box->box_top == 0 && video_box->box_bottom == 0) {
GST_LOG_OBJECT (video_box, "we are using passthrough");
gst_base_transform_set_passthrough (GST_BASE_TRANSFORM_CAST (video_box),
TRUE);
} else {
GST_LOG_OBJECT (video_box, "we are not using passthrough");
gst_base_transform_set_passthrough (GST_BASE_TRANSFORM_CAST (video_box),
FALSE);
}
return res;
}
static gboolean
gst_video_box_set_caps (GstBaseTransform * trans, GstCaps * in, GstCaps * out)
{
GstVideoBox *video_box = GST_VIDEO_BOX (trans);
gboolean ret;
g_mutex_lock (video_box->mutex);
ret =
gst_video_format_parse_caps (in, &video_box->in_format,
&video_box->in_width, &video_box->in_height);
ret &=
gst_video_format_parse_caps (out, &video_box->out_format,
&video_box->out_width, &video_box->out_height);
/* something wrong getting the caps */
if (!ret)
goto no_caps;
GST_DEBUG_OBJECT (trans, "Input w: %d h: %d", video_box->in_width,
video_box->in_height);
GST_DEBUG_OBJECT (trans, "Output w: %d h: %d", video_box->out_width,
video_box->out_height);
if (video_box->autocrop)
gst_video_box_autocrop (video_box);
/* recalc the transformation strategy */
ret = video_box_recalc_transform (video_box);
g_mutex_unlock (video_box->mutex);
return ret;
/* ERRORS */
no_caps:
{
GST_DEBUG_OBJECT (video_box,
"Invalid caps: %" GST_PTR_FORMAT " -> %" GST_PTR_FORMAT, in, out);
g_mutex_unlock (video_box->mutex);
return FALSE;
}
}
static gboolean
gst_video_box_get_unit_size (GstBaseTransform * trans, GstCaps * caps,
guint * size)
{
GstVideoBox *video_box = GST_VIDEO_BOX (trans);
GstVideoFormat format;
gint width, height;
gboolean ret;
g_assert (size);
ret = gst_video_format_parse_caps (caps, &format, &width, &height);
if (!ret) {
GST_ERROR_OBJECT (video_box, "Invalid caps: %" GST_PTR_FORMAT, caps);
return FALSE;
}
*size = gst_video_format_get_size (format, width, height);
GST_LOG_OBJECT (video_box, "Returning from _unit_size %d", *size);
return TRUE;
}
static const guint8 yuv_colors_Y[VIDEO_BOX_FILL_LAST] = { 16, 150, 29 };
static const guint8 yuv_colors_U[VIDEO_BOX_FILL_LAST] = { 128, 46, 255 };
static const guint8 yuv_colors_V[VIDEO_BOX_FILL_LAST] = { 128, 21, 107 };
static void
gst_video_box_copy_plane_i420 (GstVideoBox * video_box, const guint8 * src,
guint8 * dest, gint br, gint bl, gint bt, gint bb, gint src_crop_width,
gint src_crop_height, gint src_stride, gint dest_width, gint dest_stride,
guint8 fill_color)
{
gint j;
/* top border */
for (j = 0; j < bt; j++) {
oil_splat_u8_ns (dest, &fill_color, dest_width);
dest += dest_stride;
}
/* copy and add left and right border */
for (j = 0; j < src_crop_height; j++) {
oil_splat_u8_ns (dest, &fill_color, bl);
oil_memcpy (dest + bl, src, src_crop_width);
oil_splat_u8_ns (dest + bl + src_crop_width, &fill_color, br);
dest += dest_stride;
src += src_stride;
}
/* bottom border */
for (j = 0; j < bb; j++) {
oil_splat_u8_ns (dest, &fill_color, dest_width);
dest += dest_stride;
}
}
static void
gst_video_box_apply_alpha (guint8 * dest, guint8 alpha)
{
if (dest[0] != 0)
dest[0] = alpha;
}
static void
gst_video_box_ayuv_ayuv (GstVideoBox * video_box, const guint8 * src,
guint8 * dest)
{
gint dblen = video_box->out_height * video_box->out_width;
guint32 *destb = (guint32 *) dest;
const guint32 *srcb = (const guint32 *) src;
guint8 b_alpha = (guint8) (video_box->border_alpha * 255);
guint8 i_alpha = (guint8) (video_box->alpha * 255);
gint br, bl, bt, bb, crop_w, crop_h;
gint i;
guint32 *loc = destb;
guint32 empty_pixel;
GST_LOG_OBJECT (video_box, "Processing AYUV -> AYUV data");
crop_h = 0;
crop_w = 0;
empty_pixel = GUINT32_FROM_BE ((b_alpha << 24) |
(yuv_colors_Y[video_box->fill_type] << 16) |
(yuv_colors_U[video_box->fill_type] << 8) |
yuv_colors_V[video_box->fill_type]);
br = video_box->box_right;
bl = video_box->box_left;
bt = video_box->box_top;
bb = video_box->box_bottom;
if (br >= 0 && bl >= 0) {
crop_w = video_box->in_width - (br + bl);
} else if (br >= 0 && bl < 0) {
crop_w = video_box->in_width - (br);
} else if (br < 0 && bl >= 0) {
crop_w = video_box->in_width - (bl);
} else if (br < 0 && bl < 0) {
crop_w = video_box->in_width;
}
if (bb >= 0 && bt >= 0) {
crop_h = video_box->in_height - (bb + bt);
} else if (bb >= 0 && bt < 0) {
crop_h = video_box->in_height - (bb);
} else if (bb < 0 && bt >= 0) {
crop_h = video_box->in_height - (bt);
} else if (bb < 0 && bt < 0) {
crop_h = video_box->in_height;
}
GST_DEBUG_OBJECT (video_box, "Borders are: L:%d, R:%d, T:%d, B:%d", bl, br,
bt, bb);
GST_DEBUG_OBJECT (video_box, "Alpha value is: %d", i_alpha);
if (crop_h <= 0 || crop_w <= 0) {
oil_splat_u32_ns (destb, &empty_pixel, dblen);
} else {
const guint32 *src_loc = srcb;
/* Top border */
if (bt < 0) {
oil_splat_u32_ns (loc, &empty_pixel, (-bt) * video_box->out_width);
loc = loc + ((-bt) * video_box->out_width);
} else {
src_loc = src_loc + (bt * video_box->in_width);
}
if (bl >= 0)
src_loc += bl;
for (i = 0; i < crop_h; i++) {
gint j;
/* Left border */
if (bl < 0) {
oil_splat_u32_ns (loc, &empty_pixel, -bl);
loc += (-bl);
}
/* Cropped area */
oil_copy_u8 ((guint8 *) loc, (guint8 *) src_loc, crop_w * 4);
for (j = 0; j < crop_w; j++)
gst_video_box_apply_alpha ((guint8 *) & loc[j], i_alpha);
src_loc += video_box->in_width;
loc += crop_w;
/* Right border */
if (br < 0) {
oil_splat_u32_ns (loc, &empty_pixel, -br);
loc += (-br);
}
}
/* Bottom border */
if (bb < 0) {
oil_splat_u32_ns (loc, &empty_pixel, (-bb) * video_box->out_width);
}
}
GST_LOG_OBJECT (video_box, "image created");
}
static gpointer
gst_video_box_clear (gpointer dest, gint size)
{
guint8 nil = 255;
oil_splat_u8_ns (dest, &nil, size);
return dest;
}
static gint
UVfloor (gint j)
{
return floor (((float) j) / 2);
}
static gint
UVceil (gint j)
{
return ceil (((float) j) / 2);
}
static void
gst_video_box_ayuv_i420 (GstVideoBox * video_box, const guint8 * src,
guint8 * dest)
{
gint br, bl, bt, bb, crop_w, crop_h, rest;
gint Ysize, Usize, Vsize;
guint8 *Ydest, *Udest, *Vdest;
guint8 *Utemp, *Vtemp;
guint32 empty_px_values[3];
gint i, j;
guint Ywidth, Uwidth, Vwidth;
GST_LOG_OBJECT (video_box, "AYUV to I420 conversion");
crop_h = 0;
crop_w = 0;
rest = 0;
empty_px_values[0] = yuv_colors_Y[video_box->fill_type];
empty_px_values[1] = yuv_colors_U[video_box->fill_type];
empty_px_values[2] = yuv_colors_V[video_box->fill_type];
Ywidth =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 0,
video_box->out_width);
Uwidth =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 1,
video_box->out_width);
Vwidth =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 2,
video_box->out_width);
Ydest =
dest + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 0,
video_box->out_width, video_box->out_height);
Udest =
dest + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 1,
video_box->out_width, video_box->out_height);
Vdest =
dest + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 2,
video_box->out_width, video_box->out_height);
Ysize =
Ywidth * gst_video_format_get_component_height (GST_VIDEO_FORMAT_I420, 0,
video_box->out_height);
Usize =
Ywidth * gst_video_format_get_component_height (GST_VIDEO_FORMAT_I420, 1,
video_box->out_height);
Vsize =
Ywidth * gst_video_format_get_component_height (GST_VIDEO_FORMAT_I420, 2,
video_box->out_height);
br = video_box->box_right;
bl = video_box->box_left;
bt = video_box->box_top;
bb = video_box->box_bottom;
if (br >= 0 && bl >= 0) {
rest = Ywidth - video_box->out_width;
crop_w = video_box->in_width - (bl + br);
} else if (br >= 0 && bl < 0) {
rest = Ywidth - video_box->out_width;
crop_w = video_box->in_width - (br);
} else if (br < 0 && bl >= 0) {
rest = Ywidth - video_box->out_width;
crop_w = video_box->in_width - (bl);
} else if (br < 0 && bl < 0) {
rest = Ywidth - video_box->out_width;
crop_w = video_box->in_width;
}
if (bb >= 0 && bt >= 0) {
crop_h = video_box->in_height - (bb + bt);
} else if (bb >= 0 && bt < 0) {
crop_h = video_box->in_height - (bb);
} else if (bb < 0 && bt >= 0) {
crop_h = video_box->in_height - (bt);
} else if (bb < 0 && bt < 0) {
crop_h = video_box->in_height;
}
Utemp = g_malloc0 (Uwidth);
Vtemp = g_malloc0 (Vwidth);
GST_LOG_OBJECT (video_box, "Borders are: L:%d, R:%d, T:%d, B:%d", bl, br, bt,
bb);
GST_LOG_OBJECT (video_box, "Starting conversion");
if (crop_h <= 0 || crop_w <= 0) {
oil_splat_u8_ns (Ydest, (guint8 *) & empty_px_values[0], Ysize);
oil_splat_u8_ns (Udest, (guint8 *) & empty_px_values[1], Usize);
oil_splat_u8_ns (Vdest, (guint8 *) & empty_px_values[2], Vsize);
} else {
gboolean sumbuff = FALSE;
const guint32 *src_loc1;
gint a = 0;
src_loc1 = (const guint32 *) src;
if (bt < 0) {
oil_splat_u8_ns (Ydest, (guint8 *) & empty_px_values[0], (-bt) * Ywidth);
oil_splat_u8_ns (Udest, (guint8 *) & empty_px_values[1],
(UVfloor (-bt) * Uwidth) + 7);
oil_splat_u8_ns (Vdest, (guint8 *) & empty_px_values[2],
UVfloor (-bt) * Vwidth);
if ((-bt) % 2 > 0) {
oil_splat_u8_ns (Utemp, (guint8 *) & empty_px_values[1], Uwidth);
oil_splat_u8_ns (Vtemp, (guint8 *) & empty_px_values[2], Vwidth);
sumbuff = TRUE;
}
Ydest += ((-bt) * Ywidth);
Udest += (UVfloor (-bt) * Uwidth);
Vdest += (UVfloor (-bt) * Vwidth);
} else {
src_loc1 = src_loc1 + (bt * video_box->in_width);
}
if (bl >= 0)
src_loc1 += bl;
GST_LOG_OBJECT (video_box, "Cropped area");
GST_LOG_OBJECT (video_box, "Ydest value: %p Ywidth: %u", Ydest, Ywidth);
GST_LOG_OBJECT (video_box, "Udest value: %p Uwidth: %u", Udest, Uwidth);
GST_LOG_OBJECT (video_box, "Vdest value: %p Vwidth: %u", Vdest, Vwidth);
GST_LOG_OBJECT (video_box, "Rest: %d", rest);
for (i = 0; i < crop_h; i++) {
a = 0;
if (sumbuff) {
/* left border */
if (bl < 0) {
oil_splat_u8_ns (Ydest, (guint8 *) & empty_px_values[0], -bl);
for (j = 0; j < -bl; j++) {
Utemp[UVfloor (j)] = (Utemp[UVfloor (j)] + empty_px_values[1]) / 2;
Vtemp[UVfloor (j)] = (Vtemp[UVfloor (j)] + empty_px_values[2]) / 2;
}
Ydest += -bl;
a = -bl;
}
for (j = 0; j < crop_w; j++) {
/* check ARCH */
Ydest[j] = ((guint8 *) & src_loc1[j])[1];
Utemp[UVfloor (a + j)] =
(Utemp[UVfloor (a + j)] + ((guint8 *) & src_loc1[j])[2]) / 2;
Vtemp[UVfloor (a + j)] =
(Vtemp[UVfloor (a + j)] + ((guint8 *) & src_loc1[j])[3]) / 2;
}
Ydest += crop_w;
/* right border */
if (br < 0) {
oil_splat_u8_ns (Ydest, (guint8 *) & empty_px_values[0], -br);
for (j = 0; j < -br; j++) {
Utemp[UVfloor (a + crop_w + j)] =
(Utemp[UVfloor (a + crop_w + j)] + empty_px_values[1]) / 2;
Vtemp[UVfloor (a + crop_w + j)] =
(Vtemp[UVfloor (a + crop_w + j)] + empty_px_values[2]) / 2;
}
Ydest += -br;
}
oil_copy_u8 (Udest, Utemp, Uwidth);
oil_copy_u8 (Vdest, Vtemp, Vwidth);
Udest += Uwidth;
Vdest += Vwidth;
Ydest += rest;
gst_video_box_clear (Utemp, Uwidth);
gst_video_box_clear (Vtemp, Vwidth);
src_loc1 += video_box->in_width;
sumbuff = FALSE;
} else {
/* left border */
a = 0;
if (bl < 0) {
oil_splat_u8_ns (Ydest, (guint8 *) & empty_px_values[0], -bl);
oil_splat_u8_ns (Vtemp, (guint8 *) & empty_px_values[1],
UVceil (-bl));
oil_splat_u8_ns (Utemp, (guint8 *) & empty_px_values[2],
UVceil (-bl));
Ydest += -bl;
a = -bl;
}
for (j = 0; j < crop_w; j++) {
/* check ARCH */
Ydest[j] = ((guint8 *) & src_loc1[j])[1];
if ((a + j) % 2 > 0) {
Utemp[UVfloor (a + j)] =
(Utemp[UVfloor (a + j)] + ((guint8 *) & src_loc1[j])[2]) / 2;
Vtemp[UVfloor (a + j)] =
(Vtemp[UVfloor (a + j)] + ((guint8 *) & src_loc1[j])[3]) / 2;
} else {
Utemp[UVfloor (a + j)] = ((guint8 *) & src_loc1[j])[2];
Vtemp[UVfloor (a + j)] = ((guint8 *) & src_loc1[j])[3];
}
}
Ydest += crop_w;
/* right border */
if (br < 0) {
j = 0;
if ((a + crop_w) % 2 > 0) {
Utemp[UVfloor (a + crop_w)] =
(Utemp[UVfloor (a + crop_w)] + empty_px_values[1]) / 2;
Vtemp[UVfloor (a + crop_w)] =
(Vtemp[UVfloor (a + crop_w)] + empty_px_values[2]) / 2;
a++;
j = -1;
}
oil_splat_u8_ns (Ydest, (guint8 *) & empty_px_values[0], -br);
oil_splat_u8_ns (&Utemp[UVfloor (a + crop_w)],
(guint8 *) & empty_px_values[1], UVceil ((-br) + j));
oil_splat_u8_ns (&Vtemp[UVfloor (a + crop_w)],
(guint8 *) & empty_px_values[2], UVceil ((-br) + j));
Ydest += -br;
}
Ydest += rest;
src_loc1 += video_box->in_width;
sumbuff = TRUE;
}
}
/* bottom border */
if (bb < 0) {
oil_splat_u8_ns (Ydest, (guint8 *) & empty_px_values[0], (-bb) * Ywidth);
if (sumbuff) {
for (i = 0; i < Uwidth; i++) {
Utemp[i] = (Utemp[i] + empty_px_values[1]) / 2;
}
for (i = 0; i < Vwidth; i++) {
Vtemp[i] = (Vtemp[i] + empty_px_values[2]) / 2;
}
oil_copy_u8 (Udest, Utemp, Uwidth);
oil_copy_u8 (Vdest, Vtemp, Vwidth);
Udest += Uwidth;
Vdest += Vwidth;
sumbuff = FALSE;
}
oil_splat_u8_ns (Udest, (guint8 *) & empty_px_values[1],
(UVfloor ((-bb))) * Uwidth);
oil_splat_u8_ns (Vdest, (guint8 *) & empty_px_values[2],
(UVfloor ((-bb))) * Vwidth);
}
if (sumbuff) {
oil_copy_u8 (Udest, Utemp, Uwidth);
oil_copy_u8 (Vdest, Vtemp, Vwidth);
}
}
GST_LOG_OBJECT (video_box, "image created");
g_free (Utemp);
g_free (Vtemp);
}
static void
gst_video_box_i420_ayuv (GstVideoBox * video_box, const guint8 * src,
guint8 * dest)
{
const guint8 *srcY, *srcU, *srcV;
gint crop_width, crop_width2, crop_height;
gint out_width, out_height;
gint src_stridey, src_strideu, src_stridev;
gint br, bl, bt, bb;
gint colorY, colorU, colorV;
gint i, j;
guint8 b_alpha = (guint8) (video_box->border_alpha * 255);
guint8 i_alpha = (guint8) (video_box->alpha * 255);
guint32 *destp;
guint32 *destb = (guint32 *) dest;
guint32 ayuv;
br = video_box->border_right;
bl = video_box->border_left;
bt = video_box->border_top;
bb = video_box->border_bottom;
out_width = video_box->out_width;
out_height = video_box->out_height;
src_stridey =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 0,
video_box->in_width);
src_strideu =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 1,
video_box->in_width);
src_stridev =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 2,
video_box->in_width);
crop_width = video_box->in_width;
crop_width -= (video_box->crop_left + video_box->crop_right);
crop_width2 = crop_width / 2;
crop_height = video_box->in_height;
crop_height -= (video_box->crop_top + video_box->crop_bottom);
srcY =
src + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 0,
video_box->in_width, video_box->in_height);
srcY += src_stridey * video_box->crop_top + video_box->crop_left;
srcU =
src + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 1,
video_box->in_width, video_box->in_height);
srcU += src_strideu * (video_box->crop_top / 2) + (video_box->crop_left / 2);
srcV =
src + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 2,
video_box->in_width, video_box->in_height);
srcV += src_stridev * (video_box->crop_top / 2) + (video_box->crop_left / 2);
colorY = yuv_colors_Y[video_box->fill_type];
colorU = yuv_colors_U[video_box->fill_type];
colorV = yuv_colors_V[video_box->fill_type];
ayuv =
GUINT32_FROM_BE ((b_alpha << 24) | (colorY << 16) | (colorU << 8) |
colorV);
/* top border */
if (bt) {
size_t nb_pixels = bt * out_width;
oil_splat_u32_ns (destb, &ayuv, nb_pixels);
destb += nb_pixels;
}
for (i = 0; i < crop_height; i++) {
destp = destb;
/* left border */
if (bl) {
oil_splat_u32_ns (destp, &ayuv, bl);
destp += bl;
}
dest = (guint8 *) destp;
/* center */
/* We can splat the alpha channel for the whole line */
oil_splat_u8 (dest, 4, &i_alpha, crop_width);
for (j = 0; j < crop_width2; j++) {
dest++;
*dest++ = *srcY++;
*dest++ = *srcU;
*dest++ = *srcV;
dest++;
*dest++ = *srcY++;
*dest++ = *srcU++;
*dest++ = *srcV++;
}
if (i % 2 == 0) {
srcU -= crop_width2;
srcV -= crop_width2;
} else {
srcU += src_strideu - crop_width2;
srcV += src_stridev - crop_width2;
}
srcY += src_stridey - (crop_width2 * 2);
destp = (guint32 *) dest;
/* right border */
if (br) {
oil_splat_u32_ns (destp, &ayuv, br);
}
destb += out_width;
}
/* bottom border */
if (bb) {
size_t nb_pixels = bb * out_width;
oil_splat_u32_ns (destb, &ayuv, nb_pixels);
}
}
static void
gst_video_box_i420_i420 (GstVideoBox * video_box, const guint8 * src,
guint8 * dest)
{
const guint8 *srcY, *srcU, *srcV;
guint8 *destY, *destU, *destV;
gint crop_width, crop_height;
gint out_width, out_height;
gint src_width, src_height;
gint src_stride, dest_stride;
gint br, bl, bt, bb;
br = video_box->border_right;
bl = video_box->border_left;
bt = video_box->border_top;
bb = video_box->border_bottom;
out_width = video_box->out_width;
out_height = video_box->out_height;
src_width = video_box->in_width;
src_height = video_box->in_height;
crop_width = src_width - (video_box->crop_left + video_box->crop_right);
crop_height = src_height - (video_box->crop_top + video_box->crop_bottom);
/* Y plane */
src_stride =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 0, src_width);
dest_stride =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 0, out_width);
destY =
dest + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 0,
out_width, out_height);
srcY =
src + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 0,
src_width, src_height);
srcY += src_stride * video_box->crop_top + video_box->crop_left;
gst_video_box_copy_plane_i420 (video_box, srcY, destY, br, bl, bt, bb,
crop_width, crop_height, src_stride, out_width, dest_stride,
yuv_colors_Y[video_box->fill_type]);
br /= 2;
bb /= 2;
bl /= 2;
bt /= 2;
/* we need to round up to make sure we draw all the U and V lines */
crop_width = (crop_width + 1) / 2;
crop_height = (crop_height + 1) / 2;
/* U plane */
src_stride =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 1, src_width);
dest_stride =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 1, out_width);
destU =
dest + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 1,
out_width, out_height);
srcU =
src + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 1,
src_width, src_height);
srcU += src_stride * (video_box->crop_top / 2) + (video_box->crop_left / 2);
gst_video_box_copy_plane_i420 (video_box, srcU, destU, br, bl, bt, bb,
crop_width, crop_height, src_stride, out_width / 2, dest_stride,
yuv_colors_U[video_box->fill_type]);
/* V plane */
src_stride =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 2, src_width);
dest_stride =
gst_video_format_get_row_stride (GST_VIDEO_FORMAT_I420, 2, out_width);
destV =
dest + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 2,
out_width, out_height);
srcV =
src + gst_video_format_get_component_offset (GST_VIDEO_FORMAT_I420, 2,
src_width, src_height);
srcV += src_stride * (video_box->crop_top / 2) + (video_box->crop_left / 2);
gst_video_box_copy_plane_i420 (video_box, srcV, destV, br, bl, bt, bb,
crop_width, crop_height, src_stride, out_width / 2, dest_stride,
yuv_colors_V[video_box->fill_type]);
}
static GstFlowReturn
gst_video_box_transform (GstBaseTransform * trans, GstBuffer * in,
GstBuffer * out)
{
GstVideoBox *video_box = GST_VIDEO_BOX (trans);
const guint8 *indata;
guint8 *outdata;
GstClockTime timestamp, stream_time;
timestamp = GST_BUFFER_TIMESTAMP (in);
stream_time =
gst_segment_to_stream_time (&trans->segment, GST_FORMAT_TIME, timestamp);
GST_DEBUG_OBJECT (video_box, "sync to %" GST_TIME_FORMAT,
GST_TIME_ARGS (timestamp));
if (GST_CLOCK_TIME_IS_VALID (stream_time))
gst_object_sync_values (G_OBJECT (video_box), stream_time);
indata = GST_BUFFER_DATA (in);
outdata = GST_BUFFER_DATA (out);
g_mutex_lock (video_box->mutex);
switch (video_box->in_format) {
case GST_VIDEO_FORMAT_AYUV:
switch (video_box->out_format) {
case GST_VIDEO_FORMAT_AYUV:
gst_video_box_ayuv_ayuv (video_box, indata, outdata);
break;
case GST_VIDEO_FORMAT_I420:
gst_video_box_ayuv_i420 (video_box, indata, outdata);
break;
default:
goto invalid_format;
}
break;
case GST_VIDEO_FORMAT_I420:
switch (video_box->out_format) {
case GST_VIDEO_FORMAT_AYUV:
gst_video_box_i420_ayuv (video_box, indata, outdata);
break;
case GST_VIDEO_FORMAT_I420:
gst_video_box_i420_i420 (video_box, indata, outdata);
break;
default:
goto invalid_format;
}
break;
default:
goto invalid_format;
}
g_mutex_unlock (video_box->mutex);
return GST_FLOW_OK;
/* ERRORS */
invalid_format:
{
g_mutex_unlock (video_box->mutex);
return GST_FLOW_ERROR;
}
}
/* FIXME: 0.11 merge with videocrop plugin */
static gboolean
plugin_init (GstPlugin * plugin)
{
oil_init ();
gst_controller_init (NULL, NULL);
GST_DEBUG_CATEGORY_INIT (videobox_debug, "videobox", 0,
"Resizes a video by adding borders or cropping");
return gst_element_register (plugin, "videobox", GST_RANK_NONE,
GST_TYPE_VIDEO_BOX);
}
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
GST_VERSION_MINOR,
"videobox",
"resizes a video by adding borders or cropping",
plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)
Reference in a new issue View git blame Copy permalink