Mercurial > libavcodec.hg
annotate png.c @ 5076:796c2a5481ad libavcodec
make decoder decode the first frame properly if theres more than just one
frame input
author | michael |
---|---|
date | Sun, 27 May 2007 22:34:49 +0000 |
parents | 6166fbf375cc |
children | bff60ecc02f9 |
rev | line source |
---|---|
2342 | 1 /* |
2 * PNG image format | |
3 * Copyright (c) 2003 Fabrice Bellard. | |
4 * | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3777
diff
changeset
|
5 * This file is part of FFmpeg. |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3777
diff
changeset
|
6 * |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3777
diff
changeset
|
7 * FFmpeg is free software; you can redistribute it and/or |
2342 | 8 * modify it under the terms of the GNU Lesser General Public |
9 * License as published by the Free Software Foundation; either | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3777
diff
changeset
|
10 * version 2.1 of the License, or (at your option) any later version. |
2342 | 11 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3777
diff
changeset
|
12 * FFmpeg is distributed in the hope that it will be useful, |
2342 | 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3777
diff
changeset
|
18 * License along with FFmpeg; if not, write to the Free Software |
3036
0b546eab515d
Update licensing information: The FSF changed postal address.
diego
parents:
2967
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
2342 | 20 */ |
21 #include "avcodec.h" | |
5067 | 22 #include "bytestream.h" |
2342 | 23 |
24 /* TODO: | |
25 * - add 2, 4 and 16 bit depth support | |
26 * - use filters when generating a png (better compression) | |
27 */ | |
28 | |
29 #include <zlib.h> | |
30 | |
31 //#define DEBUG | |
32 | |
33 #define PNG_COLOR_MASK_PALETTE 1 | |
34 #define PNG_COLOR_MASK_COLOR 2 | |
35 #define PNG_COLOR_MASK_ALPHA 4 | |
36 | |
37 #define PNG_COLOR_TYPE_GRAY 0 | |
38 #define PNG_COLOR_TYPE_PALETTE (PNG_COLOR_MASK_COLOR | PNG_COLOR_MASK_PALETTE) | |
39 #define PNG_COLOR_TYPE_RGB (PNG_COLOR_MASK_COLOR) | |
40 #define PNG_COLOR_TYPE_RGB_ALPHA (PNG_COLOR_MASK_COLOR | PNG_COLOR_MASK_ALPHA) | |
41 #define PNG_COLOR_TYPE_GRAY_ALPHA (PNG_COLOR_MASK_ALPHA) | |
42 | |
43 #define PNG_FILTER_VALUE_NONE 0 | |
44 #define PNG_FILTER_VALUE_SUB 1 | |
45 #define PNG_FILTER_VALUE_UP 2 | |
46 #define PNG_FILTER_VALUE_AVG 3 | |
47 #define PNG_FILTER_VALUE_PAETH 4 | |
48 | |
49 #define PNG_IHDR 0x0001 | |
50 #define PNG_IDAT 0x0002 | |
51 #define PNG_ALLIMAGE 0x0004 | |
52 #define PNG_PLTE 0x0008 | |
53 | |
54 #define NB_PASSES 7 | |
55 | |
56 #define IOBUF_SIZE 4096 | |
57 | |
58 typedef struct PNGContext { | |
59 uint8_t *bytestream; | |
60 uint8_t *bytestream_start; | |
61 uint8_t *bytestream_end; | |
62 AVFrame picture; | |
63 | |
64 int state; | |
65 int width, height; | |
66 int bit_depth; | |
67 int color_type; | |
68 int compression_type; | |
69 int interlace_type; | |
70 int filter_type; | |
71 int channels; | |
72 int bits_per_pixel; | |
73 int bpp; | |
2967 | 74 |
2342 | 75 uint8_t *image_buf; |
76 int image_linesize; | |
77 uint32_t palette[256]; | |
78 uint8_t *crow_buf; | |
79 uint8_t *last_row; | |
80 uint8_t *tmp_row; | |
81 int pass; | |
82 int crow_size; /* compressed row size (include filter type) */ | |
83 int row_size; /* decompressed row size */ | |
84 int pass_row_size; /* decompress row size of the current pass */ | |
85 int y; | |
86 z_stream zstream; | |
87 uint8_t buf[IOBUF_SIZE]; | |
88 } PNGContext; | |
89 | |
90 static const uint8_t pngsig[8] = {137, 80, 78, 71, 13, 10, 26, 10}; | |
91 | |
92 /* Mask to determine which y pixels are valid in a pass */ | |
93 static const uint8_t png_pass_ymask[NB_PASSES] = { | |
94 0x80, 0x80, 0x08, 0x88, 0x22, 0xaa, 0x55, | |
95 }; | |
96 | |
97 /* Mask to determine which y pixels can be written in a pass */ | |
98 static const uint8_t png_pass_dsp_ymask[NB_PASSES] = { | |
99 0xff, 0xff, 0x0f, 0xcc, 0x33, 0xff, 0x55, | |
100 }; | |
101 | |
102 /* minimum x value */ | |
103 static const uint8_t png_pass_xmin[NB_PASSES] = { | |
104 0, 4, 0, 2, 0, 1, 0 | |
105 }; | |
106 | |
107 /* x shift to get row width */ | |
108 static const uint8_t png_pass_xshift[NB_PASSES] = { | |
109 3, 3, 2, 2, 1, 1, 0 | |
110 }; | |
111 | |
112 /* Mask to determine which pixels are valid in a pass */ | |
113 static const uint8_t png_pass_mask[NB_PASSES] = { | |
114 0x80, 0x08, 0x88, 0x22, 0xaa, 0x55, 0xff | |
115 }; | |
116 | |
117 /* Mask to determine which pixels to overwrite while displaying */ | |
2967 | 118 static const uint8_t png_pass_dsp_mask[NB_PASSES] = { |
2342 | 119 0xff, 0x0f, 0xff, 0x33, 0xff, 0x55, 0xff |
120 }; | |
121 #if 0 | |
122 static int png_probe(AVProbeData *pd) | |
123 { | |
124 if (pd->buf_size >= 8 && | |
125 memcmp(pd->buf, pngsig, 8) == 0) | |
126 return AVPROBE_SCORE_MAX; | |
127 else | |
128 return 0; | |
129 } | |
130 #endif | |
131 static void *png_zalloc(void *opaque, unsigned int items, unsigned int size) | |
132 { | |
2422 | 133 if(items >= UINT_MAX / size) |
134 return NULL; | |
2342 | 135 return av_malloc(items * size); |
136 } | |
137 | |
138 static void png_zfree(void *opaque, void *ptr) | |
139 { | |
140 av_free(ptr); | |
141 } | |
142 | |
143 static int png_get_nb_channels(int color_type) | |
144 { | |
145 int channels; | |
146 channels = 1; | |
147 if ((color_type & (PNG_COLOR_MASK_COLOR | PNG_COLOR_MASK_PALETTE)) == | |
148 PNG_COLOR_MASK_COLOR) | |
149 channels = 3; | |
150 if (color_type & PNG_COLOR_MASK_ALPHA) | |
151 channels++; | |
152 return channels; | |
153 } | |
154 | |
155 /* compute the row size of an interleaved pass */ | |
156 static int png_pass_row_size(int pass, int bits_per_pixel, int width) | |
157 { | |
158 int shift, xmin, pass_width; | |
159 | |
160 xmin = png_pass_xmin[pass]; | |
161 if (width <= xmin) | |
162 return 0; | |
163 shift = png_pass_xshift[pass]; | |
164 pass_width = (width - xmin + (1 << shift) - 1) >> shift; | |
165 return (pass_width * bits_per_pixel + 7) >> 3; | |
166 } | |
167 | |
168 /* NOTE: we try to construct a good looking image at each pass. width | |
169 is the original image width. We also do pixel format convertion at | |
170 this stage */ | |
2967 | 171 static void png_put_interlaced_row(uint8_t *dst, int width, |
172 int bits_per_pixel, int pass, | |
2342 | 173 int color_type, const uint8_t *src) |
174 { | |
175 int x, mask, dsp_mask, j, src_x, b, bpp; | |
176 uint8_t *d; | |
177 const uint8_t *s; | |
2967 | 178 |
2342 | 179 mask = png_pass_mask[pass]; |
180 dsp_mask = png_pass_dsp_mask[pass]; | |
181 switch(bits_per_pixel) { | |
182 case 1: | |
183 /* we must intialize the line to zero before writing to it */ | |
184 if (pass == 0) | |
185 memset(dst, 0, (width + 7) >> 3); | |
186 src_x = 0; | |
187 for(x = 0; x < width; x++) { | |
188 j = (x & 7); | |
189 if ((dsp_mask << j) & 0x80) { | |
190 b = (src[src_x >> 3] >> (7 - (src_x & 7))) & 1; | |
191 dst[x >> 3] |= b << (7 - j); | |
192 } | |
193 if ((mask << j) & 0x80) | |
194 src_x++; | |
195 } | |
196 break; | |
197 default: | |
198 bpp = bits_per_pixel >> 3; | |
199 d = dst; | |
200 s = src; | |
201 if (color_type == PNG_COLOR_TYPE_RGB_ALPHA) { | |
202 for(x = 0; x < width; x++) { | |
203 j = x & 7; | |
204 if ((dsp_mask << j) & 0x80) { | |
205 *(uint32_t *)d = (s[3] << 24) | (s[0] << 16) | (s[1] << 8) | s[2]; | |
206 } | |
207 d += bpp; | |
208 if ((mask << j) & 0x80) | |
209 s += bpp; | |
210 } | |
211 } else { | |
212 for(x = 0; x < width; x++) { | |
213 j = x & 7; | |
214 if ((dsp_mask << j) & 0x80) { | |
215 memcpy(d, s, bpp); | |
216 } | |
217 d += bpp; | |
218 if ((mask << j) & 0x80) | |
219 s += bpp; | |
220 } | |
221 } | |
222 break; | |
223 } | |
224 } | |
225 | |
3777 | 226 #ifdef CONFIG_ENCODERS |
2967 | 227 static void png_get_interlaced_row(uint8_t *dst, int row_size, |
228 int bits_per_pixel, int pass, | |
2342 | 229 const uint8_t *src, int width) |
230 { | |
231 int x, mask, dst_x, j, b, bpp; | |
232 uint8_t *d; | |
233 const uint8_t *s; | |
234 | |
235 mask = png_pass_mask[pass]; | |
236 switch(bits_per_pixel) { | |
237 case 1: | |
238 memset(dst, 0, row_size); | |
239 dst_x = 0; | |
240 for(x = 0; x < width; x++) { | |
241 j = (x & 7); | |
242 if ((mask << j) & 0x80) { | |
243 b = (src[x >> 3] >> (7 - j)) & 1; | |
244 dst[dst_x >> 3] |= b << (7 - (dst_x & 7)); | |
245 dst_x++; | |
246 } | |
247 } | |
248 break; | |
249 default: | |
250 bpp = bits_per_pixel >> 3; | |
251 d = dst; | |
252 s = src; | |
253 for(x = 0; x < width; x++) { | |
254 j = x & 7; | |
255 if ((mask << j) & 0x80) { | |
256 memcpy(d, s, bpp); | |
257 d += bpp; | |
258 } | |
259 s += bpp; | |
260 } | |
261 break; | |
262 } | |
263 } | |
3777 | 264 #endif |
2342 | 265 |
266 /* XXX: optimize */ | |
267 /* NOTE: 'dst' can be equal to 'last' */ | |
2967 | 268 static void png_filter_row(uint8_t *dst, int filter_type, |
2342 | 269 uint8_t *src, uint8_t *last, int size, int bpp) |
270 { | |
271 int i, p; | |
272 | |
273 switch(filter_type) { | |
274 case PNG_FILTER_VALUE_NONE: | |
275 memcpy(dst, src, size); | |
276 break; | |
277 case PNG_FILTER_VALUE_SUB: | |
278 for(i = 0; i < bpp; i++) { | |
279 dst[i] = src[i]; | |
280 } | |
281 for(i = bpp; i < size; i++) { | |
282 p = dst[i - bpp]; | |
283 dst[i] = p + src[i]; | |
284 } | |
285 break; | |
286 case PNG_FILTER_VALUE_UP: | |
287 for(i = 0; i < size; i++) { | |
288 p = last[i]; | |
289 dst[i] = p + src[i]; | |
290 } | |
291 break; | |
292 case PNG_FILTER_VALUE_AVG: | |
293 for(i = 0; i < bpp; i++) { | |
294 p = (last[i] >> 1); | |
295 dst[i] = p + src[i]; | |
296 } | |
297 for(i = bpp; i < size; i++) { | |
298 p = ((dst[i - bpp] + last[i]) >> 1); | |
299 dst[i] = p + src[i]; | |
300 } | |
301 break; | |
302 case PNG_FILTER_VALUE_PAETH: | |
303 for(i = 0; i < bpp; i++) { | |
304 p = last[i]; | |
305 dst[i] = p + src[i]; | |
306 } | |
307 for(i = bpp; i < size; i++) { | |
308 int a, b, c, pa, pb, pc; | |
309 | |
310 a = dst[i - bpp]; | |
311 b = last[i]; | |
312 c = last[i - bpp]; | |
313 | |
314 p = b - c; | |
315 pc = a - c; | |
316 | |
317 pa = abs(p); | |
318 pb = abs(pc); | |
319 pc = abs(p + pc); | |
320 | |
321 if (pa <= pb && pa <= pc) | |
322 p = a; | |
323 else if (pb <= pc) | |
324 p = b; | |
325 else | |
326 p = c; | |
327 dst[i] = p + src[i]; | |
328 } | |
329 break; | |
330 } | |
331 } | |
332 | |
3777 | 333 #ifdef CONFIG_ENCODERS |
4515 | 334 static void convert_from_rgb32(uint8_t *dst, const uint8_t *src, int width) |
2342 | 335 { |
336 uint8_t *d; | |
337 int j; | |
338 unsigned int v; | |
2967 | 339 |
2342 | 340 d = dst; |
341 for(j = 0; j < width; j++) { | |
3347
82277c821113
Add const to (mostly) char* and make some functions static, which aren't used
diego
parents:
3177
diff
changeset
|
342 v = ((const uint32_t *)src)[j]; |
2342 | 343 d[0] = v >> 16; |
344 d[1] = v >> 8; | |
345 d[2] = v; | |
346 d[3] = v >> 24; | |
347 d += 4; | |
348 } | |
349 } | |
3777 | 350 #endif |
2342 | 351 |
3777 | 352 #ifdef CONFIG_DECODERS |
4515 | 353 static void convert_to_rgb32(uint8_t *dst, const uint8_t *src, int width) |
2342 | 354 { |
355 int j; | |
356 unsigned int r, g, b, a; | |
357 | |
358 for(j = 0;j < width; j++) { | |
359 r = src[0]; | |
360 g = src[1]; | |
361 b = src[2]; | |
362 a = src[3]; | |
363 *(uint32_t *)dst = (a << 24) | (r << 16) | (g << 8) | b; | |
364 dst += 4; | |
365 src += 4; | |
366 } | |
367 } | |
368 | |
369 /* process exactly one decompressed row */ | |
370 static void png_handle_row(PNGContext *s) | |
371 { | |
372 uint8_t *ptr, *last_row; | |
373 int got_line; | |
2967 | 374 |
2342 | 375 if (!s->interlace_type) { |
376 ptr = s->image_buf + s->image_linesize * s->y; | |
377 /* need to swap bytes correctly for RGB_ALPHA */ | |
378 if (s->color_type == PNG_COLOR_TYPE_RGB_ALPHA) { | |
2967 | 379 png_filter_row(s->tmp_row, s->crow_buf[0], s->crow_buf + 1, |
2342 | 380 s->last_row, s->row_size, s->bpp); |
381 memcpy(s->last_row, s->tmp_row, s->row_size); | |
4515 | 382 convert_to_rgb32(ptr, s->tmp_row, s->width); |
2342 | 383 } else { |
384 /* in normal case, we avoid one copy */ | |
385 if (s->y == 0) | |
386 last_row = s->last_row; | |
387 else | |
388 last_row = ptr - s->image_linesize; | |
2967 | 389 |
390 png_filter_row(ptr, s->crow_buf[0], s->crow_buf + 1, | |
2342 | 391 last_row, s->row_size, s->bpp); |
392 } | |
393 s->y++; | |
394 if (s->y == s->height) { | |
395 s->state |= PNG_ALLIMAGE; | |
396 } | |
397 } else { | |
398 got_line = 0; | |
399 for(;;) { | |
400 ptr = s->image_buf + s->image_linesize * s->y; | |
401 if ((png_pass_ymask[s->pass] << (s->y & 7)) & 0x80) { | |
402 /* if we already read one row, it is time to stop to | |
403 wait for the next one */ | |
404 if (got_line) | |
405 break; | |
2967 | 406 png_filter_row(s->tmp_row, s->crow_buf[0], s->crow_buf + 1, |
2342 | 407 s->last_row, s->pass_row_size, s->bpp); |
408 memcpy(s->last_row, s->tmp_row, s->pass_row_size); | |
409 got_line = 1; | |
410 } | |
411 if ((png_pass_dsp_ymask[s->pass] << (s->y & 7)) & 0x80) { | |
4515 | 412 /* NOTE: RGB32 is handled directly in png_put_interlaced_row */ |
2967 | 413 png_put_interlaced_row(ptr, s->width, s->bits_per_pixel, s->pass, |
2342 | 414 s->color_type, s->last_row); |
415 } | |
416 s->y++; | |
417 if (s->y == s->height) { | |
418 for(;;) { | |
419 if (s->pass == NB_PASSES - 1) { | |
420 s->state |= PNG_ALLIMAGE; | |
421 goto the_end; | |
422 } else { | |
423 s->pass++; | |
424 s->y = 0; | |
2967 | 425 s->pass_row_size = png_pass_row_size(s->pass, |
426 s->bits_per_pixel, | |
2342 | 427 s->width); |
428 s->crow_size = s->pass_row_size + 1; | |
429 if (s->pass_row_size != 0) | |
430 break; | |
431 /* skip pass if empty row */ | |
432 } | |
433 } | |
434 } | |
435 } | |
436 the_end: ; | |
437 } | |
438 } | |
439 | |
440 static int png_decode_idat(PNGContext *s, int length) | |
441 { | |
442 int ret; | |
443 s->zstream.avail_in = length; | |
444 s->zstream.next_in = s->bytestream; | |
445 s->bytestream += length; | |
2967 | 446 |
2342 | 447 if(s->bytestream > s->bytestream_end) |
448 return -1; | |
449 | |
450 /* decode one line if possible */ | |
451 while (s->zstream.avail_in > 0) { | |
452 ret = inflate(&s->zstream, Z_PARTIAL_FLUSH); | |
453 if (ret != Z_OK && ret != Z_STREAM_END) { | |
454 return -1; | |
455 } | |
456 if (s->zstream.avail_out == 0) { | |
457 if (!(s->state & PNG_ALLIMAGE)) { | |
458 png_handle_row(s); | |
459 } | |
460 s->zstream.avail_out = s->crow_size; | |
461 s->zstream.next_out = s->crow_buf; | |
462 } | |
463 } | |
464 return 0; | |
465 } | |
466 | |
2967 | 467 static int decode_frame(AVCodecContext *avctx, |
2342 | 468 void *data, int *data_size, |
469 uint8_t *buf, int buf_size) | |
470 { | |
471 PNGContext * const s = avctx->priv_data; | |
472 AVFrame *picture = data; | |
473 AVFrame * const p= (AVFrame*)&s->picture; | |
474 uint32_t tag, length; | |
475 int ret, crc; | |
476 | |
477 s->bytestream_start= | |
478 s->bytestream= buf; | |
479 s->bytestream_end= buf + buf_size; | |
480 | |
481 /* check signature */ | |
482 if (memcmp(s->bytestream, pngsig, 8) != 0) | |
483 return -1; | |
484 s->bytestream+= 8; | |
485 s->y= | |
486 s->state=0; | |
487 // memset(s, 0, sizeof(PNGContext)); | |
488 /* init the zlib */ | |
489 s->zstream.zalloc = png_zalloc; | |
490 s->zstream.zfree = png_zfree; | |
491 s->zstream.opaque = NULL; | |
492 ret = inflateInit(&s->zstream); | |
493 if (ret != Z_OK) | |
494 return -1; | |
495 for(;;) { | |
2347
c6280d48be02
When bswap_32 is a macro, png images fail to decode properly, patch by (Milan Cutka <cutka>at<szm>dot<sk>)
michael
parents:
2342
diff
changeset
|
496 int tag32; |
2342 | 497 if (s->bytestream >= s->bytestream_end) |
498 goto fail; | |
5067 | 499 length = bytestream_get_be32(&s->bytestream); |
2342 | 500 if (length > 0x7fffffff) |
501 goto fail; | |
5067 | 502 tag32 = bytestream_get_be32(&s->bytestream); |
2347
c6280d48be02
When bswap_32 is a macro, png images fail to decode properly, patch by (Milan Cutka <cutka>at<szm>dot<sk>)
michael
parents:
2342
diff
changeset
|
503 tag = bswap_32(tag32); |
2342 | 504 #ifdef DEBUG |
3177 | 505 av_log(avctx, AV_LOG_DEBUG, "png: tag=%c%c%c%c length=%u\n", |
2342 | 506 (tag & 0xff), |
507 ((tag >> 8) & 0xff), | |
508 ((tag >> 16) & 0xff), | |
509 ((tag >> 24) & 0xff), length); | |
510 #endif | |
511 switch(tag) { | |
512 case MKTAG('I', 'H', 'D', 'R'): | |
513 if (length != 13) | |
514 goto fail; | |
5067 | 515 s->width = bytestream_get_be32(&s->bytestream); |
516 s->height = bytestream_get_be32(&s->bytestream); | |
2422 | 517 if(avcodec_check_dimensions(avctx, s->width, s->height)){ |
518 s->width= s->height= 0; | |
519 goto fail; | |
520 } | |
2342 | 521 s->bit_depth = *s->bytestream++; |
522 s->color_type = *s->bytestream++; | |
523 s->compression_type = *s->bytestream++; | |
524 s->filter_type = *s->bytestream++; | |
525 s->interlace_type = *s->bytestream++; | |
5067 | 526 crc = bytestream_get_be32(&s->bytestream); |
2342 | 527 s->state |= PNG_IHDR; |
528 #ifdef DEBUG | |
3177 | 529 av_log(avctx, AV_LOG_DEBUG, "width=%d height=%d depth=%d color_type=%d compression_type=%d filter_type=%d interlace_type=%d\n", |
2967 | 530 s->width, s->height, s->bit_depth, s->color_type, |
2342 | 531 s->compression_type, s->filter_type, s->interlace_type); |
532 #endif | |
533 break; | |
534 case MKTAG('I', 'D', 'A', 'T'): | |
535 if (!(s->state & PNG_IHDR)) | |
536 goto fail; | |
537 if (!(s->state & PNG_IDAT)) { | |
538 /* init image info */ | |
539 avctx->width = s->width; | |
540 avctx->height = s->height; | |
541 | |
542 s->channels = png_get_nb_channels(s->color_type); | |
543 s->bits_per_pixel = s->bit_depth * s->channels; | |
544 s->bpp = (s->bits_per_pixel + 7) >> 3; | |
545 s->row_size = (avctx->width * s->bits_per_pixel + 7) >> 3; | |
546 | |
2967 | 547 if (s->bit_depth == 8 && |
2342 | 548 s->color_type == PNG_COLOR_TYPE_RGB) { |
549 avctx->pix_fmt = PIX_FMT_RGB24; | |
2967 | 550 } else if (s->bit_depth == 8 && |
2342 | 551 s->color_type == PNG_COLOR_TYPE_RGB_ALPHA) { |
4494
ce643a22f049
Replace deprecated PIX_FMT names by the newer variants.
diego
parents:
4379
diff
changeset
|
552 avctx->pix_fmt = PIX_FMT_RGB32; |
2967 | 553 } else if (s->bit_depth == 8 && |
2342 | 554 s->color_type == PNG_COLOR_TYPE_GRAY) { |
555 avctx->pix_fmt = PIX_FMT_GRAY8; | |
4067 | 556 } else if (s->bit_depth == 16 && |
557 s->color_type == PNG_COLOR_TYPE_GRAY) { | |
558 avctx->pix_fmt = PIX_FMT_GRAY16BE; | |
2967 | 559 } else if (s->bit_depth == 1 && |
2342 | 560 s->color_type == PNG_COLOR_TYPE_GRAY) { |
561 avctx->pix_fmt = PIX_FMT_MONOBLACK; | |
562 } else if (s->color_type == PNG_COLOR_TYPE_PALETTE) { | |
563 avctx->pix_fmt = PIX_FMT_PAL8; | |
564 } else { | |
565 goto fail; | |
566 } | |
567 if(p->data[0]) | |
568 avctx->release_buffer(avctx, p); | |
2967 | 569 |
2342 | 570 p->reference= 0; |
571 if(avctx->get_buffer(avctx, p) < 0){ | |
572 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); | |
573 goto fail; | |
574 } | |
575 p->pict_type= FF_I_TYPE; | |
576 p->key_frame= 1; | |
577 p->interlaced_frame = !!s->interlace_type; | |
578 | |
579 /* compute the compressed row size */ | |
580 if (!s->interlace_type) { | |
581 s->crow_size = s->row_size + 1; | |
582 } else { | |
583 s->pass = 0; | |
2967 | 584 s->pass_row_size = png_pass_row_size(s->pass, |
585 s->bits_per_pixel, | |
2342 | 586 s->width); |
587 s->crow_size = s->pass_row_size + 1; | |
588 } | |
589 #ifdef DEBUG | |
3177 | 590 av_log(avctx, AV_LOG_DEBUG, "row_size=%d crow_size =%d\n", |
2342 | 591 s->row_size, s->crow_size); |
592 #endif | |
593 s->image_buf = p->data[0]; | |
594 s->image_linesize = p->linesize[0]; | |
595 /* copy the palette if needed */ | |
596 if (s->color_type == PNG_COLOR_TYPE_PALETTE) | |
597 memcpy(p->data[1], s->palette, 256 * sizeof(uint32_t)); | |
598 /* empty row is used if differencing to the first row */ | |
599 s->last_row = av_mallocz(s->row_size); | |
600 if (!s->last_row) | |
601 goto fail; | |
602 if (s->interlace_type || | |
603 s->color_type == PNG_COLOR_TYPE_RGB_ALPHA) { | |
604 s->tmp_row = av_malloc(s->row_size); | |
605 if (!s->tmp_row) | |
606 goto fail; | |
607 } | |
608 /* compressed row */ | |
609 s->crow_buf = av_malloc(s->row_size + 1); | |
610 if (!s->crow_buf) | |
611 goto fail; | |
612 s->zstream.avail_out = s->crow_size; | |
613 s->zstream.next_out = s->crow_buf; | |
614 } | |
615 s->state |= PNG_IDAT; | |
616 if (png_decode_idat(s, length) < 0) | |
617 goto fail; | |
618 /* skip crc */ | |
5067 | 619 crc = bytestream_get_be32(&s->bytestream); |
2342 | 620 break; |
621 case MKTAG('P', 'L', 'T', 'E'): | |
622 { | |
623 int n, i, r, g, b; | |
2967 | 624 |
2342 | 625 if ((length % 3) != 0 || length > 256 * 3) |
626 goto skip_tag; | |
627 /* read the palette */ | |
628 n = length / 3; | |
629 for(i=0;i<n;i++) { | |
630 r = *s->bytestream++; | |
631 g = *s->bytestream++; | |
632 b = *s->bytestream++; | |
633 s->palette[i] = (0xff << 24) | (r << 16) | (g << 8) | b; | |
634 } | |
635 for(;i<256;i++) { | |
636 s->palette[i] = (0xff << 24); | |
637 } | |
638 s->state |= PNG_PLTE; | |
5067 | 639 crc = bytestream_get_be32(&s->bytestream); |
2342 | 640 } |
641 break; | |
642 case MKTAG('t', 'R', 'N', 'S'): | |
643 { | |
644 int v, i; | |
645 | |
646 /* read the transparency. XXX: Only palette mode supported */ | |
647 if (s->color_type != PNG_COLOR_TYPE_PALETTE || | |
648 length > 256 || | |
649 !(s->state & PNG_PLTE)) | |
650 goto skip_tag; | |
651 for(i=0;i<length;i++) { | |
652 v = *s->bytestream++; | |
653 s->palette[i] = (s->palette[i] & 0x00ffffff) | (v << 24); | |
654 } | |
5067 | 655 crc = bytestream_get_be32(&s->bytestream); |
2342 | 656 } |
657 break; | |
658 case MKTAG('I', 'E', 'N', 'D'): | |
659 if (!(s->state & PNG_ALLIMAGE)) | |
660 goto fail; | |
5067 | 661 crc = bytestream_get_be32(&s->bytestream); |
2342 | 662 goto exit_loop; |
663 default: | |
664 /* skip tag */ | |
665 skip_tag: | |
666 s->bytestream += length + 4; | |
667 break; | |
668 } | |
669 } | |
670 exit_loop: | |
671 *picture= *(AVFrame*)&s->picture; | |
672 *data_size = sizeof(AVPicture); | |
673 | |
674 ret = s->bytestream - s->bytestream_start; | |
675 the_end: | |
676 inflateEnd(&s->zstream); | |
677 av_freep(&s->crow_buf); | |
678 av_freep(&s->last_row); | |
679 av_freep(&s->tmp_row); | |
680 return ret; | |
681 fail: | |
682 ret = -1; | |
683 goto the_end; | |
684 } | |
3777 | 685 #endif |
2342 | 686 |
3777 | 687 #ifdef CONFIG_ENCODERS |
2342 | 688 static void png_write_chunk(uint8_t **f, uint32_t tag, |
689 const uint8_t *buf, int length) | |
690 { | |
691 uint32_t crc; | |
692 uint8_t tagbuf[4]; | |
693 | |
5067 | 694 bytestream_put_be32(f, length); |
2342 | 695 crc = crc32(0, Z_NULL, 0); |
696 tagbuf[0] = tag; | |
697 tagbuf[1] = tag >> 8; | |
698 tagbuf[2] = tag >> 16; | |
699 tagbuf[3] = tag >> 24; | |
700 crc = crc32(crc, tagbuf, 4); | |
5067 | 701 bytestream_put_be32(f, bswap_32(tag)); |
2342 | 702 if (length > 0) { |
703 crc = crc32(crc, buf, length); | |
704 memcpy(*f, buf, length); | |
705 *f += length; | |
706 } | |
5067 | 707 bytestream_put_be32(f, crc); |
2342 | 708 } |
709 | |
710 /* XXX: do filtering */ | |
711 static int png_write_row(PNGContext *s, const uint8_t *data, int size) | |
712 { | |
713 int ret; | |
714 | |
715 s->zstream.avail_in = size; | |
716 s->zstream.next_in = (uint8_t *)data; | |
717 while (s->zstream.avail_in > 0) { | |
718 ret = deflate(&s->zstream, Z_NO_FLUSH); | |
719 if (ret != Z_OK) | |
720 return -1; | |
721 if (s->zstream.avail_out == 0) { | |
2422 | 722 if(s->bytestream_end - s->bytestream > IOBUF_SIZE + 100) |
723 png_write_chunk(&s->bytestream, MKTAG('I', 'D', 'A', 'T'), s->buf, IOBUF_SIZE); | |
2342 | 724 s->zstream.avail_out = IOBUF_SIZE; |
725 s->zstream.next_out = s->buf; | |
726 } | |
727 } | |
728 return 0; | |
729 } | |
3777 | 730 #endif /* CONFIG_ENCODERS */ |
2342 | 731 |
732 static int common_init(AVCodecContext *avctx){ | |
733 PNGContext *s = avctx->priv_data; | |
734 | |
735 avcodec_get_frame_defaults((AVFrame*)&s->picture); | |
736 avctx->coded_frame= (AVFrame*)&s->picture; | |
737 // s->avctx= avctx; | |
738 | |
739 return 0; | |
740 } | |
741 | |
3777 | 742 #ifdef CONFIG_ENCODERS |
2342 | 743 static int encode_frame(AVCodecContext *avctx, unsigned char *buf, int buf_size, void *data){ |
744 PNGContext *s = avctx->priv_data; | |
745 AVFrame *pict = data; | |
746 AVFrame * const p= (AVFrame*)&s->picture; | |
747 int bit_depth, color_type, y, len, row_size, ret, is_progressive; | |
748 int bits_per_pixel, pass_row_size; | |
749 uint8_t *ptr; | |
750 uint8_t *crow_buf = NULL; | |
751 uint8_t *tmp_buf = NULL; | |
752 | |
753 *p = *pict; | |
754 p->pict_type= FF_I_TYPE; | |
755 p->key_frame= 1; | |
2967 | 756 |
2342 | 757 s->bytestream_start= |
758 s->bytestream= buf; | |
759 s->bytestream_end= buf+buf_size; | |
760 | |
761 is_progressive = !!(avctx->flags & CODEC_FLAG_INTERLACED_DCT); | |
762 switch(avctx->pix_fmt) { | |
4494
ce643a22f049
Replace deprecated PIX_FMT names by the newer variants.
diego
parents:
4379
diff
changeset
|
763 case PIX_FMT_RGB32: |
2342 | 764 bit_depth = 8; |
765 color_type = PNG_COLOR_TYPE_RGB_ALPHA; | |
766 break; | |
767 case PIX_FMT_RGB24: | |
768 bit_depth = 8; | |
769 color_type = PNG_COLOR_TYPE_RGB; | |
770 break; | |
771 case PIX_FMT_GRAY8: | |
772 bit_depth = 8; | |
773 color_type = PNG_COLOR_TYPE_GRAY; | |
774 break; | |
775 case PIX_FMT_MONOBLACK: | |
776 bit_depth = 1; | |
777 color_type = PNG_COLOR_TYPE_GRAY; | |
778 break; | |
779 case PIX_FMT_PAL8: | |
780 bit_depth = 8; | |
781 color_type = PNG_COLOR_TYPE_PALETTE; | |
782 break; | |
783 default: | |
784 return -1; | |
785 } | |
786 bits_per_pixel = png_get_nb_channels(color_type) * bit_depth; | |
787 row_size = (avctx->width * bits_per_pixel + 7) >> 3; | |
788 | |
789 s->zstream.zalloc = png_zalloc; | |
790 s->zstream.zfree = png_zfree; | |
791 s->zstream.opaque = NULL; | |
792 ret = deflateInit2(&s->zstream, Z_DEFAULT_COMPRESSION, | |
793 Z_DEFLATED, 15, 8, Z_DEFAULT_STRATEGY); | |
794 if (ret != Z_OK) | |
795 return -1; | |
796 crow_buf = av_malloc(row_size + 1); | |
797 if (!crow_buf) | |
798 goto fail; | |
799 if (is_progressive) { | |
800 tmp_buf = av_malloc(row_size + 1); | |
801 if (!tmp_buf) | |
802 goto fail; | |
803 } | |
804 | |
805 /* write png header */ | |
806 memcpy(s->bytestream, pngsig, 8); | |
807 s->bytestream += 8; | |
2967 | 808 |
5067 | 809 AV_WB32(s->buf, avctx->width); |
810 AV_WB32(s->buf + 4, avctx->height); | |
2342 | 811 s->buf[8] = bit_depth; |
812 s->buf[9] = color_type; | |
813 s->buf[10] = 0; /* compression type */ | |
814 s->buf[11] = 0; /* filter type */ | |
815 s->buf[12] = is_progressive; /* interlace type */ | |
2967 | 816 |
2342 | 817 png_write_chunk(&s->bytestream, MKTAG('I', 'H', 'D', 'R'), s->buf, 13); |
818 | |
819 /* put the palette if needed */ | |
820 if (color_type == PNG_COLOR_TYPE_PALETTE) { | |
821 int has_alpha, alpha, i; | |
822 unsigned int v; | |
823 uint32_t *palette; | |
824 uint8_t *alpha_ptr; | |
2967 | 825 |
2342 | 826 palette = (uint32_t *)p->data[1]; |
827 ptr = s->buf; | |
828 alpha_ptr = s->buf + 256 * 3; | |
829 has_alpha = 0; | |
830 for(i = 0; i < 256; i++) { | |
831 v = palette[i]; | |
832 alpha = v >> 24; | |
4018 | 833 if (alpha && alpha != 0xff) |
2342 | 834 has_alpha = 1; |
835 *alpha_ptr++ = alpha; | |
836 ptr[0] = v >> 16; | |
837 ptr[1] = v >> 8; | |
838 ptr[2] = v; | |
839 ptr += 3; | |
840 } | |
841 png_write_chunk(&s->bytestream, MKTAG('P', 'L', 'T', 'E'), s->buf, 256 * 3); | |
842 if (has_alpha) { | |
843 png_write_chunk(&s->bytestream, MKTAG('t', 'R', 'N', 'S'), s->buf + 256 * 3, 256); | |
844 } | |
845 } | |
846 | |
847 /* now put each row */ | |
848 s->zstream.avail_out = IOBUF_SIZE; | |
849 s->zstream.next_out = s->buf; | |
850 if (is_progressive) { | |
851 uint8_t *ptr1; | |
852 int pass; | |
853 | |
854 for(pass = 0; pass < NB_PASSES; pass++) { | |
855 /* NOTE: a pass is completely omited if no pixels would be | |
856 output */ | |
857 pass_row_size = png_pass_row_size(pass, bits_per_pixel, avctx->width); | |
858 if (pass_row_size > 0) { | |
859 for(y = 0; y < avctx->height; y++) { | |
860 if ((png_pass_ymask[pass] << (y & 7)) & 0x80) { | |
861 ptr = p->data[0] + y * p->linesize[0]; | |
862 if (color_type == PNG_COLOR_TYPE_RGB_ALPHA) { | |
4515 | 863 convert_from_rgb32(tmp_buf, ptr, avctx->width); |
2342 | 864 ptr1 = tmp_buf; |
865 } else { | |
866 ptr1 = ptr; | |
867 } | |
2967 | 868 png_get_interlaced_row(crow_buf + 1, pass_row_size, |
869 bits_per_pixel, pass, | |
2342 | 870 ptr1, avctx->width); |
871 crow_buf[0] = PNG_FILTER_VALUE_NONE; | |
872 png_write_row(s, crow_buf, pass_row_size + 1); | |
873 } | |
874 } | |
875 } | |
876 } | |
877 } else { | |
878 for(y = 0; y < avctx->height; y++) { | |
879 ptr = p->data[0] + y * p->linesize[0]; | |
880 if (color_type == PNG_COLOR_TYPE_RGB_ALPHA) | |
4515 | 881 convert_from_rgb32(crow_buf + 1, ptr, avctx->width); |
2342 | 882 else |
883 memcpy(crow_buf + 1, ptr, row_size); | |
884 crow_buf[0] = PNG_FILTER_VALUE_NONE; | |
885 png_write_row(s, crow_buf, row_size + 1); | |
886 } | |
887 } | |
888 /* compress last bytes */ | |
889 for(;;) { | |
890 ret = deflate(&s->zstream, Z_FINISH); | |
891 if (ret == Z_OK || ret == Z_STREAM_END) { | |
892 len = IOBUF_SIZE - s->zstream.avail_out; | |
2422 | 893 if (len > 0 && s->bytestream_end - s->bytestream > len + 100) { |
2342 | 894 png_write_chunk(&s->bytestream, MKTAG('I', 'D', 'A', 'T'), s->buf, len); |
895 } | |
896 s->zstream.avail_out = IOBUF_SIZE; | |
897 s->zstream.next_out = s->buf; | |
898 if (ret == Z_STREAM_END) | |
899 break; | |
900 } else { | |
901 goto fail; | |
902 } | |
903 } | |
904 png_write_chunk(&s->bytestream, MKTAG('I', 'E', 'N', 'D'), NULL, 0); | |
905 | |
906 ret = s->bytestream - s->bytestream_start; | |
907 the_end: | |
908 av_free(crow_buf); | |
909 av_free(tmp_buf); | |
910 deflateEnd(&s->zstream); | |
911 return ret; | |
912 fail: | |
913 ret = -1; | |
914 goto the_end; | |
915 } | |
3777 | 916 #endif |
2342 | 917 |
3777 | 918 #ifdef CONFIG_PNG_DECODER |
2342 | 919 AVCodec png_decoder = { |
920 "png", | |
921 CODEC_TYPE_VIDEO, | |
922 CODEC_ID_PNG, | |
923 sizeof(PNGContext), | |
924 common_init, | |
925 NULL, | |
926 NULL, //decode_end, | |
927 decode_frame, | |
928 0 /*CODEC_CAP_DR1*/ /*| CODEC_CAP_DRAW_HORIZ_BAND*/, | |
929 NULL | |
930 }; | |
3777 | 931 #endif |
2342 | 932 |
2661
b2846918585c
a few #ifdef CONFIG_X_ENCODER, patch by (Roine Gustafsson <roine users.sourceforge net]
michael
parents:
2453
diff
changeset
|
933 #ifdef CONFIG_PNG_ENCODER |
2342 | 934 AVCodec png_encoder = { |
935 "png", | |
936 CODEC_TYPE_VIDEO, | |
937 CODEC_ID_PNG, | |
938 sizeof(PNGContext), | |
939 common_init, | |
940 encode_frame, | |
941 NULL, //encode_end, | |
4494
ce643a22f049
Replace deprecated PIX_FMT names by the newer variants.
diego
parents:
4379
diff
changeset
|
942 .pix_fmts= (enum PixelFormat[]){PIX_FMT_RGB24, PIX_FMT_RGB32, PIX_FMT_PAL8, PIX_FMT_GRAY8, PIX_FMT_MONOBLACK, -1}, |
2342 | 943 }; |
2661
b2846918585c
a few #ifdef CONFIG_X_ENCODER, patch by (Roine Gustafsson <roine users.sourceforge net]
michael
parents:
2453
diff
changeset
|
944 #endif // CONFIG_PNG_ENCODER |