Commit 479a1cc2 authored by Jean-Paul Saman's avatar Jean-Paul Saman Committed by Jean-Paul Saman

Add scene filter which provides the same functionality as the vout_output...

Add scene filter which provides the same functionality as the vout_output image, but which can be inserted in a playback or stream out chain.
parent c3baee8d
......@@ -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])
......
......@@ -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
/*****************************************************************************
* 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 );
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment