/*****************************************************************************
* image.c : wrapper for image reading/writing facilities
*****************************************************************************
- * Copyright (C) 2004 VideoLAN
+ * Copyright (C) 2004 the VideoLAN team
* $Id$
*
* Author: Gildas Bazin <gbazin@videolan.org>
*
* 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.
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
*****************************************************************************/
/**
#include <vlc/decoder.h>
#include <vlc_filter.h>
#include <vlc_image.h>
+#include <vlc_stream.h>
+#include <charset.h>
static picture_t *ImageRead( image_handler_t *, block_t *,
video_format_t *, video_format_t * );
static int ImageWriteUrl( image_handler_t *, picture_t *,
video_format_t *, video_format_t *, const char * );
+static picture_t *ImageConvert( image_handler_t *, picture_t *,
+ video_format_t *, video_format_t * );
+static picture_t *ImageFilter( image_handler_t *, picture_t *,
+ video_format_t *, const char *psz_module );
+
static decoder_t *CreateDecoder( vlc_object_t *, video_format_t * );
static void DeleteDecoder( decoder_t * );
+static encoder_t *CreateEncoder( vlc_object_t *, video_format_t *,
+ video_format_t * );
+static void DeleteEncoder( encoder_t * );
static filter_t *CreateFilter( vlc_object_t *, es_format_t *,
- video_format_t * );
+ video_format_t *, const char * );
static void DeleteFilter( filter_t * );
static vlc_fourcc_t Ext2Fourcc( const char * );
-static const char *Fourcc2Ext( vlc_fourcc_t );
+/*static const char *Fourcc2Ext( vlc_fourcc_t );*/
/**
* Create an image_handler_t instance
p_image->pf_read_url = ImageReadUrl;
p_image->pf_write = ImageWrite;
p_image->pf_write_url = ImageWriteUrl;
+ p_image->pf_convert = ImageConvert;
+ p_image->pf_filter = ImageFilter;
return p_image;
}
if( !p_image ) return;
if( p_image->p_dec ) DeleteDecoder( p_image->p_dec );
+ if( p_image->p_enc ) DeleteEncoder( p_image->p_enc );
if( p_image->p_filter ) DeleteFilter( p_image->p_filter );
free( p_image );
+ p_image = NULL;
}
/**
video_format_t *p_fmt_in,
video_format_t *p_fmt_out )
{
- picture_t *p_pic;
+ picture_t *p_pic = NULL, *p_tmp;
/* Check if we can reuse the current decoder */
if( p_image->p_dec &&
}
p_block->i_pts = p_block->i_dts = mdate();
- p_pic = p_image->p_dec->pf_decode_video( p_image->p_dec, &p_block );
- p_image->p_dec->pf_decode_video( p_image->p_dec, &p_block );
+ while( (p_tmp = p_image->p_dec->pf_decode_video( p_image->p_dec, &p_block ))
+ != NULL )
+ {
+ if ( p_pic != NULL )
+ p_pic->pf_release( p_pic );
+ p_pic = p_tmp;
+ }
- if( !p_pic )
+ if( p_pic == NULL )
{
- msg_Dbg( p_image->p_parent, "no image decoded" );
+ msg_Warn( p_image->p_parent, "no image decoded" );
return 0;
}
if( !p_fmt_out->i_chroma )
p_fmt_out->i_chroma = p_image->p_dec->fmt_out.video.i_chroma;
+ if( !p_fmt_out->i_width && p_fmt_out->i_height )
+ p_fmt_out->i_width = p_fmt_out->i_height
+ * p_image->p_dec->fmt_out.video.i_aspect
+ / VOUT_ASPECT_FACTOR;
+ if( !p_fmt_out->i_height && p_fmt_out->i_width )
+ p_fmt_out->i_height = p_fmt_out->i_width * VOUT_ASPECT_FACTOR
+ / p_image->p_dec->fmt_out.video.i_aspect;
if( !p_fmt_out->i_width )
p_fmt_out->i_width = p_image->p_dec->fmt_out.video.i_width;
if( !p_fmt_out->i_height )
if( p_image->p_filter )
if( p_image->p_filter->fmt_in.video.i_chroma !=
p_image->p_dec->fmt_out.video.i_chroma ||
- p_image->p_filter->fmt_in.video.i_width !=
- p_image->p_dec->fmt_out.video.i_width ||
- p_image->p_filter->fmt_in.video.i_height !=
- p_image->p_dec->fmt_out.video.i_height ||
- p_image->p_filter->fmt_out.video.i_chroma != p_fmt_out->i_chroma ||
- p_image->p_filter->fmt_out.video.i_width != p_fmt_out->i_width ||
- p_image->p_filter->fmt_out.video.i_height != p_fmt_out->i_height )
+ p_image->p_filter->fmt_out.video.i_chroma != p_fmt_out->i_chroma )
{
/* We need to restart a new filter */
DeleteFilter( p_image->p_filter );
{
p_image->p_filter =
CreateFilter( p_image->p_parent, &p_image->p_dec->fmt_out,
- p_fmt_out );
+ p_fmt_out, NULL );
if( !p_image->p_filter )
{
return NULL;
}
}
+ else
+ {
+ /* Filters should handle on-the-fly size changes */
+ p_image->p_filter->fmt_in = p_image->p_dec->fmt_out;
+ p_image->p_filter->fmt_out = p_image->p_dec->fmt_out;
+ p_image->p_filter->fmt_out.i_codec = p_fmt_out->i_chroma;
+ p_image->p_filter->fmt_out.video = *p_fmt_out;
+ }
p_pic = p_image->p_filter->pf_video_filter( p_image->p_filter, p_pic );
*p_fmt_out = p_image->p_filter->fmt_out.video;
{
block_t *p_block;
picture_t *p_pic;
- FILE *file;
+ stream_t *p_stream = NULL;
int i_size;
- file = fopen( psz_url, "rb" );
- if( !file )
+ p_stream = stream_UrlNew( p_image->p_parent, psz_url );
+
+ if( !p_stream )
{
- msg_Dbg( p_image->p_parent, "could not open file %s for reading",
+ msg_Dbg( p_image->p_parent, "could not open %s for reading",
psz_url );
return NULL;
}
- fseek( file, 0, SEEK_END );
- i_size = ftell( file );
- fseek( file, 0, SEEK_SET );
+ i_size = stream_Size( p_stream );
p_block = block_New( p_image->p_parent, i_size );
- fread( p_block->p_buffer, sizeof(char), i_size, file );
- fclose( file );
+
+ stream_Read( p_stream, p_block->p_buffer, i_size );
+ stream_Delete( p_stream );
if( !p_fmt_in->i_chroma )
{
*
*/
+void PicRelease( picture_t *p_pic ){};
+
static block_t *ImageWrite( image_handler_t *p_image, picture_t *p_pic,
video_format_t *p_fmt_in,
video_format_t *p_fmt_out )
{
- return NULL;
+ block_t *p_block;
+ void (*pf_release)( picture_t * );
+
+ /* Check if we can reuse the current encoder */
+ if( p_image->p_enc &&
+ ( p_image->p_enc->fmt_out.i_codec != p_fmt_out->i_chroma ||
+ p_image->p_enc->fmt_out.video.i_width != p_fmt_out->i_width ||
+ p_image->p_enc->fmt_out.video.i_height != p_fmt_out->i_height ) )
+ {
+ DeleteEncoder( p_image->p_enc );
+ p_image->p_enc = 0;
+ }
+
+ /* Start an encoder */
+ if( !p_image->p_enc )
+ {
+ p_image->p_enc = CreateEncoder( p_image->p_parent,
+ p_fmt_in, p_fmt_out );
+ if( !p_image->p_enc ) return NULL;
+ }
+
+ /* Check if we need chroma conversion or resizing */
+ if( p_image->p_enc->fmt_in.video.i_chroma != p_fmt_in->i_chroma ||
+ p_image->p_enc->fmt_in.video.i_width != p_fmt_in->i_width ||
+ p_image->p_enc->fmt_in.video.i_height != p_fmt_in->i_height )
+ {
+ picture_t *p_tmp_pic;
+
+ if( p_image->p_filter )
+ if( p_image->p_filter->fmt_in.video.i_chroma != p_fmt_in->i_chroma ||
+ p_image->p_filter->fmt_out.video.i_chroma !=
+ p_image->p_enc->fmt_in.video.i_chroma )
+ {
+ /* We need to restart a new filter */
+ DeleteFilter( p_image->p_filter );
+ p_image->p_filter = 0;
+ }
+
+ /* Start a filter */
+ if( !p_image->p_filter )
+ {
+ es_format_t fmt_in;
+ es_format_Init( &fmt_in, VIDEO_ES, p_fmt_in->i_chroma );
+ fmt_in.video = *p_fmt_in;
+
+ p_image->p_filter =
+ CreateFilter( p_image->p_parent, &fmt_in,
+ &p_image->p_enc->fmt_in.video, NULL );
+
+ if( !p_image->p_filter )
+ {
+ return NULL;
+ }
+ }
+ else
+ {
+ /* Filters should handle on-the-fly size changes */
+ p_image->p_filter->fmt_in.i_codec = p_fmt_in->i_chroma;
+ p_image->p_filter->fmt_out.video = *p_fmt_in;
+ p_image->p_filter->fmt_out.i_codec =p_image->p_enc->fmt_in.i_codec;
+ p_image->p_filter->fmt_out.video = p_image->p_enc->fmt_in.video;
+ }
+
+ pf_release = p_pic->pf_release;
+ p_pic->pf_release = PicRelease; /* Small hack */
+ p_tmp_pic =
+ p_image->p_filter->pf_video_filter( p_image->p_filter, p_pic );
+ p_pic->pf_release = pf_release;
+
+ p_block = p_image->p_enc->pf_encode_video( p_image->p_enc, p_tmp_pic );
+
+ p_image->p_filter->pf_vout_buffer_del( p_image->p_filter, p_tmp_pic );
+ }
+ else
+ {
+ p_block = p_image->p_enc->pf_encode_video( p_image->p_enc, p_pic );
+ }
+
+ if( !p_block )
+ {
+ msg_Dbg( p_image->p_parent, "no image encoded" );
+ return 0;
+ }
+
+ return p_block;
}
static int ImageWriteUrl( image_handler_t *p_image, picture_t *p_pic,
video_format_t *p_fmt_in, video_format_t *p_fmt_out,
const char *psz_url )
{
- return VLC_EGENERIC;
+ block_t *p_block;
+ FILE *file;
+
+ if( !p_fmt_out->i_chroma )
+ {
+ /* Try to guess format from file name */
+ p_fmt_out->i_chroma = Ext2Fourcc( psz_url );
+ }
+
+ file = utf8_fopen( psz_url, "wb" );
+ if( !file )
+ {
+ msg_Dbg( p_image->p_parent, "could not open file %s for writing",
+ psz_url );
+ return VLC_EGENERIC;
+ }
+
+ p_block = ImageWrite( p_image, p_pic, p_fmt_in, p_fmt_out );
+
+ if( p_block )
+ {
+ fwrite( p_block->p_buffer, sizeof(char), p_block->i_buffer, file );
+ block_Release( p_block );
+ }
+
+ fclose( file );
+
+ return p_block ? VLC_SUCCESS : VLC_EGENERIC;
+}
+
+/**
+ * Convert an image to a different format
+ *
+ */
+
+static picture_t *ImageConvert( image_handler_t *p_image, picture_t *p_pic,
+ video_format_t *p_fmt_in,
+ video_format_t *p_fmt_out )
+{
+ void (*pf_release)( picture_t * );
+ picture_t *p_pif;
+
+ if( !p_fmt_out->i_width && !p_fmt_out->i_height &&
+ p_fmt_out->i_sar_num && p_fmt_out->i_sar_den &&
+ p_fmt_out->i_sar_num * p_fmt_in->i_sar_den !=
+ p_fmt_out->i_sar_den * p_fmt_in->i_sar_num )
+ {
+ p_fmt_out->i_width =
+ p_fmt_in->i_sar_num * (int64_t)p_fmt_out->i_sar_den *
+ p_fmt_in->i_width / p_fmt_in->i_sar_den / p_fmt_out->i_sar_num;
+ p_fmt_out->i_visible_width =
+ p_fmt_in->i_sar_num * (int64_t)p_fmt_out->i_sar_den *
+ p_fmt_in->i_visible_width / p_fmt_in->i_sar_den /
+ p_fmt_out->i_sar_num;
+ }
+
+ if( !p_fmt_out->i_chroma ) p_fmt_out->i_chroma = p_fmt_in->i_chroma;
+ if( !p_fmt_out->i_width )
+ p_fmt_out->i_width = p_fmt_out->i_visible_width = p_fmt_in->i_width;
+ if( !p_fmt_out->i_height )
+ p_fmt_out->i_height = p_fmt_out->i_visible_height = p_fmt_in->i_height;
+ if( !p_fmt_out->i_sar_num ) p_fmt_out->i_sar_num = p_fmt_in->i_sar_num;
+ if( !p_fmt_out->i_sar_den ) p_fmt_out->i_sar_den = p_fmt_in->i_sar_den;
+ if( !p_fmt_out->i_aspect ) p_fmt_out->i_aspect = p_fmt_in->i_aspect;
+
+ if( p_image->p_filter )
+ if( p_image->p_filter->fmt_in.video.i_chroma != p_fmt_in->i_chroma ||
+ p_image->p_filter->fmt_out.video.i_chroma != p_fmt_out->i_chroma )
+ {
+ /* We need to restart a new filter */
+ DeleteFilter( p_image->p_filter );
+ p_image->p_filter = NULL;
+ }
+
+ /* Start a filter */
+ if( !p_image->p_filter )
+ {
+ es_format_t fmt_in;
+ es_format_Init( &fmt_in, VIDEO_ES, p_fmt_in->i_chroma );
+ fmt_in.video = *p_fmt_in;
+
+ p_image->p_filter =
+ CreateFilter( p_image->p_parent, &fmt_in, p_fmt_out, NULL );
+
+ if( !p_image->p_filter )
+ {
+ return NULL;
+ }
+ }
+ else
+ {
+ /* Filters should handle on-the-fly size changes */
+ p_image->p_filter->fmt_in.video = *p_fmt_in;
+ p_image->p_filter->fmt_out.video = *p_fmt_out;
+ }
+
+ pf_release = p_pic->pf_release;
+ p_pic->pf_release = PicRelease; /* Small hack */
+ p_pif = p_image->p_filter->pf_video_filter( p_image->p_filter, p_pic );
+ p_pic->pf_release = pf_release;
+
+ if( p_fmt_in->i_chroma == p_fmt_out->i_chroma &&
+ p_fmt_in->i_width == p_fmt_out->i_width &&
+ p_fmt_in->i_height == p_fmt_out->i_height )
+ {
+ /* Duplicate image */
+ p_pif = p_image->p_filter->pf_vout_buffer_new( p_image->p_filter );
+ if( p_pif ) vout_CopyPicture( p_image->p_parent, p_pif, p_pic );
+ }
+
+ return p_pif;
+}
+
+/**
+ * Filter an image with a psz_module filter
+ *
+ */
+
+static picture_t *ImageFilter( image_handler_t *p_image, picture_t *p_pic,
+ video_format_t *p_fmt, const char *psz_module )
+{
+ void (*pf_release)( picture_t * );
+ picture_t *p_pif;
+
+ /* Start a filter */
+ if( !p_image->p_filter )
+ {
+ es_format_t fmt;
+ es_format_Init( &fmt, VIDEO_ES, p_fmt->i_chroma );
+ fmt.video = *p_fmt;
+
+ p_image->p_filter =
+ CreateFilter( p_image->p_parent, &fmt, &fmt.video, psz_module );
+
+ if( !p_image->p_filter )
+ {
+ return NULL;
+ }
+ }
+ else
+ {
+ /* Filters should handle on-the-fly size changes */
+ p_image->p_filter->fmt_in.video = *p_fmt;
+ p_image->p_filter->fmt_out.video = *p_fmt;
+ }
+
+ pf_release = p_pic->pf_release;
+ p_pic->pf_release = PicRelease; /* Small hack */
+ p_pif = p_image->p_filter->pf_video_filter( p_image->p_filter, p_pic );
+ p_pic->pf_release = pf_release;
+
+ return p_pif;
}
/**
{ VLC_FOURCC('p','g','m','y'), "pgmyuv" },
{ VLC_FOURCC('p','b','m',' '), "pbm" },
{ VLC_FOURCC('p','a','m',' '), "pam" },
+ { VLC_FOURCC('t','g','a',' '), "tga" },
+ { VLC_FOURCC('b','m','p',' '), "bmp" },
+ { VLC_FOURCC('p','n','m',' '), "pnm" },
+ { VLC_FOURCC('x','p','m',' '), "xpm" },
+ { VLC_FOURCC('x','c','f',' '), "xcf" },
+ { VLC_FOURCC('p','c','x',' '), "pcx" },
+ { VLC_FOURCC('g','i','f',' '), "gif" },
+ { VLC_FOURCC('t','i','f','f'), "tif" },
+ { VLC_FOURCC('t','i','f','f'), "tiff" },
+ { VLC_FOURCC('l','b','m',' '), "lbm" },
{ 0, NULL }
};
return 0;
}
+/*
static const char *Fourcc2Ext( vlc_fourcc_t i_codec )
{
int i;
return NULL;
}
+*/
static void video_release_buffer( picture_t *p_pic )
{
static decoder_t *CreateDecoder( vlc_object_t *p_this, video_format_t *fmt )
{
decoder_t *p_dec;
- static es_format_t null_es_format = {0};
p_dec = vlc_object_create( p_this, VLC_OBJECT_DECODER );
if( p_dec == NULL )
}
p_dec->p_module = NULL;
- es_format_Copy( &p_dec->fmt_in, &null_es_format );
- es_format_Copy( &p_dec->fmt_out, &null_es_format );
+ es_format_Init( &p_dec->fmt_in, VIDEO_ES, fmt->i_chroma );
+ es_format_Init( &p_dec->fmt_out, VIDEO_ES, 0 );
p_dec->fmt_in.video = *fmt;
- p_dec->fmt_in.i_cat = VIDEO_ES;
- p_dec->fmt_in.i_codec = fmt->i_chroma;
p_dec->b_pace_control = VLC_TRUE;
p_dec->pf_vout_buffer_new = video_new_buffer;
es_format_Clean( &p_dec->fmt_out );
vlc_object_destroy( p_dec );
+ p_dec = NULL;
+}
+
+static encoder_t *CreateEncoder( vlc_object_t *p_this, video_format_t *fmt_in,
+ video_format_t *fmt_out )
+{
+ encoder_t *p_enc;
+
+ p_enc = vlc_object_create( p_this, VLC_OBJECT_ENCODER );
+ if( p_enc == NULL )
+ {
+ msg_Err( p_this, "out of memory" );
+ return NULL;
+ }
+
+ p_enc->p_module = NULL;
+ es_format_Init( &p_enc->fmt_in, VIDEO_ES, fmt_in->i_chroma );
+ p_enc->fmt_in.video = *fmt_in;
+ if( fmt_out->i_width > 0 && fmt_out->i_height > 0 )
+ {
+ p_enc->fmt_in.video.i_width = fmt_out->i_width;
+ p_enc->fmt_in.video.i_height = fmt_out->i_height;
+
+ if( fmt_out->i_visible_width > 0 &&
+ fmt_out->i_visible_height > 0 )
+ {
+ p_enc->fmt_in.video.i_visible_width = fmt_out->i_visible_width;
+ p_enc->fmt_in.video.i_visible_height = fmt_out->i_visible_height;
+ }
+ else
+ {
+ p_enc->fmt_in.video.i_visible_width = fmt_out->i_width;
+ p_enc->fmt_in.video.i_visible_height = fmt_out->i_height;
+ }
+ }
+ else if( fmt_out->i_sar_num && fmt_out->i_sar_den &&
+ fmt_out->i_sar_num * fmt_in->i_sar_den !=
+ fmt_out->i_sar_den * fmt_in->i_sar_num )
+ {
+ p_enc->fmt_in.video.i_width =
+ fmt_in->i_sar_num * (int64_t)fmt_out->i_sar_den * fmt_in->i_width /
+ fmt_in->i_sar_den / fmt_out->i_sar_num;
+ p_enc->fmt_in.video.i_visible_width =
+ fmt_in->i_sar_num * (int64_t)fmt_out->i_sar_den *
+ fmt_in->i_visible_width / fmt_in->i_sar_den / fmt_out->i_sar_num;
+ }
+
+ p_enc->fmt_in.video.i_frame_rate = 25;
+ p_enc->fmt_in.video.i_frame_rate_base = 1;
+
+ es_format_Init( &p_enc->fmt_out, VIDEO_ES, fmt_out->i_chroma );
+ p_enc->fmt_out.video = *fmt_out;
+ p_enc->fmt_out.video.i_width = p_enc->fmt_in.video.i_width;
+ p_enc->fmt_out.video.i_height = p_enc->fmt_in.video.i_height;
+
+ vlc_object_attach( p_enc, p_this );
+
+ /* Find a suitable decoder module */
+ p_enc->p_module = module_Need( p_enc, "encoder", 0, 0 );
+ if( !p_enc->p_module )
+ {
+ msg_Err( p_enc, "no suitable encoder module for fourcc `%4.4s'.\n"
+ "VLC probably does not support this image format.",
+ (char*)&p_enc->fmt_out.i_codec );
+
+ DeleteEncoder( p_enc );
+ return NULL;
+ }
+ p_enc->fmt_in.video.i_chroma = p_enc->fmt_in.i_codec;
+
+ return p_enc;
+}
+
+static void DeleteEncoder( encoder_t * p_enc )
+{
+ vlc_object_detach( p_enc );
+
+ if( p_enc->p_module ) module_Unneed( p_enc, p_enc->p_module );
+
+ es_format_Clean( &p_enc->fmt_in );
+ es_format_Clean( &p_enc->fmt_out );
+
+ vlc_object_destroy( p_enc );
+ p_enc = NULL;
}
static filter_t *CreateFilter( vlc_object_t *p_this, es_format_t *p_fmt_in,
- video_format_t *p_fmt_out )
+ video_format_t *p_fmt_out,
+ const char *psz_module )
{
filter_t *p_filter;
p_filter->fmt_out = *p_fmt_in;
p_filter->fmt_out.i_codec = p_fmt_out->i_chroma;
p_filter->fmt_out.video = *p_fmt_out;
- p_filter->p_module = module_Need( p_filter, "video filter2", 0, 0 );
+ p_filter->p_module = module_Need( p_filter, "video filter2",
+ psz_module, 0 );
if( !p_filter->p_module )
{
es_format_Clean( &p_filter->fmt_out );
vlc_object_destroy( p_filter );
+ p_filter = NULL;
}