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