Use LinkableValueNode members functions when possible in the derived valuenodes.
[synfig.git] / libpng-1.2.5 / pngwutil.c
1
2 /* pngwutil.c - utilities to write a PNG file
3  *
4  * libpng 1.2.5 - October 3, 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 #define PNG_INTERNAL
12 #include "png.h"
13 #ifdef PNG_WRITE_SUPPORTED
14
15 /* Place a 32-bit number into a buffer in PNG byte order.  We work
16  * with unsigned numbers for convenience, although one supported
17  * ancillary chunk uses signed (two's complement) numbers.
18  */
19 void /* PRIVATE */
20 png_save_uint_32(png_bytep buf, png_uint_32 i)
21 {
22    buf[0] = (png_byte)((i >> 24) & 0xff);
23    buf[1] = (png_byte)((i >> 16) & 0xff);
24    buf[2] = (png_byte)((i >> 8) & 0xff);
25    buf[3] = (png_byte)(i & 0xff);
26 }
27
28 #if defined(PNG_WRITE_pCAL_SUPPORTED) || defined(PNG_WRITE_oFFs_SUPPORTED)
29 /* The png_save_int_32 function assumes integers are stored in two's
30  * complement format.  If this isn't the case, then this routine needs to
31  * be modified to write data in two's complement format.
32  */
33 void /* PRIVATE */
34 png_save_int_32(png_bytep buf, png_int_32 i)
35 {
36    buf[0] = (png_byte)((i >> 24) & 0xff);
37    buf[1] = (png_byte)((i >> 16) & 0xff);
38    buf[2] = (png_byte)((i >> 8) & 0xff);
39    buf[3] = (png_byte)(i & 0xff);
40 }
41 #endif
42
43 /* Place a 16-bit number into a buffer in PNG byte order.
44  * The parameter is declared unsigned int, not png_uint_16,
45  * just to avoid potential problems on pre-ANSI C compilers.
46  */
47 void /* PRIVATE */
48 png_save_uint_16(png_bytep buf, unsigned int i)
49 {
50    buf[0] = (png_byte)((i >> 8) & 0xff);
51    buf[1] = (png_byte)(i & 0xff);
52 }
53
54 /* Write a PNG chunk all at once.  The type is an array of ASCII characters
55  * representing the chunk name.  The array must be at least 4 bytes in
56  * length, and does not need to be null terminated.  To be safe, pass the
57  * pre-defined chunk names here, and if you need a new one, define it
58  * where the others are defined.  The length is the length of the data.
59  * All the data must be present.  If that is not possible, use the
60  * png_write_chunk_start(), png_write_chunk_data(), and png_write_chunk_end()
61  * functions instead.
62  */
63 void PNGAPI
64 png_write_chunk(png_structp png_ptr, png_bytep chunk_name,
65    png_bytep data, png_size_t length)
66 {
67    png_write_chunk_start(png_ptr, chunk_name, (png_uint_32)length);
68    png_write_chunk_data(png_ptr, data, length);
69    png_write_chunk_end(png_ptr);
70 }
71
72 /* Write the start of a PNG chunk.  The type is the chunk type.
73  * The total_length is the sum of the lengths of all the data you will be
74  * passing in png_write_chunk_data().
75  */
76 void PNGAPI
77 png_write_chunk_start(png_structp png_ptr, png_bytep chunk_name,
78    png_uint_32 length)
79 {
80    png_byte buf[4];
81    png_debug2(0, "Writing %s chunk (%lu bytes)\n", chunk_name, length);
82
83    /* write the length */
84    png_save_uint_32(buf, length);
85    png_write_data(png_ptr, buf, (png_size_t)4);
86
87    /* write the chunk name */
88    png_write_data(png_ptr, chunk_name, (png_size_t)4);
89    /* reset the crc and run it over the chunk name */
90    png_reset_crc(png_ptr);
91    png_calculate_crc(png_ptr, chunk_name, (png_size_t)4);
92 }
93
94 /* Write the data of a PNG chunk started with png_write_chunk_start().
95  * Note that multiple calls to this function are allowed, and that the
96  * sum of the lengths from these calls *must* add up to the total_length
97  * given to png_write_chunk_start().
98  */
99 void PNGAPI
100 png_write_chunk_data(png_structp png_ptr, png_bytep data, png_size_t length)
101 {
102    /* write the data, and run the CRC over it */
103    if (data != NULL && length > 0)
104    {
105       png_calculate_crc(png_ptr, data, length);
106       png_write_data(png_ptr, data, length);
107    }
108 }
109
110 /* Finish a chunk started with png_write_chunk_start(). */
111 void PNGAPI
112 png_write_chunk_end(png_structp png_ptr)
113 {
114    png_byte buf[4];
115
116    /* write the crc */
117    png_save_uint_32(buf, png_ptr->crc);
118
119    png_write_data(png_ptr, buf, (png_size_t)4);
120 }
121
122 /* Simple function to write the signature.  If we have already written
123  * the magic bytes of the signature, or more likely, the PNG stream is
124  * being embedded into another stream and doesn't need its own signature,
125  * we should call png_set_sig_bytes() to tell libpng how many of the
126  * bytes have already been written.
127  */
128 void /* PRIVATE */
129 png_write_sig(png_structp png_ptr)
130 {
131    png_byte png_signature[8] = {137, 80, 78, 71, 13, 10, 26, 10};
132    /* write the rest of the 8 byte signature */
133    png_write_data(png_ptr, &png_signature[png_ptr->sig_bytes],
134       (png_size_t)8 - png_ptr->sig_bytes);
135    if(png_ptr->sig_bytes < 3)
136       png_ptr->mode |= PNG_HAVE_PNG_SIGNATURE;
137 }
138
139 #if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_iCCP_SUPPORTED)
140 /*
141  * This pair of functions encapsulates the operation of (a) compressing a
142  * text string, and (b) issuing it later as a series of chunk data writes.
143  * The compression_state structure is shared context for these functions
144  * set up by the caller in order to make the whole mess thread-safe.
145  */
146
147 typedef struct
148 {
149     char *input;   /* the uncompressed input data */
150     int input_len;   /* its length */
151     int num_output_ptr; /* number of output pointers used */
152     int max_output_ptr; /* size of output_ptr */
153     png_charpp output_ptr; /* array of pointers to output */
154 } compression_state;
155
156 /* compress given text into storage in the png_ptr structure */
157 static int /* PRIVATE */
158 png_text_compress(png_structp png_ptr,
159         png_charp text, png_size_t text_len, int compression,
160         compression_state *comp)
161 {
162    int ret;
163
164    comp->num_output_ptr = comp->max_output_ptr = 0;
165    comp->output_ptr = NULL;
166    comp->input = NULL;
167
168    /* we may just want to pass the text right through */
169    if (compression == PNG_TEXT_COMPRESSION_NONE)
170    {
171        comp->input = text;
172        comp->input_len = text_len;
173        return((int)text_len);
174    }
175
176    if (compression >= PNG_TEXT_COMPRESSION_LAST)
177    {
178 #if !defined(PNG_NO_STDIO) && !defined(_WIN32_WCE)
179       char msg[50];
180       sprintf(msg, "Unknown compression type %d", compression);
181       png_warning(png_ptr, msg);
182 #else
183       png_warning(png_ptr, "Unknown compression type");
184 #endif
185    }
186
187    /* We can't write the chunk until we find out how much data we have,
188     * which means we need to run the compressor first and save the
189     * output.  This shouldn't be a problem, as the vast majority of
190     * comments should be reasonable, but we will set up an array of
191     * malloc'd pointers to be sure.
192     *
193     * If we knew the application was well behaved, we could simplify this
194     * greatly by assuming we can always malloc an output buffer large
195     * enough to hold the compressed text ((1001 * text_len / 1000) + 12)
196     * and malloc this directly.  The only time this would be a bad idea is
197     * if we can't malloc more than 64K and we have 64K of random input
198     * data, or if the input string is incredibly large (although this
199     * wouldn't cause a failure, just a slowdown due to swapping).
200     */
201
202    /* set up the compression buffers */
203    png_ptr->zstream.avail_in = (uInt)text_len;
204    png_ptr->zstream.next_in = (Bytef *)text;
205    png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
206    png_ptr->zstream.next_out = (Bytef *)png_ptr->zbuf;
207
208    /* this is the same compression loop as in png_write_row() */
209    do
210    {
211       /* compress the data */
212       ret = deflate(&png_ptr->zstream, Z_NO_FLUSH);
213       if (ret != Z_OK)
214       {
215          /* error */
216          if (png_ptr->zstream.msg != NULL)
217             png_error(png_ptr, png_ptr->zstream.msg);
218          else
219             png_error(png_ptr, "zlib error");
220       }
221       /* check to see if we need more room */
222       if (!png_ptr->zstream.avail_out && png_ptr->zstream.avail_in)
223       {
224          /* make sure the output array has room */
225          if (comp->num_output_ptr >= comp->max_output_ptr)
226          {
227             int old_max;
228
229             old_max = comp->max_output_ptr;
230             comp->max_output_ptr = comp->num_output_ptr + 4;
231             if (comp->output_ptr != NULL)
232             {
233                png_charpp old_ptr;
234
235                old_ptr = comp->output_ptr;
236                comp->output_ptr = (png_charpp)png_malloc(png_ptr,
237                   (png_uint_32)(comp->max_output_ptr * sizeof (png_charpp)));
238                png_memcpy(comp->output_ptr, old_ptr, old_max
239                   * sizeof (png_charp));
240                png_free(png_ptr, old_ptr);
241             }
242             else
243                comp->output_ptr = (png_charpp)png_malloc(png_ptr,
244                   (png_uint_32)(comp->max_output_ptr * sizeof (png_charp)));
245          }
246
247          /* save the data */
248          comp->output_ptr[comp->num_output_ptr] = (png_charp)png_malloc(png_ptr,
249             (png_uint_32)png_ptr->zbuf_size);
250          png_memcpy(comp->output_ptr[comp->num_output_ptr], png_ptr->zbuf,
251             png_ptr->zbuf_size);
252          comp->num_output_ptr++;
253
254          /* and reset the buffer */
255          png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
256          png_ptr->zstream.next_out = png_ptr->zbuf;
257       }
258    /* continue until we don't have any more to compress */
259    } while (png_ptr->zstream.avail_in);
260
261    /* finish the compression */
262    do
263    {
264       /* tell zlib we are finished */
265       ret = deflate(&png_ptr->zstream, Z_FINISH);
266
267       if (ret == Z_OK)
268       {
269          /* check to see if we need more room */
270          if (!(png_ptr->zstream.avail_out))
271          {
272             /* check to make sure our output array has room */
273             if (comp->num_output_ptr >= comp->max_output_ptr)
274             {
275                int old_max;
276
277                old_max = comp->max_output_ptr;
278                comp->max_output_ptr = comp->num_output_ptr + 4;
279                if (comp->output_ptr != NULL)
280                {
281                   png_charpp old_ptr;
282
283                   old_ptr = comp->output_ptr;
284                   /* This could be optimized to realloc() */
285                   comp->output_ptr = (png_charpp)png_malloc(png_ptr,
286                      (png_uint_32)(comp->max_output_ptr * sizeof (png_charpp)));
287                   png_memcpy(comp->output_ptr, old_ptr,
288                      old_max * sizeof (png_charp));
289                   png_free(png_ptr, old_ptr);
290                }
291                else
292                   comp->output_ptr = (png_charpp)png_malloc(png_ptr,
293                      (png_uint_32)(comp->max_output_ptr * sizeof (png_charp)));
294             }
295
296             /* save off the data */
297             comp->output_ptr[comp->num_output_ptr] =
298                (png_charp)png_malloc(png_ptr, (png_uint_32)png_ptr->zbuf_size);
299             png_memcpy(comp->output_ptr[comp->num_output_ptr], png_ptr->zbuf,
300                png_ptr->zbuf_size);
301             comp->num_output_ptr++;
302
303             /* and reset the buffer pointers */
304             png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
305             png_ptr->zstream.next_out = png_ptr->zbuf;
306          }
307       }
308       else if (ret != Z_STREAM_END)
309       {
310          /* we got an error */
311          if (png_ptr->zstream.msg != NULL)
312             png_error(png_ptr, png_ptr->zstream.msg);
313          else
314             png_error(png_ptr, "zlib error");
315       }
316    } while (ret != Z_STREAM_END);
317
318    /* text length is number of buffers plus last buffer */
319    text_len = png_ptr->zbuf_size * comp->num_output_ptr;
320    if (png_ptr->zstream.avail_out < png_ptr->zbuf_size)
321       text_len += png_ptr->zbuf_size - (png_size_t)png_ptr->zstream.avail_out;
322
323    return((int)text_len);
324 }
325
326 /* ship the compressed text out via chunk writes */
327 static void /* PRIVATE */
328 png_write_compressed_data_out(png_structp png_ptr, compression_state *comp)
329 {
330    int i;
331
332    /* handle the no-compression case */
333    if (comp->input)
334    {
335        png_write_chunk_data(png_ptr, (png_bytep)comp->input,
336                             (png_size_t)comp->input_len);
337        return;
338    }
339
340    /* write saved output buffers, if any */
341    for (i = 0; i < comp->num_output_ptr; i++)
342    {
343       png_write_chunk_data(png_ptr,(png_bytep)comp->output_ptr[i],
344          png_ptr->zbuf_size);
345       png_free(png_ptr, comp->output_ptr[i]);
346       comp->output_ptr[i]=NULL;
347    }
348    if (comp->max_output_ptr != 0)
349       png_free(png_ptr, comp->output_ptr);
350       comp->output_ptr=NULL;
351    /* write anything left in zbuf */
352    if (png_ptr->zstream.avail_out < (png_uint_32)png_ptr->zbuf_size)
353       png_write_chunk_data(png_ptr, png_ptr->zbuf,
354          png_ptr->zbuf_size - png_ptr->zstream.avail_out);
355
356    /* reset zlib for another zTXt/iTXt or the image data */
357    deflateReset(&png_ptr->zstream);
358
359 }
360 #endif
361
362 /* Write the IHDR chunk, and update the png_struct with the necessary
363  * information.  Note that the rest of this code depends upon this
364  * information being correct.
365  */
366 void /* PRIVATE */
367 png_write_IHDR(png_structp png_ptr, png_uint_32 width, png_uint_32 height,
368    int bit_depth, int color_type, int compression_type, int filter_type,
369    int interlace_type)
370 {
371 #ifdef PNG_USE_LOCAL_ARRAYS
372    PNG_IHDR;
373 #endif
374    png_byte buf[13]; /* buffer to store the IHDR info */
375
376    png_debug(1, "in png_write_IHDR\n");
377    /* Check that we have valid input data from the application info */
378    switch (color_type)
379    {
380       case PNG_COLOR_TYPE_GRAY:
381          switch (bit_depth)
382          {
383             case 1:
384             case 2:
385             case 4:
386             case 8:
387             case 16: png_ptr->channels = 1; break;
388             default: png_error(png_ptr,"Invalid bit depth for grayscale image");
389          }
390          break;
391       case PNG_COLOR_TYPE_RGB:
392          if (bit_depth != 8 && bit_depth != 16)
393             png_error(png_ptr, "Invalid bit depth for RGB image");
394          png_ptr->channels = 3;
395          break;
396       case PNG_COLOR_TYPE_PALETTE:
397          switch (bit_depth)
398          {
399             case 1:
400             case 2:
401             case 4:
402             case 8: png_ptr->channels = 1; break;
403             default: png_error(png_ptr, "Invalid bit depth for paletted image");
404          }
405          break;
406       case PNG_COLOR_TYPE_GRAY_ALPHA:
407          if (bit_depth != 8 && bit_depth != 16)
408             png_error(png_ptr, "Invalid bit depth for grayscale+alpha image");
409          png_ptr->channels = 2;
410          break;
411       case PNG_COLOR_TYPE_RGB_ALPHA:
412          if (bit_depth != 8 && bit_depth != 16)
413             png_error(png_ptr, "Invalid bit depth for RGBA image");
414          png_ptr->channels = 4;
415          break;
416       default:
417          png_error(png_ptr, "Invalid image color type specified");
418    }
419
420    if (compression_type != PNG_COMPRESSION_TYPE_BASE)
421    {
422       png_warning(png_ptr, "Invalid compression type specified");
423       compression_type = PNG_COMPRESSION_TYPE_BASE;
424    }
425
426    /* Write filter_method 64 (intrapixel differencing) only if
427     * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and
428     * 2. Libpng did not write a PNG signature (this filter_method is only
429     *    used in PNG datastreams that are embedded in MNG datastreams) and
430     * 3. The application called png_permit_mng_features with a mask that
431     *    included PNG_FLAG_MNG_FILTER_64 and
432     * 4. The filter_method is 64 and
433     * 5. The color_type is RGB or RGBA
434     */
435    if (
436 #if defined(PNG_MNG_FEATURES_SUPPORTED)
437       !((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
438       ((png_ptr->mode&PNG_HAVE_PNG_SIGNATURE) == 0) &&
439       (color_type == PNG_COLOR_TYPE_RGB ||
440        color_type == PNG_COLOR_TYPE_RGB_ALPHA) &&
441       (filter_type == PNG_INTRAPIXEL_DIFFERENCING)) &&
442 #endif
443       filter_type != PNG_FILTER_TYPE_BASE)
444    {
445       png_warning(png_ptr, "Invalid filter type specified");
446       filter_type = PNG_FILTER_TYPE_BASE;
447    }
448
449 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
450    if (interlace_type != PNG_INTERLACE_NONE &&
451       interlace_type != PNG_INTERLACE_ADAM7)
452    {
453       png_warning(png_ptr, "Invalid interlace type specified");
454       interlace_type = PNG_INTERLACE_ADAM7;
455    }
456 #else
457    interlace_type=PNG_INTERLACE_NONE;
458 #endif
459
460    /* save off the relevent information */
461    png_ptr->bit_depth = (png_byte)bit_depth;
462    png_ptr->color_type = (png_byte)color_type;
463    png_ptr->interlaced = (png_byte)interlace_type;
464 #if defined(PNG_MNG_FEATURES_SUPPORTED)
465    png_ptr->filter_type = (png_byte)filter_type;
466 #endif
467    png_ptr->width = width;
468    png_ptr->height = height;
469
470    png_ptr->pixel_depth = (png_byte)(bit_depth * png_ptr->channels);
471    png_ptr->rowbytes = ((width * (png_size_t)png_ptr->pixel_depth + 7) >> 3);
472    /* set the usr info, so any transformations can modify it */
473    png_ptr->usr_width = png_ptr->width;
474    png_ptr->usr_bit_depth = png_ptr->bit_depth;
475    png_ptr->usr_channels = png_ptr->channels;
476
477    /* pack the header information into the buffer */
478    png_save_uint_32(buf, width);
479    png_save_uint_32(buf + 4, height);
480    buf[8] = (png_byte)bit_depth;
481    buf[9] = (png_byte)color_type;
482    buf[10] = (png_byte)compression_type;
483    buf[11] = (png_byte)filter_type;
484    buf[12] = (png_byte)interlace_type;
485
486    /* write the chunk */
487    png_write_chunk(png_ptr, (png_bytep)png_IHDR, buf, (png_size_t)13);
488
489    /* initialize zlib with PNG info */
490    png_ptr->zstream.zalloc = png_zalloc;
491    png_ptr->zstream.zfree = png_zfree;
492    png_ptr->zstream.opaque = (voidpf)png_ptr;
493    if (!(png_ptr->do_filter))
494    {
495       if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE ||
496          png_ptr->bit_depth < 8)
497          png_ptr->do_filter = PNG_FILTER_NONE;
498       else
499          png_ptr->do_filter = PNG_ALL_FILTERS;
500    }
501    if (!(png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_STRATEGY))
502    {
503       if (png_ptr->do_filter != PNG_FILTER_NONE)
504          png_ptr->zlib_strategy = Z_FILTERED;
505       else
506          png_ptr->zlib_strategy = Z_DEFAULT_STRATEGY;
507    }
508    if (!(png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_LEVEL))
509       png_ptr->zlib_level = Z_DEFAULT_COMPRESSION;
510    if (!(png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_MEM_LEVEL))
511       png_ptr->zlib_mem_level = 8;
512    if (!(png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_WINDOW_BITS))
513       png_ptr->zlib_window_bits = 15;
514    if (!(png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_METHOD))
515       png_ptr->zlib_method = 8;
516    deflateInit2(&png_ptr->zstream, png_ptr->zlib_level,
517       png_ptr->zlib_method, png_ptr->zlib_window_bits,
518       png_ptr->zlib_mem_level, png_ptr->zlib_strategy);
519    png_ptr->zstream.next_out = png_ptr->zbuf;
520    png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
521
522    png_ptr->mode = PNG_HAVE_IHDR;
523 }
524
525 /* write the palette.  We are careful not to trust png_color to be in the
526  * correct order for PNG, so people can redefine it to any convenient
527  * structure.
528  */
529 void /* PRIVATE */
530 png_write_PLTE(png_structp png_ptr, png_colorp palette, png_uint_32 num_pal)
531 {
532 #ifdef PNG_USE_LOCAL_ARRAYS
533    PNG_PLTE;
534 #endif
535    png_uint_32 i;
536    png_colorp pal_ptr;
537    png_byte buf[3];
538
539    png_debug(1, "in png_write_PLTE\n");
540    if ((
541 #if defined(PNG_MNG_FEATURES_SUPPORTED)
542         !(png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE) &&
543 #endif
544         num_pal == 0) || num_pal > 256)
545    {
546      if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
547      {
548         png_error(png_ptr, "Invalid number of colors in palette");
549      }
550      else
551      {
552         png_warning(png_ptr, "Invalid number of colors in palette");
553         return;
554      }
555    }
556
557    if (!(png_ptr->color_type&PNG_COLOR_MASK_COLOR))
558    {
559       png_warning(png_ptr,
560         "Ignoring request to write a PLTE chunk in grayscale PNG");
561       return;
562    }
563
564    png_ptr->num_palette = (png_uint_16)num_pal;
565    png_debug1(3, "num_palette = %d\n", png_ptr->num_palette);
566
567    png_write_chunk_start(png_ptr, (png_bytep)png_PLTE, num_pal * 3);
568 #ifndef PNG_NO_POINTER_INDEXING
569    for (i = 0, pal_ptr = palette; i < num_pal; i++, pal_ptr++)
570    {
571       buf[0] = pal_ptr->red;
572       buf[1] = pal_ptr->green;
573       buf[2] = pal_ptr->blue;
574       png_write_chunk_data(png_ptr, buf, (png_size_t)3);
575    }
576 #else
577    /* This is a little slower but some buggy compilers need to do this instead */
578    pal_ptr=palette;
579    for (i = 0; i < num_pal; i++)
580    {
581       buf[0] = pal_ptr[i].red;
582       buf[1] = pal_ptr[i].green;
583       buf[2] = pal_ptr[i].blue;
584       png_write_chunk_data(png_ptr, buf, (png_size_t)3);
585    }
586 #endif
587    png_write_chunk_end(png_ptr);
588    png_ptr->mode |= PNG_HAVE_PLTE;
589 }
590
591 /* write an IDAT chunk */
592 void /* PRIVATE */
593 png_write_IDAT(png_structp png_ptr, png_bytep data, png_size_t length)
594 {
595 #ifdef PNG_USE_LOCAL_ARRAYS
596    PNG_IDAT;
597 #endif
598    png_debug(1, "in png_write_IDAT\n");
599    png_write_chunk(png_ptr, (png_bytep)png_IDAT, data, length);
600    png_ptr->mode |= PNG_HAVE_IDAT;
601 }
602
603 /* write an IEND chunk */
604 void /* PRIVATE */
605 png_write_IEND(png_structp png_ptr)
606 {
607 #ifdef PNG_USE_LOCAL_ARRAYS
608    PNG_IEND;
609 #endif
610    png_debug(1, "in png_write_IEND\n");
611    png_write_chunk(png_ptr, (png_bytep)png_IEND, png_bytep_NULL,
612      (png_size_t)0);
613    png_ptr->mode |= PNG_HAVE_IEND;
614 }
615
616 #if defined(PNG_WRITE_gAMA_SUPPORTED)
617 /* write a gAMA chunk */
618 #ifdef PNG_FLOATING_POINT_SUPPORTED
619 void /* PRIVATE */
620 png_write_gAMA(png_structp png_ptr, double file_gamma)
621 {
622 #ifdef PNG_USE_LOCAL_ARRAYS
623    PNG_gAMA;
624 #endif
625    png_uint_32 igamma;
626    png_byte buf[4];
627
628    png_debug(1, "in png_write_gAMA\n");
629    /* file_gamma is saved in 1/100,000ths */
630    igamma = (png_uint_32)(file_gamma * 100000.0 + 0.5);
631    png_save_uint_32(buf, igamma);
632    png_write_chunk(png_ptr, (png_bytep)png_gAMA, buf, (png_size_t)4);
633 }
634 #endif
635 #ifdef PNG_FIXED_POINT_SUPPORTED
636 void /* PRIVATE */
637 png_write_gAMA_fixed(png_structp png_ptr, png_fixed_point file_gamma)
638 {
639 #ifdef PNG_USE_LOCAL_ARRAYS
640    PNG_gAMA;
641 #endif
642    png_byte buf[4];
643
644    png_debug(1, "in png_write_gAMA\n");
645    /* file_gamma is saved in 1/100,000ths */
646    png_save_uint_32(buf, (png_uint_32)file_gamma);
647    png_write_chunk(png_ptr, (png_bytep)png_gAMA, buf, (png_size_t)4);
648 }
649 #endif
650 #endif
651
652 #if defined(PNG_WRITE_sRGB_SUPPORTED)
653 /* write a sRGB chunk */
654 void /* PRIVATE */
655 png_write_sRGB(png_structp png_ptr, int srgb_intent)
656 {
657 #ifdef PNG_USE_LOCAL_ARRAYS
658    PNG_sRGB;
659 #endif
660    png_byte buf[1];
661
662    png_debug(1, "in png_write_sRGB\n");
663    if(srgb_intent >= PNG_sRGB_INTENT_LAST)
664          png_warning(png_ptr,
665             "Invalid sRGB rendering intent specified");
666    buf[0]=(png_byte)srgb_intent;
667    png_write_chunk(png_ptr, (png_bytep)png_sRGB, buf, (png_size_t)1);
668 }
669 #endif
670
671 #if defined(PNG_WRITE_iCCP_SUPPORTED)
672 /* write an iCCP chunk */
673 void /* PRIVATE */
674 png_write_iCCP(png_structp png_ptr, png_charp name, int compression_type,
675    png_charp profile, int profile_len)
676 {
677 #ifdef PNG_USE_LOCAL_ARRAYS
678    PNG_iCCP;
679 #endif
680    png_size_t name_len;
681    png_charp new_name;
682    compression_state comp;
683
684    png_debug(1, "in png_write_iCCP\n");
685    if (name == NULL || (name_len = png_check_keyword(png_ptr, name,
686       &new_name)) == 0)
687    {
688       png_warning(png_ptr, "Empty keyword in iCCP chunk");
689       return;
690    }
691
692    if (compression_type != PNG_COMPRESSION_TYPE_BASE)
693       png_warning(png_ptr, "Unknown compression type in iCCP chunk");
694
695    if (profile == NULL)
696       profile_len = 0;
697
698    if (profile_len)
699        profile_len = png_text_compress(png_ptr, profile, (png_size_t)profile_len,
700           PNG_COMPRESSION_TYPE_BASE, &comp);
701
702    /* make sure we include the NULL after the name and the compression type */
703    png_write_chunk_start(png_ptr, (png_bytep)png_iCCP,
704           (png_uint_32)name_len+profile_len+2);
705    new_name[name_len+1]=0x00;
706    png_write_chunk_data(png_ptr, (png_bytep)new_name, name_len + 2);
707
708    if (profile_len)
709       png_write_compressed_data_out(png_ptr, &comp);
710
711    png_write_chunk_end(png_ptr);
712    png_free(png_ptr, new_name);
713 }
714 #endif
715
716 #if defined(PNG_WRITE_sPLT_SUPPORTED)
717 /* write a sPLT chunk */
718 void /* PRIVATE */
719 png_write_sPLT(png_structp png_ptr, png_sPLT_tp spalette)
720 {
721 #ifdef PNG_USE_LOCAL_ARRAYS
722    PNG_sPLT;
723 #endif
724    png_size_t name_len;
725    png_charp new_name;
726    png_byte entrybuf[10];
727    int entry_size = (spalette->depth == 8 ? 6 : 10);
728    int palette_size = entry_size * spalette->nentries;
729    png_sPLT_entryp ep;
730 #ifdef PNG_NO_POINTER_INDEXING
731    int i;
732 #endif
733
734    png_debug(1, "in png_write_sPLT\n");
735    if (spalette->name == NULL || (name_len = png_check_keyword(png_ptr,
736       spalette->name, &new_name))==0)
737    {
738       png_warning(png_ptr, "Empty keyword in sPLT chunk");
739       return;
740    }
741
742    /* make sure we include the NULL after the name */
743    png_write_chunk_start(png_ptr, (png_bytep)png_sPLT,
744           (png_uint_32)(name_len + 2 + palette_size));
745    png_write_chunk_data(png_ptr, (png_bytep)new_name, name_len + 1);
746    png_write_chunk_data(png_ptr, (png_bytep)&spalette->depth, 1);
747
748    /* loop through each palette entry, writing appropriately */
749 #ifndef PNG_NO_POINTER_INDEXING
750    for (ep = spalette->entries; ep<spalette->entries+spalette->nentries; ep++)
751    {
752        if (spalette->depth == 8)
753        {
754            entrybuf[0] = (png_byte)ep->red;
755            entrybuf[1] = (png_byte)ep->green;
756            entrybuf[2] = (png_byte)ep->blue;
757            entrybuf[3] = (png_byte)ep->alpha;
758            png_save_uint_16(entrybuf + 4, ep->frequency);
759        }
760        else
761        {
762            png_save_uint_16(entrybuf + 0, ep->red);
763            png_save_uint_16(entrybuf + 2, ep->green);
764            png_save_uint_16(entrybuf + 4, ep->blue);
765            png_save_uint_16(entrybuf + 6, ep->alpha);
766            png_save_uint_16(entrybuf + 8, ep->frequency);
767        }
768        png_write_chunk_data(png_ptr, entrybuf, (png_size_t)entry_size);
769    }
770 #else
771    ep=spalette->entries;
772    for (i=0; i>spalette->nentries; i++)
773    {
774        if (spalette->depth == 8)
775        {
776            entrybuf[0] = (png_byte)ep[i].red;
777            entrybuf[1] = (png_byte)ep[i].green;
778            entrybuf[2] = (png_byte)ep[i].blue;
779            entrybuf[3] = (png_byte)ep[i].alpha;
780            png_save_uint_16(entrybuf + 4, ep[i].frequency);
781        }
782        else
783        {
784            png_save_uint_16(entrybuf + 0, ep[i].red);
785            png_save_uint_16(entrybuf + 2, ep[i].green);
786            png_save_uint_16(entrybuf + 4, ep[i].blue);
787            png_save_uint_16(entrybuf + 6, ep[i].alpha);
788            png_save_uint_16(entrybuf + 8, ep[i].frequency);
789        }
790        png_write_chunk_data(png_ptr, entrybuf, entry_size);
791    }
792 #endif
793
794    png_write_chunk_end(png_ptr);
795    png_free(png_ptr, new_name);
796 }
797 #endif
798
799 #if defined(PNG_WRITE_sBIT_SUPPORTED)
800 /* write the sBIT chunk */
801 void /* PRIVATE */
802 png_write_sBIT(png_structp png_ptr, png_color_8p sbit, int color_type)
803 {
804 #ifdef PNG_USE_LOCAL_ARRAYS
805    PNG_sBIT;
806 #endif
807    png_byte buf[4];
808    png_size_t size;
809
810    png_debug(1, "in png_write_sBIT\n");
811    /* make sure we don't depend upon the order of PNG_COLOR_8 */
812    if (color_type & PNG_COLOR_MASK_COLOR)
813    {
814       png_byte maxbits;
815
816       maxbits = (png_byte)(color_type==PNG_COLOR_TYPE_PALETTE ? 8 :
817                 png_ptr->usr_bit_depth);
818       if (sbit->red == 0 || sbit->red > maxbits ||
819           sbit->green == 0 || sbit->green > maxbits ||
820           sbit->blue == 0 || sbit->blue > maxbits)
821       {
822          png_warning(png_ptr, "Invalid sBIT depth specified");
823          return;
824       }
825       buf[0] = sbit->red;
826       buf[1] = sbit->green;
827       buf[2] = sbit->blue;
828       size = 3;
829    }
830    else
831    {
832       if (sbit->gray == 0 || sbit->gray > png_ptr->usr_bit_depth)
833       {
834          png_warning(png_ptr, "Invalid sBIT depth specified");
835          return;
836       }
837       buf[0] = sbit->gray;
838       size = 1;
839    }
840
841    if (color_type & PNG_COLOR_MASK_ALPHA)
842    {
843       if (sbit->alpha == 0 || sbit->alpha > png_ptr->usr_bit_depth)
844       {
845          png_warning(png_ptr, "Invalid sBIT depth specified");
846          return;
847       }
848       buf[size++] = sbit->alpha;
849    }
850
851    png_write_chunk(png_ptr, (png_bytep)png_sBIT, buf, size);
852 }
853 #endif
854
855 #if defined(PNG_WRITE_cHRM_SUPPORTED)
856 /* write the cHRM chunk */
857 #ifdef PNG_FLOATING_POINT_SUPPORTED
858 void /* PRIVATE */
859 png_write_cHRM(png_structp png_ptr, double white_x, double white_y,
860    double red_x, double red_y, double green_x, double green_y,
861    double blue_x, double blue_y)
862 {
863 #ifdef PNG_USE_LOCAL_ARRAYS
864    PNG_cHRM;
865 #endif
866    png_byte buf[32];
867    png_uint_32 itemp;
868
869    png_debug(1, "in png_write_cHRM\n");
870    /* each value is saved in 1/100,000ths */
871    if (white_x < 0 || white_x > 0.8 || white_y < 0 || white_y > 0.8 ||
872        white_x + white_y > 1.0)
873    {
874       png_warning(png_ptr, "Invalid cHRM white point specified");
875 #if !defined(PNG_NO_CONSOLE_IO)
876       fprintf(stderr,"white_x=%f, white_y=%f\n",white_x, white_y);
877 #endif
878       return;
879    }
880    itemp = (png_uint_32)(white_x * 100000.0 + 0.5);
881    png_save_uint_32(buf, itemp);
882    itemp = (png_uint_32)(white_y * 100000.0 + 0.5);
883    png_save_uint_32(buf + 4, itemp);
884
885    if (red_x < 0 || red_x > 0.8 || red_y < 0 || red_y > 0.8 ||
886        red_x + red_y > 1.0)
887    {
888       png_warning(png_ptr, "Invalid cHRM red point specified");
889       return;
890    }
891    itemp = (png_uint_32)(red_x * 100000.0 + 0.5);
892    png_save_uint_32(buf + 8, itemp);
893    itemp = (png_uint_32)(red_y * 100000.0 + 0.5);
894    png_save_uint_32(buf + 12, itemp);
895
896    if (green_x < 0 || green_x > 0.8 || green_y < 0 || green_y > 0.8 ||
897        green_x + green_y > 1.0)
898    {
899       png_warning(png_ptr, "Invalid cHRM green point specified");
900       return;
901    }
902    itemp = (png_uint_32)(green_x * 100000.0 + 0.5);
903    png_save_uint_32(buf + 16, itemp);
904    itemp = (png_uint_32)(green_y * 100000.0 + 0.5);
905    png_save_uint_32(buf + 20, itemp);
906
907    if (blue_x < 0 || blue_x > 0.8 || blue_y < 0 || blue_y > 0.8 ||
908        blue_x + blue_y > 1.0)
909    {
910       png_warning(png_ptr, "Invalid cHRM blue point specified");
911       return;
912    }
913    itemp = (png_uint_32)(blue_x * 100000.0 + 0.5);
914    png_save_uint_32(buf + 24, itemp);
915    itemp = (png_uint_32)(blue_y * 100000.0 + 0.5);
916    png_save_uint_32(buf + 28, itemp);
917
918    png_write_chunk(png_ptr, (png_bytep)png_cHRM, buf, (png_size_t)32);
919 }
920 #endif
921 #ifdef PNG_FIXED_POINT_SUPPORTED
922 void /* PRIVATE */
923 png_write_cHRM_fixed(png_structp png_ptr, png_fixed_point white_x,
924    png_fixed_point white_y, png_fixed_point red_x, png_fixed_point red_y,
925    png_fixed_point green_x, png_fixed_point green_y, png_fixed_point blue_x,
926    png_fixed_point blue_y)
927 {
928 #ifdef PNG_USE_LOCAL_ARRAYS
929    PNG_cHRM;
930 #endif
931    png_byte buf[32];
932
933    png_debug(1, "in png_write_cHRM\n");
934    /* each value is saved in 1/100,000ths */
935    if (white_x > 80000L || white_y > 80000L || white_x + white_y > 100000L)
936    {
937       png_warning(png_ptr, "Invalid fixed cHRM white point specified");
938 #if !defined(PNG_NO_CONSOLE_IO)
939       fprintf(stderr,"white_x=%ld, white_y=%ld\n",white_x, white_y);
940 #endif
941       return;
942    }
943    png_save_uint_32(buf, (png_uint_32)white_x);
944    png_save_uint_32(buf + 4, (png_uint_32)white_y);
945
946    if (red_x > 80000L || red_y > 80000L || red_x + red_y > 100000L)
947    {
948       png_warning(png_ptr, "Invalid cHRM fixed red point specified");
949       return;
950    }
951    png_save_uint_32(buf + 8, (png_uint_32)red_x);
952    png_save_uint_32(buf + 12, (png_uint_32)red_y);
953
954    if (green_x > 80000L || green_y > 80000L || green_x + green_y > 100000L)
955    {
956       png_warning(png_ptr, "Invalid fixed cHRM green point specified");
957       return;
958    }
959    png_save_uint_32(buf + 16, (png_uint_32)green_x);
960    png_save_uint_32(buf + 20, (png_uint_32)green_y);
961
962    if (blue_x > 80000L || blue_y > 80000L || blue_x + blue_y > 100000L)
963    {
964       png_warning(png_ptr, "Invalid fixed cHRM blue point specified");
965       return;
966    }
967    png_save_uint_32(buf + 24, (png_uint_32)blue_x);
968    png_save_uint_32(buf + 28, (png_uint_32)blue_y);
969
970    png_write_chunk(png_ptr, (png_bytep)png_cHRM, buf, (png_size_t)32);
971 }
972 #endif
973 #endif
974
975 #if defined(PNG_WRITE_tRNS_SUPPORTED)
976 /* write the tRNS chunk */
977 void /* PRIVATE */
978 png_write_tRNS(png_structp png_ptr, png_bytep trans, png_color_16p tran,
979    int num_trans, int color_type)
980 {
981 #ifdef PNG_USE_LOCAL_ARRAYS
982    PNG_tRNS;
983 #endif
984    png_byte buf[6];
985
986    png_debug(1, "in png_write_tRNS\n");
987    if (color_type == PNG_COLOR_TYPE_PALETTE)
988    {
989       if (num_trans <= 0 || num_trans > (int)png_ptr->num_palette)
990       {
991          png_warning(png_ptr,"Invalid number of transparent colors specified");
992          return;
993       }
994       /* write the chunk out as it is */
995       png_write_chunk(png_ptr, (png_bytep)png_tRNS, trans, (png_size_t)num_trans);
996    }
997    else if (color_type == PNG_COLOR_TYPE_GRAY)
998    {
999       /* one 16 bit value */
1000       if(tran->gray >= (1 << png_ptr->bit_depth))
1001       {
1002          png_warning(png_ptr,
1003            "Ignoring attempt to write tRNS chunk out-of-range for bit_depth");
1004          return;
1005       }
1006       png_save_uint_16(buf, tran->gray);
1007       png_write_chunk(png_ptr, (png_bytep)png_tRNS, buf, (png_size_t)2);
1008    }
1009    else if (color_type == PNG_COLOR_TYPE_RGB)
1010    {
1011       /* three 16 bit values */
1012       png_save_uint_16(buf, tran->red);
1013       png_save_uint_16(buf + 2, tran->green);
1014       png_save_uint_16(buf + 4, tran->blue);
1015       if(png_ptr->bit_depth == 8 && (buf[0] | buf[2] | buf[4]))
1016          {
1017             png_warning(png_ptr,
1018               "Ignoring attempt to write 16-bit tRNS chunk when bit_depth is 8");
1019             return;
1020          }
1021       png_write_chunk(png_ptr, (png_bytep)png_tRNS, buf, (png_size_t)6);
1022    }
1023    else
1024    {
1025       png_warning(png_ptr, "Can't write tRNS with an alpha channel");
1026    }
1027 }
1028 #endif
1029
1030 #if defined(PNG_WRITE_bKGD_SUPPORTED)
1031 /* write the background chunk */
1032 void /* PRIVATE */
1033 png_write_bKGD(png_structp png_ptr, png_color_16p back, int color_type)
1034 {
1035 #ifdef PNG_USE_LOCAL_ARRAYS
1036    PNG_bKGD;
1037 #endif
1038    png_byte buf[6];
1039
1040    png_debug(1, "in png_write_bKGD\n");
1041    if (color_type == PNG_COLOR_TYPE_PALETTE)
1042    {
1043       if (
1044 #if defined(PNG_MNG_FEATURES_SUPPORTED)
1045           (png_ptr->num_palette ||
1046           (!(png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE))) &&
1047 #endif
1048          back->index > png_ptr->num_palette)
1049       {
1050          png_warning(png_ptr, "Invalid background palette index");
1051          return;
1052       }
1053       buf[0] = back->index;
1054       png_write_chunk(png_ptr, (png_bytep)png_bKGD, buf, (png_size_t)1);
1055    }
1056    else if (color_type & PNG_COLOR_MASK_COLOR)
1057    {
1058       png_save_uint_16(buf, back->red);
1059       png_save_uint_16(buf + 2, back->green);
1060       png_save_uint_16(buf + 4, back->blue);
1061       if(png_ptr->bit_depth == 8 && (buf[0] | buf[2] | buf[4]))
1062          {
1063             png_warning(png_ptr,
1064               "Ignoring attempt to write 16-bit bKGD chunk when bit_depth is 8");
1065             return;
1066          }
1067       png_write_chunk(png_ptr, (png_bytep)png_bKGD, buf, (png_size_t)6);
1068    }
1069    else
1070    {
1071       if(back->gray >= (1 << png_ptr->bit_depth))
1072       {
1073          png_warning(png_ptr,
1074            "Ignoring attempt to write bKGD chunk out-of-range for bit_depth");
1075          return;
1076       }
1077       png_save_uint_16(buf, back->gray);
1078       png_write_chunk(png_ptr, (png_bytep)png_bKGD, buf, (png_size_t)2);
1079    }
1080 }
1081 #endif
1082
1083 #if defined(PNG_WRITE_hIST_SUPPORTED)
1084 /* write the histogram */
1085 void /* PRIVATE */
1086 png_write_hIST(png_structp png_ptr, png_uint_16p hist, int num_hist)
1087 {
1088 #ifdef PNG_USE_LOCAL_ARRAYS
1089    PNG_hIST;
1090 #endif
1091    int i;
1092    png_byte buf[3];
1093
1094    png_debug(1, "in png_write_hIST\n");
1095    if (num_hist > (int)png_ptr->num_palette)
1096    {
1097       png_debug2(3, "num_hist = %d, num_palette = %d\n", num_hist,
1098          png_ptr->num_palette);
1099       png_warning(png_ptr, "Invalid number of histogram entries specified");
1100       return;
1101    }
1102
1103    png_write_chunk_start(png_ptr, (png_bytep)png_hIST, (png_uint_32)(num_hist * 2));
1104    for (i = 0; i < num_hist; i++)
1105    {
1106       png_save_uint_16(buf, hist[i]);
1107       png_write_chunk_data(png_ptr, buf, (png_size_t)2);
1108    }
1109    png_write_chunk_end(png_ptr);
1110 }
1111 #endif
1112
1113 #if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_pCAL_SUPPORTED) || \
1114     defined(PNG_WRITE_iCCP_SUPPORTED) || defined(PNG_WRITE_sPLT_SUPPORTED)
1115 /* Check that the tEXt or zTXt keyword is valid per PNG 1.0 specification,
1116  * and if invalid, correct the keyword rather than discarding the entire
1117  * chunk.  The PNG 1.0 specification requires keywords 1-79 characters in
1118  * length, forbids leading or trailing whitespace, multiple internal spaces,
1119  * and the non-break space (0x80) from ISO 8859-1.  Returns keyword length.
1120  *
1121  * The new_key is allocated to hold the corrected keyword and must be freed
1122  * by the calling routine.  This avoids problems with trying to write to
1123  * static keywords without having to have duplicate copies of the strings.
1124  */
1125 png_size_t /* PRIVATE */
1126 png_check_keyword(png_structp png_ptr, png_charp key, png_charpp new_key)
1127 {
1128    png_size_t key_len;
1129    png_charp kp, dp;
1130    int kflag;
1131    int kwarn=0;
1132
1133    png_debug(1, "in png_check_keyword\n");
1134    *new_key = NULL;
1135
1136    if (key == NULL || (key_len = png_strlen(key)) == 0)
1137    {
1138       png_warning(png_ptr, "zero length keyword");
1139       return ((png_size_t)0);
1140    }
1141
1142    png_debug1(2, "Keyword to be checked is '%s'\n", key);
1143
1144    *new_key = (png_charp)png_malloc(png_ptr, (png_uint_32)(key_len + 2));
1145
1146    /* Replace non-printing characters with a blank and print a warning */
1147    for (kp = key, dp = *new_key; *kp != '\0'; kp++, dp++)
1148    {
1149       if (*kp < 0x20 || (*kp > 0x7E && (png_byte)*kp < 0xA1))
1150       {
1151 #if !defined(PNG_NO_STDIO) && !defined(_WIN32_WCE)
1152          char msg[40];
1153
1154          sprintf(msg, "invalid keyword character 0x%02X", *kp);
1155          png_warning(png_ptr, msg);
1156 #else
1157          png_warning(png_ptr, "invalid character in keyword");
1158 #endif
1159          *dp = ' ';
1160       }
1161       else
1162       {
1163          *dp = *kp;
1164       }
1165    }
1166    *dp = '\0';
1167
1168    /* Remove any trailing white space. */
1169    kp = *new_key + key_len - 1;
1170    if (*kp == ' ')
1171    {
1172       png_warning(png_ptr, "trailing spaces removed from keyword");
1173
1174       while (*kp == ' ')
1175       {
1176         *(kp--) = '\0';
1177         key_len--;
1178       }
1179    }
1180
1181    /* Remove any leading white space. */
1182    kp = *new_key;
1183    if (*kp == ' ')
1184    {
1185       png_warning(png_ptr, "leading spaces removed from keyword");
1186
1187       while (*kp == ' ')
1188       {
1189         kp++;
1190         key_len--;
1191       }
1192    }
1193
1194    png_debug1(2, "Checking for multiple internal spaces in '%s'\n", kp);
1195
1196    /* Remove multiple internal spaces. */
1197    for (kflag = 0, dp = *new_key; *kp != '\0'; kp++)
1198    {
1199       if (*kp == ' ' && kflag == 0)
1200       {
1201          *(dp++) = *kp;
1202          kflag = 1;
1203       }
1204       else if (*kp == ' ')
1205       {
1206          key_len--;
1207          kwarn=1;
1208       }
1209       else
1210       {
1211          *(dp++) = *kp;
1212          kflag = 0;
1213       }
1214    }
1215    *dp = '\0';
1216    if(kwarn)
1217       png_warning(png_ptr, "extra interior spaces removed from keyword");
1218
1219    if (key_len == 0)
1220    {
1221       png_free(png_ptr, *new_key);
1222       *new_key=NULL;
1223       png_warning(png_ptr, "Zero length keyword");
1224    }
1225
1226    if (key_len > 79)
1227    {
1228       png_warning(png_ptr, "keyword length must be 1 - 79 characters");
1229       new_key[79] = '\0';
1230       key_len = 79;
1231    }
1232
1233    return (key_len);
1234 }
1235 #endif
1236
1237 #if defined(PNG_WRITE_tEXt_SUPPORTED)
1238 /* write a tEXt chunk */
1239 void /* PRIVATE */
1240 png_write_tEXt(png_structp png_ptr, png_charp key, png_charp text,
1241    png_size_t text_len)
1242 {
1243 #ifdef PNG_USE_LOCAL_ARRAYS
1244    PNG_tEXt;
1245 #endif
1246    png_size_t key_len;
1247    png_charp new_key;
1248
1249    png_debug(1, "in png_write_tEXt\n");
1250    if (key == NULL || (key_len = png_check_keyword(png_ptr, key, &new_key))==0)
1251    {
1252       png_warning(png_ptr, "Empty keyword in tEXt chunk");
1253       return;
1254    }
1255
1256    if (text == NULL || *text == '\0')
1257       text_len = 0;
1258    else
1259       text_len = png_strlen(text);
1260
1261    /* make sure we include the 0 after the key */
1262    png_write_chunk_start(png_ptr, (png_bytep)png_tEXt, (png_uint_32)key_len+text_len+1);
1263    /*
1264     * We leave it to the application to meet PNG-1.0 requirements on the
1265     * contents of the text.  PNG-1.0 through PNG-1.2 discourage the use of
1266     * any non-Latin-1 characters except for NEWLINE.  ISO PNG will forbid them.
1267     * The NUL character is forbidden by PNG-1.0 through PNG-1.2 and ISO PNG.
1268     */
1269    png_write_chunk_data(png_ptr, (png_bytep)new_key, key_len + 1);
1270    if (text_len)
1271       png_write_chunk_data(png_ptr, (png_bytep)text, text_len);
1272
1273    png_write_chunk_end(png_ptr);
1274    png_free(png_ptr, new_key);
1275 }
1276 #endif
1277
1278 #if defined(PNG_WRITE_zTXt_SUPPORTED)
1279 /* write a compressed text chunk */
1280 void /* PRIVATE */
1281 png_write_zTXt(png_structp png_ptr, png_charp key, png_charp text,
1282    png_size_t text_len, int compression)
1283 {
1284 #ifdef PNG_USE_LOCAL_ARRAYS
1285    PNG_zTXt;
1286 #endif
1287    png_size_t key_len;
1288    char buf[1];
1289    png_charp new_key;
1290    compression_state comp;
1291
1292    png_debug(1, "in png_write_zTXt\n");
1293
1294    if (key == NULL || (key_len = png_check_keyword(png_ptr, key, &new_key))==0)
1295    {
1296       png_warning(png_ptr, "Empty keyword in zTXt chunk");
1297       return;
1298    }
1299
1300    if (text == NULL || *text == '\0' || compression==PNG_TEXT_COMPRESSION_NONE)
1301    {
1302       png_write_tEXt(png_ptr, new_key, text, (png_size_t)0);
1303       png_free(png_ptr, new_key);
1304       return;
1305    }
1306
1307    text_len = png_strlen(text);
1308
1309    png_free(png_ptr, new_key);
1310
1311    /* compute the compressed data; do it now for the length */
1312    text_len = png_text_compress(png_ptr, text, text_len, compression,
1313        &comp);
1314
1315    /* write start of chunk */
1316    png_write_chunk_start(png_ptr, (png_bytep)png_zTXt, (png_uint_32)
1317       (key_len+text_len+2));
1318    /* write key */
1319    png_write_chunk_data(png_ptr, (png_bytep)key, key_len + 1);
1320    buf[0] = (png_byte)compression;
1321    /* write compression */
1322    png_write_chunk_data(png_ptr, (png_bytep)buf, (png_size_t)1);
1323    /* write the compressed data */
1324    png_write_compressed_data_out(png_ptr, &comp);
1325
1326    /* close the chunk */
1327    png_write_chunk_end(png_ptr);
1328 }
1329 #endif
1330
1331 #if defined(PNG_WRITE_iTXt_SUPPORTED)
1332 /* write an iTXt chunk */
1333 void /* PRIVATE */
1334 png_write_iTXt(png_structp png_ptr, int compression, png_charp key,
1335     png_charp lang, png_charp lang_key, png_charp text)
1336 {
1337 #ifdef PNG_USE_LOCAL_ARRAYS
1338    PNG_iTXt;
1339 #endif
1340    png_size_t lang_len, key_len, lang_key_len, text_len;
1341    png_charp new_lang, new_key;
1342    png_byte cbuf[2];
1343    compression_state comp;
1344
1345    png_debug(1, "in png_write_iTXt\n");
1346
1347    if (key == NULL || (key_len = png_check_keyword(png_ptr, key, &new_key))==0)
1348    {
1349       png_warning(png_ptr, "Empty keyword in iTXt chunk");
1350       return;
1351    }
1352    if (lang == NULL || (lang_len = png_check_keyword(png_ptr, lang, &new_lang))==0)
1353    {
1354       png_warning(png_ptr, "Empty language field in iTXt chunk");
1355       new_lang = NULL;
1356       lang_len = 0;      
1357    }
1358
1359    if (lang_key == NULL)
1360      lang_key_len = 0;
1361    else
1362      lang_key_len = png_strlen(lang_key);
1363
1364    if (text == NULL)
1365       text_len = 0;
1366    else
1367      text_len = png_strlen(text);
1368
1369    /* compute the compressed data; do it now for the length */
1370    text_len = png_text_compress(png_ptr, text, text_len, compression-2,
1371       &comp);
1372
1373
1374    /* make sure we include the compression flag, the compression byte,
1375     * and the NULs after the key, lang, and lang_key parts */
1376
1377    png_write_chunk_start(png_ptr, (png_bytep)png_iTXt,
1378           (png_uint_32)(
1379         5 /* comp byte, comp flag, terminators for key, lang and lang_key */
1380         + key_len
1381         + lang_len
1382         + lang_key_len
1383         + text_len));
1384
1385    /*
1386     * We leave it to the application to meet PNG-1.0 requirements on the
1387     * contents of the text.  PNG-1.0 through PNG-1.2 discourage the use of
1388     * any non-Latin-1 characters except for NEWLINE.  ISO PNG will forbid them.
1389     * The NUL character is forbidden by PNG-1.0 through PNG-1.2 and ISO PNG.
1390     */
1391    png_write_chunk_data(png_ptr, (png_bytep)new_key, key_len + 1);
1392
1393    /* set the compression flag */
1394    if (compression == PNG_ITXT_COMPRESSION_NONE || \
1395        compression == PNG_TEXT_COMPRESSION_NONE)
1396        cbuf[0] = 0;
1397    else /* compression == PNG_ITXT_COMPRESSION_zTXt */
1398        cbuf[0] = 1;
1399    /* set the compression method */
1400    cbuf[1] = 0;
1401    png_write_chunk_data(png_ptr, cbuf, 2);
1402
1403    cbuf[0] = 0;
1404    png_write_chunk_data(png_ptr, (new_lang ? (png_bytep)new_lang : cbuf), lang_len + 1);
1405    png_write_chunk_data(png_ptr, (lang_key ? (png_bytep)lang_key : cbuf), lang_key_len + 1);
1406    png_write_compressed_data_out(png_ptr, &comp);
1407
1408    png_write_chunk_end(png_ptr);
1409    png_free(png_ptr, new_key);
1410    if (new_lang)
1411      png_free(png_ptr, new_lang);
1412 }
1413 #endif
1414
1415 #if defined(PNG_WRITE_oFFs_SUPPORTED)
1416 /* write the oFFs chunk */
1417 void /* PRIVATE */
1418 png_write_oFFs(png_structp png_ptr, png_int_32 x_offset, png_int_32 y_offset,
1419    int unit_type)
1420 {
1421 #ifdef PNG_USE_LOCAL_ARRAYS
1422    PNG_oFFs;
1423 #endif
1424    png_byte buf[9];
1425
1426    png_debug(1, "in png_write_oFFs\n");
1427    if (unit_type >= PNG_OFFSET_LAST)
1428       png_warning(png_ptr, "Unrecognized unit type for oFFs chunk");
1429
1430    png_save_int_32(buf, x_offset);
1431    png_save_int_32(buf + 4, y_offset);
1432    buf[8] = (png_byte)unit_type;
1433
1434    png_write_chunk(png_ptr, (png_bytep)png_oFFs, buf, (png_size_t)9);
1435 }
1436 #endif
1437
1438 #if defined(PNG_WRITE_pCAL_SUPPORTED)
1439 /* write the pCAL chunk (described in the PNG extensions document) */
1440 void /* PRIVATE */
1441 png_write_pCAL(png_structp png_ptr, png_charp purpose, png_int_32 X0,
1442    png_int_32 X1, int type, int nparams, png_charp units, png_charpp params)
1443 {
1444 #ifdef PNG_USE_LOCAL_ARRAYS
1445    PNG_pCAL;
1446 #endif
1447    png_size_t purpose_len, units_len, total_len;
1448    png_uint_32p params_len;
1449    png_byte buf[10];
1450    png_charp new_purpose;
1451    int i;
1452
1453    png_debug1(1, "in png_write_pCAL (%d parameters)\n", nparams);
1454    if (type >= PNG_EQUATION_LAST)
1455       png_warning(png_ptr, "Unrecognized equation type for pCAL chunk");
1456
1457    purpose_len = png_check_keyword(png_ptr, purpose, &new_purpose) + 1;
1458    png_debug1(3, "pCAL purpose length = %d\n", (int)purpose_len);
1459    units_len = png_strlen(units) + (nparams == 0 ? 0 : 1);
1460    png_debug1(3, "pCAL units length = %d\n", (int)units_len);
1461    total_len = purpose_len + units_len + 10;
1462
1463    params_len = (png_uint_32p)png_malloc(png_ptr, (png_uint_32)(nparams
1464       *sizeof(png_uint_32)));
1465
1466    /* Find the length of each parameter, making sure we don't count the
1467       null terminator for the last parameter. */
1468    for (i = 0; i < nparams; i++)
1469    {
1470       params_len[i] = png_strlen(params[i]) + (i == nparams - 1 ? 0 : 1);
1471       png_debug2(3, "pCAL parameter %d length = %lu\n", i, params_len[i]);
1472       total_len += (png_size_t)params_len[i];
1473    }
1474
1475    png_debug1(3, "pCAL total length = %d\n", (int)total_len);
1476    png_write_chunk_start(png_ptr, (png_bytep)png_pCAL, (png_uint_32)total_len);
1477    png_write_chunk_data(png_ptr, (png_bytep)new_purpose, purpose_len);
1478    png_save_int_32(buf, X0);
1479    png_save_int_32(buf + 4, X1);
1480    buf[8] = (png_byte)type;
1481    buf[9] = (png_byte)nparams;
1482    png_write_chunk_data(png_ptr, buf, (png_size_t)10);
1483    png_write_chunk_data(png_ptr, (png_bytep)units, (png_size_t)units_len);
1484
1485    png_free(png_ptr, new_purpose);
1486
1487    for (i = 0; i < nparams; i++)
1488    {
1489       png_write_chunk_data(png_ptr, (png_bytep)params[i],
1490          (png_size_t)params_len[i]);
1491    }
1492
1493    png_free(png_ptr, params_len);
1494    png_write_chunk_end(png_ptr);
1495 }
1496 #endif
1497
1498 #if defined(PNG_WRITE_sCAL_SUPPORTED)
1499 /* write the sCAL chunk */
1500 #if defined(PNG_FLOATING_POINT_SUPPORTED) && !defined(PNG_NO_STDIO)
1501 void /* PRIVATE */
1502 png_write_sCAL(png_structp png_ptr, int unit, double width,double height)
1503 {
1504 #ifdef PNG_USE_LOCAL_ARRAYS
1505    PNG_sCAL;
1506 #endif
1507    png_size_t total_len;
1508    char wbuf[32], hbuf[32];
1509
1510    png_debug(1, "in png_write_sCAL\n");
1511
1512 #if defined(_WIN32_WCE)
1513 /* sprintf() function is not supported on WindowsCE */
1514    {
1515       wchar_t wc_buf[32];
1516       swprintf(wc_buf, TEXT("%12.12e"), width);
1517       WideCharToMultiByte(CP_ACP, 0, wc_buf, -1, wbuf, 32, NULL, NULL);
1518       swprintf(wc_buf, TEXT("%12.12e"), height);
1519       WideCharToMultiByte(CP_ACP, 0, wc_buf, -1, hbuf, 32, NULL, NULL);
1520    }
1521 #else
1522    sprintf(wbuf, "%12.12e", width);
1523    sprintf(hbuf, "%12.12e", height);
1524 #endif
1525    total_len = 1 + png_strlen(wbuf)+1 + png_strlen(hbuf);
1526
1527    png_debug1(3, "sCAL total length = %d\n", (int)total_len);
1528    png_write_chunk_start(png_ptr, (png_bytep)png_sCAL, (png_uint_32)total_len);
1529    png_write_chunk_data(png_ptr, (png_bytep)&unit, 1);
1530    png_write_chunk_data(png_ptr, (png_bytep)wbuf, png_strlen(wbuf)+1);
1531    png_write_chunk_data(png_ptr, (png_bytep)hbuf, png_strlen(hbuf));
1532
1533    png_write_chunk_end(png_ptr);
1534 }
1535 #else
1536 #ifdef PNG_FIXED_POINT_SUPPORTED
1537 void /* PRIVATE */
1538 png_write_sCAL_s(png_structp png_ptr, int unit, png_charp width,
1539    png_charp height)
1540 {
1541 #ifdef PNG_USE_LOCAL_ARRAYS
1542    PNG_sCAL;
1543 #endif
1544    png_size_t total_len;
1545    char wbuf[32], hbuf[32];
1546
1547    png_debug(1, "in png_write_sCAL_s\n");
1548
1549    png_strcpy(wbuf,(const char *)width);
1550    png_strcpy(hbuf,(const char *)height);
1551    total_len = 1 + png_strlen(wbuf)+1 + png_strlen(hbuf);
1552
1553    png_debug1(3, "sCAL total length = %d\n", total_len);
1554    png_write_chunk_start(png_ptr, (png_bytep)png_sCAL, (png_uint_32)total_len);
1555    png_write_chunk_data(png_ptr, (png_bytep)&unit, 1);
1556    png_write_chunk_data(png_ptr, (png_bytep)wbuf, png_strlen(wbuf)+1);
1557    png_write_chunk_data(png_ptr, (png_bytep)hbuf, png_strlen(hbuf));
1558
1559    png_write_chunk_end(png_ptr);
1560 }
1561 #endif
1562 #endif
1563 #endif
1564
1565 #if defined(PNG_WRITE_pHYs_SUPPORTED)
1566 /* write the pHYs chunk */
1567 void /* PRIVATE */
1568 png_write_pHYs(png_structp png_ptr, png_uint_32 x_pixels_per_unit,
1569    png_uint_32 y_pixels_per_unit,
1570    int unit_type)
1571 {
1572 #ifdef PNG_USE_LOCAL_ARRAYS
1573    PNG_pHYs;
1574 #endif
1575    png_byte buf[9];
1576
1577    png_debug(1, "in png_write_pHYs\n");
1578    if (unit_type >= PNG_RESOLUTION_LAST)
1579       png_warning(png_ptr, "Unrecognized unit type for pHYs chunk");
1580
1581    png_save_uint_32(buf, x_pixels_per_unit);
1582    png_save_uint_32(buf + 4, y_pixels_per_unit);
1583    buf[8] = (png_byte)unit_type;
1584
1585    png_write_chunk(png_ptr, (png_bytep)png_pHYs, buf, (png_size_t)9);
1586 }
1587 #endif
1588
1589 #if defined(PNG_WRITE_tIME_SUPPORTED)
1590 /* Write the tIME chunk.  Use either png_convert_from_struct_tm()
1591  * or png_convert_from_time_t(), or fill in the structure yourself.
1592  */
1593 void /* PRIVATE */
1594 png_write_tIME(png_structp png_ptr, png_timep mod_time)
1595 {
1596 #ifdef PNG_USE_LOCAL_ARRAYS
1597    PNG_tIME;
1598 #endif
1599    png_byte buf[7];
1600
1601    png_debug(1, "in png_write_tIME\n");
1602    if (mod_time->month  > 12 || mod_time->month  < 1 ||
1603        mod_time->day    > 31 || mod_time->day    < 1 ||
1604        mod_time->hour   > 23 || mod_time->second > 60)
1605    {
1606       png_warning(png_ptr, "Invalid time specified for tIME chunk");
1607       return;
1608    }
1609
1610    png_save_uint_16(buf, mod_time->year);
1611    buf[2] = mod_time->month;
1612    buf[3] = mod_time->day;
1613    buf[4] = mod_time->hour;
1614    buf[5] = mod_time->minute;
1615    buf[6] = mod_time->second;
1616
1617    png_write_chunk(png_ptr, (png_bytep)png_tIME, buf, (png_size_t)7);
1618 }
1619 #endif
1620
1621 /* initializes the row writing capability of libpng */
1622 void /* PRIVATE */
1623 png_write_start_row(png_structp png_ptr)
1624 {
1625 #ifdef PNG_USE_LOCAL_ARRAYS
1626    /* arrays to facilitate easy interlacing - use pass (0 - 6) as index */
1627
1628    /* start of interlace block */
1629    int png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
1630
1631    /* offset to next interlace block */
1632    int png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
1633
1634    /* start of interlace block in the y direction */
1635    int png_pass_ystart[7] = {0, 0, 4, 0, 2, 0, 1};
1636
1637    /* offset to next interlace block in the y direction */
1638    int png_pass_yinc[7] = {8, 8, 8, 4, 4, 2, 2};
1639 #endif
1640
1641    png_size_t buf_size;
1642
1643    png_debug(1, "in png_write_start_row\n");
1644    buf_size = (png_size_t)(((png_ptr->width * png_ptr->usr_channels *
1645                             png_ptr->usr_bit_depth + 7) >> 3) + 1);
1646
1647    /* set up row buffer */
1648    png_ptr->row_buf = (png_bytep)png_malloc(png_ptr, (png_uint_32)buf_size);
1649    png_ptr->row_buf[0] = PNG_FILTER_VALUE_NONE;
1650
1651    /* set up filtering buffer, if using this filter */
1652    if (png_ptr->do_filter & PNG_FILTER_SUB)
1653    {
1654       png_ptr->sub_row = (png_bytep)png_malloc(png_ptr,
1655          (png_ptr->rowbytes + 1));
1656       png_ptr->sub_row[0] = PNG_FILTER_VALUE_SUB;
1657    }
1658
1659    /* We only need to keep the previous row if we are using one of these. */
1660    if (png_ptr->do_filter & (PNG_FILTER_AVG | PNG_FILTER_UP | PNG_FILTER_PAETH))
1661    {
1662      /* set up previous row buffer */
1663       png_ptr->prev_row = (png_bytep)png_malloc(png_ptr, (png_uint_32)buf_size);
1664       png_memset(png_ptr->prev_row, 0, buf_size);
1665
1666       if (png_ptr->do_filter & PNG_FILTER_UP)
1667       {
1668          png_ptr->up_row = (png_bytep )png_malloc(png_ptr,
1669             (png_ptr->rowbytes + 1));
1670          png_ptr->up_row[0] = PNG_FILTER_VALUE_UP;
1671       }
1672
1673       if (png_ptr->do_filter & PNG_FILTER_AVG)
1674       {
1675          png_ptr->avg_row = (png_bytep)png_malloc(png_ptr,
1676             (png_ptr->rowbytes + 1));
1677          png_ptr->avg_row[0] = PNG_FILTER_VALUE_AVG;
1678       }
1679
1680       if (png_ptr->do_filter & PNG_FILTER_PAETH)
1681       {
1682          png_ptr->paeth_row = (png_bytep )png_malloc(png_ptr,
1683             (png_ptr->rowbytes + 1));
1684          png_ptr->paeth_row[0] = PNG_FILTER_VALUE_PAETH;
1685       }
1686    }
1687
1688 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
1689    /* if interlaced, we need to set up width and height of pass */
1690    if (png_ptr->interlaced)
1691    {
1692       if (!(png_ptr->transformations & PNG_INTERLACE))
1693       {
1694          png_ptr->num_rows = (png_ptr->height + png_pass_yinc[0] - 1 -
1695             png_pass_ystart[0]) / png_pass_yinc[0];
1696          png_ptr->usr_width = (png_ptr->width + png_pass_inc[0] - 1 -
1697             png_pass_start[0]) / png_pass_inc[0];
1698       }
1699       else
1700       {
1701          png_ptr->num_rows = png_ptr->height;
1702          png_ptr->usr_width = png_ptr->width;
1703       }
1704    }
1705    else
1706 #endif
1707    {
1708       png_ptr->num_rows = png_ptr->height;
1709       png_ptr->usr_width = png_ptr->width;
1710    }
1711    png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
1712    png_ptr->zstream.next_out = png_ptr->zbuf;
1713 }
1714
1715 /* Internal use only.  Called when finished processing a row of data. */
1716 void /* PRIVATE */
1717 png_write_finish_row(png_structp png_ptr)
1718 {
1719 #ifdef PNG_USE_LOCAL_ARRAYS
1720    /* arrays to facilitate easy interlacing - use pass (0 - 6) as index */
1721
1722    /* start of interlace block */
1723    int png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
1724
1725    /* offset to next interlace block */
1726    int png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
1727
1728    /* start of interlace block in the y direction */
1729    int png_pass_ystart[7] = {0, 0, 4, 0, 2, 0, 1};
1730
1731    /* offset to next interlace block in the y direction */
1732    int png_pass_yinc[7] = {8, 8, 8, 4, 4, 2, 2};
1733 #endif
1734
1735    int ret;
1736
1737    png_debug(1, "in png_write_finish_row\n");
1738    /* next row */
1739    png_ptr->row_number++;
1740
1741    /* see if we are done */
1742    if (png_ptr->row_number < png_ptr->num_rows)
1743       return;
1744
1745 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
1746    /* if interlaced, go to next pass */
1747    if (png_ptr->interlaced)
1748    {
1749       png_ptr->row_number = 0;
1750       if (png_ptr->transformations & PNG_INTERLACE)
1751       {
1752          png_ptr->pass++;
1753       }
1754       else
1755       {
1756          /* loop until we find a non-zero width or height pass */
1757          do
1758          {
1759             png_ptr->pass++;
1760             if (png_ptr->pass >= 7)
1761                break;
1762             png_ptr->usr_width = (png_ptr->width +
1763                png_pass_inc[png_ptr->pass] - 1 -
1764                png_pass_start[png_ptr->pass]) /
1765                png_pass_inc[png_ptr->pass];
1766             png_ptr->num_rows = (png_ptr->height +
1767                png_pass_yinc[png_ptr->pass] - 1 -
1768                png_pass_ystart[png_ptr->pass]) /
1769                png_pass_yinc[png_ptr->pass];
1770             if (png_ptr->transformations & PNG_INTERLACE)
1771                break;
1772          } while (png_ptr->usr_width == 0 || png_ptr->num_rows == 0);
1773
1774       }
1775
1776       /* reset the row above the image for the next pass */
1777       if (png_ptr->pass < 7)
1778       {
1779          if (png_ptr->prev_row != NULL)
1780             png_memset(png_ptr->prev_row, 0,
1781                (png_size_t) (((png_uint_32)png_ptr->usr_channels *
1782                (png_uint_32)png_ptr->usr_bit_depth *
1783                png_ptr->width + 7) >> 3) + 1);
1784          return;
1785       }
1786    }
1787 #endif
1788
1789    /* if we get here, we've just written the last row, so we need
1790       to flush the compressor */
1791    do
1792    {
1793       /* tell the compressor we are done */
1794       ret = deflate(&png_ptr->zstream, Z_FINISH);
1795       /* check for an error */
1796       if (ret == Z_OK)
1797       {
1798          /* check to see if we need more room */
1799          if (!(png_ptr->zstream.avail_out))
1800          {
1801             png_write_IDAT(png_ptr, png_ptr->zbuf, png_ptr->zbuf_size);
1802             png_ptr->zstream.next_out = png_ptr->zbuf;
1803             png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
1804          }
1805       }
1806       else if (ret != Z_STREAM_END)
1807       {
1808          if (png_ptr->zstream.msg != NULL)
1809             png_error(png_ptr, png_ptr->zstream.msg);
1810          else
1811             png_error(png_ptr, "zlib error");
1812       }
1813    } while (ret != Z_STREAM_END);
1814
1815    /* write any extra space */
1816    if (png_ptr->zstream.avail_out < png_ptr->zbuf_size)
1817    {
1818       png_write_IDAT(png_ptr, png_ptr->zbuf, png_ptr->zbuf_size -
1819          png_ptr->zstream.avail_out);
1820    }
1821
1822    deflateReset(&png_ptr->zstream);
1823 }
1824
1825 #if defined(PNG_WRITE_INTERLACING_SUPPORTED)
1826 /* Pick out the correct pixels for the interlace pass.
1827  * The basic idea here is to go through the row with a source
1828  * pointer and a destination pointer (sp and dp), and copy the
1829  * correct pixels for the pass.  As the row gets compacted,
1830  * sp will always be >= dp, so we should never overwrite anything.
1831  * See the default: case for the easiest code to understand.
1832  */
1833 void /* PRIVATE */
1834 png_do_write_interlace(png_row_infop row_info, png_bytep row, int pass)
1835 {
1836 #ifdef PNG_USE_LOCAL_ARRAYS
1837    /* arrays to facilitate easy interlacing - use pass (0 - 6) as index */
1838
1839    /* start of interlace block */
1840    int png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
1841
1842    /* offset to next interlace block */
1843    int png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
1844 #endif
1845
1846    png_debug(1, "in png_do_write_interlace\n");
1847    /* we don't have to do anything on the last pass (6) */
1848 #if defined(PNG_USELESS_TESTS_SUPPORTED)
1849    if (row != NULL && row_info != NULL && pass < 6)
1850 #else
1851    if (pass < 6)
1852 #endif
1853    {
1854       /* each pixel depth is handled separately */
1855       switch (row_info->pixel_depth)
1856       {
1857          case 1:
1858          {
1859             png_bytep sp;
1860             png_bytep dp;
1861             int shift;
1862             int d;
1863             int value;
1864             png_uint_32 i;
1865             png_uint_32 row_width = row_info->width;
1866
1867             dp = row;
1868             d = 0;
1869             shift = 7;
1870             for (i = png_pass_start[pass]; i < row_width;
1871                i += png_pass_inc[pass])
1872             {
1873                sp = row + (png_size_t)(i >> 3);
1874                value = (int)(*sp >> (7 - (int)(i & 0x07))) & 0x01;
1875                d |= (value << shift);
1876
1877                if (shift == 0)
1878                {
1879                   shift = 7;
1880                   *dp++ = (png_byte)d;
1881                   d = 0;
1882                }
1883                else
1884                   shift--;
1885
1886             }
1887             if (shift != 7)
1888                *dp = (png_byte)d;
1889             break;
1890          }
1891          case 2:
1892          {
1893             png_bytep sp;
1894             png_bytep dp;
1895             int shift;
1896             int d;
1897             int value;
1898             png_uint_32 i;
1899             png_uint_32 row_width = row_info->width;
1900
1901             dp = row;
1902             shift = 6;
1903             d = 0;
1904             for (i = png_pass_start[pass]; i < row_width;
1905                i += png_pass_inc[pass])
1906             {
1907                sp = row + (png_size_t)(i >> 2);
1908                value = (*sp >> ((3 - (int)(i & 0x03)) << 1)) & 0x03;
1909                d |= (value << shift);
1910
1911                if (shift == 0)
1912                {
1913                   shift = 6;
1914                   *dp++ = (png_byte)d;
1915                   d = 0;
1916                }
1917                else
1918                   shift -= 2;
1919             }
1920             if (shift != 6)
1921                    *dp = (png_byte)d;
1922             break;
1923          }
1924          case 4:
1925          {
1926             png_bytep sp;
1927             png_bytep dp;
1928             int shift;
1929             int d;
1930             int value;
1931             png_uint_32 i;
1932             png_uint_32 row_width = row_info->width;
1933
1934             dp = row;
1935             shift = 4;
1936             d = 0;
1937             for (i = png_pass_start[pass]; i < row_width;
1938                i += png_pass_inc[pass])
1939             {
1940                sp = row + (png_size_t)(i >> 1);
1941                value = (*sp >> ((1 - (int)(i & 0x01)) << 2)) & 0x0f;
1942                d |= (value << shift);
1943
1944                if (shift == 0)
1945                {
1946                   shift = 4;
1947                   *dp++ = (png_byte)d;
1948                   d = 0;
1949                }
1950                else
1951                   shift -= 4;
1952             }
1953             if (shift != 4)
1954                *dp = (png_byte)d;
1955             break;
1956          }
1957          default:
1958          {
1959             png_bytep sp;
1960             png_bytep dp;
1961             png_uint_32 i;
1962             png_uint_32 row_width = row_info->width;
1963             png_size_t pixel_bytes;
1964
1965             /* start at the beginning */
1966             dp = row;
1967             /* find out how many bytes each pixel takes up */
1968             pixel_bytes = (row_info->pixel_depth >> 3);
1969             /* loop through the row, only looking at the pixels that
1970                matter */
1971             for (i = png_pass_start[pass]; i < row_width;
1972                i += png_pass_inc[pass])
1973             {
1974                /* find out where the original pixel is */
1975                sp = row + (png_size_t)i * pixel_bytes;
1976                /* move the pixel */
1977                if (dp != sp)
1978                   png_memcpy(dp, sp, pixel_bytes);
1979                /* next pixel */
1980                dp += pixel_bytes;
1981             }
1982             break;
1983          }
1984       }
1985       /* set new row width */
1986       row_info->width = (row_info->width +
1987          png_pass_inc[pass] - 1 -
1988          png_pass_start[pass]) /
1989          png_pass_inc[pass];
1990          row_info->rowbytes = ((row_info->width *
1991             row_info->pixel_depth + 7) >> 3);
1992    }
1993 }
1994 #endif
1995
1996 /* This filters the row, chooses which filter to use, if it has not already
1997  * been specified by the application, and then writes the row out with the
1998  * chosen filter.
1999  */
2000 #define PNG_MAXSUM (~((png_uint_32)0) >> 1)
2001 #define PNG_HISHIFT 10
2002 #define PNG_LOMASK ((png_uint_32)0xffffL)
2003 #define PNG_HIMASK ((png_uint_32)(~PNG_LOMASK >> PNG_HISHIFT))
2004 void /* PRIVATE */
2005 png_write_find_filter(png_structp png_ptr, png_row_infop row_info)
2006 {
2007    png_bytep prev_row, best_row, row_buf;
2008    png_uint_32 mins, bpp;
2009    png_byte filter_to_do = png_ptr->do_filter;
2010    png_uint_32 row_bytes = row_info->rowbytes;
2011 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
2012    int num_p_filters = (int)png_ptr->num_prev_filters;
2013 #endif
2014
2015    png_debug(1, "in png_write_find_filter\n");
2016    /* find out how many bytes offset each pixel is */
2017    bpp = (row_info->pixel_depth + 7) / 8;
2018
2019    prev_row = png_ptr->prev_row;
2020    best_row = row_buf = png_ptr->row_buf;
2021    mins = PNG_MAXSUM;
2022
2023    /* The prediction method we use is to find which method provides the
2024     * smallest value when summing the absolute values of the distances
2025     * from zero, using anything >= 128 as negative numbers.  This is known
2026     * as the "minimum sum of absolute differences" heuristic.  Other
2027     * heuristics are the "weighted minimum sum of absolute differences"
2028     * (experimental and can in theory improve compression), and the "zlib
2029     * predictive" method (not implemented yet), which does test compressions
2030     * of lines using different filter methods, and then chooses the
2031     * (series of) filter(s) that give minimum compressed data size (VERY
2032     * computationally expensive).
2033     *
2034     * GRR 980525:  consider also
2035     *   (1) minimum sum of absolute differences from running average (i.e.,
2036     *       keep running sum of non-absolute differences & count of bytes)
2037     *       [track dispersion, too?  restart average if dispersion too large?]
2038     *  (1b) minimum sum of absolute differences from sliding average, probably
2039     *       with window size <= deflate window (usually 32K)
2040     *   (2) minimum sum of squared differences from zero or running average
2041     *       (i.e., ~ root-mean-square approach)
2042     */
2043
2044
2045    /* We don't need to test the 'no filter' case if this is the only filter
2046     * that has been chosen, as it doesn't actually do anything to the data.
2047     */
2048    if ((filter_to_do & PNG_FILTER_NONE) &&
2049        filter_to_do != PNG_FILTER_NONE)
2050    {
2051       png_bytep rp;
2052       png_uint_32 sum = 0;
2053       png_uint_32 i;
2054       int v;
2055
2056       for (i = 0, rp = row_buf + 1; i < row_bytes; i++, rp++)
2057       {
2058          v = *rp;
2059          sum += (v < 128) ? v : 256 - v;
2060       }
2061
2062 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
2063       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
2064       {
2065          png_uint_32 sumhi, sumlo;
2066          int j;
2067          sumlo = sum & PNG_LOMASK;
2068          sumhi = (sum >> PNG_HISHIFT) & PNG_HIMASK; /* Gives us some footroom */
2069
2070          /* Reduce the sum if we match any of the previous rows */
2071          for (j = 0; j < num_p_filters; j++)
2072          {
2073             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_NONE)
2074             {
2075                sumlo = (sumlo * png_ptr->filter_weights[j]) >>
2076                   PNG_WEIGHT_SHIFT;
2077                sumhi = (sumhi * png_ptr->filter_weights[j]) >>
2078                   PNG_WEIGHT_SHIFT;
2079             }
2080          }
2081
2082          /* Factor in the cost of this filter (this is here for completeness,
2083           * but it makes no sense to have a "cost" for the NONE filter, as
2084           * it has the minimum possible computational cost - none).
2085           */
2086          sumlo = (sumlo * png_ptr->filter_costs[PNG_FILTER_VALUE_NONE]) >>
2087             PNG_COST_SHIFT;
2088          sumhi = (sumhi * png_ptr->filter_costs[PNG_FILTER_VALUE_NONE]) >>
2089             PNG_COST_SHIFT;
2090
2091          if (sumhi > PNG_HIMASK)
2092             sum = PNG_MAXSUM;
2093          else
2094             sum = (sumhi << PNG_HISHIFT) + sumlo;
2095       }
2096 #endif
2097       mins = sum;
2098    }
2099
2100    /* sub filter */
2101    if (filter_to_do == PNG_FILTER_SUB)
2102    /* it's the only filter so no testing is needed */
2103    {
2104       png_bytep rp, lp, dp;
2105       png_uint_32 i;
2106       for (i = 0, rp = row_buf + 1, dp = png_ptr->sub_row + 1; i < bpp;
2107            i++, rp++, dp++)
2108       {
2109          *dp = *rp;
2110       }
2111       for (lp = row_buf + 1; i < row_bytes;
2112          i++, rp++, lp++, dp++)
2113       {
2114          *dp = (png_byte)(((int)*rp - (int)*lp) & 0xff);
2115       }
2116       best_row = png_ptr->sub_row;
2117    }
2118
2119    else if (filter_to_do & PNG_FILTER_SUB)
2120    {
2121       png_bytep rp, dp, lp;
2122       png_uint_32 sum = 0, lmins = mins;
2123       png_uint_32 i;
2124       int v;
2125
2126 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
2127       /* We temporarily increase the "minimum sum" by the factor we
2128        * would reduce the sum of this filter, so that we can do the
2129        * early exit comparison without scaling the sum each time.
2130        */
2131       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
2132       {
2133          int j;
2134          png_uint_32 lmhi, lmlo;
2135          lmlo = lmins & PNG_LOMASK;
2136          lmhi = (lmins >> PNG_HISHIFT) & PNG_HIMASK;
2137
2138          for (j = 0; j < num_p_filters; j++)
2139          {
2140             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_SUB)
2141             {
2142                lmlo = (lmlo * png_ptr->inv_filter_weights[j]) >>
2143                   PNG_WEIGHT_SHIFT;
2144                lmhi = (lmhi * png_ptr->inv_filter_weights[j]) >>
2145                   PNG_WEIGHT_SHIFT;
2146             }
2147          }
2148
2149          lmlo = (lmlo * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_SUB]) >>
2150             PNG_COST_SHIFT;
2151          lmhi = (lmhi * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_SUB]) >>
2152             PNG_COST_SHIFT;
2153
2154          if (lmhi > PNG_HIMASK)
2155             lmins = PNG_MAXSUM;
2156          else
2157             lmins = (lmhi << PNG_HISHIFT) + lmlo;
2158       }
2159 #endif
2160
2161       for (i = 0, rp = row_buf + 1, dp = png_ptr->sub_row + 1; i < bpp;
2162            i++, rp++, dp++)
2163       {
2164          v = *dp = *rp;
2165
2166          sum += (v < 128) ? v : 256 - v;
2167       }
2168       for (lp = row_buf + 1; i < row_info->rowbytes;
2169          i++, rp++, lp++, dp++)
2170       {
2171          v = *dp = (png_byte)(((int)*rp - (int)*lp) & 0xff);
2172
2173          sum += (v < 128) ? v : 256 - v;
2174
2175          if (sum > lmins)  /* We are already worse, don't continue. */
2176             break;
2177       }
2178
2179 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
2180       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
2181       {
2182          int j;
2183          png_uint_32 sumhi, sumlo;
2184          sumlo = sum & PNG_LOMASK;
2185          sumhi = (sum >> PNG_HISHIFT) & PNG_HIMASK;
2186
2187          for (j = 0; j < num_p_filters; j++)
2188          {
2189             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_SUB)
2190             {
2191                sumlo = (sumlo * png_ptr->inv_filter_weights[j]) >>
2192                   PNG_WEIGHT_SHIFT;
2193                sumhi = (sumhi * png_ptr->inv_filter_weights[j]) >>
2194                   PNG_WEIGHT_SHIFT;
2195             }
2196          }
2197
2198          sumlo = (sumlo * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_SUB]) >>
2199             PNG_COST_SHIFT;
2200          sumhi = (sumhi * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_SUB]) >>
2201             PNG_COST_SHIFT;
2202
2203          if (sumhi > PNG_HIMASK)
2204             sum = PNG_MAXSUM;
2205          else
2206             sum = (sumhi << PNG_HISHIFT) + sumlo;
2207       }
2208 #endif
2209
2210       if (sum < mins)
2211       {
2212          mins = sum;
2213          best_row = png_ptr->sub_row;
2214       }
2215    }
2216
2217    /* up filter */
2218    if (filter_to_do == PNG_FILTER_UP)
2219    {
2220       png_bytep rp, dp, pp;
2221       png_uint_32 i;
2222
2223       for (i = 0, rp = row_buf + 1, dp = png_ptr->up_row + 1,
2224            pp = prev_row + 1; i < row_bytes;
2225            i++, rp++, pp++, dp++)
2226       {
2227          *dp = (png_byte)(((int)*rp - (int)*pp) & 0xff);
2228       }
2229       best_row = png_ptr->up_row;
2230    }
2231
2232    else if (filter_to_do & PNG_FILTER_UP)
2233    {
2234       png_bytep rp, dp, pp;
2235       png_uint_32 sum = 0, lmins = mins;
2236       png_uint_32 i;
2237       int v;
2238
2239
2240 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
2241       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
2242       {
2243          int j;
2244          png_uint_32 lmhi, lmlo;
2245          lmlo = lmins & PNG_LOMASK;
2246          lmhi = (lmins >> PNG_HISHIFT) & PNG_HIMASK;
2247
2248          for (j = 0; j < num_p_filters; j++)
2249          {
2250             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_UP)
2251             {
2252                lmlo = (lmlo * png_ptr->inv_filter_weights[j]) >>
2253                   PNG_WEIGHT_SHIFT;
2254                lmhi = (lmhi * png_ptr->inv_filter_weights[j]) >>
2255                   PNG_WEIGHT_SHIFT;
2256             }
2257          }
2258
2259          lmlo = (lmlo * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_UP]) >>
2260             PNG_COST_SHIFT;
2261          lmhi = (lmhi * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_UP]) >>
2262             PNG_COST_SHIFT;
2263
2264          if (lmhi > PNG_HIMASK)
2265             lmins = PNG_MAXSUM;
2266          else
2267             lmins = (lmhi << PNG_HISHIFT) + lmlo;
2268       }
2269 #endif
2270
2271       for (i = 0, rp = row_buf + 1, dp = png_ptr->up_row + 1,
2272            pp = prev_row + 1; i < row_bytes; i++)
2273       {
2274          v = *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
2275
2276          sum += (v < 128) ? v : 256 - v;
2277
2278          if (sum > lmins)  /* We are already worse, don't continue. */
2279             break;
2280       }
2281
2282 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
2283       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
2284       {
2285          int j;
2286          png_uint_32 sumhi, sumlo;
2287          sumlo = sum & PNG_LOMASK;
2288          sumhi = (sum >> PNG_HISHIFT) & PNG_HIMASK;
2289
2290          for (j = 0; j < num_p_filters; j++)
2291          {
2292             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_UP)
2293             {
2294                sumlo = (sumlo * png_ptr->filter_weights[j]) >>
2295                   PNG_WEIGHT_SHIFT;
2296                sumhi = (sumhi * png_ptr->filter_weights[j]) >>
2297                   PNG_WEIGHT_SHIFT;
2298             }
2299          }
2300
2301          sumlo = (sumlo * png_ptr->filter_costs[PNG_FILTER_VALUE_UP]) >>
2302             PNG_COST_SHIFT;
2303          sumhi = (sumhi * png_ptr->filter_costs[PNG_FILTER_VALUE_UP]) >>
2304             PNG_COST_SHIFT;
2305
2306          if (sumhi > PNG_HIMASK)
2307             sum = PNG_MAXSUM;
2308          else
2309             sum = (sumhi << PNG_HISHIFT) + sumlo;
2310       }
2311 #endif
2312
2313       if (sum < mins)
2314       {
2315          mins = sum;
2316          best_row = png_ptr->up_row;
2317       }
2318    }
2319
2320    /* avg filter */
2321    if (filter_to_do == PNG_FILTER_AVG)
2322    {
2323       png_bytep rp, dp, pp, lp;
2324       png_uint_32 i;
2325       for (i = 0, rp = row_buf + 1, dp = png_ptr->avg_row + 1,
2326            pp = prev_row + 1; i < bpp; i++)
2327       {
2328          *dp++ = (png_byte)(((int)*rp++ - ((int)*pp++ / 2)) & 0xff);
2329       }
2330       for (lp = row_buf + 1; i < row_bytes; i++)
2331       {
2332          *dp++ = (png_byte)(((int)*rp++ - (((int)*pp++ + (int)*lp++) / 2))
2333                  & 0xff);
2334       }
2335       best_row = png_ptr->avg_row;
2336    }
2337
2338    else if (filter_to_do & PNG_FILTER_AVG)
2339    {
2340       png_bytep rp, dp, pp, lp;
2341       png_uint_32 sum = 0, lmins = mins;
2342       png_uint_32 i;
2343       int v;
2344
2345 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
2346       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
2347       {
2348          int j;
2349          png_uint_32 lmhi, lmlo;
2350          lmlo = lmins & PNG_LOMASK;
2351          lmhi = (lmins >> PNG_HISHIFT) & PNG_HIMASK;
2352
2353          for (j = 0; j < num_p_filters; j++)
2354          {
2355             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_AVG)
2356             {
2357                lmlo = (lmlo * png_ptr->inv_filter_weights[j]) >>
2358                   PNG_WEIGHT_SHIFT;
2359                lmhi = (lmhi * png_ptr->inv_filter_weights[j]) >>
2360                   PNG_WEIGHT_SHIFT;
2361             }
2362          }
2363
2364          lmlo = (lmlo * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_AVG]) >>
2365             PNG_COST_SHIFT;
2366          lmhi = (lmhi * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_AVG]) >>
2367             PNG_COST_SHIFT;
2368
2369          if (lmhi > PNG_HIMASK)
2370             lmins = PNG_MAXSUM;
2371          else
2372             lmins = (lmhi << PNG_HISHIFT) + lmlo;
2373       }
2374 #endif
2375
2376       for (i = 0, rp = row_buf + 1, dp = png_ptr->avg_row + 1,
2377            pp = prev_row + 1; i < bpp; i++)
2378       {
2379          v = *dp++ = (png_byte)(((int)*rp++ - ((int)*pp++ / 2)) & 0xff);
2380
2381          sum += (v < 128) ? v : 256 - v;
2382       }
2383       for (lp = row_buf + 1; i < row_bytes; i++)
2384       {
2385          v = *dp++ =
2386           (png_byte)(((int)*rp++ - (((int)*pp++ + (int)*lp++) / 2)) & 0xff);
2387
2388          sum += (v < 128) ? v : 256 - v;
2389
2390          if (sum > lmins)  /* We are already worse, don't continue. */
2391             break;
2392       }
2393
2394 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
2395       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
2396       {
2397          int j;
2398          png_uint_32 sumhi, sumlo;
2399          sumlo = sum & PNG_LOMASK;
2400          sumhi = (sum >> PNG_HISHIFT) & PNG_HIMASK;
2401
2402          for (j = 0; j < num_p_filters; j++)
2403          {
2404             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_NONE)
2405             {
2406                sumlo = (sumlo * png_ptr->filter_weights[j]) >>
2407                   PNG_WEIGHT_SHIFT;
2408                sumhi = (sumhi * png_ptr->filter_weights[j]) >>
2409                   PNG_WEIGHT_SHIFT;
2410             }
2411          }
2412
2413          sumlo = (sumlo * png_ptr->filter_costs[PNG_FILTER_VALUE_AVG]) >>
2414             PNG_COST_SHIFT;
2415          sumhi = (sumhi * png_ptr->filter_costs[PNG_FILTER_VALUE_AVG]) >>
2416             PNG_COST_SHIFT;
2417
2418          if (sumhi > PNG_HIMASK)
2419             sum = PNG_MAXSUM;
2420          else
2421             sum = (sumhi << PNG_HISHIFT) + sumlo;
2422       }
2423 #endif
2424
2425       if (sum < mins)
2426       {
2427          mins = sum;
2428          best_row = png_ptr->avg_row;
2429       }
2430    }
2431
2432    /* Paeth filter */
2433    if (filter_to_do == PNG_FILTER_PAETH)
2434    {
2435       png_bytep rp, dp, pp, cp, lp;
2436       png_uint_32 i;
2437       for (i = 0, rp = row_buf + 1, dp = png_ptr->paeth_row + 1,
2438            pp = prev_row + 1; i < bpp; i++)
2439       {
2440          *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
2441       }
2442
2443       for (lp = row_buf + 1, cp = prev_row + 1; i < row_bytes; i++)
2444       {
2445          int a, b, c, pa, pb, pc, p;
2446
2447          b = *pp++;
2448          c = *cp++;
2449          a = *lp++;
2450
2451          p = b - c;
2452          pc = a - c;
2453
2454 #ifdef PNG_USE_ABS
2455          pa = abs(p);
2456          pb = abs(pc);
2457          pc = abs(p + pc);
2458 #else
2459          pa = p < 0 ? -p : p;
2460          pb = pc < 0 ? -pc : pc;
2461          pc = (p + pc) < 0 ? -(p + pc) : p + pc;
2462 #endif
2463
2464          p = (pa <= pb && pa <=pc) ? a : (pb <= pc) ? b : c;
2465
2466          *dp++ = (png_byte)(((int)*rp++ - p) & 0xff);
2467       }
2468       best_row = png_ptr->paeth_row;
2469    }
2470
2471    else if (filter_to_do & PNG_FILTER_PAETH)
2472    {
2473       png_bytep rp, dp, pp, cp, lp;
2474       png_uint_32 sum = 0, lmins = mins;
2475       png_uint_32 i;
2476       int v;
2477
2478 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
2479       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
2480       {
2481          int j;
2482          png_uint_32 lmhi, lmlo;
2483          lmlo = lmins & PNG_LOMASK;
2484          lmhi = (lmins >> PNG_HISHIFT) & PNG_HIMASK;
2485
2486          for (j = 0; j < num_p_filters; j++)
2487          {
2488             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_PAETH)
2489             {
2490                lmlo = (lmlo * png_ptr->inv_filter_weights[j]) >>
2491                   PNG_WEIGHT_SHIFT;
2492                lmhi = (lmhi * png_ptr->inv_filter_weights[j]) >>
2493                   PNG_WEIGHT_SHIFT;
2494             }
2495          }
2496
2497          lmlo = (lmlo * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_PAETH]) >>
2498             PNG_COST_SHIFT;
2499          lmhi = (lmhi * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_PAETH]) >>
2500             PNG_COST_SHIFT;
2501
2502          if (lmhi > PNG_HIMASK)
2503             lmins = PNG_MAXSUM;
2504          else
2505             lmins = (lmhi << PNG_HISHIFT) + lmlo;
2506       }
2507 #endif
2508
2509       for (i = 0, rp = row_buf + 1, dp = png_ptr->paeth_row + 1,
2510            pp = prev_row + 1; i < bpp; i++)
2511       {
2512          v = *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
2513
2514          sum += (v < 128) ? v : 256 - v;
2515       }
2516
2517       for (lp = row_buf + 1, cp = prev_row + 1; i < row_bytes; i++)
2518       {
2519          int a, b, c, pa, pb, pc, p;
2520
2521          b = *pp++;
2522          c = *cp++;
2523          a = *lp++;
2524
2525 #ifndef PNG_SLOW_PAETH
2526          p = b - c;
2527          pc = a - c;
2528 #ifdef PNG_USE_ABS
2529          pa = abs(p);
2530          pb = abs(pc);
2531          pc = abs(p + pc);
2532 #else
2533          pa = p < 0 ? -p : p;
2534          pb = pc < 0 ? -pc : pc;
2535          pc = (p + pc) < 0 ? -(p + pc) : p + pc;
2536 #endif
2537          p = (pa <= pb && pa <=pc) ? a : (pb <= pc) ? b : c;
2538 #else /* PNG_SLOW_PAETH */
2539          p = a + b - c;
2540          pa = abs(p - a);
2541          pb = abs(p - b);
2542          pc = abs(p - c);
2543          if (pa <= pb && pa <= pc)
2544             p = a;
2545          else if (pb <= pc)
2546             p = b;
2547          else
2548             p = c;
2549 #endif /* PNG_SLOW_PAETH */
2550
2551          v = *dp++ = (png_byte)(((int)*rp++ - p) & 0xff);
2552
2553          sum += (v < 128) ? v : 256 - v;
2554
2555          if (sum > lmins)  /* We are already worse, don't continue. */
2556             break;
2557       }
2558
2559 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
2560       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
2561       {
2562          int j;
2563          png_uint_32 sumhi, sumlo;
2564          sumlo = sum & PNG_LOMASK;
2565          sumhi = (sum >> PNG_HISHIFT) & PNG_HIMASK;
2566
2567          for (j = 0; j < num_p_filters; j++)
2568          {
2569             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_PAETH)
2570             {
2571                sumlo = (sumlo * png_ptr->filter_weights[j]) >>
2572                   PNG_WEIGHT_SHIFT;
2573                sumhi = (sumhi * png_ptr->filter_weights[j]) >>
2574                   PNG_WEIGHT_SHIFT;
2575             }
2576          }
2577
2578          sumlo = (sumlo * png_ptr->filter_costs[PNG_FILTER_VALUE_PAETH]) >>
2579             PNG_COST_SHIFT;
2580          sumhi = (sumhi * png_ptr->filter_costs[PNG_FILTER_VALUE_PAETH]) >>
2581             PNG_COST_SHIFT;
2582
2583          if (sumhi > PNG_HIMASK)
2584             sum = PNG_MAXSUM;
2585          else
2586             sum = (sumhi << PNG_HISHIFT) + sumlo;
2587       }
2588 #endif
2589
2590       if (sum < mins)
2591       {
2592          best_row = png_ptr->paeth_row;
2593       }
2594    }
2595
2596    /* Do the actual writing of the filtered row data from the chosen filter. */
2597
2598    png_write_filtered_row(png_ptr, best_row);
2599
2600 #if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
2601    /* Save the type of filter we picked this time for future calculations */
2602    if (png_ptr->num_prev_filters > 0)
2603    {
2604       int j;
2605       for (j = 1; j < num_p_filters; j++)
2606       {
2607          png_ptr->prev_filters[j] = png_ptr->prev_filters[j - 1];
2608       }
2609       png_ptr->prev_filters[j] = best_row[0];
2610    }
2611 #endif
2612 }
2613
2614
2615 /* Do the actual writing of a previously filtered row. */
2616 void /* PRIVATE */
2617 png_write_filtered_row(png_structp png_ptr, png_bytep filtered_row)
2618 {
2619    png_debug(1, "in png_write_filtered_row\n");
2620    png_debug1(2, "filter = %d\n", filtered_row[0]);
2621    /* set up the zlib input buffer */
2622
2623    png_ptr->zstream.next_in = filtered_row;
2624    png_ptr->zstream.avail_in = (uInt)png_ptr->row_info.rowbytes + 1;
2625    /* repeat until we have compressed all the data */
2626    do
2627    {
2628       int ret; /* return of zlib */
2629
2630       /* compress the data */
2631       ret = deflate(&png_ptr->zstream, Z_NO_FLUSH);
2632       /* check for compression errors */
2633       if (ret != Z_OK)
2634       {
2635          if (png_ptr->zstream.msg != NULL)
2636             png_error(png_ptr, png_ptr->zstream.msg);
2637          else
2638             png_error(png_ptr, "zlib error");
2639       }
2640
2641       /* see if it is time to write another IDAT */
2642       if (!(png_ptr->zstream.avail_out))
2643       {
2644          /* write the IDAT and reset the zlib output buffer */
2645          png_write_IDAT(png_ptr, png_ptr->zbuf, png_ptr->zbuf_size);
2646          png_ptr->zstream.next_out = png_ptr->zbuf;
2647          png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
2648       }
2649    /* repeat until all data has been compressed */
2650    } while (png_ptr->zstream.avail_in);
2651
2652    /* swap the current and previous rows */
2653    if (png_ptr->prev_row != NULL)
2654    {
2655       png_bytep tptr;
2656
2657       tptr = png_ptr->prev_row;
2658       png_ptr->prev_row = png_ptr->row_buf;
2659       png_ptr->row_buf = tptr;
2660    }
2661
2662    /* finish row - updates counters and flushes zlib if last row */
2663    png_write_finish_row(png_ptr);
2664
2665 #if defined(PNG_WRITE_FLUSH_SUPPORTED)
2666    png_ptr->flush_rows++;
2667
2668    if (png_ptr->flush_dist > 0 &&
2669        png_ptr->flush_rows >= png_ptr->flush_dist)
2670    {
2671       png_write_flush(png_ptr);
2672    }
2673 #endif
2674 }
2675 #endif /* PNG_WRITE_SUPPORTED */