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