]> git.saurik.com Git - wxWidgets.git/blob - src/png/pngwrite.c
cleanup - reformat (take 2)
[wxWidgets.git] / src / png / pngwrite.c
1
2 /* pngwrite.c - general routines to write a PNG file
3 *
4 * libpng 1.2.7 - September 12, 2004
5 * For conditions of distribution and use, see copyright notice in png.h
6 * Copyright (c) 1998-2004 Glenn Randers-Pehrson
7 * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)
8 * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)
9 */
10
11 /* get internal access to png.h */
12 #define PNG_INTERNAL
13 #include "png.h"
14 #ifdef PNG_WRITE_SUPPORTED
15
16 /* Writes all the PNG information. This is the suggested way to use the
17 * library. If you have a new chunk to add, make a function to write it,
18 * and put it in the correct location here. If you want the chunk written
19 * after the image data, put it in png_write_end(). I strongly encourage
20 * you to supply a PNG_INFO_ flag, and check info_ptr->valid before writing
21 * the chunk, as that will keep the code from breaking if you want to just
22 * write a plain PNG file. If you have long comments, I suggest writing
23 * them in png_write_end(), and compressing them.
24 */
25 void PNGAPI
26 png_write_info_before_PLTE(png_structp png_ptr, png_infop info_ptr)
27 {
28 png_debug(1, "in png_write_info_before_PLTE\n");
29 if (!(png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE))
30 {
31 png_write_sig(png_ptr); /* write PNG signature */
32 #if defined(PNG_MNG_FEATURES_SUPPORTED)
33 if((png_ptr->mode&PNG_HAVE_PNG_SIGNATURE)&&(png_ptr->mng_features_permitted))
34 {
35 png_warning(png_ptr,"MNG features are not allowed in a PNG datastream\n");
36 png_ptr->mng_features_permitted=0;
37 }
38 #endif
39 /* write IHDR information. */
40 png_write_IHDR(png_ptr, info_ptr->width, info_ptr->height,
41 info_ptr->bit_depth, info_ptr->color_type, info_ptr->compression_type,
42 info_ptr->filter_type,
43 #if defined(PNG_WRITE_INTERLACING_SUPPORTED)
44 info_ptr->interlace_type);
45 #else
46 0);
47 #endif
48 /* the rest of these check to see if the valid field has the appropriate
49 flag set, and if it does, writes the chunk. */
50 #if defined(PNG_WRITE_gAMA_SUPPORTED)
51 if (info_ptr->valid & PNG_INFO_gAMA)
52 {
53 # ifdef PNG_FLOATING_POINT_SUPPORTED
54 png_write_gAMA(png_ptr, info_ptr->gamma);
55 #else
56 #ifdef PNG_FIXED_POINT_SUPPORTED
57 png_write_gAMA_fixed(png_ptr, info_ptr->int_gamma);
58 # endif
59 #endif
60 }
61 #endif
62 #if defined(PNG_WRITE_sRGB_SUPPORTED)
63 if (info_ptr->valid & PNG_INFO_sRGB)
64 png_write_sRGB(png_ptr, (int)info_ptr->srgb_intent);
65 #endif
66 #if defined(PNG_WRITE_iCCP_SUPPORTED)
67 if (info_ptr->valid & PNG_INFO_iCCP)
68 png_write_iCCP(png_ptr, info_ptr->iccp_name, PNG_COMPRESSION_TYPE_BASE,
69 info_ptr->iccp_profile, (int)info_ptr->iccp_proflen);
70 #endif
71 #if defined(PNG_WRITE_sBIT_SUPPORTED)
72 if (info_ptr->valid & PNG_INFO_sBIT)
73 png_write_sBIT(png_ptr, &(info_ptr->sig_bit), info_ptr->color_type);
74 #endif
75 #if defined(PNG_WRITE_cHRM_SUPPORTED)
76 if (info_ptr->valid & PNG_INFO_cHRM)
77 {
78 #ifdef PNG_FLOATING_POINT_SUPPORTED
79 png_write_cHRM(png_ptr,
80 info_ptr->x_white, info_ptr->y_white,
81 info_ptr->x_red, info_ptr->y_red,
82 info_ptr->x_green, info_ptr->y_green,
83 info_ptr->x_blue, info_ptr->y_blue);
84 #else
85 # ifdef PNG_FIXED_POINT_SUPPORTED
86 png_write_cHRM_fixed(png_ptr,
87 info_ptr->int_x_white, info_ptr->int_y_white,
88 info_ptr->int_x_red, info_ptr->int_y_red,
89 info_ptr->int_x_green, info_ptr->int_y_green,
90 info_ptr->int_x_blue, info_ptr->int_y_blue);
91 # endif
92 #endif
93 }
94 #endif
95 #if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
96 if (info_ptr->unknown_chunks_num)
97 {
98 png_unknown_chunk *up;
99
100 png_debug(5, "writing extra chunks\n");
101
102 for (up = info_ptr->unknown_chunks;
103 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
104 up++)
105 {
106 int keep=png_handle_as_unknown(png_ptr, up->name);
107 if (keep != PNG_HANDLE_CHUNK_NEVER &&
108 up->location && !(up->location & PNG_HAVE_PLTE) &&
109 !(up->location & PNG_HAVE_IDAT) &&
110 ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS ||
111 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
112 {
113 png_write_chunk(png_ptr, up->name, up->data, up->size);
114 }
115 }
116 }
117 #endif
118 png_ptr->mode |= PNG_WROTE_INFO_BEFORE_PLTE;
119 }
120 }
121
122 void PNGAPI
123 png_write_info(png_structp png_ptr, png_infop info_ptr)
124 {
125 #if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_sPLT_SUPPORTED)
126 int i;
127 #endif
128
129 png_debug(1, "in png_write_info\n");
130
131 png_write_info_before_PLTE(png_ptr, info_ptr);
132
133 if (info_ptr->valid & PNG_INFO_PLTE)
134 png_write_PLTE(png_ptr, info_ptr->palette,
135 (png_uint_32)info_ptr->num_palette);
136 else if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
137 png_error(png_ptr, "Valid palette required for paletted images\n");
138
139 #if defined(PNG_WRITE_tRNS_SUPPORTED)
140 if (info_ptr->valid & PNG_INFO_tRNS)
141 {
142 #if defined(PNG_WRITE_INVERT_ALPHA_SUPPORTED)
143 /* invert the alpha channel (in tRNS) */
144 if ((png_ptr->transformations & PNG_INVERT_ALPHA) &&
145 info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
146 {
147 int j;
148 for (j=0; j<(int)info_ptr->num_trans; j++)
149 info_ptr->trans[j] = (png_byte)(255 - info_ptr->trans[j]);
150 }
151 #endif
152 png_write_tRNS(png_ptr, info_ptr->trans, &(info_ptr->trans_values),
153 info_ptr->num_trans, info_ptr->color_type);
154 }
155 #endif
156 #if defined(PNG_WRITE_bKGD_SUPPORTED)
157 if (info_ptr->valid & PNG_INFO_bKGD)
158 png_write_bKGD(png_ptr, &(info_ptr->background), info_ptr->color_type);
159 #endif
160 #if defined(PNG_WRITE_hIST_SUPPORTED)
161 if (info_ptr->valid & PNG_INFO_hIST)
162 png_write_hIST(png_ptr, info_ptr->hist, info_ptr->num_palette);
163 #endif
164 #if defined(PNG_WRITE_oFFs_SUPPORTED)
165 if (info_ptr->valid & PNG_INFO_oFFs)
166 png_write_oFFs(png_ptr, info_ptr->x_offset, info_ptr->y_offset,
167 info_ptr->offset_unit_type);
168 #endif
169 #if defined(PNG_WRITE_pCAL_SUPPORTED)
170 if (info_ptr->valid & PNG_INFO_pCAL)
171 png_write_pCAL(png_ptr, info_ptr->pcal_purpose, info_ptr->pcal_X0,
172 info_ptr->pcal_X1, info_ptr->pcal_type, info_ptr->pcal_nparams,
173 info_ptr->pcal_units, info_ptr->pcal_params);
174 #endif
175 #if defined(PNG_WRITE_sCAL_SUPPORTED)
176 if (info_ptr->valid & PNG_INFO_sCAL)
177 #if defined(PNG_FLOATING_POINT_SUPPORTED) && !defined(PNG_NO_STDIO)
178 png_write_sCAL(png_ptr, (int)info_ptr->scal_unit,
179 info_ptr->scal_pixel_width, info_ptr->scal_pixel_height);
180 #else
181 #ifdef PNG_FIXED_POINT_SUPPORTED
182 png_write_sCAL_s(png_ptr, (int)info_ptr->scal_unit,
183 info_ptr->scal_s_width, info_ptr->scal_s_height);
184 #else
185 png_warning(png_ptr,
186 "png_write_sCAL not supported; sCAL chunk not written.\n");
187 #endif
188 #endif
189 #endif
190 #if defined(PNG_WRITE_pHYs_SUPPORTED)
191 if (info_ptr->valid & PNG_INFO_pHYs)
192 png_write_pHYs(png_ptr, info_ptr->x_pixels_per_unit,
193 info_ptr->y_pixels_per_unit, info_ptr->phys_unit_type);
194 #endif
195 #if defined(PNG_WRITE_tIME_SUPPORTED)
196 if (info_ptr->valid & PNG_INFO_tIME)
197 {
198 png_write_tIME(png_ptr, &(info_ptr->mod_time));
199 png_ptr->mode |= PNG_WROTE_tIME;
200 }
201 #endif
202 #if defined(PNG_WRITE_sPLT_SUPPORTED)
203 if (info_ptr->valid & PNG_INFO_sPLT)
204 for (i = 0; i < (int)info_ptr->splt_palettes_num; i++)
205 png_write_sPLT(png_ptr, info_ptr->splt_palettes + i);
206 #endif
207 #if defined(PNG_WRITE_TEXT_SUPPORTED)
208 /* Check to see if we need to write text chunks */
209 for (i = 0; i < info_ptr->num_text; i++)
210 {
211 png_debug2(2, "Writing header text chunk %d, type %d\n", i,
212 info_ptr->text[i].compression);
213 /* an internationalized chunk? */
214 if (info_ptr->text[i].compression > 0)
215 {
216 #if defined(PNG_WRITE_iTXt_SUPPORTED)
217 /* write international chunk */
218 png_write_iTXt(png_ptr,
219 info_ptr->text[i].compression,
220 info_ptr->text[i].key,
221 info_ptr->text[i].lang,
222 info_ptr->text[i].lang_key,
223 info_ptr->text[i].text);
224 #else
225 png_warning(png_ptr, "Unable to write international text\n");
226 #endif
227 /* Mark this chunk as written */
228 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
229 }
230 /* If we want a compressed text chunk */
231 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_zTXt)
232 {
233 #if defined(PNG_WRITE_zTXt_SUPPORTED)
234 /* write compressed chunk */
235 png_write_zTXt(png_ptr, info_ptr->text[i].key,
236 info_ptr->text[i].text, 0,
237 info_ptr->text[i].compression);
238 #else
239 png_warning(png_ptr, "Unable to write compressed text\n");
240 #endif
241 /* Mark this chunk as written */
242 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
243 }
244 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
245 {
246 #if defined(PNG_WRITE_tEXt_SUPPORTED)
247 /* write uncompressed chunk */
248 png_write_tEXt(png_ptr, info_ptr->text[i].key,
249 info_ptr->text[i].text,
250 0);
251 #else
252 png_warning(png_ptr, "Unable to write uncompressed text\n");
253 #endif
254 /* Mark this chunk as written */
255 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
256 }
257 }
258 #endif
259 #if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
260 if (info_ptr->unknown_chunks_num)
261 {
262 png_unknown_chunk *up;
263
264 png_debug(5, "writing extra chunks\n");
265
266 for (up = info_ptr->unknown_chunks;
267 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
268 up++)
269 {
270 int keep=png_handle_as_unknown(png_ptr, up->name);
271 if (keep != PNG_HANDLE_CHUNK_NEVER &&
272 up->location && (up->location & PNG_HAVE_PLTE) &&
273 !(up->location & PNG_HAVE_IDAT) &&
274 ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS ||
275 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
276 {
277 png_write_chunk(png_ptr, up->name, up->data, up->size);
278 }
279 }
280 }
281 #endif
282 }
283
284 /* Writes the end of the PNG file. If you don't want to write comments or
285 * time information, you can pass NULL for info. If you already wrote these
286 * in png_write_info(), do not write them again here. If you have long
287 * comments, I suggest writing them here, and compressing them.
288 */
289 void PNGAPI
290 png_write_end(png_structp png_ptr, png_infop info_ptr)
291 {
292 png_debug(1, "in png_write_end\n");
293 if (!(png_ptr->mode & PNG_HAVE_IDAT))
294 png_error(png_ptr, "No IDATs written into file");
295
296 /* see if user wants us to write information chunks */
297 if (info_ptr != NULL)
298 {
299 #if defined(PNG_WRITE_TEXT_SUPPORTED)
300 int i; /* local index variable */
301 #endif
302 #if defined(PNG_WRITE_tIME_SUPPORTED)
303 /* check to see if user has supplied a time chunk */
304 if ((info_ptr->valid & PNG_INFO_tIME) &&
305 !(png_ptr->mode & PNG_WROTE_tIME))
306 png_write_tIME(png_ptr, &(info_ptr->mod_time));
307 #endif
308 #if defined(PNG_WRITE_TEXT_SUPPORTED)
309 /* loop through comment chunks */
310 for (i = 0; i < info_ptr->num_text; i++)
311 {
312 png_debug2(2, "Writing trailer text chunk %d, type %d\n", i,
313 info_ptr->text[i].compression);
314 /* an internationalized chunk? */
315 if (info_ptr->text[i].compression > 0)
316 {
317 #if defined(PNG_WRITE_iTXt_SUPPORTED)
318 /* write international chunk */
319 png_write_iTXt(png_ptr,
320 info_ptr->text[i].compression,
321 info_ptr->text[i].key,
322 info_ptr->text[i].lang,
323 info_ptr->text[i].lang_key,
324 info_ptr->text[i].text);
325 #else
326 png_warning(png_ptr, "Unable to write international text\n");
327 #endif
328 /* Mark this chunk as written */
329 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
330 }
331 else if (info_ptr->text[i].compression >= PNG_TEXT_COMPRESSION_zTXt)
332 {
333 #if defined(PNG_WRITE_zTXt_SUPPORTED)
334 /* write compressed chunk */
335 png_write_zTXt(png_ptr, info_ptr->text[i].key,
336 info_ptr->text[i].text, 0,
337 info_ptr->text[i].compression);
338 #else
339 png_warning(png_ptr, "Unable to write compressed text\n");
340 #endif
341 /* Mark this chunk as written */
342 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
343 }
344 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
345 {
346 #if defined(PNG_WRITE_tEXt_SUPPORTED)
347 /* write uncompressed chunk */
348 png_write_tEXt(png_ptr, info_ptr->text[i].key,
349 info_ptr->text[i].text, 0);
350 #else
351 png_warning(png_ptr, "Unable to write uncompressed text\n");
352 #endif
353
354 /* Mark this chunk as written */
355 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
356 }
357 }
358 #endif
359 #if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
360 if (info_ptr->unknown_chunks_num)
361 {
362 png_unknown_chunk *up;
363
364 png_debug(5, "writing extra chunks\n");
365
366 for (up = info_ptr->unknown_chunks;
367 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
368 up++)
369 {
370 int keep=png_handle_as_unknown(png_ptr, up->name);
371 if (keep != PNG_HANDLE_CHUNK_NEVER &&
372 up->location && (up->location & PNG_AFTER_IDAT) &&
373 ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS ||
374 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
375 {
376 png_write_chunk(png_ptr, up->name, up->data, up->size);
377 }
378 }
379 }
380 #endif
381 }
382
383 png_ptr->mode |= PNG_AFTER_IDAT;
384
385 /* write end of PNG file */
386 png_write_IEND(png_ptr);
387 #if 0
388 /* This flush, added in libpng-1.0.8, causes some applications to crash
389 because they do not set png_ptr->output_flush_fn */
390 png_flush(png_ptr);
391 #endif
392 }
393
394 #if defined(PNG_WRITE_tIME_SUPPORTED)
395 #if !defined(_WIN32_WCE)
396 /* "time.h" functions are not supported on WindowsCE */
397 void PNGAPI
398 png_convert_from_struct_tm(png_timep ptime, struct tm FAR * ttime)
399 {
400 png_debug(1, "in png_convert_from_struct_tm\n");
401 ptime->year = (png_uint_16)(1900 + ttime->tm_year);
402 ptime->month = (png_byte)(ttime->tm_mon + 1);
403 ptime->day = (png_byte)ttime->tm_mday;
404 ptime->hour = (png_byte)ttime->tm_hour;
405 ptime->minute = (png_byte)ttime->tm_min;
406 ptime->second = (png_byte)ttime->tm_sec;
407 }
408
409 void PNGAPI
410 png_convert_from_time_t(png_timep ptime, time_t ttime)
411 {
412 struct tm *tbuf;
413
414 png_debug(1, "in png_convert_from_time_t\n");
415 tbuf = gmtime(&ttime);
416 png_convert_from_struct_tm(ptime, tbuf);
417 }
418 #endif
419 #endif
420
421 /* Initialize png_ptr structure, and allocate any memory needed */
422 png_structp PNGAPI
423 png_create_write_struct(png_const_charp user_png_ver, png_voidp error_ptr,
424 png_error_ptr error_fn, png_error_ptr warn_fn)
425 {
426 #ifdef PNG_USER_MEM_SUPPORTED
427 return (png_create_write_struct_2(user_png_ver, error_ptr, error_fn,
428 warn_fn, png_voidp_NULL, png_malloc_ptr_NULL, png_free_ptr_NULL));
429 }
430
431 /* Alternate initialize png_ptr structure, and allocate any memory needed */
432 png_structp PNGAPI
433 png_create_write_struct_2(png_const_charp user_png_ver, png_voidp error_ptr,
434 png_error_ptr error_fn, png_error_ptr warn_fn, png_voidp mem_ptr,
435 png_malloc_ptr malloc_fn, png_free_ptr free_fn)
436 {
437 #endif /* PNG_USER_MEM_SUPPORTED */
438 png_structp png_ptr;
439 #ifdef PNG_SETJMP_SUPPORTED
440 #ifdef USE_FAR_KEYWORD
441 jmp_buf jmpbuf;
442 #endif
443 #endif
444 int i;
445 png_debug(1, "in png_create_write_struct\n");
446 #ifdef PNG_USER_MEM_SUPPORTED
447 png_ptr = (png_structp)png_create_struct_2(PNG_STRUCT_PNG,
448 (png_malloc_ptr)malloc_fn, (png_voidp)mem_ptr);
449 #else
450 png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG);
451 #endif /* PNG_USER_MEM_SUPPORTED */
452 if (png_ptr == NULL)
453 return (NULL);
454
455 #if !defined(PNG_1_0_X)
456 #ifdef PNG_ASSEMBLER_CODE_SUPPORTED
457 png_init_mmx_flags(png_ptr); /* 1.2.0 addition */
458 #endif
459 #endif /* PNG_1_0_X */
460
461 /* added at libpng-1.2.6 */
462 #ifdef PNG_SET_USER_LIMITS_SUPPORTED
463 png_ptr->user_width_max=PNG_USER_WIDTH_MAX;
464 png_ptr->user_height_max=PNG_USER_HEIGHT_MAX;
465 #endif
466
467 #ifdef PNG_SETJMP_SUPPORTED
468 #ifdef USE_FAR_KEYWORD
469 if (setjmp(jmpbuf))
470 #else
471 if (setjmp(png_ptr->jmpbuf))
472 #endif
473 {
474 png_free(png_ptr, png_ptr->zbuf);
475 png_ptr->zbuf=NULL;
476 png_destroy_struct(png_ptr);
477 return (NULL);
478 }
479 #ifdef USE_FAR_KEYWORD
480 png_memcpy(png_ptr->jmpbuf,jmpbuf,png_sizeof(jmp_buf));
481 #endif
482 #endif
483
484 #ifdef PNG_USER_MEM_SUPPORTED
485 png_set_mem_fn(png_ptr, mem_ptr, malloc_fn, free_fn);
486 #endif /* PNG_USER_MEM_SUPPORTED */
487 png_set_error_fn(png_ptr, error_ptr, error_fn, warn_fn);
488
489 i=0;
490 do
491 {
492 if(user_png_ver[i] != png_libpng_ver[i])
493 png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH;
494 } while (png_libpng_ver[i++]);
495
496 if (png_ptr->flags & PNG_FLAG_LIBRARY_MISMATCH)
497 {
498 /* Libpng 0.90 and later are binary incompatible with libpng 0.89, so
499 * we must recompile any applications that use any older library version.
500 * For versions after libpng 1.0, we will be compatible, so we need
501 * only check the first digit.
502 */
503 if (user_png_ver == NULL || user_png_ver[0] != png_libpng_ver[0] ||
504 (user_png_ver[0] == '1' && user_png_ver[2] != png_libpng_ver[2]) ||
505 (user_png_ver[0] == '0' && user_png_ver[2] < '9'))
506 {
507 #if !defined(PNG_NO_STDIO) && !defined(_WIN32_WCE)
508 char msg[80];
509 if (user_png_ver)
510 {
511 sprintf(msg, "Application was compiled with png.h from libpng-%.20s",
512 user_png_ver);
513 png_warning(png_ptr, msg);
514 }
515 sprintf(msg, "Application is running with png.c from libpng-%.20s",
516 png_libpng_ver);
517 png_warning(png_ptr, msg);
518 #endif
519 #ifdef PNG_ERROR_NUMBERS_SUPPORTED
520 png_ptr->flags=0;
521 #endif
522 png_error(png_ptr,
523 "Incompatible libpng version in application and library");
524 }
525 }
526
527 /* initialize zbuf - compression buffer */
528 png_ptr->zbuf_size = PNG_ZBUF_SIZE;
529 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr,
530 (png_uint_32)png_ptr->zbuf_size);
531
532 png_set_write_fn(png_ptr, png_voidp_NULL, png_rw_ptr_NULL,
533 png_flush_ptr_NULL);
534
535 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
536 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
537 1, png_doublep_NULL, png_doublep_NULL);
538 #endif
539
540 #ifdef PNG_SETJMP_SUPPORTED
541 /* Applications that neglect to set up their own setjmp() and then encounter
542 a png_error() will longjmp here. Since the jmpbuf is then meaningless we
543 abort instead of returning. */
544 #ifdef USE_FAR_KEYWORD
545 if (setjmp(jmpbuf))
546 PNG_ABORT();
547 png_memcpy(png_ptr->jmpbuf,jmpbuf,png_sizeof(jmp_buf));
548 #else
549 if (setjmp(png_ptr->jmpbuf))
550 PNG_ABORT();
551 #endif
552 #endif
553 return (png_ptr);
554 }
555
556 /* Initialize png_ptr structure, and allocate any memory needed */
557 #undef png_write_init
558 void PNGAPI
559 png_write_init(png_structp png_ptr)
560 {
561 /* We only come here via pre-1.0.7-compiled applications */
562 png_write_init_2(png_ptr, "1.0.6 or earlier", 0, 0);
563 }
564
565 void PNGAPI
566 png_write_init_2(png_structp png_ptr, png_const_charp user_png_ver,
567 png_size_t png_struct_size, png_size_t png_info_size)
568 {
569 /* We only come here via pre-1.0.12-compiled applications */
570 #if !defined(PNG_NO_STDIO) && !defined(_WIN32_WCE)
571 if(png_sizeof(png_struct) > png_struct_size ||
572 png_sizeof(png_info) > png_info_size)
573 {
574 char msg[80];
575 png_ptr->warning_fn=NULL;
576 if (user_png_ver)
577 {
578 sprintf(msg, "Application was compiled with png.h from libpng-%.20s",
579 user_png_ver);
580 png_warning(png_ptr, msg);
581 }
582 sprintf(msg, "Application is running with png.c from libpng-%.20s",
583 png_libpng_ver);
584 png_warning(png_ptr, msg);
585 }
586 #endif
587 if(png_sizeof(png_struct) > png_struct_size)
588 {
589 png_ptr->error_fn=NULL;
590 #ifdef PNG_ERROR_NUMBERS_SUPPORTED
591 png_ptr->flags=0;
592 #endif
593 png_error(png_ptr,
594 "The png struct allocated by the application for writing is too small.");
595 }
596 if(png_sizeof(png_info) > png_info_size)
597 {
598 png_ptr->error_fn=NULL;
599 #ifdef PNG_ERROR_NUMBERS_SUPPORTED
600 png_ptr->flags=0;
601 #endif
602 png_error(png_ptr,
603 "The info struct allocated by the application for writing is too small.");
604 }
605 png_write_init_3(&png_ptr, user_png_ver, png_struct_size);
606 }
607
608
609 void PNGAPI
610 png_write_init_3(png_structpp ptr_ptr, png_const_charp user_png_ver,
611 png_size_t png_struct_size)
612 {
613 png_structp png_ptr=*ptr_ptr;
614 #ifdef PNG_SETJMP_SUPPORTED
615 jmp_buf tmp_jmp; /* to save current jump buffer */
616 #endif
617 int i = 0;
618 do
619 {
620 if (user_png_ver[i] != png_libpng_ver[i])
621 {
622 #ifdef PNG_LEGACY_SUPPORTED
623 png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH;
624 #else
625 png_ptr->warning_fn=NULL;
626 png_warning(png_ptr,
627 "Application uses deprecated png_write_init() and should be recompiled.");
628 break;
629 #endif
630 }
631 } while (png_libpng_ver[i++]);
632
633 png_debug(1, "in png_write_init_3\n");
634
635 #ifdef PNG_SETJMP_SUPPORTED
636 /* save jump buffer and error functions */
637 png_memcpy(tmp_jmp, png_ptr->jmpbuf, png_sizeof (jmp_buf));
638 #endif
639
640 if (png_sizeof(png_struct) > png_struct_size)
641 {
642 png_destroy_struct(png_ptr);
643 png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG);
644 *ptr_ptr = png_ptr;
645 }
646
647 /* reset all variables to 0 */
648 png_memset(png_ptr, 0, png_sizeof (png_struct));
649
650 /* added at libpng-1.2.6 */
651 #ifdef PNG_SET_USER_LIMITS_SUPPORTED
652 png_ptr->user_width_max=PNG_USER_WIDTH_MAX;
653 png_ptr->user_height_max=PNG_USER_HEIGHT_MAX;
654 #endif
655
656 #if !defined(PNG_1_0_X)
657 #ifdef PNG_ASSEMBLER_CODE_SUPPORTED
658 png_init_mmx_flags(png_ptr); /* 1.2.0 addition */
659 #endif
660 #endif /* PNG_1_0_X */
661
662 #ifdef PNG_SETJMP_SUPPORTED
663 /* restore jump buffer */
664 png_memcpy(png_ptr->jmpbuf, tmp_jmp, png_sizeof (jmp_buf));
665 #endif
666
667 png_set_write_fn(png_ptr, png_voidp_NULL, png_rw_ptr_NULL,
668 png_flush_ptr_NULL);
669
670 /* initialize zbuf - compression buffer */
671 png_ptr->zbuf_size = PNG_ZBUF_SIZE;
672 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr,
673 (png_uint_32)png_ptr->zbuf_size);
674
675 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
676 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
677 1, png_doublep_NULL, png_doublep_NULL);
678 #endif
679 }
680
681 /* Write a few rows of image data. If the image is interlaced,
682 * either you will have to write the 7 sub images, or, if you
683 * have called png_set_interlace_handling(), you will have to
684 * "write" the image seven times.
685 */
686 void PNGAPI
687 png_write_rows(png_structp png_ptr, png_bytepp row,
688 png_uint_32 num_rows)
689 {
690 png_uint_32 i; /* row counter */
691 png_bytepp rp; /* row pointer */
692
693 png_debug(1, "in png_write_rows\n");
694 /* loop through the rows */
695 for (i = 0, rp = row; i < num_rows; i++, rp++)
696 {
697 png_write_row(png_ptr, *rp);
698 }
699 }
700
701 /* Write the image. You only need to call this function once, even
702 * if you are writing an interlaced image.
703 */
704 void PNGAPI
705 png_write_image(png_structp png_ptr, png_bytepp image)
706 {
707 png_uint_32 i; /* row index */
708 int pass, num_pass; /* pass variables */
709 png_bytepp rp; /* points to current row */
710
711 png_debug(1, "in png_write_image\n");
712 #if defined(PNG_WRITE_INTERLACING_SUPPORTED)
713 /* intialize interlace handling. If image is not interlaced,
714 this will set pass to 1 */
715 num_pass = png_set_interlace_handling(png_ptr);
716 #else
717 num_pass = 1;
718 #endif
719 /* loop through passes */
720 for (pass = 0; pass < num_pass; pass++)
721 {
722 /* loop through image */
723 for (i = 0, rp = image; i < png_ptr->height; i++, rp++)
724 {
725 png_write_row(png_ptr, *rp);
726 }
727 }
728 }
729
730 /* called by user to write a row of image data */
731 void PNGAPI
732 png_write_row(png_structp png_ptr, png_bytep row)
733 {
734 png_debug2(1, "in png_write_row (row %ld, pass %d)\n",
735 png_ptr->row_number, png_ptr->pass);
736 /* initialize transformations and other stuff if first time */
737 if (png_ptr->row_number == 0 && png_ptr->pass == 0)
738 {
739 /* make sure we wrote the header info */
740 if (!(png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE))
741 png_error(png_ptr,
742 "png_write_info was never called before png_write_row.");
743
744 /* check for transforms that have been set but were defined out */
745 #if !defined(PNG_WRITE_INVERT_SUPPORTED) && defined(PNG_READ_INVERT_SUPPORTED)
746 if (png_ptr->transformations & PNG_INVERT_MONO)
747 png_warning(png_ptr, "PNG_WRITE_INVERT_SUPPORTED is not defined.");
748 #endif
749 #if !defined(PNG_WRITE_FILLER_SUPPORTED) && defined(PNG_READ_FILLER_SUPPORTED)
750 if (png_ptr->transformations & PNG_FILLER)
751 png_warning(png_ptr, "PNG_WRITE_FILLER_SUPPORTED is not defined.");
752 #endif
753 #if !defined(PNG_WRITE_PACKSWAP_SUPPORTED) && defined(PNG_READ_PACKSWAP_SUPPORTED)
754 if (png_ptr->transformations & PNG_PACKSWAP)
755 png_warning(png_ptr, "PNG_WRITE_PACKSWAP_SUPPORTED is not defined.");
756 #endif
757 #if !defined(PNG_WRITE_PACK_SUPPORTED) && defined(PNG_READ_PACK_SUPPORTED)
758 if (png_ptr->transformations & PNG_PACK)
759 png_warning(png_ptr, "PNG_WRITE_PACK_SUPPORTED is not defined.");
760 #endif
761 #if !defined(PNG_WRITE_SHIFT_SUPPORTED) && defined(PNG_READ_SHIFT_SUPPORTED)
762 if (png_ptr->transformations & PNG_SHIFT)
763 png_warning(png_ptr, "PNG_WRITE_SHIFT_SUPPORTED is not defined.");
764 #endif
765 #if !defined(PNG_WRITE_BGR_SUPPORTED) && defined(PNG_READ_BGR_SUPPORTED)
766 if (png_ptr->transformations & PNG_BGR)
767 png_warning(png_ptr, "PNG_WRITE_BGR_SUPPORTED is not defined.");
768 #endif
769 #if !defined(PNG_WRITE_SWAP_SUPPORTED) && defined(PNG_READ_SWAP_SUPPORTED)
770 if (png_ptr->transformations & PNG_SWAP_BYTES)
771 png_warning(png_ptr, "PNG_WRITE_SWAP_SUPPORTED is not defined.");
772 #endif
773
774 png_write_start_row(png_ptr);
775 }
776
777 #if defined(PNG_WRITE_INTERLACING_SUPPORTED)
778 /* if interlaced and not interested in row, return */
779 if (png_ptr->interlaced && (png_ptr->transformations & PNG_INTERLACE))
780 {
781 switch (png_ptr->pass)
782 {
783 case 0:
784 if (png_ptr->row_number & 0x07)
785 {
786 png_write_finish_row(png_ptr);
787 return;
788 }
789 break;
790 case 1:
791 if ((png_ptr->row_number & 0x07) || png_ptr->width < 5)
792 {
793 png_write_finish_row(png_ptr);
794 return;
795 }
796 break;
797 case 2:
798 if ((png_ptr->row_number & 0x07) != 4)
799 {
800 png_write_finish_row(png_ptr);
801 return;
802 }
803 break;
804 case 3:
805 if ((png_ptr->row_number & 0x03) || png_ptr->width < 3)
806 {
807 png_write_finish_row(png_ptr);
808 return;
809 }
810 break;
811 case 4:
812 if ((png_ptr->row_number & 0x03) != 2)
813 {
814 png_write_finish_row(png_ptr);
815 return;
816 }
817 break;
818 case 5:
819 if ((png_ptr->row_number & 0x01) || png_ptr->width < 2)
820 {
821 png_write_finish_row(png_ptr);
822 return;
823 }
824 break;
825 case 6:
826 if (!(png_ptr->row_number & 0x01))
827 {
828 png_write_finish_row(png_ptr);
829 return;
830 }
831 break;
832 }
833 }
834 #endif
835
836 /* set up row info for transformations */
837 png_ptr->row_info.color_type = png_ptr->color_type;
838 png_ptr->row_info.width = png_ptr->usr_width;
839 png_ptr->row_info.channels = png_ptr->usr_channels;
840 png_ptr->row_info.bit_depth = png_ptr->usr_bit_depth;
841 png_ptr->row_info.pixel_depth = (png_byte)(png_ptr->row_info.bit_depth *
842 png_ptr->row_info.channels);
843
844 png_ptr->row_info.rowbytes = PNG_ROWBYTES(png_ptr->row_info.pixel_depth,
845 png_ptr->row_info.width);
846
847 png_debug1(3, "row_info->color_type = %d\n", png_ptr->row_info.color_type);
848 png_debug1(3, "row_info->width = %lu\n", png_ptr->row_info.width);
849 png_debug1(3, "row_info->channels = %d\n", png_ptr->row_info.channels);
850 png_debug1(3, "row_info->bit_depth = %d\n", png_ptr->row_info.bit_depth);
851 png_debug1(3, "row_info->pixel_depth = %d\n", png_ptr->row_info.pixel_depth);
852 png_debug1(3, "row_info->rowbytes = %lu\n", png_ptr->row_info.rowbytes);
853
854 /* Copy user's row into buffer, leaving room for filter byte. */
855 png_memcpy_check(png_ptr, png_ptr->row_buf + 1, row,
856 png_ptr->row_info.rowbytes);
857
858 #if defined(PNG_WRITE_INTERLACING_SUPPORTED)
859 /* handle interlacing */
860 if (png_ptr->interlaced && png_ptr->pass < 6 &&
861 (png_ptr->transformations & PNG_INTERLACE))
862 {
863 png_do_write_interlace(&(png_ptr->row_info),
864 png_ptr->row_buf + 1, png_ptr->pass);
865 /* this should always get caught above, but still ... */
866 if (!(png_ptr->row_info.width))
867 {
868 png_write_finish_row(png_ptr);
869 return;
870 }
871 }
872 #endif
873
874 /* handle other transformations */
875 if (png_ptr->transformations)
876 png_do_write_transformations(png_ptr);
877
878 #if defined(PNG_MNG_FEATURES_SUPPORTED)
879 /* Write filter_method 64 (intrapixel differencing) only if
880 * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and
881 * 2. Libpng did not write a PNG signature (this filter_method is only
882 * used in PNG datastreams that are embedded in MNG datastreams) and
883 * 3. The application called png_permit_mng_features with a mask that
884 * included PNG_FLAG_MNG_FILTER_64 and
885 * 4. The filter_method is 64 and
886 * 5. The color_type is RGB or RGBA
887 */
888 if((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
889 (png_ptr->filter_type == PNG_INTRAPIXEL_DIFFERENCING))
890 {
891 /* Intrapixel differencing */
892 png_do_write_intrapixel(&(png_ptr->row_info), png_ptr->row_buf + 1);
893 }
894 #endif
895
896 /* Find a filter if necessary, filter the row and write it out. */
897 png_write_find_filter(png_ptr, &(png_ptr->row_info));
898
899 if (png_ptr->write_row_fn != NULL)
900 (*(png_ptr->write_row_fn))(png_ptr, png_ptr->row_number, png_ptr->pass);
901 }
902
903 #if defined(PNG_WRITE_FLUSH_SUPPORTED)
904 /* Set the automatic flush interval or 0 to turn flushing off */
905 void PNGAPI
906 png_set_flush(png_structp png_ptr, int nrows)
907 {
908 png_debug(1, "in png_set_flush\n");
909 png_ptr->flush_dist = (nrows < 0 ? 0 : nrows);
910 }
911
912 /* flush the current output buffers now */
913 void PNGAPI
914 png_write_flush(png_structp png_ptr)
915 {
916 int wrote_IDAT;
917
918 png_debug(1, "in png_write_flush\n");
919 /* We have already written out all of the data */
920 if (png_ptr->row_number >= png_ptr->num_rows)
921 return;
922
923 do
924 {
925 int ret;
926
927 /* compress the data */
928 ret = deflate(&png_ptr->zstream, Z_SYNC_FLUSH);
929 wrote_IDAT = 0;
930
931 /* check for compression errors */
932 if (ret != Z_OK)
933 {
934 if (png_ptr->zstream.msg != NULL)
935 png_error(png_ptr, png_ptr->zstream.msg);
936 else
937 png_error(png_ptr, "zlib error");
938 }
939
940 if (!(png_ptr->zstream.avail_out))
941 {
942 /* write the IDAT and reset the zlib output buffer */
943 png_write_IDAT(png_ptr, png_ptr->zbuf,
944 png_ptr->zbuf_size);
945 png_ptr->zstream.next_out = png_ptr->zbuf;
946 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
947 wrote_IDAT = 1;
948 }
949 } while(wrote_IDAT == 1);
950
951 /* If there is any data left to be output, write it into a new IDAT */
952 if (png_ptr->zbuf_size != png_ptr->zstream.avail_out)
953 {
954 /* write the IDAT and reset the zlib output buffer */
955 png_write_IDAT(png_ptr, png_ptr->zbuf,
956 png_ptr->zbuf_size - png_ptr->zstream.avail_out);
957 png_ptr->zstream.next_out = png_ptr->zbuf;
958 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
959 }
960 png_ptr->flush_rows = 0;
961 png_flush(png_ptr);
962 }
963 #endif /* PNG_WRITE_FLUSH_SUPPORTED */
964
965 /* free all memory used by the write */
966 void PNGAPI
967 png_destroy_write_struct(png_structpp png_ptr_ptr, png_infopp info_ptr_ptr)
968 {
969 png_structp png_ptr = NULL;
970 png_infop info_ptr = NULL;
971 #ifdef PNG_USER_MEM_SUPPORTED
972 png_free_ptr free_fn = NULL;
973 png_voidp mem_ptr = NULL;
974 #endif
975
976 png_debug(1, "in png_destroy_write_struct\n");
977 if (png_ptr_ptr != NULL)
978 {
979 png_ptr = *png_ptr_ptr;
980 #ifdef PNG_USER_MEM_SUPPORTED
981 free_fn = png_ptr->free_fn;
982 mem_ptr = png_ptr->mem_ptr;
983 #endif
984 }
985
986 if (info_ptr_ptr != NULL)
987 info_ptr = *info_ptr_ptr;
988
989 if (info_ptr != NULL)
990 {
991 png_free_data(png_ptr, info_ptr, PNG_FREE_ALL, -1);
992
993 #if defined(PNG_UNKNOWN_CHUNKS_SUPPORTED)
994 if (png_ptr->num_chunk_list)
995 {
996 png_free(png_ptr, png_ptr->chunk_list);
997 png_ptr->chunk_list=NULL;
998 png_ptr->num_chunk_list=0;
999 }
1000 #endif
1001
1002 #ifdef PNG_USER_MEM_SUPPORTED
1003 png_destroy_struct_2((png_voidp)info_ptr, (png_free_ptr)free_fn,
1004 (png_voidp)mem_ptr);
1005 #else
1006 png_destroy_struct((png_voidp)info_ptr);
1007 #endif
1008 *info_ptr_ptr = NULL;
1009 }
1010
1011 if (png_ptr != NULL)
1012 {
1013 png_write_destroy(png_ptr);
1014 #ifdef PNG_USER_MEM_SUPPORTED
1015 png_destroy_struct_2((png_voidp)png_ptr, (png_free_ptr)free_fn,
1016 (png_voidp)mem_ptr);
1017 #else
1018 png_destroy_struct((png_voidp)png_ptr);
1019 #endif
1020 *png_ptr_ptr = NULL;
1021 }
1022 }
1023
1024
1025 /* Free any memory used in png_ptr struct (old method) */
1026 void /* PRIVATE */
1027 png_write_destroy(png_structp png_ptr)
1028 {
1029 #ifdef PNG_SETJMP_SUPPORTED
1030 jmp_buf tmp_jmp; /* save jump buffer */
1031 #endif
1032 png_error_ptr error_fn;
1033 png_error_ptr warning_fn;
1034 png_voidp error_ptr;
1035 #ifdef PNG_USER_MEM_SUPPORTED
1036 png_free_ptr free_fn;
1037 #endif
1038
1039 png_debug(1, "in png_write_destroy\n");
1040 /* free any memory zlib uses */
1041 deflateEnd(&png_ptr->zstream);
1042
1043 /* free our memory. png_free checks NULL for us. */
1044 png_free(png_ptr, png_ptr->zbuf);
1045 png_free(png_ptr, png_ptr->row_buf);
1046 png_free(png_ptr, png_ptr->prev_row);
1047 png_free(png_ptr, png_ptr->sub_row);
1048 png_free(png_ptr, png_ptr->up_row);
1049 png_free(png_ptr, png_ptr->avg_row);
1050 png_free(png_ptr, png_ptr->paeth_row);
1051
1052 #if defined(PNG_TIME_RFC1123_SUPPORTED)
1053 png_free(png_ptr, png_ptr->time_buffer);
1054 #endif
1055
1056 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
1057 png_free(png_ptr, png_ptr->prev_filters);
1058 png_free(png_ptr, png_ptr->filter_weights);
1059 png_free(png_ptr, png_ptr->inv_filter_weights);
1060 png_free(png_ptr, png_ptr->filter_costs);
1061 png_free(png_ptr, png_ptr->inv_filter_costs);
1062 #endif
1063
1064 #ifdef PNG_SETJMP_SUPPORTED
1065 /* reset structure */
1066 png_memcpy(tmp_jmp, png_ptr->jmpbuf, png_sizeof (jmp_buf));
1067 #endif
1068
1069 error_fn = png_ptr->error_fn;
1070 warning_fn = png_ptr->warning_fn;
1071 error_ptr = png_ptr->error_ptr;
1072 #ifdef PNG_USER_MEM_SUPPORTED
1073 free_fn = png_ptr->free_fn;
1074 #endif
1075
1076 png_memset(png_ptr, 0, png_sizeof (png_struct));
1077
1078 png_ptr->error_fn = error_fn;
1079 png_ptr->warning_fn = warning_fn;
1080 png_ptr->error_ptr = error_ptr;
1081 #ifdef PNG_USER_MEM_SUPPORTED
1082 png_ptr->free_fn = free_fn;
1083 #endif
1084
1085 #ifdef PNG_SETJMP_SUPPORTED
1086 png_memcpy(png_ptr->jmpbuf, tmp_jmp, png_sizeof (jmp_buf));
1087 #endif
1088 }
1089
1090 /* Allow the application to select one or more row filters to use. */
1091 void PNGAPI
1092 png_set_filter(png_structp png_ptr, int method, int filters)
1093 {
1094 png_debug(1, "in png_set_filter\n");
1095 #if defined(PNG_MNG_FEATURES_SUPPORTED)
1096 if((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
1097 (method == PNG_INTRAPIXEL_DIFFERENCING))
1098 method = PNG_FILTER_TYPE_BASE;
1099 #endif
1100 if (method == PNG_FILTER_TYPE_BASE)
1101 {
1102 switch (filters & (PNG_ALL_FILTERS | 0x07))
1103 {
1104 case 5:
1105 case 6:
1106 case 7: png_warning(png_ptr, "Unknown row filter for method 0");
1107 case PNG_FILTER_VALUE_NONE: png_ptr->do_filter=PNG_FILTER_NONE; break;
1108 case PNG_FILTER_VALUE_SUB: png_ptr->do_filter=PNG_FILTER_SUB; break;
1109 case PNG_FILTER_VALUE_UP: png_ptr->do_filter=PNG_FILTER_UP; break;
1110 case PNG_FILTER_VALUE_AVG: png_ptr->do_filter=PNG_FILTER_AVG; break;
1111 case PNG_FILTER_VALUE_PAETH: png_ptr->do_filter=PNG_FILTER_PAETH;break;
1112 default: png_ptr->do_filter = (png_byte)filters; break;
1113 }
1114
1115 /* If we have allocated the row_buf, this means we have already started
1116 * with the image and we should have allocated all of the filter buffers
1117 * that have been selected. If prev_row isn't already allocated, then
1118 * it is too late to start using the filters that need it, since we
1119 * will be missing the data in the previous row. If an application
1120 * wants to start and stop using particular filters during compression,
1121 * it should start out with all of the filters, and then add and
1122 * remove them after the start of compression.
1123 */
1124 if (png_ptr->row_buf != NULL)
1125 {
1126 if ((png_ptr->do_filter & PNG_FILTER_SUB) && png_ptr->sub_row == NULL)
1127 {
1128 png_ptr->sub_row = (png_bytep)png_malloc(png_ptr,
1129 (png_ptr->rowbytes + 1));
1130 png_ptr->sub_row[0] = PNG_FILTER_VALUE_SUB;
1131 }
1132
1133 if ((png_ptr->do_filter & PNG_FILTER_UP) && png_ptr->up_row == NULL)
1134 {
1135 if (png_ptr->prev_row == NULL)
1136 {
1137 png_warning(png_ptr, "Can't add Up filter after starting");
1138 png_ptr->do_filter &= ~PNG_FILTER_UP;
1139 }
1140 else
1141 {
1142 png_ptr->up_row = (png_bytep)png_malloc(png_ptr,
1143 (png_ptr->rowbytes + 1));
1144 png_ptr->up_row[0] = PNG_FILTER_VALUE_UP;
1145 }
1146 }
1147
1148 if ((png_ptr->do_filter & PNG_FILTER_AVG) && png_ptr->avg_row == NULL)
1149 {
1150 if (png_ptr->prev_row == NULL)
1151 {
1152 png_warning(png_ptr, "Can't add Average filter after starting");
1153 png_ptr->do_filter &= ~PNG_FILTER_AVG;
1154 }
1155 else
1156 {
1157 png_ptr->avg_row = (png_bytep)png_malloc(png_ptr,
1158 (png_ptr->rowbytes + 1));
1159 png_ptr->avg_row[0] = PNG_FILTER_VALUE_AVG;
1160 }
1161 }
1162
1163 if ((png_ptr->do_filter & PNG_FILTER_PAETH) &&
1164 png_ptr->paeth_row == NULL)
1165 {
1166 if (png_ptr->prev_row == NULL)
1167 {
1168 png_warning(png_ptr, "Can't add Paeth filter after starting");
1169 png_ptr->do_filter &= (png_byte)(~PNG_FILTER_PAETH);
1170 }
1171 else
1172 {
1173 png_ptr->paeth_row = (png_bytep)png_malloc(png_ptr,
1174 (png_ptr->rowbytes + 1));
1175 png_ptr->paeth_row[0] = PNG_FILTER_VALUE_PAETH;
1176 }
1177 }
1178
1179 if (png_ptr->do_filter == PNG_NO_FILTERS)
1180 png_ptr->do_filter = PNG_FILTER_NONE;
1181 }
1182 }
1183 else
1184 png_error(png_ptr, "Unknown custom filter method");
1185 }
1186
1187 /* This allows us to influence the way in which libpng chooses the "best"
1188 * filter for the current scanline. While the "minimum-sum-of-absolute-
1189 * differences metric is relatively fast and effective, there is some
1190 * question as to whether it can be improved upon by trying to keep the
1191 * filtered data going to zlib more consistent, hopefully resulting in
1192 * better compression.
1193 */
1194 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED) /* GRR 970116 */
1195 void PNGAPI
1196 png_set_filter_heuristics(png_structp png_ptr, int heuristic_method,
1197 int num_weights, png_doublep filter_weights,
1198 png_doublep filter_costs)
1199 {
1200 int i;
1201
1202 png_debug(1, "in png_set_filter_heuristics\n");
1203 if (heuristic_method >= PNG_FILTER_HEURISTIC_LAST)
1204 {
1205 png_warning(png_ptr, "Unknown filter heuristic method");
1206 return;
1207 }
1208
1209 if (heuristic_method == PNG_FILTER_HEURISTIC_DEFAULT)
1210 {
1211 heuristic_method = PNG_FILTER_HEURISTIC_UNWEIGHTED;
1212 }
1213
1214 if (num_weights < 0 || filter_weights == NULL ||
1215 heuristic_method == PNG_FILTER_HEURISTIC_UNWEIGHTED)
1216 {
1217 num_weights = 0;
1218 }
1219
1220 png_ptr->num_prev_filters = (png_byte)num_weights;
1221 png_ptr->heuristic_method = (png_byte)heuristic_method;
1222
1223 if (num_weights > 0)
1224 {
1225 if (png_ptr->prev_filters == NULL)
1226 {
1227 png_ptr->prev_filters = (png_bytep)png_malloc(png_ptr,
1228 (png_uint_32)(png_sizeof(png_byte) * num_weights));
1229
1230 /* To make sure that the weighting starts out fairly */
1231 for (i = 0; i < num_weights; i++)
1232 {
1233 png_ptr->prev_filters[i] = 255;
1234 }
1235 }
1236
1237 if (png_ptr->filter_weights == NULL)
1238 {
1239 png_ptr->filter_weights = (png_uint_16p)png_malloc(png_ptr,
1240 (png_uint_32)(png_sizeof(png_uint_16) * num_weights));
1241
1242 png_ptr->inv_filter_weights = (png_uint_16p)png_malloc(png_ptr,
1243 (png_uint_32)(png_sizeof(png_uint_16) * num_weights));
1244 for (i = 0; i < num_weights; i++)
1245 {
1246 png_ptr->inv_filter_weights[i] =
1247 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
1248 }
1249 }
1250
1251 for (i = 0; i < num_weights; i++)
1252 {
1253 if (filter_weights[i] < 0.0)
1254 {
1255 png_ptr->inv_filter_weights[i] =
1256 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
1257 }
1258 else
1259 {
1260 png_ptr->inv_filter_weights[i] =
1261 (png_uint_16)((double)PNG_WEIGHT_FACTOR*filter_weights[i]+0.5);
1262 png_ptr->filter_weights[i] =
1263 (png_uint_16)((double)PNG_WEIGHT_FACTOR/filter_weights[i]+0.5);
1264 }
1265 }
1266 }
1267
1268 /* If, in the future, there are other filter methods, this would
1269 * need to be based on png_ptr->filter.
1270 */
1271 if (png_ptr->filter_costs == NULL)
1272 {
1273 png_ptr->filter_costs = (png_uint_16p)png_malloc(png_ptr,
1274 (png_uint_32)(png_sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
1275
1276 png_ptr->inv_filter_costs = (png_uint_16p)png_malloc(png_ptr,
1277 (png_uint_32)(png_sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
1278
1279 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
1280 {
1281 png_ptr->inv_filter_costs[i] =
1282 png_ptr->filter_costs[i] = PNG_COST_FACTOR;
1283 }
1284 }
1285
1286 /* Here is where we set the relative costs of the different filters. We
1287 * should take the desired compression level into account when setting
1288 * the costs, so that Paeth, for instance, has a high relative cost at low
1289 * compression levels, while it has a lower relative cost at higher
1290 * compression settings. The filter types are in order of increasing
1291 * relative cost, so it would be possible to do this with an algorithm.
1292 */
1293 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
1294 {
1295 if (filter_costs == NULL || filter_costs[i] < 0.0)
1296 {
1297 png_ptr->inv_filter_costs[i] =
1298 png_ptr->filter_costs[i] = PNG_COST_FACTOR;
1299 }
1300 else if (filter_costs[i] >= 1.0)
1301 {
1302 png_ptr->inv_filter_costs[i] =
1303 (png_uint_16)((double)PNG_COST_FACTOR / filter_costs[i] + 0.5);
1304 png_ptr->filter_costs[i] =
1305 (png_uint_16)((double)PNG_COST_FACTOR * filter_costs[i] + 0.5);
1306 }
1307 }
1308 }
1309 #endif /* PNG_WRITE_WEIGHTED_FILTER_SUPPORTED */
1310
1311 void PNGAPI
1312 png_set_compression_level(png_structp png_ptr, int level)
1313 {
1314 png_debug(1, "in png_set_compression_level\n");
1315 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_LEVEL;
1316 png_ptr->zlib_level = level;
1317 }
1318
1319 void PNGAPI
1320 png_set_compression_mem_level(png_structp png_ptr, int mem_level)
1321 {
1322 png_debug(1, "in png_set_compression_mem_level\n");
1323 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_MEM_LEVEL;
1324 png_ptr->zlib_mem_level = mem_level;
1325 }
1326
1327 void PNGAPI
1328 png_set_compression_strategy(png_structp png_ptr, int strategy)
1329 {
1330 png_debug(1, "in png_set_compression_strategy\n");
1331 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_STRATEGY;
1332 png_ptr->zlib_strategy = strategy;
1333 }
1334
1335 void PNGAPI
1336 png_set_compression_window_bits(png_structp png_ptr, int window_bits)
1337 {
1338 if (window_bits > 15)
1339 png_warning(png_ptr, "Only compression windows <= 32k supported by PNG");
1340 else if (window_bits < 8)
1341 png_warning(png_ptr, "Only compression windows >= 256 supported by PNG");
1342 #ifndef WBITS_8_OK
1343 /* avoid libpng bug with 256-byte windows */
1344 if (window_bits == 8)
1345 {
1346 png_warning(png_ptr, "Compression window is being reset to 512");
1347 window_bits=9;
1348 }
1349 #endif
1350 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_WINDOW_BITS;
1351 png_ptr->zlib_window_bits = window_bits;
1352 }
1353
1354 void PNGAPI
1355 png_set_compression_method(png_structp png_ptr, int method)
1356 {
1357 png_debug(1, "in png_set_compression_method\n");
1358 if (method != 8)
1359 png_warning(png_ptr, "Only compression method 8 is supported by PNG");
1360 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_METHOD;
1361 png_ptr->zlib_method = method;
1362 }
1363
1364 void PNGAPI
1365 png_set_write_status_fn(png_structp png_ptr, png_write_status_ptr write_row_fn)
1366 {
1367 png_ptr->write_row_fn = write_row_fn;
1368 }
1369
1370 #if defined(PNG_WRITE_USER_TRANSFORM_SUPPORTED)
1371 void PNGAPI
1372 png_set_write_user_transform_fn(png_structp png_ptr, png_user_transform_ptr
1373 write_user_transform_fn)
1374 {
1375 png_debug(1, "in png_set_write_user_transform_fn\n");
1376 png_ptr->transformations |= PNG_USER_TRANSFORM;
1377 png_ptr->write_user_transform_fn = write_user_transform_fn;
1378 }
1379 #endif
1380
1381
1382 #if defined(PNG_INFO_IMAGE_SUPPORTED)
1383 void PNGAPI
1384 png_write_png(png_structp png_ptr, png_infop info_ptr,
1385 int transforms, voidp params)
1386 {
1387 #if defined(PNG_WRITE_INVERT_ALPHA_SUPPORTED)
1388 /* invert the alpha channel from opacity to transparency */
1389 if (transforms & PNG_TRANSFORM_INVERT_ALPHA)
1390 png_set_invert_alpha(png_ptr);
1391 #endif
1392
1393 /* Write the file header information. */
1394 png_write_info(png_ptr, info_ptr);
1395
1396 /* ------ these transformations don't touch the info structure ------- */
1397
1398 #if defined(PNG_WRITE_INVERT_SUPPORTED)
1399 /* invert monochrome pixels */
1400 if (transforms & PNG_TRANSFORM_INVERT_MONO)
1401 png_set_invert_mono(png_ptr);
1402 #endif
1403
1404 #if defined(PNG_WRITE_SHIFT_SUPPORTED)
1405 /* Shift the pixels up to a legal bit depth and fill in
1406 * as appropriate to correctly scale the image.
1407 */
1408 if ((transforms & PNG_TRANSFORM_SHIFT)
1409 && (info_ptr->valid & PNG_INFO_sBIT))
1410 png_set_shift(png_ptr, &info_ptr->sig_bit);
1411 #endif
1412
1413 #if defined(PNG_WRITE_PACK_SUPPORTED)
1414 /* pack pixels into bytes */
1415 if (transforms & PNG_TRANSFORM_PACKING)
1416 png_set_packing(png_ptr);
1417 #endif
1418
1419 #if defined(PNG_WRITE_SWAP_ALPHA_SUPPORTED)
1420 /* swap location of alpha bytes from ARGB to RGBA */
1421 if (transforms & PNG_TRANSFORM_SWAP_ALPHA)
1422 png_set_swap_alpha(png_ptr);
1423 #endif
1424
1425 #if defined(PNG_WRITE_FILLER_SUPPORTED)
1426 /* Get rid of filler (OR ALPHA) bytes, pack XRGB/RGBX/ARGB/RGBA into
1427 * RGB (4 channels -> 3 channels). The second parameter is not used.
1428 */
1429 if (transforms & PNG_TRANSFORM_STRIP_FILLER)
1430 png_set_filler(png_ptr, 0, PNG_FILLER_BEFORE);
1431 #endif
1432
1433 #if defined(PNG_WRITE_BGR_SUPPORTED)
1434 /* flip BGR pixels to RGB */
1435 if (transforms & PNG_TRANSFORM_BGR)
1436 png_set_bgr(png_ptr);
1437 #endif
1438
1439 #if defined(PNG_WRITE_SWAP_SUPPORTED)
1440 /* swap bytes of 16-bit files to most significant byte first */
1441 if (transforms & PNG_TRANSFORM_SWAP_ENDIAN)
1442 png_set_swap(png_ptr);
1443 #endif
1444
1445 #if defined(PNG_WRITE_PACKSWAP_SUPPORTED)
1446 /* swap bits of 1, 2, 4 bit packed pixel formats */
1447 if (transforms & PNG_TRANSFORM_PACKSWAP)
1448 png_set_packswap(png_ptr);
1449 #endif
1450
1451 /* ----------------------- end of transformations ------------------- */
1452
1453 /* write the bits */
1454 if (info_ptr->valid & PNG_INFO_IDAT)
1455 png_write_image(png_ptr, info_ptr->row_pointers);
1456
1457 /* It is REQUIRED to call this to finish writing the rest of the file */
1458 png_write_end(png_ptr, info_ptr);
1459
1460 if(transforms == 0 || params == NULL)
1461 /* quiet compiler warnings */ return;
1462 }
1463 #endif
1464 #endif /* PNG_WRITE_SUPPORTED */