Mercurial > libavformat.hg
changeset 44:7c5da7e7239a libavformat
added basic png handling
author | bellard |
---|---|
date | Sat, 01 Feb 2003 20:53:57 +0000 |
parents | a23fb606308a |
children | 2036f790ece3 |
files | allformats.c avformat.h png.c |
diffstat | 3 files changed, 242 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- a/allformats.c Sat Feb 01 20:53:07 2003 +0000 +++ b/allformats.c Sat Feb 01 20:53:57 2003 +0000 @@ -75,6 +75,7 @@ av_register_image_format(&ppm_image_format); av_register_image_format(&pgmyuv_image_format); av_register_image_format(&yuv_image_format); + av_register_image_format(&png_image_format); /* file protocols */ register_protocol(&file_protocol);
--- a/avformat.h Sat Feb 01 20:53:07 2003 +0000 +++ b/avformat.h Sat Feb 01 20:53:57 2003 +0000 @@ -235,6 +235,7 @@ extern AVImageFormat ppm_image_format; extern AVImageFormat pgmyuv_image_format; extern AVImageFormat yuv_image_format; +extern AVImageFormat png_image_format; /* XXX: use automatic init with either ELF sections or C file parser */ /* modules */
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/png.c Sat Feb 01 20:53:57 2003 +0000 @@ -0,0 +1,240 @@ +/* + * PNG image format + * Copyright (c) 2003 Fabrice Bellard. + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2 of the License, or (at your option) any later version. + * + * This library 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ +#include "avformat.h" + +#include "libpng/png.h" + +extern const uint8_t png_sig[]; + +static int png_probe(AVProbeData *pd) +{ + if (pd->buf_size >= 8 && + memcmp(pd->buf, png_sig, 8) == 0) + return AVPROBE_SCORE_MAX; + else + return 0; +} + +png_voidp PNGAPI +png_memset_check (png_structp png_ptr, png_voidp s1, int value, + png_uint_32 length) +{ + png_size_t size; + + size = (png_size_t)length; + if ((png_uint_32)size != length) + png_error(png_ptr,"Overflow in png_memset_check."); + + return (png_memset (s1, value, size)); + +} + +png_voidp PNGAPI +png_memcpy_check (png_structp png_ptr, png_voidp s1, png_voidp s2, + png_uint_32 length) +{ + png_size_t size; + + size = (png_size_t)length; + if ((png_uint_32)size != length) + png_error(png_ptr,"Overflow in png_memcpy_check."); + + return(png_memcpy (s1, s2, size)); +} + +void png_error (png_struct *png_ptr, const char *message) +{ + longjmp(png_ptr->jmpbuf, 0); +} + +void png_warning (png_struct *png_ptr, const char *message) +{ +} + +void PNGAPI +png_chunk_error(png_structp png_ptr, png_const_charp error_message) +{ + char msg[18+64]; + // png_format_buffer(png_ptr, msg, error_message); + png_error(png_ptr, msg); +} + +void PNGAPI +png_chunk_warning(png_structp png_ptr, png_const_charp warning_message) +{ + char msg[18+64]; + // png_format_buffer(png_ptr, msg, warning_message); + png_warning(png_ptr, msg); +} + +void png_read_data (png_struct *png_ptr, png_byte *data, png_uint_32 length) +{ + int ret; + + ret = get_buffer(png_ptr->io_ptr, data, length); + if (ret != length) + png_error (png_ptr, "Read Error"); +} + +void *png_malloc (png_struct *png_ptr, png_uint_32 size) +{ + return av_malloc(size); +} + +void png_free (png_struct *png_ptr, void *ptr) +{ + return av_free(ptr); +} + +static int png_read(ByteIOContext *f, + int (*alloc_cb)(void *opaque, AVImageInfo *info), void *opaque) +{ + png_struct png_struct1, *png_ptr; + png_info png_info1, png_info2, *info_ptr, *end_info; + AVImageInfo info1, *info = &info1; + int y, height, ret, row_size; + uint8_t *ptr; + + png_ptr = &png_struct1; + png_read_init(png_ptr); + + info_ptr = &png_info1; + end_info = &png_info2; + + png_info_init(info_ptr); + png_info_init(end_info); + + png_ptr->io_ptr = f; + + if (setjmp(png_jmpbuf(png_ptr))) { + png_read_destroy (png_ptr, info_ptr, NULL); + return -1; + } + + png_read_info (png_ptr, info_ptr); + + /* init image info */ + info->width = info_ptr->width; + info->height = info_ptr->height; + + if (info_ptr->bit_depth == 8 && + info_ptr->color_type == PNG_COLOR_TYPE_RGB) { + info->pix_fmt = PIX_FMT_RGB24; + row_size = info_ptr->width * 3; + } else if (info_ptr->bit_depth == 8 && + info_ptr->color_type == PNG_COLOR_TYPE_GRAY) { + info->pix_fmt = PIX_FMT_GRAY8; + row_size = info_ptr->width; + } else if (info_ptr->bit_depth == 1 && + info_ptr->color_type == PNG_COLOR_TYPE_GRAY) { + info->pix_fmt = PIX_FMT_MONOBLACK; + row_size = (info_ptr->width + 7) >> 3; + } else { + png_read_destroy (png_ptr, info_ptr, NULL); + return -1; + } + ret = alloc_cb(opaque, info); + if (ret) { + png_read_destroy (png_ptr, info_ptr, NULL); + return ret; + } + + /* now read the whole image */ + png_start_read_image (png_ptr); + + height = info->height; + ptr = info->pict.data[0]; + for (y = 0; y < height; y++) { + png_read_row (png_ptr, NULL, NULL); + memcpy(ptr, png_ptr->row_buf + 1, row_size); + ptr += info->pict.linesize[0]; + } + + png_read_destroy (png_ptr, info_ptr, NULL); + return 0; +} + +void +png_write_data(png_structp png_ptr, png_bytep data, png_size_t length) +{ + put_buffer(png_ptr->io_ptr, data, length); +} + +static int png_write(ByteIOContext *pb, AVImageInfo *info) +{ + png_struct png_struct1, *png_ptr; + png_info png_info1, *info_ptr; + int w, h, y; + uint8_t *ptr; + + png_ptr = &png_struct1; + info_ptr = &png_info1; + + png_write_init(png_ptr); + png_info_init(info_ptr); + + png_ptr->io_ptr = pb; + + if (setjmp(png_ptr->jmpbuf)) { + png_write_destroy(png_ptr); + return -1; + } + + w = info->width; + h = info->height; + + info_ptr->width = w; + info_ptr->height = h; + switch(info->pix_fmt) { + case PIX_FMT_RGB24: + info_ptr->bit_depth = 8; + info_ptr->color_type = PNG_COLOR_TYPE_RGB; + break; + case PIX_FMT_GRAY8: + info_ptr->bit_depth = 8; + info_ptr->color_type = PNG_COLOR_TYPE_GRAY; + break; + case PIX_FMT_MONOBLACK: + info_ptr->bit_depth = 1; + info_ptr->color_type = PNG_COLOR_TYPE_GRAY; + break; + default: + return -1; + } + png_write_info(png_ptr, info_ptr); + + ptr = info->pict.data[0]; + for(y=0;y<h;y++) { + png_write_row(png_ptr, ptr); + ptr += info->pict.linesize[0]; + } + png_write_end(png_ptr, info_ptr); + png_write_destroy(png_ptr); + put_flush_packet(pb); + return 0; +} + +AVImageFormat png_image_format = { + "png", + "png", + png_probe, + png_read, + (1 << PIX_FMT_RGB24) | (1 << PIX_FMT_GRAY8) | (1 << PIX_FMT_MONOBLACK), + png_write, +};