From 3c6a230219821f0fdb7883de5707772fdd9a77d7 Mon Sep 17 00:00:00 2001 From: Jeremy Simon Date: Tue, 7 May 2002 22:29:08 +0000 Subject: add gstsnapshot : drop a frame to a png file when the plugin recieved a signal or at a given frame Original commit message from CVS: add gstsnapshot : drop a frame to a png file when the plugin recieved a signal or at a given frame --- configure.ac | 9 ++ ext/snapshot/Makefile.am | 10 ++ ext/snapshot/gstsnapshot.c | 396 +++++++++++++++++++++++++++++++++++++++++++++ ext/snapshot/gstsnapshot.h | 83 ++++++++++ 4 files changed, 498 insertions(+) create mode 100644 ext/snapshot/Makefile.am create mode 100644 ext/snapshot/gstsnapshot.c create mode 100644 ext/snapshot/gstsnapshot.h diff --git a/configure.ac b/configure.ac index 3afe7dc9..a674cb5b 100644 --- a/configure.ac +++ b/configure.ac @@ -572,6 +572,14 @@ GST_CHECK_FEATURE(SMOOTHWAVE, [smoothwave plugin], smoothwave, [ if test $HAVE_GTK = "yes"; then HAVE_SMOOTHWAVE=yes; fi; ]) + +dnl *** snapshot *** +translit(dnm, m, l) AM_CONDITIONAL(USE_LIBPNG, true) +GST_CHECK_FEATURE(LIBPNG, [snapshot plugin], snapshot, [ + GST_CHECK_LIBHEADER(LIBPNG, png, png_read_info, -lz, png.h, LIBPNG_LIBS="-lpng -lz") + AC_SUBST(LIBPNG_LIBS) +]) + dnl *** tarkin *** dnl for now the sources are included in the plugin dnl and should be moved to ext-libs/ perhaps @@ -833,6 +841,7 @@ ext/sdl/Makefile ext/shout/Makefile ext/sidplay/Makefile ext/smoothwave/Makefile +ext/snapshot/Makefile ext/vorbis/Makefile ext/tarkin/Makefile ext/xmms/Makefile diff --git a/ext/snapshot/Makefile.am b/ext/snapshot/Makefile.am new file mode 100644 index 00000000..057a650e --- /dev/null +++ b/ext/snapshot/Makefile.am @@ -0,0 +1,10 @@ +plugindir = $(libdir)/gst + +plugin_LTLIBRARIES = libgstsnapshot.la + +libgstsnapshot_la_SOURCES = gstsnapshot.c ../hermes/yuv2rgb.c ../hermes/yuv2yuv.c +libgstsnapshot_la_CFLAGS = $(GST_CFLAGS) +libgstsnapshot_la_LIBADD = $(GST_LIBS) $(LIBPNG_LIBS) +libgstsnapshot_la_LDFLAGS = @GST_PLUGIN_LDFLAGS@ + +noinst_HEADERS = gstsnapshot.h diff --git a/ext/snapshot/gstsnapshot.c b/ext/snapshot/gstsnapshot.c new file mode 100644 index 00000000..20dc2648 --- /dev/null +++ b/ext/snapshot/gstsnapshot.c @@ -0,0 +1,396 @@ +/* GStreamer + * Copyright (C) <1999> Erik Walthinsen + * + * 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 +#include + +#include "gstsnapshot.h" + +#define MAX_HEIGHT 2048 + + +static GstElementDetails snapshot_details = { + "snapshot", + "Filter/Effect", + "Dump a frame to a png file", + VERSION, + "Jeremy SIMON ", + "(C) 2002", +}; + +GST_PAD_TEMPLATE_FACTORY (snapshot_src_factory, + "src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_CAPS_NEW ( + "snapshot_src", + "video/raw", + "format", GST_PROPS_LIST ( + GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")), + GST_PROPS_FOURCC (GST_STR_FOURCC ("YUY2")) + ), + "width", GST_PROPS_INT_RANGE(0, G_MAXINT), + "height", GST_PROPS_INT_RANGE(0, G_MAXINT) + ) +) + +GST_PAD_TEMPLATE_FACTORY (snapshot_sink_factory, + "sink", + GST_PAD_SINK, + GST_PAD_ALWAYS, + GST_CAPS_NEW ( + "snapshot_src", + "video/raw", + "format", GST_PROPS_LIST ( + GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")), + GST_PROPS_FOURCC (GST_STR_FOURCC ("YUY2")) + ), + "width", GST_PROPS_INT_RANGE(0, G_MAXINT), + "height", GST_PROPS_INT_RANGE(0, G_MAXINT) + ) +) + + +/* Shot signals and args */ +enum { + /* FILL ME */ + SNAPSHOT_SIGNAL, + LAST_SIGNAL +}; + +enum { + ARG_0, + ARG_FRAME, + ARG_LOCATION +}; + +static GType gst_snapshot_get_type (void); +static void gst_snapshot_class_init (GstShotClass *klass); +static void gst_snapshot_init (GstShot *snapshot); + +static void gst_snapshot_chain (GstPad *pad, GstBuffer *buf); + +static void gst_snapshot_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec); +static void gst_snapshot_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec); +static void snapshot_handler(GstElement *element); + + +static GstElementClass *parent_class = NULL; +static guint gst_snapshot_signals[LAST_SIGNAL] = { 0 }; + + +static void user_error_fn(png_structp png_ptr, png_const_charp error_msg) +{ + g_warning("%s", error_msg); +} + +static void user_warning_fn(png_structp png_ptr, png_const_charp warning_msg) +{ + g_warning("%s", warning_msg); +} + +GType +gst_snapshot_get_type (void) +{ + static GType snapshot_type = 0; + + if (!snapshot_type) { + static const GTypeInfo snapshot_info = { + sizeof(GstShotClass), NULL, NULL, (GClassInitFunc)gst_snapshot_class_init, + NULL, + NULL, + sizeof(GstShot), + 0, + (GInstanceInitFunc)gst_snapshot_init, + }; + snapshot_type = g_type_register_static(GST_TYPE_ELEMENT, "GstShot", &snapshot_info, 0); + } + return snapshot_type; +} + +static void +gst_snapshot_class_init (GstShotClass *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_FRAME, + g_param_spec_long("frame","frame","frame", + 0, G_MAXLONG, 0, G_PARAM_READWRITE)); + g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_LOCATION, + g_param_spec_string("location","location","location", + 0,G_PARAM_READWRITE)); + + gst_snapshot_signals[SNAPSHOT_SIGNAL] = + g_signal_new("snapshot", G_TYPE_FROM_CLASS(klass), + G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET(GstShotClass, snapshot), + NULL, NULL, g_cclosure_marshal_VOID__VOID, G_TYPE_NONE, 0); + + klass->snapshot = snapshot_handler; + + gobject_class->set_property = gst_snapshot_set_property; + gobject_class->get_property = gst_snapshot_get_property; +} + +static void +snapshot_handler(GstElement *element) +{ + GstShot *snapshot; + + snapshot = GST_SNAPSHOT( element ); + snapshot->snapshot_asked=TRUE; +} + + +static gboolean +gst_snapshot_sinkconnect (GstPad *pad, GstCaps *caps) +{ + GstShot *filter; + GstCaps *from_caps, *to_caps; + + filter = GST_SNAPSHOT (gst_pad_get_parent (pad)); + + if (!GST_CAPS_IS_FIXED (caps)) + return GST_PAD_CONNECT_DELAYED; + + gst_caps_get_int (caps, "width", &filter->width); + gst_caps_get_int (caps, "height", &filter->height); + gst_caps_get_fourcc_int (caps, "format", &filter->format); + filter->to_bpp = 24; + + + to_caps = GST_CAPS_NEW ( + "snapshot_conversion", + "video/raw", + "format", GST_PROPS_FOURCC (GST_STR_FOURCC ("RGB ")), + "width", GST_PROPS_INT( filter->width ), + "height", GST_PROPS_INT( filter->height ), + "red_mask", GST_PROPS_INT (0xFF0000), + "green_mask", GST_PROPS_INT (0x00FF00), + "blue_mask", GST_PROPS_INT (0x0000FF), + "bpp", GST_PROPS_INT( 24 ) + ); + + switch ( filter->format ) + { + case GST_MAKE_FOURCC('Y','U','Y','2'): + case GST_MAKE_FOURCC('I','4','2','0'): + from_caps = GST_CAPS_NEW ( + "snapshot_from", + "video/raw", + "format", GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")), + "width", GST_PROPS_INT( filter->width ), + "height", GST_PROPS_INT( filter->height ), + "red_mask", GST_PROPS_INT (0xFF0000), + "green_mask", GST_PROPS_INT (0x00FF00), + "blue_mask", GST_PROPS_INT (0x0000FF), + "bpp", GST_PROPS_INT( 24 ) + ); + + filter->converter = gst_colorspace_yuv2rgb_get_converter ( from_caps, to_caps ); + break; + + default : + break; + } + + filter->png_struct_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, (png_voidp)NULL, user_error_fn, user_warning_fn); + if ( filter->png_struct_ptr == NULL ) + g_warning( "Failed to initialize png structure"); + + filter->png_info_ptr = png_create_info_struct( filter->png_struct_ptr ); + + if (setjmp( filter->png_struct_ptr->jmpbuf)) + png_destroy_write_struct(&filter->png_struct_ptr, &filter->png_info_ptr); + + gst_pad_try_set_caps (filter->srcpad, caps); + + return GST_PAD_CONNECT_OK; +} + +static void +gst_snapshot_init (GstShot *snapshot) +{ + snapshot->sinkpad = gst_pad_new_from_template (GST_PAD_TEMPLATE_GET (snapshot_sink_factory), "sink"); + gst_pad_set_connect_function (snapshot->sinkpad, gst_snapshot_sinkconnect); + gst_pad_set_chain_function (snapshot->sinkpad, gst_snapshot_chain); + gst_element_add_pad (GST_ELEMENT (snapshot), snapshot->sinkpad); + + snapshot->srcpad = gst_pad_new_from_template (GST_PAD_TEMPLATE_GET (snapshot_src_factory), "src"); + gst_element_add_pad (GST_ELEMENT (snapshot), snapshot->srcpad); + + snapshot->cur_frame = 0; + snapshot->frame=-1; + snapshot->snapshot_asked=FALSE; +} + + +static void +gst_snapshot_chain (GstPad *pad, GstBuffer *buf) +{ + GstShot *snapshot; + guchar *data, *data_to_convert, *buffer_i420, *data_converted; + gulong size,image_size; + GstBuffer *outbuf; + gint i; + png_byte *row_pointers[ MAX_HEIGHT ]; + FILE *fp; + + g_return_if_fail(pad != NULL); + g_return_if_fail(GST_IS_PAD(pad)); + g_return_if_fail(buf != NULL); + + snapshot = GST_SNAPSHOT (GST_OBJECT_PARENT (pad)); + + data = GST_BUFFER_DATA(buf); + size = GST_BUFFER_SIZE(buf); + + GST_DEBUG (0,"snapshot: have buffer of %d\n", GST_BUFFER_SIZE(buf)); + + outbuf = gst_buffer_new(); + GST_BUFFER_DATA(outbuf) = g_malloc(GST_BUFFER_SIZE(buf)); + GST_BUFFER_SIZE(outbuf) = GST_BUFFER_SIZE(buf); + + snapshot->cur_frame++; + if ( snapshot->cur_frame == snapshot->frame || snapshot->snapshot_asked == TRUE ) + { + snapshot->snapshot_asked = FALSE; + image_size = snapshot->width * snapshot->height; + data_converted = g_malloc ((image_size * (snapshot->to_bpp/8)) ); + + if ( snapshot->format == GST_MAKE_FOURCC('Y','U','Y','2') ) + { + GST_DEBUG(0, "YUY2 => RGB\n"); + buffer_i420 = g_malloc ((image_size * (snapshot->to_bpp/8)) ); + gst_colorspace_yuy2_to_i420( data, buffer_i420, snapshot->width, snapshot->height); + data_to_convert = buffer_i420; + } + else + data_to_convert = data; + + gst_colorspace_convert (snapshot->converter, data_to_convert, data_converted); + + GST_INFO(0,"dumpfile : %s\n", snapshot->location ); + fp = fopen( snapshot->location, "wb" ); + if ( fp == NULL ) + g_warning(" Can not open %s\n", snapshot->location ); + else + { + png_set_filter( snapshot->png_struct_ptr, 0, PNG_FILTER_NONE | PNG_FILTER_VALUE_NONE ); + png_init_io(snapshot->png_struct_ptr, fp); + png_set_compression_level( snapshot->png_struct_ptr, 9); + png_set_IHDR( + snapshot->png_struct_ptr, + snapshot->png_info_ptr, + snapshot->width, + snapshot->height, + snapshot->to_bpp/3, + PNG_COLOR_TYPE_RGB, + PNG_INTERLACE_NONE, + PNG_COMPRESSION_TYPE_DEFAULT, + PNG_FILTER_TYPE_DEFAULT + ); + + for ( i = 0; i < snapshot->height; i++ ) + row_pointers[i] = data_converted + (snapshot->width * i * snapshot->to_bpp/8 ); + + png_write_info( snapshot->png_struct_ptr, snapshot->png_info_ptr ); + png_write_image( snapshot->png_struct_ptr, row_pointers ); + png_write_end( snapshot->png_struct_ptr, NULL ); + png_destroy_info_struct ( snapshot->png_struct_ptr, &snapshot->png_info_ptr ); + png_destroy_write_struct( &snapshot->png_struct_ptr, (png_infopp)NULL ); + fclose( fp ); + } + } + + gst_pad_push(snapshot->srcpad,buf ); +} + +static void +gst_snapshot_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec) +{ + GstShot *snapshot; + + g_return_if_fail(GST_IS_SNAPSHOT(object)); + snapshot = GST_SNAPSHOT(object); + + switch (prop_id) { + case ARG_LOCATION: + snapshot->location = g_strdup(g_value_get_string (value)); + break; + case ARG_FRAME: + snapshot->frame = g_value_get_long(value); + break; + default: + break; + } +} + +static void +gst_snapshot_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec) +{ + GstShot *snapshot; + + g_return_if_fail(GST_IS_SNAPSHOT(object)); + snapshot = GST_SNAPSHOT(object); + + switch (prop_id) { + case ARG_LOCATION: + g_value_set_string(value, snapshot->location); + break; + case ARG_FRAME: + g_value_set_long(value, snapshot->frame); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + + +static gboolean +plugin_init (GModule *module, GstPlugin *plugin) +{ + GstElementFactory *factory; + + factory = gst_element_factory_new("snapshot",GST_TYPE_SNAPSHOT, + &snapshot_details); + g_return_val_if_fail(factory != NULL, FALSE); + + gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET (snapshot_sink_factory)); + gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET (snapshot_src_factory)); + + gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory)); + + + return TRUE; +} + +GstPluginDesc plugin_desc = { + GST_VERSION_MAJOR, + GST_VERSION_MINOR, + "snapshot", + plugin_init +}; + diff --git a/ext/snapshot/gstsnapshot.h b/ext/snapshot/gstsnapshot.h new file mode 100644 index 00000000..d0d7221c --- /dev/null +++ b/ext/snapshot/gstsnapshot.h @@ -0,0 +1,83 @@ +/* Gnome-Streamer + * Copyright (C) <1999> Erik Walthinsen + * + * 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. + */ + + +#ifndef __GST_SNAPSHOT_H__ +#define __GST_SNAPSHOT_H__ + + +#include +#include +#include + +#include "../hermes/yuv2rgb.h" +#include "../hermes/yuv2yuv.h" + + +#ifdef __cplusplus +extern "C" { +#endif /* __cplusplus */ + + +#define GST_TYPE_SNAPSHOT \ + (gst_snapshot_get_type()) +#define GST_SNAPSHOT(obj) \ + (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_SNAPSHOT,GstShot)) +#define GST_SNAPSHOT_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_SNAPSHOT,GstShot)) +#define GST_IS_SNAPSHOT(obj) \ + (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_SNAPSHOT)) +#define GST_IS_SNAPSHOT_CLASS(obj) \ + (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_SNAPSHOT)) + +typedef struct _GstShot GstShot; +typedef struct _GstShotClass GstShotClass; + +struct _GstShot { + GstElement element; + + guint32 format; + gint width; + gint height; + gint to_bpp; + glong frame; + glong cur_frame; + const gchar *location; + gboolean snapshot_asked; + + png_structp png_struct_ptr; + png_infop png_info_ptr; + + GstColorSpaceConverter *converter; + + GstPad *sinkpad,*srcpad; +}; + +struct _GstShotClass { + GstElementClass parent_class; + + void (*snapshot) (GstElement *elem); +}; + +#ifdef __cplusplus +} +#endif /* __cplusplus */ + + +#endif /* __GST_SNAPSHOT_H__ */ -- cgit v1.2.1