3 * Copyright (c) 2003 Fabrice Bellard.
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2 of the License, or (at your option) any later version.
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 #include "libpng/png.h"
23 extern const uint8_t png_sig[];
25 static int png_probe(AVProbeData *pd)
27 if (pd->buf_size >= 8 &&
28 memcmp(pd->buf, png_sig, 8) == 0)
29 return AVPROBE_SCORE_MAX;
35 png_memset_check (png_structp png_ptr, png_voidp s1, int value,
40 size = (png_size_t)length;
41 if ((png_uint_32)size != length)
42 png_error(png_ptr,"Overflow in png_memset_check.");
44 return (png_memset (s1, value, size));
49 png_memcpy_check (png_structp png_ptr, png_voidp s1, png_voidp s2,
54 size = (png_size_t)length;
55 if ((png_uint_32)size != length)
56 png_error(png_ptr,"Overflow in png_memcpy_check.");
58 return(png_memcpy (s1, s2, size));
61 void png_error (png_struct *png_ptr, const char *message)
63 longjmp(png_ptr->jmpbuf, 0);
66 void png_warning (png_struct *png_ptr, const char *message)
71 png_chunk_error(png_structp png_ptr, png_const_charp error_message)
74 // png_format_buffer(png_ptr, msg, error_message);
75 png_error(png_ptr, msg);
79 png_chunk_warning(png_structp png_ptr, png_const_charp warning_message)
82 // png_format_buffer(png_ptr, msg, warning_message);
83 png_warning(png_ptr, msg);
86 void png_read_data (png_struct *png_ptr, png_byte *data, png_uint_32 length)
90 ret = get_buffer(png_ptr->io_ptr, data, length);
92 png_error (png_ptr, "Read Error");
95 void *png_malloc (png_struct *png_ptr, png_uint_32 size)
97 return av_malloc(size);
100 void png_free (png_struct *png_ptr, void *ptr)
105 static int png_read(ByteIOContext *f,
106 int (*alloc_cb)(void *opaque, AVImageInfo *info), void *opaque)
108 png_struct png_struct1, *png_ptr;
109 png_info png_info1, png_info2, *info_ptr, *end_info;
110 AVImageInfo info1, *info = &info1;
111 int y, height, ret, row_size;
114 png_ptr = &png_struct1;
115 png_read_init(png_ptr);
117 info_ptr = &png_info1;
118 end_info = &png_info2;
120 png_info_init(info_ptr);
121 png_info_init(end_info);
125 if (setjmp(png_jmpbuf(png_ptr))) {
126 png_read_destroy (png_ptr, info_ptr, NULL);
130 png_read_info (png_ptr, info_ptr);
132 /* init image info */
133 info->width = info_ptr->width;
134 info->height = info_ptr->height;
136 if (info_ptr->bit_depth == 8 &&
137 info_ptr->color_type == PNG_COLOR_TYPE_RGB) {
138 info->pix_fmt = PIX_FMT_RGB24;
139 row_size = info_ptr->width * 3;
140 } else if (info_ptr->bit_depth == 8 &&
141 info_ptr->color_type == PNG_COLOR_TYPE_GRAY) {
142 info->pix_fmt = PIX_FMT_GRAY8;
143 row_size = info_ptr->width;
144 } else if (info_ptr->bit_depth == 1 &&
145 info_ptr->color_type == PNG_COLOR_TYPE_GRAY) {
146 info->pix_fmt = PIX_FMT_MONOBLACK;
147 row_size = (info_ptr->width + 7) >> 3;
149 png_read_destroy (png_ptr, info_ptr, NULL);
152 ret = alloc_cb(opaque, info);
154 png_read_destroy (png_ptr, info_ptr, NULL);
158 /* now read the whole image */
159 png_start_read_image (png_ptr);
161 height = info->height;
162 ptr = info->pict.data[0];
163 for (y = 0; y < height; y++) {
164 png_read_row (png_ptr, NULL, NULL);
165 memcpy(ptr, png_ptr->row_buf + 1, row_size);
166 ptr += info->pict.linesize[0];
169 png_read_destroy (png_ptr, info_ptr, NULL);
174 png_write_data(png_structp png_ptr, png_bytep data, png_size_t length)
176 put_buffer(png_ptr->io_ptr, data, length);
179 static int png_write(ByteIOContext *pb, AVImageInfo *info)
181 png_struct png_struct1, *png_ptr;
182 png_info png_info1, *info_ptr;
186 png_ptr = &png_struct1;
187 info_ptr = &png_info1;
189 png_write_init(png_ptr);
190 png_info_init(info_ptr);
192 png_ptr->io_ptr = pb;
194 if (setjmp(png_ptr->jmpbuf)) {
195 png_write_destroy(png_ptr);
203 info_ptr->height = h;
204 switch(info->pix_fmt) {
206 info_ptr->bit_depth = 8;
207 info_ptr->color_type = PNG_COLOR_TYPE_RGB;
210 info_ptr->bit_depth = 8;
211 info_ptr->color_type = PNG_COLOR_TYPE_GRAY;
213 case PIX_FMT_MONOBLACK:
214 info_ptr->bit_depth = 1;
215 info_ptr->color_type = PNG_COLOR_TYPE_GRAY;
220 png_write_info(png_ptr, info_ptr);
222 ptr = info->pict.data[0];
224 png_write_row(png_ptr, ptr);
225 ptr += info->pict.linesize[0];
227 png_write_end(png_ptr, info_ptr);
228 png_write_destroy(png_ptr);
229 put_flush_packet(pb);
233 AVImageFormat png_image_format = {
238 (1 << PIX_FMT_RGB24) | (1 << PIX_FMT_GRAY8) | (1 << PIX_FMT_MONOBLACK),