Mercurial > mplayer.hg
view libvo/vo_pnm.c @ 15132:f3617970ae22
Break overly long lines into something more manageable.
author | diego |
---|---|
date | Mon, 11 Apr 2005 22:59:33 +0000 |
parents | e563ad582f9e |
children | 05aa13cdf92f |
line wrap: on
line source
/* ------------------------------------------------------------------------- */ /* * vo_pnm.c, PPM/PGM/PGMYUV Video Output Driver for MPlayer * * * Written by Ivo van Poorten. (C) Copyright 2004, 2005. * Licensed under GNU General Public License version 2. * * * Changelog * * 2004-01-15 Replaced suboption parser by call to subopt-helper. * Got rid of stupid malloc_failed function. * 2004-09-16 Second draft. It now acts on VOCTRL_DRAW_IMAGE and does not * maintain a local copy of the image if the format is YV12. * Speed improvement and uses less memory. * 2004-09-09 First draft. * * */ /* ------------------------------------------------------------------------- */ /* Global Includes */ #include <stdio.h> #include <stdlib.h> #include <string.h> #include <errno.h> #include <sys/stat.h> /* ------------------------------------------------------------------------- */ /* Local Includes */ #include "config.h" #include "subopt-helper.h" #include "mp_msg.h" #include "video_out.h" #include "video_out_internal.h" #include "mplayer.h" /* for exit_player() */ #include "help_mp.h" /* ------------------------------------------------------------------------- */ /* Defines */ /* Used for temporary buffers to store file- and pathnames */ #define BUFLENGTH 512 #define PNM_ASCII_MODE 0 #define PNM_RAW_MODE 1 #define PNM_TYPE_PPM 0 #define PNM_TYPE_PGM 1 #define PNM_TYPE_PGMYUV 2 #define PNM_LINE_OF_ASCII "%03d %03d %03d %03d %03d %03d %03d %03d %03d %03d %03d %03d %03d %03d %03d\n" #define PNM_LINE15(a,b) a[b], a[b+1], a[b+2], a[b+3], a[b+4], a[b+5], a[b+6], \ a[b+7], a[b+8], a[b+9], a[b+10], a[b+11], a[b+12], \ a[b+13], a[b+14] /* ------------------------------------------------------------------------- */ /* Info */ static vo_info_t info= { "PPM/PGM/PGMYUV file", "pnm", "Ivo van Poorten (ivop@euronet.nl)", "" }; LIBVO_EXTERN (pnm) /* ------------------------------------------------------------------------- */ /* Global Variables */ int pnm_type = PNM_TYPE_PPM; int pnm_mode = PNM_RAW_MODE; char *pnm_outdir = NULL; char *pnm_subdirs = NULL; int pnm_maxfiles = 1000; char *pnm_file_extension = NULL; /* ------------------------------------------------------------------------- */ /** \brief An error occured while writing to a file. * * The program failed to write data to a file. * It displays a message and exits the player. * * \return nothing It does not return. */ void pnm_write_error(void) { mp_msg(MSGT_VO, MSGL_ERR, MSGTR_ErrorWritingFile, info.short_name); exit_player(MSGTR_Exit_error); } /* ------------------------------------------------------------------------- */ /** \brief Pre-initialisation. * * This function is called before initialising the video output driver. It * parses all suboptions and sets variables accordingly. If an error occurs * (like an option being out of range, not having any value or an unknown * option is stumbled upon) the player will exit. * * \param arg A string containing all the suboptions passed to the video * output driver. * * \return 0 All went well. */ static uint32_t preinit(const char *arg) { int ppm_type = 0, pgm_type = 0, pgmyuv_type = 0, raw_mode = 0, ascii_mode = 0; opt_t subopts[] = { {"ppm", OPT_ARG_BOOL, &ppm_type, NULL, 0}, {"pgm", OPT_ARG_BOOL, &pgm_type, NULL, 0}, {"pgmyuv", OPT_ARG_BOOL, &pgmyuv_type, NULL, 0}, {"raw", OPT_ARG_BOOL, &raw_mode, NULL, 0}, {"ascii", OPT_ARG_BOOL, &ascii_mode, NULL, 0}, {"outdir", OPT_ARG_MSTRZ, &pnm_outdir, NULL, 0}, {"subdirs", OPT_ARG_MSTRZ, &pnm_subdirs, NULL, 0}, {"maxfiles", OPT_ARG_INT, &pnm_maxfiles, (opt_test_f)int_pos, 0}, {NULL, 0, NULL, NULL, 0} }; const char *info_message = NULL; mp_msg(MSGT_VO, MSGL_INFO, "%s: %s\n", info.short_name, MSGTR_VO_ParsingSuboptions); pnm_maxfiles = 1000; pnm_outdir = strdup("."); pnm_subdirs = NULL; if (subopt_parse(arg, subopts) != 0) { return -1; } pnm_type = PNM_TYPE_PPM; pnm_mode = PNM_RAW_MODE; if (pgmyuv_type) pnm_type = PNM_TYPE_PGMYUV; if (pgm_type) pnm_type = PNM_TYPE_PGM; if (ppm_type) pnm_type = PNM_TYPE_PPM; if (ascii_mode) pnm_mode = PNM_ASCII_MODE; if (raw_mode) pnm_mode = PNM_RAW_MODE; switch (pnm_mode) { case PNM_ASCII_MODE: info_message = MSGTR_VO_PNM_ASCIIMode; break; case PNM_RAW_MODE: info_message = MSGTR_VO_PNM_RawMode; break; } mp_msg(MSGT_VO, MSGL_INFO, "%s: %s\n", info.short_name, info_message); switch (pnm_type) { case PNM_TYPE_PPM: info_message = MSGTR_VO_PNM_PPMType; break; case PNM_TYPE_PGM: info_message = MSGTR_VO_PNM_PGMType; break; case PNM_TYPE_PGMYUV: info_message = MSGTR_VO_PNM_PGMYUVType; break; } mp_msg(MSGT_VO, MSGL_INFO, "%s: %s\n", info.short_name, info_message); mp_msg(MSGT_VO, MSGL_INFO, "%s: %s\n", info.short_name, MSGTR_VO_SuboptionsParsedOK); return 0; } /* ------------------------------------------------------------------------- */ /** \brief Create a directory. * * This function creates a directory. If it already exists, it tests if * it's a directory and not something else, and if it is, it tests whether * the directory is writable or not. * * \param buf Pointer to directory name. * \param verbose Verbose on success. If verbose is non-zero, it will print * a message if it was successful in creating the directory. * * \return nothing In case anything fails, the player will exit. If it * returns, everything went well. */ void pnm_mkdir(char *buf, int verbose) { struct stat stat_p; /* Silly MING32 bug workaround */ #ifndef __MINGW32__ if ( mkdir(buf, 0755) < 0 ) { #else if ( mkdir(buf) < 0 ) { #endif 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_GenericError, strerror(errno) ); mp_msg(MSGT_VO, MSGL_ERR, "%s: %s %s\n", info.short_name, MSGTR_VO_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_ExistsButNoDirectory); exit_player(MSGTR_Exit_error); } if ( !(stat_p.st_mode & S_IWUSR) ) { mp_msg(MSGT_VO, MSGL_ERR, "%s: %s - %s\n", info.short_name, buf, MSGTR_VO_DirExistsButNotWritable); exit_player(MSGTR_Exit_error); } if (strcmp(buf, ".") != 0) { mp_msg(MSGT_VO, MSGL_INFO, "%s: %s - %s\n", info.short_name, buf, MSGTR_VO_DirExistsAndIsWritable); } break; default: mp_msg(MSGT_VO, MSGL_ERR, "%s: %s: %s\n", info.short_name, MSGTR_VO_GenericError, strerror(errno) ); mp_msg(MSGT_VO, MSGL_ERR, "%s: %s - %s\n", info.short_name, buf, MSGTR_VO_CantCreateDirectory); exit_player(MSGTR_Exit_error); } /* end switch */ } else if ( verbose ) { mp_msg(MSGT_VO, MSGL_INFO, "%s: %s - %s\n", info.short_name, buf, MSGTR_VO_DirectoryCreateSuccess); } /* end if */ } /* ------------------------------------------------------------------------- */ /** \brief Configure the video output driver. * * This functions configures the video output driver. It determines the * width and height of the image(s) and creates the output directory. * * \return 0 All went well. */ 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]; if (vo_config_count > 0 ) { /* Already configured */ return 0; } /* Create outdir. */ snprintf(buf, BUFLENGTH, "%s", pnm_outdir); pnm_mkdir(buf, 1); /* This function only returns if creation was successful. If not, the player will exit. */ if (pnm_type == PNM_TYPE_PPM) { pnm_file_extension = strdup("ppm"); } else if (pnm_type == PNM_TYPE_PGM) { pnm_file_extension = strdup("pgm"); } else if (pnm_type == PNM_TYPE_PGMYUV) { pnm_file_extension = strdup("pgmyuv"); } return 0; } /* ------------------------------------------------------------------------- */ /** \brief Write PNM file to output file * * This function writes PPM, PGM or PGMYUV data to an output file, depending * on which type was selected on the commandline. pnm_type and pnm_mode are * global variables. Depending on which mode was selected, it will write * a RAW or an ASCII file. * * \param outfile Filedescriptor of output file. * \param mpi The image to write. * * \return none The player will exit if anything goes wrong. */ void pnm_write_pnm(FILE *outfile, mp_image_t *mpi) { uint32_t w = mpi->w; uint32_t h = mpi->h; uint8_t *rgbimage = mpi->planes[0]; uint8_t *planeY = mpi->planes[0]; uint8_t *planeU = mpi->planes[1]; uint8_t *planeV = mpi->planes[2]; uint8_t *curline; uint32_t strideY = mpi->stride[0]; uint32_t strideU = mpi->stride[1]; uint32_t strideV = mpi->stride[2]; unsigned int i, j; if (pnm_mode == PNM_RAW_MODE) { if (pnm_type == PNM_TYPE_PPM) { if ( fprintf(outfile, "P6\n%d %d\n255\n", w, h) < 0 ) pnm_write_error(); if ( fwrite(rgbimage, w * 3, h, outfile) < h ) pnm_write_error(); } else if (pnm_type == PNM_TYPE_PGM) { if ( fprintf(outfile, "P5\n%d %d\n255\n", w, h) < 0 ) pnm_write_error(); for (i=0; i<h; i++) { if ( fwrite(planeY + i * strideY, w, 1, outfile) < 1 ) pnm_write_error(); } } else if (pnm_type == PNM_TYPE_PGMYUV) { if ( fprintf(outfile, "P5\n%d %d\n255\n", w, h*3/2) < 0 ) pnm_write_error(); for (i=0; i<h; i++) { if ( fwrite(planeY + i * strideY, w, 1, outfile) < 1 ) pnm_write_error(); } w = w / 2; h = h / 2; for (i=0; i<h; i++) { if ( fwrite(planeU + i * strideU, w, 1, outfile) < 1 ) pnm_write_error(); if ( fwrite(planeV + i * strideV, w, 1, outfile) < 1 ) pnm_write_error(); } } /* end if pnm_type */ } else if (pnm_mode == PNM_ASCII_MODE) { if (pnm_type == PNM_TYPE_PPM) { if ( fprintf(outfile, "P3\n%d %d\n255\n", w, h) < 0 ) pnm_write_error(); for (i=0; i <= w * h * 3 - 16 ; i += 15) { if ( fprintf(outfile, PNM_LINE_OF_ASCII, PNM_LINE15(rgbimage,i) ) < 0 ) pnm_write_error(); } while (i < (w * h * 3) ) { if ( fprintf(outfile, "%03d ", rgbimage[i]) < 0 ) pnm_write_error(); i++; } if ( fputc('\n', outfile) < 0 ) pnm_write_error(); } else if ( (pnm_type == PNM_TYPE_PGM) || (pnm_type == PNM_TYPE_PGMYUV) ) { /* different header for pgm and pgmyuv. pgmyuv is 'higher' */ if (pnm_type == PNM_TYPE_PGM) { if ( fprintf(outfile, "P2\n%d %d\n255\n", w, h) < 0 ) pnm_write_error(); } else { /* PNM_TYPE_PGMYUV */ if ( fprintf(outfile, "P2\n%d %d\n255\n", w, h*3/2) < 0 ) pnm_write_error(); } /* output Y plane for both PGM and PGMYUV */ for (j=0; j<h; j++) { curline = planeY + strideY * j; for (i=0; i <= w - 16; i+=15) { if ( fprintf(outfile, PNM_LINE_OF_ASCII, PNM_LINE15(curline,i) ) < 0 ) pnm_write_error(); } while (i < w ) { if ( fprintf(outfile, "%03d ", curline[i]) < 0 ) pnm_write_error(); i++; } if ( fputc('\n', outfile) < 0 ) pnm_write_error(); } /* also output U and V planes fpr PGMYUV */ if (pnm_type == PNM_TYPE_PGMYUV) { w = w / 2; h = h / 2; for (j=0; j<h; j++) { curline = planeU + strideU * j; for (i=0; i<= w-16; i+=15) { if ( fprintf(outfile, PNM_LINE_OF_ASCII, PNM_LINE15(curline,i) ) < 0 ) pnm_write_error(); } while (i < w ) { if ( fprintf(outfile, "%03d ", curline[i]) < 0 ) pnm_write_error(); i++; } if ( fputc('\n', outfile) < 0 ) pnm_write_error(); curline = planeV + strideV * j; for (i=0; i<= w-16; i+=15) { if ( fprintf(outfile, PNM_LINE_OF_ASCII, PNM_LINE15(curline,i) ) < 0 ) pnm_write_error(); } while (i < w ) { if ( fprintf(outfile, "%03d ", curline[i]) < 0 ) pnm_write_error(); i++; } if ( fputc('\n', outfile) < 0 ) pnm_write_error(); } } } /* end if pnm_type */ } /* end if pnm_mode */ } /* ------------------------------------------------------------------------- */ /** \brief Write a PNM image. * * This function gets called first if a PNM image has to be written to disk. * It contains the subdirectory framework and it calls pnm_write_pnm() to * actually write the image to disk. * * \param mpi The image to write. * * \return none The player will exit if anything goes wrong. */ void pnm_write_image(mp_image_t *mpi) { static int framenum = 0, framecounter = 0, subdircounter = 0; char buf[BUFLENGTH]; static char subdirname[BUFLENGTH] = ""; FILE *outfile; if (!mpi) { mp_msg(MSGT_VO, MSGL_ERR, "%s: No image data suplied to video output driver\n", info.short_name ); exit_player(MSGTR_Exit_error); } /* Start writing to new subdirectory after a certain amount of frames */ if ( framecounter == pnm_maxfiles ) { framecounter = 0; } /* If framecounter is zero (or reset to zero), increment subdirectory * number and create the subdirectory. * If pnm_subdirs is not set, do nothing. */ if ( !framecounter && pnm_subdirs ) { subdircounter++; snprintf(subdirname, BUFLENGTH, "%s%08d", pnm_subdirs, subdircounter); snprintf(buf, BUFLENGTH, "%s/%s", pnm_outdir, subdirname); pnm_mkdir(buf, 0); /* This function only returns if creation was successful. If not, the player will exit. */ } framenum++; framecounter++; /* snprintf the full pathname of the outputfile */ snprintf(buf, BUFLENGTH, "%s/%s/%08d.%s", pnm_outdir, subdirname, framenum, pnm_file_extension); if ( (outfile = fopen(buf, "wb") ) == NULL ) { mp_msg(MSGT_VO, MSGL_ERR, "\n%s: %s\n", info.short_name, MSGTR_VO_CantCreateFile); mp_msg(MSGT_VO, MSGL_ERR, "%s: %s: %s\n", info.short_name, MSGTR_VO_GenericError, strerror(errno) ); exit_player(MSGTR_Exit_error); } pnm_write_pnm(outfile, mpi); fclose(outfile); } /* ------------------------------------------------------------------------- */ static uint32_t draw_image(mp_image_t *mpi) { if (mpi->flags & MP_IMGFLAG_PLANAR) { /* Planar */ if (mpi->flags & MP_IMGFLAG_YUV) { /* Planar YUV */ pnm_write_image(mpi); return VO_TRUE; } else { /* Planar RGB */ return VO_FALSE; } } else { /* Packed */ if (mpi->flags & MP_IMGFLAG_YUV) { /* Packed YUV */ return VO_FALSE; } else { /* Packed RGB */ pnm_write_image(mpi); return VO_TRUE; } } return VO_FALSE; } /* ------------------------------------------------------------------------- */ static uint32_t draw_frame(uint8_t *src[]) { mp_msg(MSGT_VO, MSGL_V, "%s: draw_frame() is called!\n", info.short_name); return -1; } /* ------------------------------------------------------------------------- */ 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) { /* Ensure that for PPM we get Packed RGB and for PGM(YUV) we get * Planar YUV */ if (pnm_type == PNM_TYPE_PPM) { if (format == IMGFMT_RGB24) { return VFCAP_CSP_SUPPORTED|VFCAP_CSP_SUPPORTED_BY_HW; } } else if ( (pnm_type == PNM_TYPE_PGM) || (pnm_type == PNM_TYPE_PGMYUV) ) { if (format == IMGFMT_YV12) { return VFCAP_CSP_SUPPORTED|VFCAP_CSP_SUPPORTED_BY_HW; } } return 0; } /* ------------------------------------------------------------------------- */ static uint32_t control(uint32_t request, void *data, ...) { switch (request) { case VOCTRL_QUERY_FORMAT: return query_format(*((uint32_t*)data)); case VOCTRL_DRAW_IMAGE: return draw_image(data); } return VO_NOTIMPL; } /* ------------------------------------------------------------------------- */ static void uninit(void) { if (pnm_subdirs) { free(pnm_subdirs); pnm_subdirs = NULL; } if (pnm_outdir) { free(pnm_outdir); pnm_outdir = NULL; } } /* ------------------------------------------------------------------------- */ static void check_events(void) { } /* ------------------------------------------------------------------------- */ static void draw_osd(void) { } /* ------------------------------------------------------------------------- */ static void flip_page (void) { } /* ------------------------------------------------------------------------- */ #undef BUFLENGTH #undef PNM_RAW_MODE #undef PNM_ASCII_MODE #undef PNM_TYPE_PPM #undef PNM_TYPE_PGM #undef PNM_TYPE_PGMYUV /* ------------------------------------------------------------------------- */