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