mirror of
https://code.videolan.org/videolan/vlc
synced 2024-09-16 16:02:54 +02:00
244 lines
8.5 KiB
C
244 lines
8.5 KiB
C
/*****************************************************************************
|
|
* image.c : image video output
|
|
*****************************************************************************
|
|
* Copyright (C) 2004 VideoLAN
|
|
* $Id: snapshot.c 8644 2004-09-05 16:53:04Z fkuehne $
|
|
*
|
|
* Authors: 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., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
|
|
*****************************************************************************/
|
|
|
|
/*****************************************************************************
|
|
* Preamble
|
|
*****************************************************************************/
|
|
#include <stdlib.h>
|
|
|
|
#include <vlc/vlc.h>
|
|
#include <vlc/vout.h>
|
|
#include <vlc/intf.h>
|
|
|
|
#include "vlc_image.h"
|
|
|
|
/*****************************************************************************
|
|
* Local prototypes
|
|
*****************************************************************************/
|
|
static int Create ( vlc_object_t * );
|
|
static void Destroy ( vlc_object_t * );
|
|
|
|
static int Init ( vout_thread_t * );
|
|
static void End ( vout_thread_t *p_vout );
|
|
static void Display ( vout_thread_t *, picture_t * );
|
|
|
|
/*****************************************************************************
|
|
* Module descriptor
|
|
*****************************************************************************/
|
|
#define FORMAT_TEXT N_( "Image format" )
|
|
#define FORMAT_LONGTEXT N_( "Set the format of the output image." )
|
|
|
|
#define RATIO_TEXT N_( "Recording ratio" )
|
|
#define RATIO_LONGTEXT N_( "Set the ratio of images that are recorded. "\
|
|
"3 means that one image out of three is recorded." )
|
|
|
|
#define PREFIX_TEXT N_( "Filename prefix" )
|
|
#define PREFIX_LONGTEXT N_( "Set the prefix of the filename. Output filename "\
|
|
"will have the form prefixNUMBER.format" )
|
|
|
|
static char *psz_format_list[] = { "png" };
|
|
static char *psz_format_list_text[] = { N_("PNG") };
|
|
|
|
vlc_module_begin( );
|
|
set_shortname( _( "Image" ) );
|
|
set_description( _( "Image video output" ) );
|
|
set_category( CAT_VIDEO );
|
|
set_subcategory( SUBCAT_VIDEO_VOUT );
|
|
set_capability( "video output", 0 );
|
|
|
|
add_string( "image-out-format", "png", NULL, FORMAT_TEXT, FORMAT_LONGTEXT,
|
|
VLC_FALSE );
|
|
change_string_list( psz_format_list, psz_format_list_text, 0 );
|
|
add_integer( "image-out-ratio", 3 , NULL, RATIO_TEXT, RATIO_LONGTEXT,
|
|
VLC_FALSE );
|
|
add_string( "image-out-prefix", "img", NULL, PREFIX_TEXT, PREFIX_LONGTEXT,
|
|
VLC_FALSE );
|
|
set_callbacks( Create, Destroy );
|
|
vlc_module_end();
|
|
|
|
/*****************************************************************************
|
|
* vout_sys_t: video output descriptor
|
|
*****************************************************************************/
|
|
struct vout_sys_t
|
|
{
|
|
char *psz_prefix; /* Prefix */
|
|
int i_ratio; /* Image ratio */
|
|
|
|
int i_current; /* Current image */
|
|
int i_frames; /* Number of frames */
|
|
|
|
image_handler_t *p_image;
|
|
};
|
|
|
|
#define FREE( p ) if( p ) { free( p ); p = NULL; }
|
|
|
|
/*****************************************************************************
|
|
* Create: allocates video thread
|
|
*****************************************************************************
|
|
* This function allocates and initializes a vout method.
|
|
*****************************************************************************/
|
|
static int Create( vlc_object_t *p_this )
|
|
{
|
|
vout_thread_t *p_vout = ( vout_thread_t * )p_this;
|
|
|
|
/* Allocate instance and initialize some members */
|
|
p_vout->p_sys = malloc( sizeof( vout_sys_t ) );
|
|
if( ! p_vout->p_sys )
|
|
return VLC_ENOMEM;
|
|
|
|
p_vout->p_sys->psz_prefix =
|
|
var_CreateGetString( p_this, "image-out-prefix" );
|
|
p_vout->p_sys->i_ratio =
|
|
var_CreateGetInteger( p_this, "image-out-ratio" );
|
|
p_vout->p_sys->i_current = 0;
|
|
p_vout->p_sys->p_image = image_HandlerCreate( p_vout );
|
|
|
|
if( !p_vout->p_sys->p_image )
|
|
{
|
|
msg_Err( p_this, "unable to create image handler") ;
|
|
FREE( p_vout->p_sys->psz_prefix );
|
|
FREE( p_vout->p_sys );
|
|
return VLC_EGENERIC;
|
|
}
|
|
|
|
p_vout->pf_init = Init;
|
|
p_vout->pf_end = End;
|
|
p_vout->pf_manage = NULL;
|
|
p_vout->pf_render = Display;
|
|
p_vout->pf_display = NULL;
|
|
|
|
return VLC_SUCCESS;
|
|
}
|
|
|
|
/*****************************************************************************
|
|
* Init: initialize video thread
|
|
*****************************************************************************/
|
|
static int Init( vout_thread_t *p_vout )
|
|
{
|
|
int i_index;
|
|
picture_t *p_pic;
|
|
|
|
/* Initialize the output structure */
|
|
p_vout->output.i_chroma = p_vout->render.i_chroma;
|
|
p_vout->output.pf_setpalette = NULL;
|
|
p_vout->output.i_width = p_vout->render.i_width;
|
|
p_vout->output.i_height = p_vout->render.i_height;
|
|
p_vout->output.i_aspect = p_vout->output.i_width
|
|
* VOUT_ASPECT_FACTOR / p_vout->output.i_height;
|
|
|
|
p_vout->output.i_rmask = 0xff0000;
|
|
p_vout->output.i_gmask = 0x00ff00;
|
|
p_vout->output.i_bmask = 0x0000ff;
|
|
|
|
/* Try to initialize 1 direct buffer */
|
|
p_pic = NULL;
|
|
|
|
/* Find an empty picture slot */
|
|
for( i_index = 0 ; i_index < VOUT_MAX_PICTURES ; i_index++ )
|
|
{
|
|
if( p_vout->p_picture[ i_index ].i_status == FREE_PICTURE )
|
|
{
|
|
p_pic = p_vout->p_picture + i_index;
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* Allocate the picture */
|
|
if( p_pic == NULL )
|
|
{
|
|
return VLC_EGENERIC;
|
|
}
|
|
|
|
vout_AllocatePicture( VLC_OBJECT(p_vout), p_pic, p_vout->output.i_chroma,
|
|
p_vout->output.i_width, p_vout->output.i_height,
|
|
p_vout->output.i_aspect );
|
|
|
|
if( p_pic->i_planes == 0 )
|
|
{
|
|
return VLC_EGENERIC;
|
|
}
|
|
|
|
p_pic->i_status = DESTROYED_PICTURE;
|
|
p_pic->i_type = DIRECT_PICTURE;
|
|
|
|
PP_OUTPUTPICTURE[ I_OUTPUTPICTURES ] = p_pic;
|
|
I_OUTPUTPICTURES++;
|
|
return VLC_SUCCESS;
|
|
}
|
|
|
|
/*****************************************************************************
|
|
* Destroy: destroy video thread
|
|
*****************************************************************************
|
|
* Terminate an output method created by Create
|
|
*****************************************************************************/
|
|
static void Destroy( vlc_object_t *p_this )
|
|
{
|
|
vout_thread_t *p_vout = ( vout_thread_t * )p_this;
|
|
|
|
/* Destroy structure */
|
|
image_HandlerDelete( p_vout->p_sys->p_image );
|
|
FREE( p_vout->p_sys->psz_prefix );
|
|
FREE( p_vout->p_sys );
|
|
}
|
|
|
|
/*****************************************************************************
|
|
* Display: displays previously rendered output
|
|
*****************************************************************************
|
|
* This function copies the rendered picture into our circular buffer.
|
|
*****************************************************************************/
|
|
static void Display( vout_thread_t *p_vout, picture_t *p_pic )
|
|
{
|
|
video_format_t fmt_in = {0}, fmt_out = {0};
|
|
|
|
char *psz_filename;
|
|
|
|
if( p_vout->p_sys->i_frames % p_vout->p_sys->i_ratio != 0 )
|
|
{
|
|
p_vout->p_sys->i_frames++;
|
|
return;
|
|
}
|
|
p_vout->p_sys->i_frames++;
|
|
psz_filename = (char *)malloc( strlen( p_vout->p_sys->psz_prefix ) +
|
|
10 );
|
|
|
|
fmt_in.i_chroma = p_vout->render.i_chroma;
|
|
fmt_out.i_width = fmt_in.i_width = p_vout->render.i_width;
|
|
fmt_out.i_height = fmt_in.i_height = p_vout->render.i_height;
|
|
|
|
p_vout->p_sys->i_current++;
|
|
|
|
sprintf( psz_filename, "%s%.6i.%s", p_vout->p_sys->psz_prefix,
|
|
p_vout->p_sys->i_current,
|
|
"png" );
|
|
|
|
image_WriteUrl( p_vout->p_sys->p_image, p_pic,
|
|
&fmt_in, &fmt_out, psz_filename ) ;
|
|
free( psz_filename );
|
|
return;
|
|
}
|
|
|
|
|
|
static void End( vout_thread_t *p_vout )
|
|
{
|
|
}
|