1
0
mirror of https://github.com/mpv-player/mpv synced 2024-12-13 18:36:09 +00:00
mpv/libvo/vo_jpeg.c
ivo 77db87aa41 Added output to multiple directories for vo_jpeg.
Updated the manual page to describe the new options.


git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@13159 b3059339-0415-0410-9bf9-f77b7e298cf2
2004-08-26 22:18:25 +00:00

246 lines
6.8 KiB
C

/*
* vo_jpeg.c, JPEG Renderer for MPlayer
*
* Copyright 2002 by Pontscho (pontscho@makacs.poliod.hu)
* 25/04/2003: Spring cleanup -- alex
* 04/08/2004: Added multiple subdirectory support -- ivop@euronet.nl
*
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <jpeglib.h>
#include "config.h"
#include "mp_msg.h"
#include "video_out.h"
#include "video_out_internal.h"
#include "mplayer.h" /* for exit_player() */
#include "help_mp.h"
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
/* Used for temporary buffers to store file- and pathnames */
#define BUFLENGTH 512
static vo_info_t info=
{
"JPEG file",
"jpeg",
"Zoltan Ponekker (pontscho@makacs.poliod.hu)",
""
};
LIBVO_EXTERN (jpeg)
static int image_width;
static int image_height;
int jpeg_baseline = 1;
int jpeg_progressive_mode = 0;
int jpeg_optimize = 100;
int jpeg_smooth = 0;
int jpeg_quality = 75;
char * jpeg_outdir = ".";
char * jpeg_subdirs = NULL;
int jpeg_maxfiles=1000;
static int framenum=0;
static uint32_t config(uint32_t width, uint32_t height, uint32_t d_width, uint32_t d_height, uint32_t fullscreen, char *title, uint32_t format)
{
char buf[BUFLENGTH];
struct stat stat_p;
/* Create outdir. If it already exists, test if it's a writable directory */
snprintf (buf, BUFLENGTH, "%s", jpeg_outdir);
if (mkdir (buf, 0755)<0) {
switch (errno) { /* use switch in case other errors need to be caught and handled in the future */
case EEXIST:
if ( stat(buf, &stat_p) < 0 ) {
mp_msg(MSGT_VO, MSGL_ERR, "%s: %s: %s\n", info.short_name, MSGTR_VO_JPEG_GenericError, strerror(errno) );
mp_msg(MSGT_VO, MSGL_ERR, "%s: %s %s\n", info.short_name, MSGTR_VO_JPEG_UnableToAccess,buf);
exit_player(MSGTR_Exit_error);
}
if ( !S_ISDIR(stat_p.st_mode) ) {
mp_msg(MSGT_VO, MSGL_ERR, "%s: %s %s\n", info.short_name, buf, MSGTR_VO_JPEG_ExistsButNoDirectory);
exit_player(MSGTR_Exit_error);
}
if ( !(stat_p.st_mode & S_IWUSR) ) {
mp_msg(MSGT_VO, MSGL_ERR, "%s: %s\n", info.short_name, MSGTR_VO_JPEG_DirExistsButNotWritable);
exit_player(MSGTR_Exit_error);
}
mp_msg(MSGT_VO, MSGL_INFO, "%s: %s\n", info.short_name, MSGTR_VO_JPEG_DirExistsAndIsWritable);
break;
default:
mp_msg(MSGT_VO, MSGL_ERR, "%s: %s: %s\n", info.short_name, MSGTR_VO_JPEG_GenericError, strerror(errno) );
mp_msg(MSGT_VO, MSGL_ERR, "%s: %s\n", info.short_name, MSGTR_VO_JPEG_CantCreateDirectory);
exit_player(MSGTR_Exit_error);
} /* end switch */
} else {
mp_msg(MSGT_VO, MSGL_INFO, "%s: %s\n", info.short_name, MSGTR_VO_JPEG_DirectoryCreateSuccess);
} /* end if */
image_height=height;
image_width=width;
return 0;
}
static uint32_t jpeg_write( uint8_t * name,uint8_t * buffer )
{
FILE * o;
struct jpeg_compress_struct cinfo;
struct jpeg_error_mgr jerr;
JSAMPROW row_pointer[1];
int row_stride;
if ( !buffer ) return 1;
if ( (o=fopen( name,"wb" )) == NULL ) return 1;
cinfo.err=jpeg_std_error(&jerr);
jpeg_create_compress(&cinfo);
jpeg_stdio_dest( &cinfo,o );
cinfo.image_width=image_width;
cinfo.image_height=image_height;
cinfo.input_components=3;
cinfo.in_color_space=JCS_RGB;
jpeg_set_defaults( &cinfo );
jpeg_set_quality( &cinfo,jpeg_quality,jpeg_baseline );
cinfo.optimize_coding=jpeg_optimize;
cinfo.smoothing_factor=jpeg_smooth;
if ( jpeg_progressive_mode ) jpeg_simple_progression( &cinfo );
jpeg_start_compress( &cinfo,TRUE );
row_stride = image_width * 3;
while ( cinfo.next_scanline < cinfo.image_height )
{
row_pointer[0]=&buffer[ cinfo.next_scanline * row_stride ];
(void)jpeg_write_scanlines( &cinfo,row_pointer,1 );
}
jpeg_finish_compress( &cinfo );
fclose( o );
jpeg_destroy_compress( &cinfo );
return 0;
}
static uint32_t draw_frame(uint8_t * src[])
{
static uint32_t framecounter=0, subdircounter=0;
char buf[BUFLENGTH];
uint8_t *dst= src[0];
static char subdirname[BUFLENGTH] = "";
struct stat stat_p;
/* Start writing to new subdirectory after a certain amount of frames */
if ( framecounter == jpeg_maxfiles ) {
framecounter = 0;
}
/* If framecounter is zero (or reset to zero), increment subdirectory number
* and create the subdirectory.
* If jpeg_subdirs is not set, do nothing and resort to old behaviour. */
if ( !framecounter && jpeg_subdirs ) {
snprintf (subdirname, BUFLENGTH, "%s%08d", jpeg_subdirs, ++subdircounter);
snprintf (buf, BUFLENGTH, "%s/%s", jpeg_outdir, subdirname);
if (mkdir (buf, 0755)<0) {
switch (errno) { /* use switch in case other errors need to be caught and handled in the future */
case EEXIST:
if ( stat(buf, &stat_p) < 0 ) {
mp_msg(MSGT_VO, MSGL_ERR, "%s: %s: %s\n", info.short_name, MSGTR_VO_JPEG_GenericError, strerror(errno) );
mp_msg(MSGT_VO, MSGL_ERR, "%s: %s %s\n", info.short_name, MSGTR_VO_JPEG_UnableToAccess, buf);
exit_player(MSGTR_Exit_error);
}
if ( !S_ISDIR(stat_p.st_mode) ) {
mp_msg(MSGT_VO, MSGL_ERR, "\n%s: %s %s\n", info.short_name, buf, MSGTR_VO_JPEG_ExistsButNoDirectory);
exit_player(MSGTR_Exit_error);
}
if ( !(stat_p.st_mode & S_IWUSR) ) {
mp_msg(MSGT_VO, MSGL_ERR, "\n%s: %s - %s\n", info.short_name, buf, MSGTR_VO_JPEG_DirExistsButNotWritable);
exit_player(MSGTR_Exit_error);
}
mp_msg(MSGT_VO, MSGL_INFO, "\n%s: %s - %s\n", info.short_name, buf, MSGTR_VO_JPEG_DirExistsAndIsWritable);
break;
default:
mp_msg(MSGT_VO, MSGL_ERR, "%s: %s: %s\n", info.short_name, MSGTR_VO_JPEG_GenericError, strerror(errno) );
mp_msg(MSGT_VO, MSGL_ERR, "\n%s: %s - %s.\n", info.short_name, buf, MSGTR_VO_JPEG_CantCreateDirectory);
exit_player(MSGTR_Exit_error);
break;
}
} /* switch */
} /* if !framecounter && jpeg_subdirs */
framenum++;
/* snprintf the full pathname of the outputfile */
snprintf (buf, BUFLENGTH, "%s/%s/%08d.jpg", jpeg_outdir, subdirname, framenum);
framecounter++;
return jpeg_write( buf,src[0] );
}
static void draw_osd(void)
{
}
static void flip_page (void)
{
}
static uint32_t draw_slice( uint8_t *src[],int stride[],int w,int h,int x,int y )
{
return 0;
}
static uint32_t query_format(uint32_t format)
{
if (format == IMGFMT_RGB24)
return VFCAP_CSP_SUPPORTED|VFCAP_CSP_SUPPORTED_BY_HW;
return 0;
}
static void uninit(void)
{
}
static void check_events(void)
{
}
static uint32_t preinit(const char *arg)
{
return 0;
}
static uint32_t control(uint32_t request, void *data, ...)
{
switch (request)
{
case VOCTRL_QUERY_FORMAT:
return query_format(*((uint32_t*)data));
}
return VO_NOTIMPL;
}
#undef BUFLENGTH