/* * 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 #include #include #include #include #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 #include #include /* 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