diff options
Diffstat (limited to 'gst/videocrop/gstvideocrop.c')
| -rw-r--r-- | gst/videocrop/gstvideocrop.c | 419 |
1 files changed, 419 insertions, 0 deletions
diff --git a/gst/videocrop/gstvideocrop.c b/gst/videocrop/gstvideocrop.c new file mode 100644 index 00000000..d1cc5af4 --- /dev/null +++ b/gst/videocrop/gstvideocrop.c @@ -0,0 +1,419 @@ +/* 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 <string.h> + +#define GST_TYPE_VIDEO_CROP \ + (gst_video_crop_get_type()) +#define GST_VIDEO_CROP(obj) \ + (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_VIDEO_CROP,GstVideoCrop)) +#define GST_VIDEO_CROP_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_VIDEO_CROP,GstVideoCropClass)) +#define GST_IS_VIDEO_CROP(obj) \ + (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_VIDEO_CROP)) +#define GST_IS_VIDEO_CROP_CLASS(obj) \ + (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_VIDEO_CROP)) + +typedef struct _GstVideoCrop GstVideoCrop; +typedef struct _GstVideoCropClass GstVideoCropClass; + +struct _GstVideoCrop { + GstElement element; + + /* pads */ + GstPad *sinkpad; + GstPad *srcpad; + + /* caps */ + gint width, height; + gint crop_x, crop_y; + gint crop_width, crop_height; +}; + +struct _GstVideoCropClass { + GstElementClass parent_class; +}; + +/* elementfactory information */ +static GstElementDetails gst_video_crop_details = { + "video crop filter", + "Filter/Video/Crop", + "Crops video into a user defined region", + VERSION, + "Wim Taymans <wim.taymans@chello.be>", + "(C) 2002", +}; + + +/* VideoCrop signals and args */ +enum { + /* FILL ME */ + LAST_SIGNAL +}; + +enum { + ARG_0, + ARG_X, + ARG_Y, + ARG_WIDTH, + ARG_HEIGHT, + /* FILL ME */ +}; + +GST_PAD_TEMPLATE_FACTORY (video_crop_src_template_factory, + "src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_CAPS_NEW ( + "video_crop_src", + "video/raw", + "format", GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")) + ) +) + +GST_PAD_TEMPLATE_FACTORY (video_crop_sink_template_factory, + "sink", + GST_PAD_SINK, + GST_PAD_ALWAYS, + GST_CAPS_NEW ( + "video_crop_sink", + "video/raw", + "format", GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")) + ) +) + + +static void gst_video_crop_class_init (GstVideoCropClass *klass); +static void gst_video_crop_init (GstVideoCrop *video_crop); + +static void gst_video_crop_set_property (GObject *object, guint prop_id, + const GValue *value, GParamSpec *pspec); +static void gst_video_crop_get_property (GObject *object, guint prop_id, + GValue *value, GParamSpec *pspec); + +static GstPadConnectReturn + gst_video_crop_sink_connect (GstPad *pad, GstCaps *caps); +static void gst_video_crop_chain (GstPad *pad, GstBuffer *buffer); + +static GstElementStateReturn + gst_video_crop_change_state (GstElement *element); + + +static GstElementClass *parent_class = NULL; +/* static guint gst_video_crop_signals[LAST_SIGNAL] = { 0 }; */ + +GType +gst_video_crop_get_type (void) +{ + static GType video_crop_type = 0; + + if (!video_crop_type) { + static const GTypeInfo video_crop_info = { + sizeof(GstVideoCropClass), + NULL, + NULL, + (GClassInitFunc)gst_video_crop_class_init, + NULL, + NULL, + sizeof(GstVideoCrop), + 0, + (GInstanceInitFunc)gst_video_crop_init, + }; + video_crop_type = g_type_register_static(GST_TYPE_ELEMENT, "GstVideoCrop", &video_crop_info, 0); + } + return video_crop_type; +} + +static void +gst_video_crop_class_init (GstVideoCropClass *klass) +{ + GObjectClass *gobject_class; + GstElementClass *gstelement_class; + + gobject_class = (GObjectClass*) klass; + gstelement_class = (GstElementClass*) klass; + + parent_class = g_type_class_ref(GST_TYPE_ELEMENT); + + g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_X, + g_param_spec_int ("x", "X", "X offset of image", + 0, G_MAXINT, 0, G_PARAM_READWRITE)); + g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_Y, + g_param_spec_int ("y", "Y", "Y offset of image", + 0, G_MAXINT, 0, G_PARAM_READWRITE)); + g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_WIDTH, + g_param_spec_int ("width", "Width", "Width of image", + 0, G_MAXINT, 0, G_PARAM_READWRITE)); + g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_HEIGHT, + g_param_spec_int ("height", "Height", "Height of image", + 0, G_MAXINT, 0, G_PARAM_READWRITE)); + + gobject_class->set_property = gst_video_crop_set_property; + gobject_class->get_property = gst_video_crop_get_property; + + gstelement_class->change_state = gst_video_crop_change_state; +} + +static void +gst_video_crop_init (GstVideoCrop *video_crop) +{ + /* create the sink and src pads */ + video_crop->sinkpad = gst_pad_new_from_template( + GST_PAD_TEMPLATE_GET (video_crop_sink_template_factory), "sink"); + gst_element_add_pad (GST_ELEMENT (video_crop), video_crop->sinkpad); + gst_pad_set_chain_function (video_crop->sinkpad, GST_DEBUG_FUNCPTR (gst_video_crop_chain)); + gst_pad_set_connect_function (video_crop->sinkpad, GST_DEBUG_FUNCPTR (gst_video_crop_sink_connect)); + + video_crop->srcpad = gst_pad_new_from_template( + GST_PAD_TEMPLATE_GET (video_crop_src_template_factory), "src"); + gst_element_add_pad (GST_ELEMENT (video_crop), video_crop->srcpad); + + video_crop->crop_x = 0; + video_crop->crop_y = 0; + video_crop->crop_width = 128; + video_crop->crop_height = 128; + + GST_FLAG_SET (video_crop, GST_ELEMENT_EVENT_AWARE); +} + +/* do we need this function? */ +static void +gst_video_crop_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec) +{ + GstVideoCrop *video_crop; + + /* it's not null if we got it, but it might not be ours */ + g_return_if_fail (GST_IS_VIDEO_CROP (object)); + + video_crop = GST_VIDEO_CROP (object); + + switch (prop_id) { + case ARG_X: + video_crop->crop_x = g_value_get_int (value); + break; + case ARG_Y: + video_crop->crop_y = g_value_get_int (value); + break; + case ARG_WIDTH: + video_crop->crop_width = g_value_get_int (value); + break; + case ARG_HEIGHT: + video_crop->crop_height = g_value_get_int (value); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} +static void +gst_video_crop_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec) +{ + GstVideoCrop *video_crop; + + /* it's not null if we got it, but it might not be ours */ + g_return_if_fail (GST_IS_VIDEO_CROP (object)); + + video_crop = GST_VIDEO_CROP (object); + + switch (prop_id) { + case ARG_X: + g_value_set_int (value, video_crop->crop_x); + break; + case ARG_Y: + g_value_set_int (value, video_crop->crop_y); + break; + case ARG_WIDTH: + g_value_set_int (value, video_crop->crop_width); + break; + case ARG_HEIGHT: + g_value_set_int (value, video_crop->crop_height); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static GstPadConnectReturn +gst_video_crop_sink_connect (GstPad *pad, GstCaps *caps) +{ + GstVideoCrop *video_crop; + + /* we are not going to act on variable caps */ + if (!GST_CAPS_IS_FIXED (caps)) + return GST_PAD_CONNECT_DELAYED; + + video_crop = GST_VIDEO_CROP (gst_pad_get_parent (pad)); + + gst_caps_get_int (caps, "width", &video_crop->width); + gst_caps_get_int (caps, "height", &video_crop->height); + + if (video_crop->crop_width + video_crop->crop_x > video_crop->width) + video_crop->crop_width = video_crop->width - video_crop->crop_x; + if (video_crop->crop_height + video_crop->crop_y > video_crop->height) + video_crop->crop_height = video_crop->height - video_crop->crop_y; + + return GST_PAD_CONNECT_OK; +} + +static void +gst_video_crop_i420 (GstVideoCrop *video_crop, GstBuffer *src_buffer, GstBuffer *dest_buffer) +{ + guint8 *srcY, *srcU, *srcV; + guint8 *destY, *destU, *destV; + gint width = video_crop->crop_width; + gint src_stride = video_crop->width; + gint frame_size = video_crop->width * video_crop->height; + gint crop_height; + gint j; + + srcY = GST_BUFFER_DATA (src_buffer) + (src_stride * video_crop->crop_y + video_crop->crop_x); + destY = GST_BUFFER_DATA (dest_buffer); + + crop_height = video_crop->crop_height; + + /* copy Y plane first */ + for (j = crop_height; j; j--) { + memcpy (destY, srcY, width); + srcY += src_stride; + destY += width; + } + + width >>= 1; + src_stride >>= 1; + crop_height >>= 1; + + destU = destY; + destV = destU + ((video_crop->crop_width * crop_height) >> 1); + + srcU = GST_BUFFER_DATA (src_buffer) + frame_size + ((src_stride * video_crop->crop_y + video_crop->crop_x) >> 1); + srcV = srcU + (frame_size >> 2); + + /* copy U plane */ + for (j = crop_height; j; j--) { + memcpy (destU, srcU, width); + srcU += src_stride; + destU += width; + } + /* copy U plane */ + for (j = crop_height; j; j--) { + memcpy (destV, srcV, width); + srcV += src_stride; + destV += width; + } +} + +static void +gst_video_crop_chain (GstPad *pad, GstBuffer *buffer) +{ + GstVideoCrop *video_crop; + GstBuffer *outbuf; + + video_crop = GST_VIDEO_CROP (gst_pad_get_parent (pad)); + + if (GST_IS_EVENT (buffer)) { + GstEvent *event = GST_EVENT (buffer); + + switch (GST_EVENT_TYPE (event)) { + default: + gst_pad_event_default (pad, event); + break; + } + return; + } + + if (GST_PAD_CAPS (video_crop->srcpad) == NULL) { + if (!gst_pad_try_set_caps (video_crop->srcpad, + GST_CAPS_NEW ( + "video_crop_caps", + "video/raw", + "format", GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")), + "width", GST_PROPS_INT (video_crop->crop_width), + "height", GST_PROPS_INT (video_crop->crop_height) + ))) + { + gst_element_error (GST_ELEMENT (video_crop), "could not negotiate pads"); + return; + } + } + + outbuf = gst_buffer_new_and_alloc ((video_crop->crop_width * video_crop->crop_height * 3) / 2); + GST_BUFFER_TIMESTAMP (outbuf) = GST_BUFFER_TIMESTAMP (buffer); + + gst_video_crop_i420 (video_crop, buffer, outbuf); + gst_buffer_unref (buffer); + + gst_pad_push (video_crop->srcpad, outbuf); +} + +static GstElementStateReturn +gst_video_crop_change_state (GstElement *element) +{ + GstVideoCrop *video_crop; + + video_crop = GST_VIDEO_CROP (element); + + switch (GST_STATE_TRANSITION (element)) { + case GST_STATE_NULL_TO_READY: + break; + case GST_STATE_READY_TO_PAUSED: + break; + case GST_STATE_PAUSED_TO_PLAYING: + break; + case GST_STATE_PLAYING_TO_PAUSED: + break; + case GST_STATE_PAUSED_TO_READY: + break; + case GST_STATE_READY_TO_NULL: + break; + } + + parent_class->change_state (element); + + return GST_STATE_SUCCESS; +} + +static gboolean +plugin_init (GModule *module, GstPlugin *plugin) +{ + GstElementFactory *factory; + + /* create an elementfactory for the videocrop element */ + factory = gst_element_factory_new ("videocrop", GST_TYPE_VIDEO_CROP, &gst_video_crop_details); + g_return_val_if_fail (factory != NULL, FALSE); + + gst_element_factory_add_pad_template (factory, + GST_PAD_TEMPLATE_GET (video_crop_sink_template_factory)); + gst_element_factory_add_pad_template (factory, + GST_PAD_TEMPLATE_GET (video_crop_src_template_factory)); + gst_element_factory_set_rank (factory, GST_ELEMENT_RANK_PRIMARY); + + gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory)); + + return TRUE; +} + +GstPluginDesc plugin_desc = { + GST_VERSION_MAJOR, + GST_VERSION_MINOR, + "videocrop", + plugin_init +}; |