Add scene filter which provides the same functionality as the vout_output image,...
authorJean-Paul Saman <jpsaman@gandalf.kpn-communicationcentre.nl>
Fri, 1 Aug 2008 10:51:44 +0000 (12:51 +0200)
committerJean-Paul Saman <jean-paul.saman@m2x.nl>
Wed, 27 Aug 2008 12:44:30 +0000 (14:44 +0200)
configure.ac
modules/video_filter/Modules.am
modules/video_filter/scene.c [new file with mode: 0644]

index ba03506..f92ae3c 100644 (file)
@@ -648,7 +648,7 @@ AM_CONDITIONAL(BUILD_GETOPT, ${need_getopt})
 if test "${SYS}" != "mingw32" -a "${SYS}" != "mingwce"; then
 AC_TYPE_SIGNAL
 AC_CHECK_LIB(m,cos,[
-  VLC_ADD_LIBS([adjust wave ripple psychedelic gradient a52tofloat32 dtstofloat32 x264 goom visual panoramix rotate noise grain],[-lm])
+  VLC_ADD_LIBS([adjust wave ripple psychedelic gradient a52tofloat32 dtstofloat32 x264 goom visual panoramix rotate noise grain scene],[-lm])
 ])
 AC_CHECK_LIB(m,pow,[
   VLC_ADD_LIBS([avcodec avformat swscale imgresample postproc ffmpegaltivec stream_out_transrate i420_rgb faad twolame equalizer spatializer param_eq libvlc vorbis freetype mod mpc dmo quicktime realaudio realvideo galaktos opengl],[-lm])
@@ -1152,6 +1152,7 @@ VLC_ADD_PLUGIN([logo])
 VLC_ADD_PLUGIN([magnify])
 VLC_ADD_PLUGIN([puzzle])
 VLC_ADD_PLUGIN([colorthres])
+VLC_ADD_PLUGIN([scene])
 dnl Meta demuxers:
 VLC_ADD_PLUGIN([playlist])
 VLC_ADD_PLUGIN([export])
index 11643b8..31d1a77 100644 (file)
@@ -43,4 +43,5 @@ SOURCES_chain = chain.c
 SOURCES_postproc = postproc.c
 SOURCES_swscale = swscale.c ../codec/avcodec/chroma.h
 SOURCES_imgresample = imgresample.c ../codec/avcodec/chroma.h
+SOURCES_scene = scene.c
 noinst_HEADERS = filter_common.h filter_picture.h
diff --git a/modules/video_filter/scene.c b/modules/video_filter/scene.c
new file mode 100644 (file)
index 0000000..67412dd
--- /dev/null
@@ -0,0 +1,319 @@
+/*****************************************************************************
+ * scene.c : scene video filter (based on modules/video_output/image.c)
+ *****************************************************************************
+ * Copyright (C) 2004-2008 the VideoLAN team
+ * $Id$
+ *
+ * Authors: Jean-Paul Saman <jpsaman@videolan.org>
+ *          ClĂ©ment Stenac <zorglub@videolan.org>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
+ *****************************************************************************/
+
+/*****************************************************************************
+ * Preamble
+ *****************************************************************************/
+
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
+
+#include <vlc_common.h>
+#include <vlc_plugin.h>
+#include <vlc_vout.h>
+#include <vlc_block.h>
+
+#include "vlc_filter.h"
+#include "filter_picture.h"
+#include "vlc_image.h"
+#include "vlc_strings.h"
+
+/*****************************************************************************
+ * Local prototypes
+ *****************************************************************************/
+static int  Create      ( vlc_object_t * );
+static void Destroy     ( vlc_object_t * );
+
+static picture_t *Filter( filter_t *, picture_t * );
+
+static void SnapshotRatio( filter_t *p_filter, picture_t *p_pic );
+static void SavePicture( filter_t *, picture_t * );
+
+/*****************************************************************************
+ * Module descriptor
+ *****************************************************************************/
+#define FORMAT_TEXT N_( "Image format" )
+#define FORMAT_LONGTEXT N_( "Format of the output images (png or jpg)." )
+
+#define WIDTH_TEXT N_( "Image width" )
+#define WIDTH_LONGTEXT N_( "You can enforce the image width. By default " \
+                            "(-1) VLC will adapt to the video " \
+                            "characteristics.")
+
+#define HEIGHT_TEXT N_( "Image height" )
+#define HEIGHT_LONGTEXT N_( "You can enforce the image height. By default " \
+                            "(-1) VLC will adapt to the video " \
+                            "characteristics.")
+
+#define RATIO_TEXT N_( "Recording ratio" )
+#define RATIO_LONGTEXT N_( "Ratio of images to record. "\
+                           "3 means that one image out of three is recorded." )
+
+#define PREFIX_TEXT N_( "Filename prefix" )
+#define PREFIX_LONGTEXT N_( "Prefix of the output images filenames. Output " \
+                            "filenames will have the \"prefixNUMBER.format\" "\
+                            "form if replace is not true." )
+
+#define PATH_TEXT N_( "Directory path prefix" )
+#define PATH_LONGTEXT N_( "Directory path where images files should be saved." \
+                          "If not set, then images will be automatically saved in " \
+                          "users homedir." )
+
+#define REPLACE_TEXT N_( "Always write to the same file" )
+#define REPLACE_LONGTEXT N_( "Always write to the same file instead of " \
+                            "creating one file per image. In this case, " \
+                             "the number is not appended to the filename." )
+
+static const char *const psz_format_list[] = { "png", "jpeg" };
+static const char *const psz_format_list_text[] = { "PNG", "JPEG" };
+
+#define CFG_PREFIX "scene-"
+
+vlc_module_begin( );
+    set_shortname( N_( "Scene filter" ) );
+    set_description( N_( "Scene video filter" ) );
+    set_category( CAT_VIDEO );
+    set_subcategory( SUBCAT_VIDEO_VOUT );
+    set_capability( "video filter2", 0 );
+
+    /* General options */
+    add_string(  CFG_PREFIX "format", "png", NULL,
+                 FORMAT_TEXT, FORMAT_LONGTEXT, false );
+    change_string_list( psz_format_list, psz_format_list_text, 0 );
+    add_integer( CFG_PREFIX "width", 288, NULL,
+                 WIDTH_TEXT, WIDTH_LONGTEXT, true );
+    add_integer( CFG_PREFIX "height", 160, NULL,
+                 HEIGHT_TEXT, HEIGHT_LONGTEXT, true );
+    add_string(  CFG_PREFIX "prefix", "scene", NULL,
+                 PREFIX_TEXT, PREFIX_LONGTEXT, false );
+    add_string(  CFG_PREFIX "path", NULL, NULL,
+                 PATH_TEXT, PATH_LONGTEXT, false );
+    add_bool(    CFG_PREFIX "replace", false, NULL,
+                 REPLACE_TEXT, REPLACE_LONGTEXT, false );
+
+    /* Snapshot method */
+    add_integer( CFG_PREFIX "ratio", 50, NULL,
+                 RATIO_TEXT, RATIO_LONGTEXT, false );
+
+    set_callbacks( Create, Destroy );
+vlc_module_end();
+
+static const char *const ppsz_vfilter_options[] = {
+    "format", "width", "height", "ratio", "prefix", "path", "replace", NULL
+};
+
+typedef struct scene_t {
+    picture_t       *p_pic;
+    video_format_t  format;
+} scene_t;
+
+/*****************************************************************************
+ * filter_sys_t: private data
+ *****************************************************************************/
+struct filter_sys_t
+{
+    image_handler_t *p_image;
+    scene_t *p_scene;
+
+    char *psz_path;
+    char *psz_prefix;
+    char *psz_format;
+    int32_t i_width;
+    int32_t i_height;
+    int32_t i_ratio;  /* save every n-th frame */
+    int32_t i_frames; /* frames count */
+    bool  b_replace;
+};
+
+/*****************************************************************************
+ * Create: initialize and set pf_video_filter()
+ *****************************************************************************/
+static int Create( vlc_object_t *p_this )
+{
+    filter_t *p_filter = (filter_t *)p_this;
+    filter_sys_t *p_sys = NULL;
+
+    config_ChainParse( p_filter, CFG_PREFIX, ppsz_vfilter_options,
+                       p_filter->p_cfg );
+
+    p_filter->p_sys = p_sys = malloc( sizeof( filter_sys_t ) );
+    if( p_filter->p_sys == NULL )
+        return VLC_ENOMEM;
+    memset( p_sys, 0, sizeof(filter_sys_t) );
+
+    p_sys->p_scene = malloc( sizeof( scene_t ) );
+    if( !p_sys->p_scene )
+    {
+        free( p_sys );
+        return VLC_ENOMEM;
+    }
+    memset( p_sys->p_scene, 0, sizeof(scene_t) );
+
+    p_sys->p_image = image_HandlerCreate( p_this );
+    if( !p_sys->p_image )
+    {
+        msg_Err( p_this, "Couldn't get handle to image conversion routines." );
+        free( p_sys->p_scene );
+        free( p_sys );
+        return VLC_EGENERIC;
+    }
+
+    p_sys->psz_prefix = var_CreateGetString( p_this, CFG_PREFIX "prefix" );
+    p_sys->psz_path = var_GetNonEmptyString( p_this, CFG_PREFIX "path" );
+    if( p_sys->psz_path == NULL )
+    {
+        int i_ret;
+        i_ret = asprintf( &p_sys->psz_path, "%s", config_GetHomeDir());
+        if( i_ret == -1 )
+            p_sys->psz_path = NULL;
+    }
+    p_sys->psz_format = var_CreateGetString( p_this, CFG_PREFIX "format" );
+    p_sys->i_width = var_CreateGetInteger( p_this, CFG_PREFIX "width" );
+    p_sys->i_height = var_CreateGetInteger( p_this, CFG_PREFIX "height" );
+    p_sys->i_ratio = var_CreateGetInteger( p_this, CFG_PREFIX "ratio" );
+    p_sys->b_replace = var_CreateGetBool( p_this, CFG_PREFIX "replace" );
+
+    p_filter->pf_video_filter = Filter;
+
+    return VLC_SUCCESS;
+}
+
+/*****************************************************************************
+ * Destroy: destroy video filter method
+ *****************************************************************************/
+static void Destroy( vlc_object_t *p_this )
+{
+    filter_t *p_filter = (filter_t *)p_this;
+    filter_sys_t *p_sys = (filter_sys_t *) p_filter->p_sys;
+
+    image_HandlerDelete( p_sys->p_image );
+
+    if( p_sys->p_scene && p_sys->p_scene->p_pic )
+    picture_Release( p_sys->p_scene->p_pic );
+    free( p_sys->p_scene );
+    free( p_sys->psz_format );
+    free( p_sys->psz_prefix );
+    free( p_sys->psz_path );
+    free( p_sys );
+}
+
+/*****************************************************************************
+ * Filter: Apply filtering logic to picture.
+ *****************************************************************************/
+static picture_t *Filter( filter_t *p_filter, picture_t *p_pic )
+{
+    /* TODO: think of some funky algorithm to detect scene changes. */
+    SnapshotRatio( p_filter, p_pic );
+    return p_pic;
+}
+
+static void SnapshotRatio( filter_t *p_filter, picture_t *p_pic )
+{
+    filter_sys_t *p_sys = (filter_sys_t *)p_filter->p_sys;
+
+    if( !p_sys || !p_pic ) return;
+
+    if( p_sys->i_frames % p_sys->i_ratio != 0 )
+    {
+        p_sys->i_frames++;
+        return;
+    }
+    p_sys->i_frames++;
+
+    if( p_sys->p_scene )
+    {
+        if( p_sys->p_scene->p_pic )
+            picture_Release( p_sys->p_scene->p_pic );
+        p_sys->p_scene->p_pic = picture_New( p_pic->format.i_chroma,
+           p_pic->format.i_width, p_pic->format.i_height,
+           p_pic->format.i_sar_num );
+        if( p_sys->p_scene->p_pic )
+        {
+            picture_Copy( p_sys->p_scene->p_pic, p_pic );
+            SavePicture( p_filter, p_sys->p_scene->p_pic );
+        }
+    }
+}
+
+/*****************************************************************************
+ * Save Picture to disk
+ *****************************************************************************/
+static void SavePicture( filter_t *p_filter, picture_t *p_pic )
+{
+    filter_sys_t *p_sys = (filter_sys_t *)p_filter->p_sys;
+    video_format_t fmt_in, fmt_out;
+    char *psz_filename = NULL;
+    int i_ret;
+
+    memset( &fmt_in, 0, sizeof(video_format_t) );
+    memset( &fmt_out, 0, sizeof(video_format_t) );
+
+    /* Save snapshot psz_format to a memory zone */
+    fmt_in = p_pic->format;
+    fmt_out.i_sar_num = fmt_out.i_sar_den = 1;
+    fmt_out.i_width = p_sys->i_width;
+    fmt_out.i_height = p_sys->i_height;
+    fmt_out.i_chroma = VLC_FOURCC('p','n','g',' ');
+
+    if( (fmt_out.i_width == 0) && (fmt_out.i_height > 0) )
+    {
+        fmt_out.i_width = (fmt_in.i_width * fmt_out.i_height) / fmt_in.i_height;
+    }
+    else if( (fmt_out.i_height == 0) && (fmt_out.i_width > 0) )
+    {
+        fmt_out.i_height = (fmt_in.i_height * fmt_out.i_width) / fmt_in.i_width;
+    }
+    else
+    {
+        fmt_out.i_width = fmt_in.i_width;
+        fmt_out.i_height = fmt_in.i_height;
+    }
+
+    /* Save the snapshot */
+    if( p_sys->b_replace )
+        i_ret = asprintf( &psz_filename, "%s" DIR_SEP "%s.%s",
+                          p_sys->psz_path, p_sys->psz_prefix,
+                          p_sys->psz_format );
+    else
+        i_ret = asprintf( &psz_filename, "%s" DIR_SEP "%s%05d.%s",
+                          p_sys->psz_path, p_sys->psz_prefix,
+                          p_sys->i_frames, p_sys->psz_format );
+
+    if( i_ret == -1 )
+    {
+        msg_Err( p_filter, "could not create snapshot %s", psz_filename );
+        return;
+    }
+    path_sanitize( psz_filename );
+
+    /* Save the image */
+    i_ret = image_WriteUrl( p_sys->p_image, p_pic, &fmt_in, &fmt_out,
+                            psz_filename );
+    if( i_ret != VLC_SUCCESS )
+    {
+        msg_Err( p_filter, "could not create snapshot %s", psz_filename );
+    }
+    free( psz_filename );
+}