Subversion Repositories shark

Rev

Details | Last modification | View Log | RSS feed

Rev Author Line No. Line
96 giacomo 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
      cprintf("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
      cprintf("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 */