libvo/vo_png.c

   1 /*
   2  * vo_png.c, Portable Network Graphics Renderer for MPlayer
   3  *
   4  * Copyright 2001 by Felix Buenemann <atmosfear@users.sourceforge.net>
   5  *
   6  * Uses libpng (which uses zlib), so see according licenses.
   7  *
   8  */
   9
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13 #include <errno.h>
  14
  15 #include <png.h>
  16
  17 #include "config.h"
  18 #include "mp_msg.h"
  19 #include "mp_msg.h"
  20 #include "help_mp.h"
  21 #include "video_out.h"
  22 #include "video_out_internal.h"
  23 #include "subopt-helper.h"
  24
  25 static const vo_info_t info =
  26 {
  27         "PNG file",
  28         "png",
  29         "Felix Buenemann <atmosfear@users.sourceforge.net>",
  30         ""
  31 };
  32
  33 const LIBVO_EXTERN (png)
  34
  35 static int z_compression = Z_NO_COMPRESSION;
  36 static int framenum = 0;
  37
  38 struct pngdata {
  39         FILE * fp;
  40         png_structp png_ptr;
  41         png_infop info_ptr;
  42         enum {OK,ERROR} status;
  43 };
  44
  45 static int
  46 config(uint32_t width, uint32_t height, uint32_t d_width, uint32_t d_height, uint32_t flags, char *title, uint32_t format)
  47 {
  48
  49             if(z_compression == 0) {
  50                     mp_msg(MSGT_VO,MSGL_INFO, MSGTR_LIBVO_PNG_Warning1);
  51                     mp_msg(MSGT_VO,MSGL_INFO, MSGTR_LIBVO_PNG_Warning2);
  52                     mp_msg(MSGT_VO,MSGL_INFO, MSGTR_LIBVO_PNG_Warning3);
  53             }
  54
  55     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Compression level %i\n", z_compression);
  56
  57     return 0;
  58 }
  59
  60
  61 static struct pngdata create_png (char * fname, int image_width, int image_height, int swapped)
  62 {
  63     struct pngdata png;
  64
  65     /*png_structp png_ptr = png_create_write_struct
  66        (PNG_LIBPNG_VER_STRING, (png_voidp)user_error_ptr,
  67         user_error_fn, user_warning_fn);*/
  68     //png_byte *row_pointers[image_height];
  69     png.png_ptr = png_create_write_struct
  70        (PNG_LIBPNG_VER_STRING, NULL,
  71         NULL, NULL);
  72     png.info_ptr = png_create_info_struct(png.png_ptr);
  73
  74     if (!png.png_ptr) {
  75        mp_msg(MSGT_VO,MSGL_DBG2, "PNG Failed to init png pointer\n");
  76        png.status = ERROR;
  77        return png;
  78     }
  79
  80     if (!png.info_ptr) {
  81        mp_msg(MSGT_VO,MSGL_DBG2, "PNG Failed to init png infopointer\n");
  82        png_destroy_write_struct(&png.png_ptr,
  83          (png_infopp)NULL);
  84        png.status = ERROR;
  85        return png;
  86     }
  87
  88     if (setjmp(png.png_ptr->jmpbuf)) {
  89         mp_msg(MSGT_VO,MSGL_DBG2, "PNG Internal error!\n");
  90         png_destroy_write_struct(&png.png_ptr, &png.info_ptr);
  91         fclose(png.fp);
  92         png.status = ERROR;
  93         return png;
  94     }
  95
  96     png.fp = fopen (fname, "wb");
  97     if (png.fp == NULL) {
  98         mp_msg(MSGT_VO,MSGL_WARN, MSGTR_LIBVO_PNG_ErrorOpeningForWriting, strerror(errno));
  99         png.status = ERROR;
 100         return png;
 101     }
 102
 103     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Init IO\n");
 104     png_init_io(png.png_ptr, png.fp);
 105
 106     /* set the zlib compression level */
 107     png_set_compression_level(png.png_ptr, z_compression);
 108
 109
 110     /*png_set_IHDR(png_ptr, info_ptr, width, height,
 111        bit_depth, color_type, interlace_type,
 112        compression_type, filter_type)*/
 113     png_set_IHDR(png.png_ptr, png.info_ptr, image_width, image_height,
 114        8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,
 115        PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
 116
 117     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Write Info\n");
 118     png_write_info(png.png_ptr, png.info_ptr);
 119
 120     if(swapped) {
 121         mp_msg(MSGT_VO,MSGL_DBG2, "PNG Set BGR Conversion\n");
 122         png_set_bgr(png.png_ptr);
 123     }
 124
 125     png.status = OK;
 126     return png;
 127 }
 128
 129 static uint8_t destroy_png(struct pngdata png) {
 130
 131     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Write End\n");
 132     png_write_end(png.png_ptr, png.info_ptr);
 133
 134     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Destroy Write Struct\n");
 135     png_destroy_write_struct(&png.png_ptr, &png.info_ptr);
 136
 137     fclose (png.fp);
 138
 139     return 0;
 140 }
 141
 142 static uint32_t draw_image(mp_image_t* mpi){
 143     char buf[100];
 144     int k;
 145     struct pngdata png;
 146     png_byte *row_pointers[mpi->h];
 147
 148     // if -dr or -slices then do nothing:
 149     if(mpi->flags&(MP_IMGFLAG_DIRECT|MP_IMGFLAG_DRAW_CALLBACK)) return VO_TRUE;
 150
 151     snprintf (buf, 100, "%08d.png", ++framenum);
 152
 153     png = create_png(buf, mpi->w, mpi->h, IMGFMT_IS_BGR(mpi->imgfmt));
 154
 155     if(png.status){
 156             mp_msg(MSGT_VO,MSGL_WARN, MSGTR_LIBVO_PNG_ErrorInCreatePng);
 157             return 1;
 158     }
 159
 160     mp_msg(MSGT_VO,MSGL_DBG2, "PNG Creating Row Pointers\n");
 161     for ( k = 0; k < mpi->h; k++ )
 162         row_pointers[k] = mpi->planes[0]+mpi->stride[0]*k;
 163
 164     //png_write_flush(png.png_ptr);
 165     //png_set_flush(png.png_ptr, nrows);
 166
 167     if( mp_msg_test(MSGT_VO,MSGL_DBG2) ) {
 168         mp_msg(MSGT_VO,MSGL_DBG2, "PNG Writing Image Data\n"); }
 169     png_write_image(png.png_ptr, row_pointers);
 170
 171     destroy_png(png);
 172
 173     return VO_TRUE;
 174 }
 175
 176 static void draw_osd(void){}
 177
 178 static void flip_page (void){}
 179
 180 static int draw_frame(uint8_t * src[])
 181 {
 182     return -1;
 183 }
 184
 185 static int draw_slice( uint8_t *src[],int stride[],int w,int h,int x,int y )
 186 {
 187     return -1;
 188 }
 189
 190 static int
 191 query_format(uint32_t format)
 192 {
 193     switch(format){
 194     case IMGFMT_RGB|24:
 195     case IMGFMT_BGR|24:
 196         return VFCAP_CSP_SUPPORTED|VFCAP_CSP_SUPPORTED_BY_HW|VFCAP_ACCEPT_STRIDE;
 197     }
 198     return 0;
 199 }
 200
 201 static void uninit(void){}
 202
 203 static void check_events(void){}
 204
 205 static int int_zero_to_nine(int *sh)
 206 {
 207     if ( (*sh < 0) || (*sh > 9) )
 208         return 0;
 209     return 1;
 210 }
 211
 212 static opt_t subopts[] = {
 213     {"z",   OPT_ARG_INT, &z_compression, (opt_test_f)int_zero_to_nine},
 214     {NULL}
 215 };
 216
 217 static int preinit(const char *arg)
 218 {
 219     z_compression = 0;
 220     if (subopt_parse(arg, subopts) != 0) {
 221         return -1;
 222     }
 223     return 0;
 224 }
 225
 226 static int control(uint32_t request, void *data)
 227 {
 228   switch (request) {
 229   case VOCTRL_DRAW_IMAGE:
 230     return draw_image(data);
 231   case VOCTRL_QUERY_FORMAT:
 232     return query_format(*((uint32_t*)data));
 233   }
 234   return VO_NOTIMPL;
 235 }