mpv/libvo/vo_png.c

236 lines
5.6 KiB
C

/*
* vo_png.c, Portable Network Graphics Renderer for MPlayer
*
* Copyright 2001 by Felix Buenemann <atmosfear@users.sourceforge.net>
*
* Uses libpng (which uses zlib), so see according licenses.
*
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <png.h>
#include "mp_msg.h"
#include "config.h"
#include "mp_msg.h"
#include "help_mp.h"
#include "video_out.h"
#include "video_out_internal.h"
#include "subopt-helper.h"
static const vo_info_t info =
{
"PNG file",
"png",
"Felix Buenemann <atmosfear@users.sourceforge.net>",
""
};
const LIBVO_EXTERN (png)
static int z_compression = Z_NO_COMPRESSION;
static int framenum = 0;
struct pngdata {
FILE * fp;
png_structp png_ptr;
png_infop info_ptr;
enum {OK,ERROR} status;
};
static int
config(uint32_t width, uint32_t height, uint32_t d_width, uint32_t d_height, uint32_t flags, char *title, uint32_t format)
{
if(z_compression == 0) {
mp_msg(MSGT_VO,MSGL_INFO, MSGTR_LIBVO_PNG_Warning1);
mp_msg(MSGT_VO,MSGL_INFO, MSGTR_LIBVO_PNG_Warning2);
mp_msg(MSGT_VO,MSGL_INFO, MSGTR_LIBVO_PNG_Warning3);
}
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Compression level %i\n", z_compression);
return 0;
}
static struct pngdata create_png (char * fname, int image_width, int image_height, int swapped)
{
struct pngdata png;
/*png_structp png_ptr = png_create_write_struct
(PNG_LIBPNG_VER_STRING, (png_voidp)user_error_ptr,
user_error_fn, user_warning_fn);*/
//png_byte *row_pointers[image_height];
png.png_ptr = png_create_write_struct
(PNG_LIBPNG_VER_STRING, NULL,
NULL, NULL);
png.info_ptr = png_create_info_struct(png.png_ptr);
if (!png.png_ptr) {
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Failed to init png pointer\n");
png.status = ERROR;
return png;
}
if (!png.info_ptr) {
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Failed to init png infopointer\n");
png_destroy_write_struct(&png.png_ptr,
(png_infopp)NULL);
png.status = ERROR;
return png;
}
if (setjmp(png.png_ptr->jmpbuf)) {
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Internal error!\n");
png_destroy_write_struct(&png.png_ptr, &png.info_ptr);
fclose(png.fp);
png.status = ERROR;
return png;
}
png.fp = fopen (fname, "wb");
if (png.fp == NULL) {
mp_msg(MSGT_VO,MSGL_WARN, MSGTR_LIBVO_PNG_ErrorOpeningForWriting, strerror(errno));
png.status = ERROR;
return png;
}
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Init IO\n");
png_init_io(png.png_ptr, png.fp);
/* set the zlib compression level */
png_set_compression_level(png.png_ptr, z_compression);
/*png_set_IHDR(png_ptr, info_ptr, width, height,
bit_depth, color_type, interlace_type,
compression_type, filter_type)*/
png_set_IHDR(png.png_ptr, png.info_ptr, image_width, image_height,
8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,
PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Write Info\n");
png_write_info(png.png_ptr, png.info_ptr);
if(swapped) {
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Set BGR Conversion\n");
png_set_bgr(png.png_ptr);
}
png.status = OK;
return png;
}
static uint8_t destroy_png(struct pngdata png) {
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Write End\n");
png_write_end(png.png_ptr, png.info_ptr);
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Destroy Write Struct\n");
png_destroy_write_struct(&png.png_ptr, &png.info_ptr);
fclose (png.fp);
return 0;
}
static uint32_t draw_image(mp_image_t* mpi){
char buf[100];
int k;
struct pngdata png;
png_byte *row_pointers[mpi->h];
// if -dr or -slices then do nothing:
if(mpi->flags&(MP_IMGFLAG_DIRECT|MP_IMGFLAG_DRAW_CALLBACK)) return VO_TRUE;
snprintf (buf, 100, "%08d.png", ++framenum);
png = create_png(buf, mpi->w, mpi->h, IMGFMT_IS_BGR(mpi->imgfmt));
if(png.status){
mp_msg(MSGT_VO,MSGL_WARN, MSGTR_LIBVO_PNG_ErrorInCreatePng);
return 1;
}
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Creating Row Pointers\n");
for ( k = 0; k < mpi->h; k++ )
row_pointers[k] = mpi->planes[0]+mpi->stride[0]*k;
//png_write_flush(png.png_ptr);
//png_set_flush(png.png_ptr, nrows);
if( mp_msg_test(MSGT_VO,MSGL_DBG2) ) {
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Writing Image Data\n"); }
png_write_image(png.png_ptr, row_pointers);
destroy_png(png);
return VO_TRUE;
}
static void draw_osd(void){}
static void flip_page (void){}
static int draw_frame(uint8_t * src[])
{
return -1;
}
static int draw_slice( uint8_t *src[],int stride[],int w,int h,int x,int y )
{
return -1;
}
static int
query_format(uint32_t format)
{
switch(format){
case IMGFMT_RGB|24:
case IMGFMT_BGR|24:
return VFCAP_CSP_SUPPORTED|VFCAP_CSP_SUPPORTED_BY_HW|VFCAP_ACCEPT_STRIDE;
}
return 0;
}
static void uninit(void){}
static void check_events(void){}
static int int_zero_to_nine(int *sh)
{
if ( (*sh < 0) || (*sh > 9) )
return 0;
return 1;
}
static opt_t subopts[] = {
{"z", OPT_ARG_INT, &z_compression, (opt_test_f)int_zero_to_nine},
{NULL}
};
static int preinit(const char *arg)
{
z_compression = 0;
if (subopt_parse(arg, subopts) != 0) {
return -1;
}
return 0;
}
static int control(uint32_t request, void *data, ...)
{
switch (request) {
case VOCTRL_DRAW_IMAGE:
return draw_image(data);
case VOCTRL_QUERY_FORMAT:
return query_format(*((uint32_t*)data));
}
return VO_NOTIMPL;
}