libvo/vo_png.c

   1 /*
   2  * Portable Network Graphics renderer
   3  *
   4  * Copyright 2001 by Felix Buenemann <atmosfear@users.sourceforge.net>
   5  *
   6  * Uses libpng (which uses zlib), so see according licenses.
   7  *
   8  * This file is part of MPlayer.
   9  *
  10  * MPlayer is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * MPlayer is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License along
  21  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  22  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  23  */
  24
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #include <errno.h>
  29 #include <sys/stat.h>
  30 #include <sys/types.h>
  31 #include <unistd.h>
  32
  33 #include <png.h>
  34
  35 #include "config.h"
  36 #include "mp_msg.h"
  37 #include "mp_msg.h"
  38 #include "help_mp.h"
  39 #include "video_out.h"
  40 #include "video_out_internal.h"
  41 #include "subopt-helper.h"
  42 #include "mplayer.h"
  43
  44 #define BUFLENGTH 512
  45
  46 static const vo_info_t info =
  47 {
  48         "PNG file",
  49         "png",
  50         "Felix Buenemann <atmosfear@users.sourceforge.net>",
  51         ""
  52 };
  53
  54 const LIBVO_EXTERN (png)
  55
  56 static int z_compression = Z_NO_COMPRESSION;
  57 static char *png_outdir = NULL;
  58 static int framenum = 0;
  59 static int use_alpha;
  60
  61 struct pngdata {
  62         FILE * fp;
  63         png_structp png_ptr;
  64         png_infop info_ptr;
  65         enum {OK,ERROR} status;
  66 };
  67
  68 static void png_mkdir(char *buf, int verbose) {
  69     struct stat stat_p;
  70
  71 #ifndef __MINGW32__
  72     if ( mkdir(buf, 0755) < 0 ) {
  73 #else
  74     if ( mkdir(buf) < 0 ) {
  75 #endif
  76         switch (errno) { /* use switch in case other errors need to be caught
  77                             and handled in the future */
  78             case EEXIST:
  79                 if ( stat(buf, &stat_p ) < 0 ) {
  80                     mp_msg(MSGT_VO, MSGL_ERR, "%s: %s: %s\n", info.short_name,
  81                             MSGTR_VO_GenericError, strerror(errno) );
  82                     mp_msg(MSGT_VO, MSGL_ERR, "%s: %s %s\n", info.short_name,
  83                             MSGTR_VO_UnableToAccess,buf);
  84                     exit_player(MSGTR_Exit_error);
  85                 }
  86                 if ( !S_ISDIR(stat_p.st_mode) ) {
  87                     mp_msg(MSGT_VO, MSGL_ERR, "%s: %s %s\n", info.short_name,
  88                             buf, MSGTR_VO_ExistsButNoDirectory);
  89                     exit_player(MSGTR_Exit_error);
  90                 }
  91                 if ( !(stat_p.st_mode & S_IWUSR) ) {
  92                     mp_msg(MSGT_VO, MSGL_ERR, "%s: %s - %s\n", info.short_name,
  93                             buf, MSGTR_VO_DirExistsButNotWritable);
  94                     exit_player(MSGTR_Exit_error);
  95                 }
  96
  97                 mp_msg(MSGT_VO, MSGL_INFO, "%s: %s - %s\n", info.short_name,
  98                         buf, MSGTR_VO_DirExistsAndIsWritable);
  99                 break;
 100
 101             default:
 102                 mp_msg(MSGT_VO, MSGL_ERR, "%s: %s: %s\n", info.short_name,
 103                         MSGTR_VO_GenericError, strerror(errno) );
 104                 mp_msg(MSGT_VO, MSGL_ERR, "%s: %s - %s\n", info.short_name,
 105                         buf, MSGTR_VO_CantCreateDirectory);
 106                 exit_player(MSGTR_Exit_error);
 107         } /* end switch */
 108     } else if ( verbose ) {
 109         mp_msg(MSGT_VO, MSGL_INFO, "%s: %s - %s\n", info.short_name,
 110                 buf, MSGTR_VO_DirectoryCreateSuccess);
 111     } /* end if */
 112 }
 113
 114 static int
 115 config(uint32_t width, uint32_t height, uint32_t d_width, uint32_t d_height, uint32_t flags, char *title, uint32_t format)
 116 {
 117     char buf[BUFLENGTH];
 118
 119             if(z_compression == 0) {
 120                     mp_msg(MSGT_VO,MSGL_INFO, MSGTR_LIBVO_PNG_Warning1);
 121                     mp_msg(MSGT_VO,MSGL_INFO, MSGTR_LIBVO_PNG_Warning2);
 122                     mp_msg(MSGT_VO,MSGL_INFO, MSGTR_LIBVO_PNG_Warning3);
 123             }
 124
 125     snprintf(buf, BUFLENGTH, "%s", png_outdir);
 126     png_mkdir(buf, 1);
 127     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Compression level %i\n", z_compression);
 128
 129     return 0;
 130 }
 131
 132
 133 static struct pngdata create_png (char * fname, int image_width, int image_height, int swapped)
 134 {
 135     struct pngdata png;
 136
 137     /*png_structp png_ptr = png_create_write_struct
 138        (PNG_LIBPNG_VER_STRING, (png_voidp)user_error_ptr,
 139         user_error_fn, user_warning_fn);*/
 140     //png_byte *row_pointers[image_height];
 141     png.png_ptr = png_create_write_struct
 142        (PNG_LIBPNG_VER_STRING, NULL,
 143         NULL, NULL);
 144     png.info_ptr = png_create_info_struct(png.png_ptr);
 145
 146     if (!png.png_ptr) {
 147        mp_msg(MSGT_VO,MSGL_DBG2, "PNG Failed to init png pointer\n");
 148        png.status = ERROR;
 149        return png;
 150     }
 151
 152     if (!png.info_ptr) {
 153        mp_msg(MSGT_VO,MSGL_DBG2, "PNG Failed to init png infopointer\n");
 154        png_destroy_write_struct(&png.png_ptr,
 155          (png_infopp)NULL);
 156        png.status = ERROR;
 157        return png;
 158     }
 159
 160     if (setjmp(png.png_ptr->jmpbuf)) {
 161         mp_msg(MSGT_VO,MSGL_DBG2, "PNG Internal error!\n");
 162         png_destroy_write_struct(&png.png_ptr, &png.info_ptr);
 163         fclose(png.fp);
 164         png.status = ERROR;
 165         return png;
 166     }
 167
 168     png.fp = fopen (fname, "wb");
 169     if (png.fp == NULL) {
 170         mp_msg(MSGT_VO,MSGL_WARN, MSGTR_LIBVO_PNG_ErrorOpeningForWriting, strerror(errno));
 171         png.status = ERROR;
 172         return png;
 173     }
 174
 175     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Init IO\n");
 176     png_init_io(png.png_ptr, png.fp);
 177
 178     /* set the zlib compression level */
 179     png_set_compression_level(png.png_ptr, z_compression);
 180
 181
 182     /*png_set_IHDR(png_ptr, info_ptr, width, height,
 183        bit_depth, color_type, interlace_type,
 184        compression_type, filter_type)*/
 185     png_set_IHDR(png.png_ptr, png.info_ptr, image_width, image_height,
 186        8, use_alpha ? PNG_COLOR_TYPE_RGBA : PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,
 187        PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
 188
 189     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Write Info\n");
 190     png_write_info(png.png_ptr, png.info_ptr);
 191
 192     if(swapped) {
 193         mp_msg(MSGT_VO,MSGL_DBG2, "PNG Set BGR Conversion\n");
 194         png_set_bgr(png.png_ptr);
 195     }
 196
 197     png.status = OK;
 198     return png;
 199 }
 200
 201 static uint8_t destroy_png(struct pngdata png) {
 202
 203     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Write End\n");
 204     png_write_end(png.png_ptr, png.info_ptr);
 205
 206     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Destroy Write Struct\n");
 207     png_destroy_write_struct(&png.png_ptr, &png.info_ptr);
 208
 209     fclose (png.fp);
 210
 211     return 0;
 212 }
 213
 214 static uint32_t draw_image(mp_image_t* mpi){
 215     char buf[100];
 216     int k;
 217     struct pngdata png;
 218     png_byte *row_pointers[mpi->h];
 219
 220     // if -dr or -slices then do nothing:
 221     if(mpi->flags&(MP_IMGFLAG_DIRECT|MP_IMGFLAG_DRAW_CALLBACK)) return VO_TRUE;
 222
 223     snprintf (buf, 100, "%s/%08d.png", png_outdir, ++framenum);
 224
 225     png = create_png(buf, mpi->w, mpi->h, IMGFMT_IS_BGR(mpi->imgfmt));
 226
 227     if(png.status){
 228             mp_msg(MSGT_VO,MSGL_WARN, MSGTR_LIBVO_PNG_ErrorInCreatePng);
 229             return 1;
 230     }
 231
 232     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Creating Row Pointers\n");
 233     for ( k = 0; k < mpi->h; k++ )
 234         row_pointers[k] = mpi->planes[0]+mpi->stride[0]*k;
 235
 236     //png_write_flush(png.png_ptr);
 237     //png_set_flush(png.png_ptr, nrows);
 238
 239     if( mp_msg_test(MSGT_VO,MSGL_DBG2) ) {
 240         mp_msg(MSGT_VO,MSGL_DBG2, "PNG Writing Image Data\n"); }
 241     png_write_image(png.png_ptr, row_pointers);
 242
 243     destroy_png(png);
 244
 245     return VO_TRUE;
 246 }
 247
 248 static void draw_osd(void){}
 249
 250 static void flip_page (void){}
 251
 252 static int draw_frame(uint8_t * src[])
 253 {
 254     return -1;
 255 }
 256
 257 static int draw_slice( uint8_t *src[],int stride[],int w,int h,int x,int y )
 258 {
 259     return -1;
 260 }
 261
 262 static int
 263 query_format(uint32_t format)
 264 {
 265     const int supported_flags = VFCAP_CSP_SUPPORTED|VFCAP_CSP_SUPPORTED_BY_HW|VFCAP_ACCEPT_STRIDE;
 266     switch(format){
 267     case IMGFMT_RGB24:
 268     case IMGFMT_BGR24:
 269         return use_alpha ? 0 : supported_flags;
 270     case IMGFMT_RGBA:
 271     case IMGFMT_BGRA:
 272         return use_alpha ? supported_flags : 0;
 273     }
 274     return 0;
 275 }
 276
 277 static void uninit(void){
 278     if (png_outdir) {
 279         free(png_outdir);
 280         png_outdir = NULL;
 281     }
 282 }
 283
 284 static void check_events(void){}
 285
 286 static int int_zero_to_nine(int *sh)
 287 {
 288     if ( (*sh < 0) || (*sh > 9) )
 289         return 0;
 290     return 1;
 291 }
 292
 293 static opt_t subopts[] = {
 294     {"alpha", OPT_ARG_BOOL, &use_alpha, NULL, 0},
 295     {"z",   OPT_ARG_INT, &z_compression, (opt_test_f)int_zero_to_nine},
 296     {"outdir",      OPT_ARG_MSTRZ,  &png_outdir,           NULL, 0},
 297     {NULL}
 298 };
 299
 300 static int preinit(const char *arg)
 301 {
 302     z_compression = 0;
 303     png_outdir = strdup(".");
 304     use_alpha = 0;
 305     if (subopt_parse(arg, subopts) != 0) {
 306         return -1;
 307     }
 308     return 0;
 309 }
 310
 311 static int control(uint32_t request, void *data, ...)
 312 {
 313   switch (request) {
 314   case VOCTRL_DRAW_IMAGE:
 315     return draw_image(data);
 316   case VOCTRL_QUERY_FORMAT:
 317     return query_format(*((uint32_t*)data));
 318   }
 319   return VO_NOTIMPL;
 320 }