mpv/libvo/vo_png.c

203 lines
5.0 KiB
C

/*
* Portable Network Graphics renderer
*
* Copyright 2001 by Felix Buenemann <atmosfear@users.sourceforge.net>
*
* Uses libpng (which uses zlib), so see according licenses.
*
* This file is part of MPlayer.
*
* MPlayer is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* MPlayer is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with MPlayer; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include "config.h"
#include "mp_msg.h"
#include "mp_msg.h"
#include "video_out.h"
#include "video_out_internal.h"
#include "subopt-helper.h"
#include "libavcodec/avcodec.h"
#include "fmt-conversion.h"
static const vo_info_t info =
{
"PNG file",
"png",
"Felix Buenemann <atmosfear@users.sourceforge.net>",
""
};
const LIBVO_EXTERN (png)
static int z_compression;
static int framenum;
static int use_alpha;
static AVCodecContext *avctx;
static AVFrame *pic;
static uint8_t *outbuffer;
int outbuffer_size;
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_tmsg(MSGT_VO,MSGL_INFO, "[VO_PNG] Warning: compression level set to 0, compression disabled!\n");
mp_tmsg(MSGT_VO,MSGL_INFO, "[VO_PNG] Info: Use -vo png:z=<n> to set compression level from 0 to 9.\n");
mp_tmsg(MSGT_VO,MSGL_INFO, "[VO_PNG] Info: (0 = no compression, 1 = fastest, lowest - 9 best, slowest compression)\n");
}
mp_msg(MSGT_VO,MSGL_DBG2, "PNG Compression level %i\n", z_compression);
uninit();
struct AVCodec *png_codec = avcodec_find_encoder(CODEC_ID_PNG);
if (!png_codec)
goto error;
avctx = avcodec_alloc_context3(png_codec);
if (!avctx)
goto error;
pic = avcodec_alloc_frame();
if (!pic)
goto error;
avctx->width = width;
avctx->height = height;
avctx->pix_fmt = imgfmt2pixfmt(format);
avctx->compression_level = z_compression;
if (avcodec_open2(avctx, png_codec, NULL) < 0)
goto error;
return 0;
error:
uninit();
return -1;
}
static uint32_t draw_image(mp_image_t* mpi){
int buffersize;
int res;
char buf[100];
FILE *outfile;
// 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);
outfile = fopen(buf, "wb");
if (!outfile) {
mp_msg(MSGT_VO,MSGL_WARN, "\n[VO_PNG] Error opening '%s' for writing!\n", strerror(errno));
return 1;
}
avcodec_get_frame_defaults(pic);
pic->data[0] = mpi->planes[0];
pic->linesize[0] = mpi->stride[0];
buffersize = mpi->w * mpi->h * 8;
if (outbuffer_size < buffersize) {
av_freep(&outbuffer);
outbuffer = av_malloc(buffersize);
outbuffer_size = buffersize;
}
res = avcodec_encode_video(avctx, outbuffer, outbuffer_size, pic);
if(res < 0){
mp_msg(MSGT_VO,MSGL_WARN, "[VO_PNG] Error in create_png.\n");
fclose(outfile);
return 1;
}
fwrite(outbuffer, res, 1, outfile);
fclose(outfile);
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)
{
const int supported_flags = VFCAP_CSP_SUPPORTED|VFCAP_CSP_SUPPORTED_BY_HW|VFCAP_ACCEPT_STRIDE;
switch(format){
case IMGFMT_RGB24:
return use_alpha ? 0 : supported_flags;
case IMGFMT_BGR32:
return use_alpha ? supported_flags : 0;
}
return 0;
}
static void uninit(void)
{
if (avctx)
avcodec_close(avctx);
av_freep(&avctx);
av_freep(&outbuffer);
av_free(pic);
pic = NULL;
outbuffer_size = 0;
}
static void check_events(void){}
static int int_zero_to_nine(void *value)
{
int *sh = value;
return *sh >= 0 && *sh <= 9;
}
static const opt_t subopts[] = {
{"alpha", OPT_ARG_BOOL, &use_alpha, NULL},
{"z", OPT_ARG_INT, &z_compression, int_zero_to_nine},
{NULL}
};
static int preinit(const char *arg)
{
z_compression = 0;
use_alpha = 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;
}