pngwrite.c 48.3 KB
Newer Older
1

A
Andreas Dilger 已提交
2
/* pngwrite.c - general routines to write a PNG file
3
 *
4
 * Last changed in libpng 1.2.30 [July 17, 2008]
5
 * For conditions of distribution and use, see copyright notice in png.h
6
 * Copyright (c) 1998-2008 Glenn Randers-Pehrson
7 8
 * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)
 * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)
9
 */
G
Guy Schalnat 已提交
10 11 12 13

/* get internal access to png.h */
#define PNG_INTERNAL
#include "png.h"
14
#ifdef PNG_WRITE_SUPPORTED
G
Guy Schalnat 已提交
15

A
Andreas Dilger 已提交
16 17 18
/* Writes all the PNG information.  This is the suggested way to use the
 * library.  If you have a new chunk to add, make a function to write it,
 * and put it in the correct location here.  If you want the chunk written
19
 * after the image data, put it in png_write_end().  I strongly encourage
A
Andreas Dilger 已提交
20 21 22 23 24
 * you to supply a PNG_INFO_ flag, and check info_ptr->valid before writing
 * the chunk, as that will keep the code from breaking if you want to just
 * write a plain PNG file.  If you have long comments, I suggest writing
 * them in png_write_end(), and compressing them.
 */
25
void PNGAPI
26
png_write_info_before_PLTE(png_structp png_ptr, png_infop info_ptr)
G
Guy Schalnat 已提交
27
{
28
   png_debug(1, "in png_write_info_before_PLTE\n");
29 30
   if (png_ptr == NULL || info_ptr == NULL)
      return;
31 32
   if (!(png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE))
   {
G
Guy Schalnat 已提交
33
   png_write_sig(png_ptr); /* write PNG signature */
34
#if defined(PNG_MNG_FEATURES_SUPPORTED)
35
   if ((png_ptr->mode&PNG_HAVE_PNG_SIGNATURE)&&(png_ptr->mng_features_permitted))
36
   {
37
      png_warning(png_ptr, "MNG features are not allowed in a PNG datastream");
38 39 40
      png_ptr->mng_features_permitted=0;
   }
#endif
G
Guy Schalnat 已提交
41
   /* write IHDR information. */
A
Andreas Dilger 已提交
42 43
   png_write_IHDR(png_ptr, info_ptr->width, info_ptr->height,
      info_ptr->bit_depth, info_ptr->color_type, info_ptr->compression_type,
44 45 46 47 48 49
      info_ptr->filter_type,
#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
      info_ptr->interlace_type);
#else
      0);
#endif
G
Guy Schalnat 已提交
50 51
   /* the rest of these check to see if the valid field has the appropriate
      flag set, and if it does, writes the chunk. */
G
Guy Schalnat 已提交
52
#if defined(PNG_WRITE_gAMA_SUPPORTED)
A
Andreas Dilger 已提交
53
   if (info_ptr->valid & PNG_INFO_gAMA)
54 55
   {
#  ifdef PNG_FLOATING_POINT_SUPPORTED
A
Andreas Dilger 已提交
56
      png_write_gAMA(png_ptr, info_ptr->gamma);
57 58 59
#else
#ifdef PNG_FIXED_POINT_SUPPORTED
      png_write_gAMA_fixed(png_ptr, info_ptr->int_gamma);
60 61 62
#  endif
#endif
   }
G
Guy Schalnat 已提交
63
#endif
64 65
#if defined(PNG_WRITE_sRGB_SUPPORTED)
   if (info_ptr->valid & PNG_INFO_sRGB)
66
      png_write_sRGB(png_ptr, (int)info_ptr->srgb_intent);
67
#endif
68 69
#if defined(PNG_WRITE_iCCP_SUPPORTED)
   if (info_ptr->valid & PNG_INFO_iCCP)
70
      png_write_iCCP(png_ptr, info_ptr->iccp_name, PNG_COMPRESSION_TYPE_BASE,
71 72
                     info_ptr->iccp_profile, (int)info_ptr->iccp_proflen);
#endif
G
Guy Schalnat 已提交
73
#if defined(PNG_WRITE_sBIT_SUPPORTED)
A
Andreas Dilger 已提交
74 75
   if (info_ptr->valid & PNG_INFO_sBIT)
      png_write_sBIT(png_ptr, &(info_ptr->sig_bit), info_ptr->color_type);
G
Guy Schalnat 已提交
76 77
#endif
#if defined(PNG_WRITE_cHRM_SUPPORTED)
A
Andreas Dilger 已提交
78
   if (info_ptr->valid & PNG_INFO_cHRM)
79
   {
80
#ifdef PNG_FLOATING_POINT_SUPPORTED
G
Guy Schalnat 已提交
81
      png_write_cHRM(png_ptr,
A
Andreas Dilger 已提交
82 83 84 85
         info_ptr->x_white, info_ptr->y_white,
         info_ptr->x_red, info_ptr->y_red,
         info_ptr->x_green, info_ptr->y_green,
         info_ptr->x_blue, info_ptr->y_blue);
86 87 88 89 90 91 92
#else
#  ifdef PNG_FIXED_POINT_SUPPORTED
      png_write_cHRM_fixed(png_ptr,
         info_ptr->int_x_white, info_ptr->int_y_white,
         info_ptr->int_x_red, info_ptr->int_y_red,
         info_ptr->int_x_green, info_ptr->int_y_green,
         info_ptr->int_x_blue, info_ptr->int_y_blue);
93 94 95 96 97 98 99 100 101 102 103 104 105 106
#  endif
#endif
   }
#endif
#if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
   if (info_ptr->unknown_chunks_num)
   {
       png_unknown_chunk *up;

       png_debug(5, "writing extra chunks\n");

       for (up = info_ptr->unknown_chunks;
            up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
            up++)
107 108
       {
         int keep=png_handle_as_unknown(png_ptr, up->name);
109
         if (keep != PNG_HANDLE_CHUNK_NEVER &&
110 111
            up->location && !(up->location & PNG_HAVE_PLTE) &&
            !(up->location & PNG_HAVE_IDAT) &&
112
            ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS ||
113 114
            (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
         {
115 116
            if (up->size == 0)
               png_warning(png_ptr, "Writing zero-length unknown chunk");
117
            png_write_chunk(png_ptr, up->name, up->data, up->size);
118 119
         }
       }
120
   }
G
Guy Schalnat 已提交
121
#endif
122 123 124 125
      png_ptr->mode |= PNG_WROTE_INFO_BEFORE_PLTE;
   }
}

126
void PNGAPI
127 128
png_write_info(png_structp png_ptr, png_infop info_ptr)
{
129
#if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_sPLT_SUPPORTED)
130 131 132 133 134
   int i;
#endif

   png_debug(1, "in png_write_info\n");

135 136 137
   if (png_ptr == NULL || info_ptr == NULL)
      return;

138 139
   png_write_info_before_PLTE(png_ptr, info_ptr);

A
Andreas Dilger 已提交
140
   if (info_ptr->valid & PNG_INFO_PLTE)
A
Andreas Dilger 已提交
141 142
      png_write_PLTE(png_ptr, info_ptr->palette,
         (png_uint_32)info_ptr->num_palette);
A
Andreas Dilger 已提交
143
   else if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
144
      png_error(png_ptr, "Valid palette required for paletted images");
145

G
Guy Schalnat 已提交
146
#if defined(PNG_WRITE_tRNS_SUPPORTED)
A
Andreas Dilger 已提交
147
   if (info_ptr->valid & PNG_INFO_tRNS)
148 149 150
      {
#if defined(PNG_WRITE_INVERT_ALPHA_SUPPORTED)
         /* invert the alpha channel (in tRNS) */
151
         if ((png_ptr->transformations & PNG_INVERT_ALPHA) &&
152 153
            info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
         {
154
            int j;
155
            for (j=0; j<(int)info_ptr->num_trans; j++)
156
               info_ptr->trans[j] = (png_byte)(255 - info_ptr->trans[j]);
157 158
         }
#endif
A
Andreas Dilger 已提交
159 160
      png_write_tRNS(png_ptr, info_ptr->trans, &(info_ptr->trans_values),
         info_ptr->num_trans, info_ptr->color_type);
161
      }
G
Guy Schalnat 已提交
162 163
#endif
#if defined(PNG_WRITE_bKGD_SUPPORTED)
A
Andreas Dilger 已提交
164 165
   if (info_ptr->valid & PNG_INFO_bKGD)
      png_write_bKGD(png_ptr, &(info_ptr->background), info_ptr->color_type);
G
Guy Schalnat 已提交
166 167
#endif
#if defined(PNG_WRITE_hIST_SUPPORTED)
A
Andreas Dilger 已提交
168 169
   if (info_ptr->valid & PNG_INFO_hIST)
      png_write_hIST(png_ptr, info_ptr->hist, info_ptr->num_palette);
G
Guy Schalnat 已提交
170 171
#endif
#if defined(PNG_WRITE_oFFs_SUPPORTED)
A
Andreas Dilger 已提交
172 173 174
   if (info_ptr->valid & PNG_INFO_oFFs)
      png_write_oFFs(png_ptr, info_ptr->x_offset, info_ptr->y_offset,
         info_ptr->offset_unit_type);
G
Guy Schalnat 已提交
175
#endif
A
Andreas Dilger 已提交
176 177 178 179 180 181
#if defined(PNG_WRITE_pCAL_SUPPORTED)
   if (info_ptr->valid & PNG_INFO_pCAL)
      png_write_pCAL(png_ptr, info_ptr->pcal_purpose, info_ptr->pcal_X0,
         info_ptr->pcal_X1, info_ptr->pcal_type, info_ptr->pcal_nparams,
         info_ptr->pcal_units, info_ptr->pcal_params);
#endif
182 183
#if defined(PNG_WRITE_sCAL_SUPPORTED)
   if (info_ptr->valid & PNG_INFO_sCAL)
184
#if defined(PNG_FLOATING_POINT_SUPPORTED) && !defined(PNG_NO_STDIO)
185 186 187 188 189
      png_write_sCAL(png_ptr, (int)info_ptr->scal_unit,
          info_ptr->scal_pixel_width, info_ptr->scal_pixel_height);
#else
#ifdef PNG_FIXED_POINT_SUPPORTED
      png_write_sCAL_s(png_ptr, (int)info_ptr->scal_unit,
190
          info_ptr->scal_s_width, info_ptr->scal_s_height);
191 192
#else
      png_warning(png_ptr,
193
          "png_write_sCAL not supported; sCAL chunk not written.");
194
#endif
195 196
#endif
#endif
A
Andreas Dilger 已提交
197 198 199 200 201
#if defined(PNG_WRITE_pHYs_SUPPORTED)
   if (info_ptr->valid & PNG_INFO_pHYs)
      png_write_pHYs(png_ptr, info_ptr->x_pixels_per_unit,
         info_ptr->y_pixels_per_unit, info_ptr->phys_unit_type);
#endif
G
Guy Schalnat 已提交
202
#if defined(PNG_WRITE_tIME_SUPPORTED)
A
Andreas Dilger 已提交
203
   if (info_ptr->valid & PNG_INFO_tIME)
G
Guy Schalnat 已提交
204
   {
A
Andreas Dilger 已提交
205
      png_write_tIME(png_ptr, &(info_ptr->mod_time));
206
      png_ptr->mode |= PNG_WROTE_tIME;
G
Guy Schalnat 已提交
207
   }
G
Guy Schalnat 已提交
208
#endif
209 210 211 212 213
#if defined(PNG_WRITE_sPLT_SUPPORTED)
   if (info_ptr->valid & PNG_INFO_sPLT)
     for (i = 0; i < (int)info_ptr->splt_palettes_num; i++)
       png_write_sPLT(png_ptr, info_ptr->splt_palettes + i);
#endif
214
#if defined(PNG_WRITE_TEXT_SUPPORTED)
G
Guy Schalnat 已提交
215
   /* Check to see if we need to write text chunks */
A
Andreas Dilger 已提交
216
   for (i = 0; i < info_ptr->num_text; i++)
G
Guy Schalnat 已提交
217
   {
A
Andreas Dilger 已提交
218 219
      png_debug2(2, "Writing header text chunk %d, type %d\n", i,
         info_ptr->text[i].compression);
220
      /* an internationalized chunk? */
221
      if (info_ptr->text[i].compression > 0)
222 223 224 225 226 227
      {
#if defined(PNG_WRITE_iTXt_SUPPORTED)
          /* write international chunk */
          png_write_iTXt(png_ptr,
                         info_ptr->text[i].compression,
                         info_ptr->text[i].key,
228 229
                         info_ptr->text[i].lang,
                         info_ptr->text[i].lang_key,
230 231
                         info_ptr->text[i].text);
#else
232
          png_warning(png_ptr, "Unable to write international text");
233 234 235 236
#endif
          /* Mark this chunk as written */
          info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
      }
A
Andreas Dilger 已提交
237
      /* If we want a compressed text chunk */
238
      else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_zTXt)
G
Guy Schalnat 已提交
239
      {
G
Guy Schalnat 已提交
240
#if defined(PNG_WRITE_zTXt_SUPPORTED)
A
Andreas Dilger 已提交
241 242
         /* write compressed chunk */
         png_write_zTXt(png_ptr, info_ptr->text[i].key,
243
            info_ptr->text[i].text, 0,
A
Andreas Dilger 已提交
244
            info_ptr->text[i].compression);
G
Guy Schalnat 已提交
245
#else
246
         png_warning(png_ptr, "Unable to write compressed text");
G
Guy Schalnat 已提交
247
#endif
A
Andreas Dilger 已提交
248 249 250 251 252
         /* Mark this chunk as written */
         info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
      }
      else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
      {
G
Guy Schalnat 已提交
253
#if defined(PNG_WRITE_tEXt_SUPPORTED)
A
Andreas Dilger 已提交
254 255
         /* write uncompressed chunk */
         png_write_tEXt(png_ptr, info_ptr->text[i].key,
256
                         info_ptr->text[i].text,
257
                         0);
G
Guy Schalnat 已提交
258
#else
259
         png_warning(png_ptr, "Unable to write uncompressed text");
G
Guy Schalnat 已提交
260
#endif
A
Andreas Dilger 已提交
261 262
         /* Mark this chunk as written */
         info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
G
Guy Schalnat 已提交
263 264
      }
   }
G
Guy Schalnat 已提交
265
#endif
266 267 268 269 270 271 272 273 274 275
#if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
   if (info_ptr->unknown_chunks_num)
   {
       png_unknown_chunk *up;

       png_debug(5, "writing extra chunks\n");

       for (up = info_ptr->unknown_chunks;
            up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
            up++)
276 277
       {
         int keep=png_handle_as_unknown(png_ptr, up->name);
278
         if (keep != PNG_HANDLE_CHUNK_NEVER &&
279 280
            up->location && (up->location & PNG_HAVE_PLTE) &&
            !(up->location & PNG_HAVE_IDAT) &&
281
            ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS ||
282 283 284 285 286
            (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
         {
            png_write_chunk(png_ptr, up->name, up->data, up->size);
         }
       }
287 288
   }
#endif
G
Guy Schalnat 已提交
289 290
}

A
Andreas Dilger 已提交
291
/* Writes the end of the PNG file.  If you don't want to write comments or
292 293 294 295
 * time information, you can pass NULL for info.  If you already wrote these
 * in png_write_info(), do not write them again here.  If you have long
 * comments, I suggest writing them here, and compressing them.
 */
296
void PNGAPI
A
Andreas Dilger 已提交
297
png_write_end(png_structp png_ptr, png_infop info_ptr)
G
Guy Schalnat 已提交
298
{
A
Andreas Dilger 已提交
299
   png_debug(1, "in png_write_end\n");
300 301
   if (png_ptr == NULL)
      return;
G
Guy Schalnat 已提交
302 303 304
   if (!(png_ptr->mode & PNG_HAVE_IDAT))
      png_error(png_ptr, "No IDATs written into file");

G
Guy Schalnat 已提交
305
   /* see if user wants us to write information chunks */
A
Andreas Dilger 已提交
306
   if (info_ptr != NULL)
G
Guy Schalnat 已提交
307
   {
308
#if defined(PNG_WRITE_TEXT_SUPPORTED)
A
Andreas Dilger 已提交
309
      int i; /* local index variable */
310
#endif
G
Guy Schalnat 已提交
311
#if defined(PNG_WRITE_tIME_SUPPORTED)
G
Guy Schalnat 已提交
312
      /* check to see if user has supplied a time chunk */
313
      if ((info_ptr->valid & PNG_INFO_tIME) &&
314
         !(png_ptr->mode & PNG_WROTE_tIME))
A
Andreas Dilger 已提交
315
         png_write_tIME(png_ptr, &(info_ptr->mod_time));
G
Guy Schalnat 已提交
316
#endif
317
#if defined(PNG_WRITE_TEXT_SUPPORTED)
A
Andreas Dilger 已提交
318 319
      /* loop through comment chunks */
      for (i = 0; i < info_ptr->num_text; i++)
G
Guy Schalnat 已提交
320
      {
A
Andreas Dilger 已提交
321 322
         png_debug2(2, "Writing trailer text chunk %d, type %d\n", i,
            info_ptr->text[i].compression);
323 324 325 326 327 328 329 330 331 332 333 334
         /* an internationalized chunk? */
         if (info_ptr->text[i].compression > 0)
         {
#if defined(PNG_WRITE_iTXt_SUPPORTED)
             /* write international chunk */
             png_write_iTXt(png_ptr,
                         info_ptr->text[i].compression,
                         info_ptr->text[i].key,
                         info_ptr->text[i].lang,
                         info_ptr->text[i].lang_key,
                         info_ptr->text[i].text);
#else
335
             png_warning(png_ptr, "Unable to write international text");
336 337 338 339 340
#endif
             /* Mark this chunk as written */
             info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
         }
         else if (info_ptr->text[i].compression >= PNG_TEXT_COMPRESSION_zTXt)
G
Guy Schalnat 已提交
341
         {
G
Guy Schalnat 已提交
342
#if defined(PNG_WRITE_zTXt_SUPPORTED)
A
Andreas Dilger 已提交
343 344
            /* write compressed chunk */
            png_write_zTXt(png_ptr, info_ptr->text[i].key,
345
               info_ptr->text[i].text, 0,
A
Andreas Dilger 已提交
346 347
               info_ptr->text[i].compression);
#else
348
            png_warning(png_ptr, "Unable to write compressed text");
G
Guy Schalnat 已提交
349
#endif
A
Andreas Dilger 已提交
350 351 352 353 354
            /* Mark this chunk as written */
            info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
         }
         else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
         {
G
Guy Schalnat 已提交
355
#if defined(PNG_WRITE_tEXt_SUPPORTED)
A
Andreas Dilger 已提交
356 357
            /* write uncompressed chunk */
            png_write_tEXt(png_ptr, info_ptr->text[i].key,
358
               info_ptr->text[i].text, 0);
A
Andreas Dilger 已提交
359
#else
360
            png_warning(png_ptr, "Unable to write uncompressed text");
G
Guy Schalnat 已提交
361
#endif
A
Andreas Dilger 已提交
362 363 364

            /* Mark this chunk as written */
            info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
G
Guy Schalnat 已提交
365 366
         }
      }
367 368 369 370 371 372 373 374 375 376 377
#endif
#if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
   if (info_ptr->unknown_chunks_num)
   {
       png_unknown_chunk *up;

       png_debug(5, "writing extra chunks\n");

       for (up = info_ptr->unknown_chunks;
            up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
            up++)
378 379
       {
         int keep=png_handle_as_unknown(png_ptr, up->name);
380
         if (keep != PNG_HANDLE_CHUNK_NEVER &&
381
            up->location && (up->location & PNG_AFTER_IDAT) &&
382
            ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS ||
383 384
            (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
         {
385
            png_write_chunk(png_ptr, up->name, up->data, up->size);
386 387
         }
       }
388
   }
G
Guy Schalnat 已提交
389
#endif
G
Guy Schalnat 已提交
390
   }
G
Guy Schalnat 已提交
391 392 393

   png_ptr->mode |= PNG_AFTER_IDAT;

A
Andreas Dilger 已提交
394
   /* write end of PNG file */
G
Guy Schalnat 已提交
395
   png_write_IEND(png_ptr);
396 397 398 399 400 401
   /* This flush, added in libpng-1.0.8, removed from libpng-1.0.9beta03,
    * and restored again in libpng-1.2.30, may cause some applications that
    * do not set png_ptr->output_flush_fn to crash.  If your application
    * experiences this problem, please report the event to
    * png-mng-implement at lists.sf.net .
    */
402
   png_flush(png_ptr);
G
Guy Schalnat 已提交
403 404
}

G
Guy Schalnat 已提交
405
#if defined(PNG_WRITE_tIME_SUPPORTED)
406 407
#if !defined(_WIN32_WCE)
/* "time.h" functions are not supported on WindowsCE */
408
void PNGAPI
G
Guy Schalnat 已提交
409
png_convert_from_struct_tm(png_timep ptime, struct tm FAR * ttime)
G
Guy Schalnat 已提交
410
{
A
Andreas Dilger 已提交
411
   png_debug(1, "in png_convert_from_struct_tm\n");
G
Guy Schalnat 已提交
412 413 414 415 416 417
   ptime->year = (png_uint_16)(1900 + ttime->tm_year);
   ptime->month = (png_byte)(ttime->tm_mon + 1);
   ptime->day = (png_byte)ttime->tm_mday;
   ptime->hour = (png_byte)ttime->tm_hour;
   ptime->minute = (png_byte)ttime->tm_min;
   ptime->second = (png_byte)ttime->tm_sec;
G
Guy Schalnat 已提交
418 419
}

420
void PNGAPI
G
Guy Schalnat 已提交
421
png_convert_from_time_t(png_timep ptime, time_t ttime)
G
Guy Schalnat 已提交
422 423 424
{
   struct tm *tbuf;

A
Andreas Dilger 已提交
425
   png_debug(1, "in png_convert_from_time_t\n");
G
Guy Schalnat 已提交
426 427 428
   tbuf = gmtime(&ttime);
   png_convert_from_struct_tm(ptime, tbuf);
}
G
Guy Schalnat 已提交
429
#endif
430
#endif
G
Guy Schalnat 已提交
431

A
Andreas Dilger 已提交
432
/* Initialize png_ptr structure, and allocate any memory needed */
433
png_structp PNGAPI
434
png_create_write_struct(png_const_charp user_png_ver, png_voidp error_ptr,
A
Andreas Dilger 已提交
435
   png_error_ptr error_fn, png_error_ptr warn_fn)
G
Guy Schalnat 已提交
436
{
437 438
#ifdef PNG_USER_MEM_SUPPORTED
   return (png_create_write_struct_2(user_png_ver, error_ptr, error_fn,
439
      warn_fn, png_voidp_NULL, png_malloc_ptr_NULL, png_free_ptr_NULL));
440 441 442
}

/* Alternate initialize png_ptr structure, and allocate any memory needed */
443
png_structp PNGAPI
444 445 446 447 448
png_create_write_struct_2(png_const_charp user_png_ver, png_voidp error_ptr,
   png_error_ptr error_fn, png_error_ptr warn_fn, png_voidp mem_ptr,
   png_malloc_ptr malloc_fn, png_free_ptr free_fn)
{
#endif /* PNG_USER_MEM_SUPPORTED */
G
Guy Schalnat 已提交
449
   png_structp png_ptr;
450
#ifdef PNG_SETJMP_SUPPORTED
A
Andreas Dilger 已提交
451 452
#ifdef USE_FAR_KEYWORD
   jmp_buf jmpbuf;
453
#endif
A
Andreas Dilger 已提交
454
#endif
455
   int i;
A
Andreas Dilger 已提交
456
   png_debug(1, "in png_create_write_struct\n");
457
#ifdef PNG_USER_MEM_SUPPORTED
458 459
   png_ptr = (png_structp)png_create_struct_2(PNG_STRUCT_PNG,
      (png_malloc_ptr)malloc_fn, (png_voidp)mem_ptr);
460
#else
461
   png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG);
462
#endif /* PNG_USER_MEM_SUPPORTED */
463
   if (png_ptr == NULL)
464
      return (NULL);
465

466 467 468 469 470 471
   /* added at libpng-1.2.6 */
#ifdef PNG_SET_USER_LIMITS_SUPPORTED
   png_ptr->user_width_max=PNG_USER_WIDTH_MAX;
   png_ptr->user_height_max=PNG_USER_HEIGHT_MAX;
#endif

472
#ifdef PNG_SETJMP_SUPPORTED
A
Andreas Dilger 已提交
473 474 475
#ifdef USE_FAR_KEYWORD
   if (setjmp(jmpbuf))
#else
G
Guy Schalnat 已提交
476
   if (setjmp(png_ptr->jmpbuf))
A
Andreas Dilger 已提交
477
#endif
G
Guy Schalnat 已提交
478
   {
A
Andreas Dilger 已提交
479
      png_free(png_ptr, png_ptr->zbuf);
480
      png_ptr->zbuf=NULL;
G
Guy Schalnat 已提交
481
      png_destroy_struct(png_ptr);
482
      return (NULL);
G
Guy Schalnat 已提交
483
   }
A
Andreas Dilger 已提交
484
#ifdef USE_FAR_KEYWORD
485
   png_memcpy(png_ptr->jmpbuf, jmpbuf, png_sizeof(jmp_buf));
A
Andreas Dilger 已提交
486
#endif
487 488
#endif

489 490 491
#ifdef PNG_USER_MEM_SUPPORTED
   png_set_mem_fn(png_ptr, mem_ptr, malloc_fn, free_fn);
#endif /* PNG_USER_MEM_SUPPORTED */
A
Andreas Dilger 已提交
492
   png_set_error_fn(png_ptr, error_ptr, error_fn, warn_fn);
G
Guy Schalnat 已提交
493

494
   if (user_png_ver)
G
Guy Schalnat 已提交
495
   {
496 497 498
     i=0;
     do
     {
499
       if (user_png_ver[i] != png_libpng_ver[i])
500 501 502
          png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH;
     } while (png_libpng_ver[i++]);
   }
503

504
   if (png_ptr->flags & PNG_FLAG_LIBRARY_MISMATCH)
505
   {
506 507 508 509 510 511
     /* Libpng 0.90 and later are binary incompatible with libpng 0.89, so
      * we must recompile any applications that use any older library version.
      * For versions after libpng 1.0, we will be compatible, so we need
      * only check the first digit.
      */
     if (user_png_ver == NULL || user_png_ver[0] != png_libpng_ver[0] ||
512
         (user_png_ver[0] == '1' && user_png_ver[2] != png_libpng_ver[2]) ||
513 514
         (user_png_ver[0] == '0' && user_png_ver[2] < '9'))
     {
515 516 517 518
#if !defined(PNG_NO_STDIO) && !defined(_WIN32_WCE)
        char msg[80];
        if (user_png_ver)
        {
519 520
          png_snprintf(msg, 80,
             "Application was compiled with png.h from libpng-%.20s",
521 522 523
             user_png_ver);
          png_warning(png_ptr, msg);
        }
524 525
        png_snprintf(msg, 80,
           "Application  is  running with png.c from libpng-%.20s",
526 527 528 529 530 531
           png_libpng_ver);
        png_warning(png_ptr, msg);
#endif
#ifdef PNG_ERROR_NUMBERS_SUPPORTED
        png_ptr->flags=0;
#endif
532 533 534 535
        png_error(png_ptr,
           "Incompatible libpng version in application and library");
     }
   }
536

G
Guy Schalnat 已提交
537 538
   /* initialize zbuf - compression buffer */
   png_ptr->zbuf_size = PNG_ZBUF_SIZE;
539 540
   png_ptr->zbuf = (png_bytep)png_malloc(png_ptr,
      (png_uint_32)png_ptr->zbuf_size);
G
Guy Schalnat 已提交
541

542 543
   png_set_write_fn(png_ptr, png_voidp_NULL, png_rw_ptr_NULL,
      png_flush_ptr_NULL);
G
Guy Schalnat 已提交
544

545 546
#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
   png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
547
      1, png_doublep_NULL, png_doublep_NULL);
548 549
#endif

550 551 552 553 554 555 556
#ifdef PNG_SETJMP_SUPPORTED
/* Applications that neglect to set up their own setjmp() and then encounter
   a png_error() will longjmp here.  Since the jmpbuf is then meaningless we
   abort instead of returning. */
#ifdef USE_FAR_KEYWORD
   if (setjmp(jmpbuf))
      PNG_ABORT();
557
   png_memcpy(png_ptr->jmpbuf, jmpbuf, png_sizeof(jmp_buf));
558 559 560 561 562 563
#else
   if (setjmp(png_ptr->jmpbuf))
      PNG_ABORT();
#endif
#endif
   return (png_ptr);
G
Guy Schalnat 已提交
564 565
}

A
Andreas Dilger 已提交
566
/* Initialize png_ptr structure, and allocate any memory needed */
567
#if defined(PNG_1_0_X) || defined(PNG_1_2_X)
568
/* Deprecated. */
569
#undef png_write_init
570
void PNGAPI
G
Guy Schalnat 已提交
571
png_write_init(png_structp png_ptr)
572 573
{
   /* We only come here via pre-1.0.7-compiled applications */
574
   png_write_init_2(png_ptr, "1.0.6 or earlier", 0, 0);
575 576
}

577
void PNGAPI
578 579
png_write_init_2(png_structp png_ptr, png_const_charp user_png_ver,
   png_size_t png_struct_size, png_size_t png_info_size)
G
Guy Schalnat 已提交
580
{
581
   /* We only come here via pre-1.0.12-compiled applications */
582
   if (png_ptr == NULL) return;
583
#if !defined(PNG_NO_STDIO) && !defined(_WIN32_WCE)
584
   if (png_sizeof(png_struct) > png_struct_size ||
585
      png_sizeof(png_info) > png_info_size)
586 587
   {
      char msg[80];
588
      png_ptr->warning_fn=NULL;
589 590
      if (user_png_ver)
      {
591 592
        png_snprintf(msg, 80,
           "Application was compiled with png.h from libpng-%.20s",
593 594 595
           user_png_ver);
        png_warning(png_ptr, msg);
      }
596 597
      png_snprintf(msg, 80,
         "Application  is  running with png.c from libpng-%.20s",
598 599 600 601
         png_libpng_ver);
      png_warning(png_ptr, msg);
   }
#endif
602
   if (png_sizeof(png_struct) > png_struct_size)
603
     {
604
       png_ptr->error_fn=NULL;
605 606 607
#ifdef PNG_ERROR_NUMBERS_SUPPORTED
       png_ptr->flags=0;
#endif
608
       png_error(png_ptr,
609
       "The png struct allocated by the application for writing is too small.");
610
     }
611
   if (png_sizeof(png_info) > png_info_size)
612
     {
613
       png_ptr->error_fn=NULL;
614 615 616
#ifdef PNG_ERROR_NUMBERS_SUPPORTED
       png_ptr->flags=0;
#endif
617
       png_error(png_ptr,
618
       "The info struct allocated by the application for writing is too small.");
619 620 621
     }
   png_write_init_3(&png_ptr, user_png_ver, png_struct_size);
}
622
#endif /* PNG_1_0_X || PNG_1_2_X */
623 624 625 626 627 628 629


void PNGAPI
png_write_init_3(png_structpp ptr_ptr, png_const_charp user_png_ver,
   png_size_t png_struct_size)
{
   png_structp png_ptr=*ptr_ptr;
630
#ifdef PNG_SETJMP_SUPPORTED
G
Guy Schalnat 已提交
631
   jmp_buf tmp_jmp; /* to save current jump buffer */
632
#endif
633

634 635
   int i = 0;

636 637 638
   if (png_ptr == NULL)
      return;

639 640 641 642
   do
   {
     if (user_png_ver[i] != png_libpng_ver[i])
     {
643 644 645
#ifdef PNG_LEGACY_SUPPORTED
       png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH;
#else
646
       png_ptr->warning_fn=NULL;
647
       png_warning(png_ptr,
648
 "Application uses deprecated png_write_init() and should be recompiled.");
649
       break;
650
#endif
651 652
     }
   } while (png_libpng_ver[i++]);
653

654
   png_debug(1, "in png_write_init_3\n");
G
Guy Schalnat 已提交
655

656
#ifdef PNG_SETJMP_SUPPORTED
G
Guy Schalnat 已提交
657
   /* save jump buffer and error functions */
658
   png_memcpy(tmp_jmp, png_ptr->jmpbuf, png_sizeof(jmp_buf));
659
#endif
G
Guy Schalnat 已提交
660

661
   if (png_sizeof(png_struct) > png_struct_size)
662 663 664 665 666 667
     {
       png_destroy_struct(png_ptr);
       png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG);
       *ptr_ptr = png_ptr;
     }

G
Guy Schalnat 已提交
668
   /* reset all variables to 0 */
669
   png_memset(png_ptr, 0, png_sizeof(png_struct));
G
Guy Schalnat 已提交
670

671 672 673 674 675 676
   /* added at libpng-1.2.6 */
#ifdef PNG_SET_USER_LIMITS_SUPPORTED
   png_ptr->user_width_max=PNG_USER_WIDTH_MAX;
   png_ptr->user_height_max=PNG_USER_HEIGHT_MAX;
#endif

677
#ifdef PNG_SETJMP_SUPPORTED
G
Guy Schalnat 已提交
678
   /* restore jump buffer */
679
   png_memcpy(png_ptr->jmpbuf, tmp_jmp, png_sizeof(jmp_buf));
680
#endif
G
Guy Schalnat 已提交
681

682 683
   png_set_write_fn(png_ptr, png_voidp_NULL, png_rw_ptr_NULL,
      png_flush_ptr_NULL);
684

G
Guy Schalnat 已提交
685 686
   /* initialize zbuf - compression buffer */
   png_ptr->zbuf_size = PNG_ZBUF_SIZE;
687 688
   png_ptr->zbuf = (png_bytep)png_malloc(png_ptr,
      (png_uint_32)png_ptr->zbuf_size);
A
Andreas Dilger 已提交
689 690 691

#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
   png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
692
      1, png_doublep_NULL, png_doublep_NULL);
A
Andreas Dilger 已提交
693
#endif
G
Guy Schalnat 已提交
694 695
}

696 697 698 699 700
/* Write a few rows of image data.  If the image is interlaced,
 * either you will have to write the 7 sub images, or, if you
 * have called png_set_interlace_handling(), you will have to
 * "write" the image seven times.
 */
701
void PNGAPI
G
Guy Schalnat 已提交
702
png_write_rows(png_structp png_ptr, png_bytepp row,
G
Guy Schalnat 已提交
703 704 705
   png_uint_32 num_rows)
{
   png_uint_32 i; /* row counter */
G
Guy Schalnat 已提交
706
   png_bytepp rp; /* row pointer */
G
Guy Schalnat 已提交
707

A
Andreas Dilger 已提交
708
   png_debug(1, "in png_write_rows\n");
709 710 711 712

   if (png_ptr == NULL)
      return;

G
Guy Schalnat 已提交
713 714 715 716 717 718 719
   /* loop through the rows */
   for (i = 0, rp = row; i < num_rows; i++, rp++)
   {
      png_write_row(png_ptr, *rp);
   }
}

720 721 722
/* Write the image.  You only need to call this function once, even
 * if you are writing an interlaced image.
 */
723
void PNGAPI
G
Guy Schalnat 已提交
724
png_write_image(png_structp png_ptr, png_bytepp image)
G
Guy Schalnat 已提交
725 726 727
{
   png_uint_32 i; /* row index */
   int pass, num_pass; /* pass variables */
G
Guy Schalnat 已提交
728
   png_bytepp rp; /* points to current row */
G
Guy Schalnat 已提交
729

730 731 732
   if (png_ptr == NULL)
      return;

A
Andreas Dilger 已提交
733
   png_debug(1, "in png_write_image\n");
734
#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
G
Guy Schalnat 已提交
735 736 737
   /* intialize interlace handling.  If image is not interlaced,
      this will set pass to 1 */
   num_pass = png_set_interlace_handling(png_ptr);
738 739 740
#else
   num_pass = 1;
#endif
G
Guy Schalnat 已提交
741 742 743 744 745 746 747 748 749 750 751
   /* loop through passes */
   for (pass = 0; pass < num_pass; pass++)
   {
      /* loop through image */
      for (i = 0, rp = image; i < png_ptr->height; i++, rp++)
      {
         png_write_row(png_ptr, *rp);
      }
   }
}

G
Guy Schalnat 已提交
752
/* called by user to write a row of image data */
753
void PNGAPI
G
Guy Schalnat 已提交
754
png_write_row(png_structp png_ptr, png_bytep row)
G
Guy Schalnat 已提交
755
{
756 757
   if (png_ptr == NULL)
      return;
758 759
   png_debug2(1, "in png_write_row (row %ld, pass %d)\n",
      png_ptr->row_number, png_ptr->pass);
760

G
Guy Schalnat 已提交
761
   /* initialize transformations and other stuff if first time */
G
Guy Schalnat 已提交
762
   if (png_ptr->row_number == 0 && png_ptr->pass == 0)
G
Guy Schalnat 已提交
763
   {
764 765 766 767 768
   /* make sure we wrote the header info */
   if (!(png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE))
      png_error(png_ptr,
         "png_write_info was never called before png_write_row.");

769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798
   /* check for transforms that have been set but were defined out */
#if !defined(PNG_WRITE_INVERT_SUPPORTED) && defined(PNG_READ_INVERT_SUPPORTED)
   if (png_ptr->transformations & PNG_INVERT_MONO)
      png_warning(png_ptr, "PNG_WRITE_INVERT_SUPPORTED is not defined.");
#endif
#if !defined(PNG_WRITE_FILLER_SUPPORTED) && defined(PNG_READ_FILLER_SUPPORTED)
   if (png_ptr->transformations & PNG_FILLER)
      png_warning(png_ptr, "PNG_WRITE_FILLER_SUPPORTED is not defined.");
#endif
#if !defined(PNG_WRITE_PACKSWAP_SUPPORTED) && defined(PNG_READ_PACKSWAP_SUPPORTED)
   if (png_ptr->transformations & PNG_PACKSWAP)
      png_warning(png_ptr, "PNG_WRITE_PACKSWAP_SUPPORTED is not defined.");
#endif
#if !defined(PNG_WRITE_PACK_SUPPORTED) && defined(PNG_READ_PACK_SUPPORTED)
   if (png_ptr->transformations & PNG_PACK)
      png_warning(png_ptr, "PNG_WRITE_PACK_SUPPORTED is not defined.");
#endif
#if !defined(PNG_WRITE_SHIFT_SUPPORTED) && defined(PNG_READ_SHIFT_SUPPORTED)
   if (png_ptr->transformations & PNG_SHIFT)
      png_warning(png_ptr, "PNG_WRITE_SHIFT_SUPPORTED is not defined.");
#endif
#if !defined(PNG_WRITE_BGR_SUPPORTED) && defined(PNG_READ_BGR_SUPPORTED)
   if (png_ptr->transformations & PNG_BGR)
      png_warning(png_ptr, "PNG_WRITE_BGR_SUPPORTED is not defined.");
#endif
#if !defined(PNG_WRITE_SWAP_SUPPORTED) && defined(PNG_READ_SWAP_SUPPORTED)
   if (png_ptr->transformations & PNG_SWAP_BYTES)
      png_warning(png_ptr, "PNG_WRITE_SWAP_SUPPORTED is not defined.");
#endif

G
Guy Schalnat 已提交
799 800 801
      png_write_start_row(png_ptr);
   }

G
Guy Schalnat 已提交
802
#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
G
Guy Schalnat 已提交
803 804 805 806 807
   /* if interlaced and not interested in row, return */
   if (png_ptr->interlaced && (png_ptr->transformations & PNG_INTERLACE))
   {
      switch (png_ptr->pass)
      {
G
Guy Schalnat 已提交
808
         case 0:
809
            if (png_ptr->row_number & 0x07)
G
Guy Schalnat 已提交
810 811 812 813 814 815
            {
               png_write_finish_row(png_ptr);
               return;
            }
            break;
         case 1:
816
            if ((png_ptr->row_number & 0x07) || png_ptr->width < 5)
G
Guy Schalnat 已提交
817 818 819 820 821 822
            {
               png_write_finish_row(png_ptr);
               return;
            }
            break;
         case 2:
823
            if ((png_ptr->row_number & 0x07) != 4)
G
Guy Schalnat 已提交
824 825
            {
               png_write_finish_row(png_ptr);
G
Guy Schalnat 已提交
826
               return;
G
Guy Schalnat 已提交
827 828 829
            }
            break;
         case 3:
830
            if ((png_ptr->row_number & 0x03) || png_ptr->width < 3)
G
Guy Schalnat 已提交
831 832 833 834 835 836
            {
               png_write_finish_row(png_ptr);
               return;
            }
            break;
         case 4:
837
            if ((png_ptr->row_number & 0x03) != 2)
G
Guy Schalnat 已提交
838 839 840 841 842 843
            {
               png_write_finish_row(png_ptr);
               return;
            }
            break;
         case 5:
844
            if ((png_ptr->row_number & 0x01) || png_ptr->width < 2)
G
Guy Schalnat 已提交
845 846 847 848 849 850
            {
               png_write_finish_row(png_ptr);
               return;
            }
            break;
         case 6:
851
            if (!(png_ptr->row_number & 0x01))
G
Guy Schalnat 已提交
852 853 854 855 856 857 858
            {
               png_write_finish_row(png_ptr);
               return;
            }
            break;
      }
   }
G
Guy Schalnat 已提交
859
#endif
G
Guy Schalnat 已提交
860 861

   /* set up row info for transformations */
G
Guy Schalnat 已提交
862
   png_ptr->row_info.color_type = png_ptr->color_type;
G
Guy Schalnat 已提交
863 864 865
   png_ptr->row_info.width = png_ptr->usr_width;
   png_ptr->row_info.channels = png_ptr->usr_channels;
   png_ptr->row_info.bit_depth = png_ptr->usr_bit_depth;
G
Guy Schalnat 已提交
866 867
   png_ptr->row_info.pixel_depth = (png_byte)(png_ptr->row_info.bit_depth *
      png_ptr->row_info.channels);
868

869 870
   png_ptr->row_info.rowbytes = PNG_ROWBYTES(png_ptr->row_info.pixel_depth,
      png_ptr->row_info.width);
G
Guy Schalnat 已提交
871

872
   png_debug1(3, "row_info->color_type = %d\n", png_ptr->row_info.color_type);
873
   png_debug1(3, "row_info->width = %lu\n", png_ptr->row_info.width);
874 875 876
   png_debug1(3, "row_info->channels = %d\n", png_ptr->row_info.channels);
   png_debug1(3, "row_info->bit_depth = %d\n", png_ptr->row_info.bit_depth);
   png_debug1(3, "row_info->pixel_depth = %d\n", png_ptr->row_info.pixel_depth);
877
   png_debug1(3, "row_info->rowbytes = %lu\n", png_ptr->row_info.rowbytes);
A
Andreas Dilger 已提交
878 879

   /* Copy user's row into buffer, leaving room for filter byte. */
880
   png_memcpy_check(png_ptr, png_ptr->row_buf + 1, row,
881
      png_ptr->row_info.rowbytes);
G
Guy Schalnat 已提交
882

G
Guy Schalnat 已提交
883
#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
G
Guy Schalnat 已提交
884 885 886 887 888
   /* handle interlacing */
   if (png_ptr->interlaced && png_ptr->pass < 6 &&
      (png_ptr->transformations & PNG_INTERLACE))
   {
      png_do_write_interlace(&(png_ptr->row_info),
G
Guy Schalnat 已提交
889
         png_ptr->row_buf + 1, png_ptr->pass);
G
Guy Schalnat 已提交
890 891 892 893 894 895 896
      /* this should always get caught above, but still ... */
      if (!(png_ptr->row_info.width))
      {
         png_write_finish_row(png_ptr);
         return;
      }
   }
G
Guy Schalnat 已提交
897
#endif
G
Guy Schalnat 已提交
898 899 900 901 902

   /* handle other transformations */
   if (png_ptr->transformations)
      png_do_write_transformations(png_ptr);

903
#if defined(PNG_MNG_FEATURES_SUPPORTED)
904 905 906 907 908 909 910 911 912
   /* Write filter_method 64 (intrapixel differencing) only if
    * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and
    * 2. Libpng did not write a PNG signature (this filter_method is only
    *    used in PNG datastreams that are embedded in MNG datastreams) and
    * 3. The application called png_permit_mng_features with a mask that
    *    included PNG_FLAG_MNG_FILTER_64 and
    * 4. The filter_method is 64 and
    * 5. The color_type is RGB or RGBA
    */
913
   if ((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
914 915 916 917 918 919 920
      (png_ptr->filter_type == PNG_INTRAPIXEL_DIFFERENCING))
   {
      /* Intrapixel differencing */
      png_do_write_intrapixel(&(png_ptr->row_info), png_ptr->row_buf + 1);
   }
#endif

A
Andreas Dilger 已提交
921
   /* Find a filter if necessary, filter the row and write it out. */
G
Guy Schalnat 已提交
922
   png_write_find_filter(png_ptr, &(png_ptr->row_info));
923 924 925

   if (png_ptr->write_row_fn != NULL)
      (*(png_ptr->write_row_fn))(png_ptr, png_ptr->row_number, png_ptr->pass);
G
Guy Schalnat 已提交
926 927 928 929
}

#if defined(PNG_WRITE_FLUSH_SUPPORTED)
/* Set the automatic flush interval or 0 to turn flushing off */
930
void PNGAPI
G
Guy Schalnat 已提交
931
png_set_flush(png_structp png_ptr, int nrows)
G
Guy Schalnat 已提交
932
{
A
Andreas Dilger 已提交
933
   png_debug(1, "in png_set_flush\n");
934 935
   if (png_ptr == NULL)
      return;
G
Guy Schalnat 已提交
936
   png_ptr->flush_dist = (nrows < 0 ? 0 : nrows);
G
Guy Schalnat 已提交
937 938 939
}

/* flush the current output buffers now */
940
void PNGAPI
G
Guy Schalnat 已提交
941
png_write_flush(png_structp png_ptr)
G
Guy Schalnat 已提交
942
{
G
Guy Schalnat 已提交
943 944
   int wrote_IDAT;

A
Andreas Dilger 已提交
945
   png_debug(1, "in png_write_flush\n");
946 947
   if (png_ptr == NULL)
      return;
G
Guy Schalnat 已提交
948 949
   /* We have already written out all of the data */
   if (png_ptr->row_number >= png_ptr->num_rows)
G
Guy Schalnat 已提交
950 951 952 953 954 955 956
     return;

   do
   {
      int ret;

      /* compress the data */
A
Andreas Dilger 已提交
957
      ret = deflate(&png_ptr->zstream, Z_SYNC_FLUSH);
G
Guy Schalnat 已提交
958 959 960 961 962
      wrote_IDAT = 0;

      /* check for compression errors */
      if (ret != Z_OK)
      {
A
Andreas Dilger 已提交
963
         if (png_ptr->zstream.msg != NULL)
A
Andreas Dilger 已提交
964
            png_error(png_ptr, png_ptr->zstream.msg);
G
Guy Schalnat 已提交
965 966 967 968
         else
            png_error(png_ptr, "zlib error");
      }

A
Andreas Dilger 已提交
969
      if (!(png_ptr->zstream.avail_out))
G
Guy Schalnat 已提交
970 971 972 973
      {
         /* write the IDAT and reset the zlib output buffer */
         png_write_IDAT(png_ptr, png_ptr->zbuf,
                        png_ptr->zbuf_size);
A
Andreas Dilger 已提交
974 975
         png_ptr->zstream.next_out = png_ptr->zbuf;
         png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
G
Guy Schalnat 已提交
976 977 978 979 980
         wrote_IDAT = 1;
      }
   } while(wrote_IDAT == 1);

   /* If there is any data left to be output, write it into a new IDAT */
A
Andreas Dilger 已提交
981
   if (png_ptr->zbuf_size != png_ptr->zstream.avail_out)
G
Guy Schalnat 已提交
982 983 984
   {
      /* write the IDAT and reset the zlib output buffer */
      png_write_IDAT(png_ptr, png_ptr->zbuf,
A
Andreas Dilger 已提交
985 986 987
                     png_ptr->zbuf_size - png_ptr->zstream.avail_out);
      png_ptr->zstream.next_out = png_ptr->zbuf;
      png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
G
Guy Schalnat 已提交
988 989 990
   }
   png_ptr->flush_rows = 0;
   png_flush(png_ptr);
G
Guy Schalnat 已提交
991
}
G
Guy Schalnat 已提交
992
#endif /* PNG_WRITE_FLUSH_SUPPORTED */
G
Guy Schalnat 已提交
993

G
Guy Schalnat 已提交
994
/* free all memory used by the write */
995
void PNGAPI
A
Andreas Dilger 已提交
996
png_destroy_write_struct(png_structpp png_ptr_ptr, png_infopp info_ptr_ptr)
G
Guy Schalnat 已提交
997
{
A
Andreas Dilger 已提交
998 999
   png_structp png_ptr = NULL;
   png_infop info_ptr = NULL;
1000 1001
#ifdef PNG_USER_MEM_SUPPORTED
   png_free_ptr free_fn = NULL;
1002
   png_voidp mem_ptr = NULL;
1003
#endif
A
Andreas Dilger 已提交
1004

A
Andreas Dilger 已提交
1005 1006
   png_debug(1, "in png_destroy_write_struct\n");
   if (png_ptr_ptr != NULL)
1007
   {
A
Andreas Dilger 已提交
1008
      png_ptr = *png_ptr_ptr;
1009 1010
#ifdef PNG_USER_MEM_SUPPORTED
      free_fn = png_ptr->free_fn;
1011
      mem_ptr = png_ptr->mem_ptr;
1012 1013
#endif
   }
A
Andreas Dilger 已提交
1014

1015 1016 1017 1018 1019 1020 1021 1022
#ifdef PNG_USER_MEM_SUPPORTED
   if (png_ptr != NULL)
   {
      free_fn = png_ptr->free_fn;
      mem_ptr = png_ptr->mem_ptr;
   }
#endif

A
Andreas Dilger 已提交
1023
   if (info_ptr_ptr != NULL)
A
Andreas Dilger 已提交
1024 1025
      info_ptr = *info_ptr_ptr;

A
Andreas Dilger 已提交
1026
   if (info_ptr != NULL)
G
Guy Schalnat 已提交
1027
   {
1028 1029 1030
      if (png_ptr != NULL)
      {
        png_free_data(png_ptr, info_ptr, PNG_FREE_ALL, -1);
1031

1032
#if defined(PNG_UNKNOWN_CHUNKS_SUPPORTED)
1033 1034 1035 1036 1037 1038
        if (png_ptr->num_chunk_list)
        {
           png_free(png_ptr, png_ptr->chunk_list);
           png_ptr->chunk_list=NULL;
           png_ptr->num_chunk_list=0;
        }
1039
#endif
1040
      }
1041

1042
#ifdef PNG_USER_MEM_SUPPORTED
1043 1044
      png_destroy_struct_2((png_voidp)info_ptr, (png_free_ptr)free_fn,
         (png_voidp)mem_ptr);
1045
#else
A
Andreas Dilger 已提交
1046
      png_destroy_struct((png_voidp)info_ptr);
1047
#endif
1048
      *info_ptr_ptr = NULL;
G
Guy Schalnat 已提交
1049 1050
   }

A
Andreas Dilger 已提交
1051
   if (png_ptr != NULL)
G
Guy Schalnat 已提交
1052
   {
A
Andreas Dilger 已提交
1053
      png_write_destroy(png_ptr);
1054
#ifdef PNG_USER_MEM_SUPPORTED
1055 1056
      png_destroy_struct_2((png_voidp)png_ptr, (png_free_ptr)free_fn,
         (png_voidp)mem_ptr);
1057
#else
A
Andreas Dilger 已提交
1058
      png_destroy_struct((png_voidp)png_ptr);
1059
#endif
1060
      *png_ptr_ptr = NULL;
G
Guy Schalnat 已提交
1061 1062 1063
   }
}

G
Guy Schalnat 已提交
1064

A
Andreas Dilger 已提交
1065
/* Free any memory used in png_ptr struct (old method) */
1066
void /* PRIVATE */
G
Guy Schalnat 已提交
1067
png_write_destroy(png_structp png_ptr)
G
Guy Schalnat 已提交
1068
{
1069
#ifdef PNG_SETJMP_SUPPORTED
G
Guy Schalnat 已提交
1070
   jmp_buf tmp_jmp; /* save jump buffer */
1071
#endif
G
Guy Schalnat 已提交
1072 1073 1074
   png_error_ptr error_fn;
   png_error_ptr warning_fn;
   png_voidp error_ptr;
1075 1076 1077
#ifdef PNG_USER_MEM_SUPPORTED
   png_free_ptr free_fn;
#endif
G
Guy Schalnat 已提交
1078

A
Andreas Dilger 已提交
1079
   png_debug(1, "in png_write_destroy\n");
G
Guy Schalnat 已提交
1080
   /* free any memory zlib uses */
A
Andreas Dilger 已提交
1081
   deflateEnd(&png_ptr->zstream);
G
Guy Schalnat 已提交
1082

G
Guy Schalnat 已提交
1083
   /* free our memory.  png_free checks NULL for us. */
A
Andreas Dilger 已提交
1084 1085
   png_free(png_ptr, png_ptr->zbuf);
   png_free(png_ptr, png_ptr->row_buf);
1086
#ifndef PNG_NO_WRITE_FILTER
A
Andreas Dilger 已提交
1087 1088 1089 1090 1091
   png_free(png_ptr, png_ptr->prev_row);
   png_free(png_ptr, png_ptr->sub_row);
   png_free(png_ptr, png_ptr->up_row);
   png_free(png_ptr, png_ptr->avg_row);
   png_free(png_ptr, png_ptr->paeth_row);
1092
#endif
1093

1094
#if defined(PNG_TIME_RFC1123_SUPPORTED)
1095
   png_free(png_ptr, png_ptr->time_buffer);
1096 1097
#endif

A
Andreas Dilger 已提交
1098 1099 1100 1101 1102 1103
#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
   png_free(png_ptr, png_ptr->prev_filters);
   png_free(png_ptr, png_ptr->filter_weights);
   png_free(png_ptr, png_ptr->inv_filter_weights);
   png_free(png_ptr, png_ptr->filter_costs);
   png_free(png_ptr, png_ptr->inv_filter_costs);
1104
#endif
G
Guy Schalnat 已提交
1105

1106
#ifdef PNG_SETJMP_SUPPORTED
G
Guy Schalnat 已提交
1107
   /* reset structure */
1108
   png_memcpy(tmp_jmp, png_ptr->jmpbuf, png_sizeof(jmp_buf));
1109
#endif
G
Guy Schalnat 已提交
1110 1111 1112 1113

   error_fn = png_ptr->error_fn;
   warning_fn = png_ptr->warning_fn;
   error_ptr = png_ptr->error_ptr;
1114 1115 1116
#ifdef PNG_USER_MEM_SUPPORTED
   free_fn = png_ptr->free_fn;
#endif
G
Guy Schalnat 已提交
1117

1118
   png_memset(png_ptr, 0, png_sizeof(png_struct));
G
Guy Schalnat 已提交
1119 1120 1121 1122

   png_ptr->error_fn = error_fn;
   png_ptr->warning_fn = warning_fn;
   png_ptr->error_ptr = error_ptr;
1123 1124 1125
#ifdef PNG_USER_MEM_SUPPORTED
   png_ptr->free_fn = free_fn;
#endif
G
Guy Schalnat 已提交
1126

1127
#ifdef PNG_SETJMP_SUPPORTED
1128
   png_memcpy(png_ptr->jmpbuf, tmp_jmp, png_sizeof(jmp_buf));
1129
#endif
G
Guy Schalnat 已提交
1130
}
G
Guy Schalnat 已提交
1131

A
Andreas Dilger 已提交
1132
/* Allow the application to select one or more row filters to use. */
1133
void PNGAPI
G
Guy Schalnat 已提交
1134
png_set_filter(png_structp png_ptr, int method, int filters)
G
Guy Schalnat 已提交
1135
{
A
Andreas Dilger 已提交
1136
   png_debug(1, "in png_set_filter\n");
1137 1138
   if (png_ptr == NULL)
      return;
1139
#if defined(PNG_MNG_FEATURES_SUPPORTED)
1140
   if ((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
1141 1142 1143
      (method == PNG_INTRAPIXEL_DIFFERENCING))
         method = PNG_FILTER_TYPE_BASE;
#endif
A
Andreas Dilger 已提交
1144
   if (method == PNG_FILTER_TYPE_BASE)
G
Guy Schalnat 已提交
1145 1146 1147
   {
      switch (filters & (PNG_ALL_FILTERS | 0x07))
      {
1148
#ifndef PNG_NO_WRITE_FILTER
G
Guy Schalnat 已提交
1149 1150
         case 5:
         case 6:
A
Andreas Dilger 已提交
1151
         case 7: png_warning(png_ptr, "Unknown row filter for method 0");
1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163
#endif /* PNG_NO_WRITE_FILTER */
         case PNG_FILTER_VALUE_NONE:
              png_ptr->do_filter=PNG_FILTER_NONE; break;
#ifndef PNG_NO_WRITE_FILTER
         case PNG_FILTER_VALUE_SUB:
              png_ptr->do_filter=PNG_FILTER_SUB; break;
         case PNG_FILTER_VALUE_UP:
              png_ptr->do_filter=PNG_FILTER_UP; break;
         case PNG_FILTER_VALUE_AVG:
              png_ptr->do_filter=PNG_FILTER_AVG; break;
         case PNG_FILTER_VALUE_PAETH:
              png_ptr->do_filter=PNG_FILTER_PAETH; break;
G
Guy Schalnat 已提交
1164
         default: png_ptr->do_filter = (png_byte)filters; break;
1165 1166 1167
#else
         default: png_warning(png_ptr, "Unknown row filter for method 0");
#endif /* PNG_NO_WRITE_FILTER */
G
Guy Schalnat 已提交
1168 1169
      }

A
Andreas Dilger 已提交
1170 1171 1172 1173 1174 1175 1176 1177
      /* If we have allocated the row_buf, this means we have already started
       * with the image and we should have allocated all of the filter buffers
       * that have been selected.  If prev_row isn't already allocated, then
       * it is too late to start using the filters that need it, since we
       * will be missing the data in the previous row.  If an application
       * wants to start and stop using particular filters during compression,
       * it should start out with all of the filters, and then add and
       * remove them after the start of compression.
G
Guy Schalnat 已提交
1178
       */
A
Andreas Dilger 已提交
1179
      if (png_ptr->row_buf != NULL)
G
Guy Schalnat 已提交
1180
      {
1181
#ifndef PNG_NO_WRITE_FILTER
1182
         if ((png_ptr->do_filter & PNG_FILTER_SUB) && png_ptr->sub_row == NULL)
G
Guy Schalnat 已提交
1183
         {
A
Andreas Dilger 已提交
1184
            png_ptr->sub_row = (png_bytep)png_malloc(png_ptr,
1185
              (png_ptr->rowbytes + 1));
A
Andreas Dilger 已提交
1186
            png_ptr->sub_row[0] = PNG_FILTER_VALUE_SUB;
G
Guy Schalnat 已提交
1187 1188
         }

1189
         if ((png_ptr->do_filter & PNG_FILTER_UP) && png_ptr->up_row == NULL)
G
Guy Schalnat 已提交
1190
         {
A
Andreas Dilger 已提交
1191
            if (png_ptr->prev_row == NULL)
G
Guy Schalnat 已提交
1192
            {
A
Andreas Dilger 已提交
1193
               png_warning(png_ptr, "Can't add Up filter after starting");
G
Guy Schalnat 已提交
1194 1195 1196 1197
               png_ptr->do_filter &= ~PNG_FILTER_UP;
            }
            else
            {
A
Andreas Dilger 已提交
1198
               png_ptr->up_row = (png_bytep)png_malloc(png_ptr,
1199
                  (png_ptr->rowbytes + 1));
A
Andreas Dilger 已提交
1200
               png_ptr->up_row[0] = PNG_FILTER_VALUE_UP;
G
Guy Schalnat 已提交
1201 1202 1203
            }
         }

1204
         if ((png_ptr->do_filter & PNG_FILTER_AVG) && png_ptr->avg_row == NULL)
G
Guy Schalnat 已提交
1205
         {
A
Andreas Dilger 已提交
1206
            if (png_ptr->prev_row == NULL)
G
Guy Schalnat 已提交
1207
            {
A
Andreas Dilger 已提交
1208
               png_warning(png_ptr, "Can't add Average filter after starting");
G
Guy Schalnat 已提交
1209 1210 1211 1212
               png_ptr->do_filter &= ~PNG_FILTER_AVG;
            }
            else
            {
A
Andreas Dilger 已提交
1213
               png_ptr->avg_row = (png_bytep)png_malloc(png_ptr,
1214
                  (png_ptr->rowbytes + 1));
A
Andreas Dilger 已提交
1215
               png_ptr->avg_row[0] = PNG_FILTER_VALUE_AVG;
G
Guy Schalnat 已提交
1216 1217 1218
            }
         }

1219
         if ((png_ptr->do_filter & PNG_FILTER_PAETH) &&
A
Andreas Dilger 已提交
1220
             png_ptr->paeth_row == NULL)
G
Guy Schalnat 已提交
1221
         {
A
Andreas Dilger 已提交
1222
            if (png_ptr->prev_row == NULL)
G
Guy Schalnat 已提交
1223 1224
            {
               png_warning(png_ptr, "Can't add Paeth filter after starting");
1225
               png_ptr->do_filter &= (png_byte)(~PNG_FILTER_PAETH);
G
Guy Schalnat 已提交
1226 1227 1228
            }
            else
            {
1229
               png_ptr->paeth_row = (png_bytep)png_malloc(png_ptr,
1230
                  (png_ptr->rowbytes + 1));
A
Andreas Dilger 已提交
1231
               png_ptr->paeth_row[0] = PNG_FILTER_VALUE_PAETH;
G
Guy Schalnat 已提交
1232 1233 1234 1235
            }
         }

         if (png_ptr->do_filter == PNG_NO_FILTERS)
1236
#endif /* PNG_NO_WRITE_FILTER */
G
Guy Schalnat 已提交
1237 1238 1239 1240
            png_ptr->do_filter = PNG_FILTER_NONE;
      }
   }
   else
A
Andreas Dilger 已提交
1241 1242 1243 1244 1245 1246 1247 1248
      png_error(png_ptr, "Unknown custom filter method");
}

/* This allows us to influence the way in which libpng chooses the "best"
 * filter for the current scanline.  While the "minimum-sum-of-absolute-
 * differences metric is relatively fast and effective, there is some
 * question as to whether it can be improved upon by trying to keep the
 * filtered data going to zlib more consistent, hopefully resulting in
1249 1250
 * better compression.
 */
A
Andreas Dilger 已提交
1251
#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)      /* GRR 970116 */
1252
void PNGAPI
A
Andreas Dilger 已提交
1253 1254 1255 1256 1257 1258 1259
png_set_filter_heuristics(png_structp png_ptr, int heuristic_method,
   int num_weights, png_doublep filter_weights,
   png_doublep filter_costs)
{
   int i;

   png_debug(1, "in png_set_filter_heuristics\n");
1260 1261
   if (png_ptr == NULL)
      return;
A
Andreas Dilger 已提交
1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278
   if (heuristic_method >= PNG_FILTER_HEURISTIC_LAST)
   {
      png_warning(png_ptr, "Unknown filter heuristic method");
      return;
   }

   if (heuristic_method == PNG_FILTER_HEURISTIC_DEFAULT)
   {
      heuristic_method = PNG_FILTER_HEURISTIC_UNWEIGHTED;
   }

   if (num_weights < 0 || filter_weights == NULL ||
      heuristic_method == PNG_FILTER_HEURISTIC_UNWEIGHTED)
   {
      num_weights = 0;
   }

1279 1280
   png_ptr->num_prev_filters = (png_byte)num_weights;
   png_ptr->heuristic_method = (png_byte)heuristic_method;
A
Andreas Dilger 已提交
1281 1282 1283 1284 1285 1286

   if (num_weights > 0)
   {
      if (png_ptr->prev_filters == NULL)
      {
         png_ptr->prev_filters = (png_bytep)png_malloc(png_ptr,
1287
            (png_uint_32)(png_sizeof(png_byte) * num_weights));
A
Andreas Dilger 已提交
1288

1289 1290
         /* To make sure that the weighting starts out fairly */
         for (i = 0; i < num_weights; i++)
A
Andreas Dilger 已提交
1291
         {
1292
            png_ptr->prev_filters[i] = 255;
A
Andreas Dilger 已提交
1293 1294 1295 1296 1297
         }
      }

      if (png_ptr->filter_weights == NULL)
      {
1298
         png_ptr->filter_weights = (png_uint_16p)png_malloc(png_ptr,
1299
            (png_uint_32)(png_sizeof(png_uint_16) * num_weights));
A
Andreas Dilger 已提交
1300

1301
         png_ptr->inv_filter_weights = (png_uint_16p)png_malloc(png_ptr,
1302
            (png_uint_32)(png_sizeof(png_uint_16) * num_weights));
1303
         for (i = 0; i < num_weights; i++)
A
Andreas Dilger 已提交
1304
         {
1305 1306
            png_ptr->inv_filter_weights[i] =
            png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
1307
         }
A
Andreas Dilger 已提交
1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331
      }

      for (i = 0; i < num_weights; i++)
      {
         if (filter_weights[i] < 0.0)
         {
            png_ptr->inv_filter_weights[i] =
            png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
         }
         else
         {
            png_ptr->inv_filter_weights[i] =
               (png_uint_16)((double)PNG_WEIGHT_FACTOR*filter_weights[i]+0.5);
            png_ptr->filter_weights[i] =
               (png_uint_16)((double)PNG_WEIGHT_FACTOR/filter_weights[i]+0.5);
         }
      }
   }

   /* If, in the future, there are other filter methods, this would
    * need to be based on png_ptr->filter.
    */
   if (png_ptr->filter_costs == NULL)
   {
1332
      png_ptr->filter_costs = (png_uint_16p)png_malloc(png_ptr,
1333
         (png_uint_32)(png_sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
A
Andreas Dilger 已提交
1334

1335
      png_ptr->inv_filter_costs = (png_uint_16p)png_malloc(png_ptr,
1336
         (png_uint_32)(png_sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
A
Andreas Dilger 已提交
1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366

      for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
      {
         png_ptr->inv_filter_costs[i] =
         png_ptr->filter_costs[i] = PNG_COST_FACTOR;
      }
   }

   /* Here is where we set the relative costs of the different filters.  We
    * should take the desired compression level into account when setting
    * the costs, so that Paeth, for instance, has a high relative cost at low
    * compression levels, while it has a lower relative cost at higher
    * compression settings.  The filter types are in order of increasing
    * relative cost, so it would be possible to do this with an algorithm.
    */
   for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
   {
      if (filter_costs == NULL || filter_costs[i] < 0.0)
      {
         png_ptr->inv_filter_costs[i] =
         png_ptr->filter_costs[i] = PNG_COST_FACTOR;
      }
      else if (filter_costs[i] >= 1.0)
      {
         png_ptr->inv_filter_costs[i] =
            (png_uint_16)((double)PNG_COST_FACTOR / filter_costs[i] + 0.5);
         png_ptr->filter_costs[i] =
            (png_uint_16)((double)PNG_COST_FACTOR * filter_costs[i] + 0.5);
      }
   }
G
Guy Schalnat 已提交
1367
}
A
Andreas Dilger 已提交
1368
#endif /* PNG_WRITE_WEIGHTED_FILTER_SUPPORTED */
G
Guy Schalnat 已提交
1369

1370
void PNGAPI
G
Guy Schalnat 已提交
1371
png_set_compression_level(png_structp png_ptr, int level)
G
Guy Schalnat 已提交
1372
{
A
Andreas Dilger 已提交
1373
   png_debug(1, "in png_set_compression_level\n");
1374 1375
   if (png_ptr == NULL)
      return;
G
Guy Schalnat 已提交
1376
   png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_LEVEL;
G
Guy Schalnat 已提交
1377 1378 1379
   png_ptr->zlib_level = level;
}

1380
void PNGAPI
G
Guy Schalnat 已提交
1381
png_set_compression_mem_level(png_structp png_ptr, int mem_level)
G
Guy Schalnat 已提交
1382
{
A
Andreas Dilger 已提交
1383
   png_debug(1, "in png_set_compression_mem_level\n");
1384 1385
   if (png_ptr == NULL)
      return;
G
Guy Schalnat 已提交
1386
   png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_MEM_LEVEL;
G
Guy Schalnat 已提交
1387
   png_ptr->zlib_mem_level = mem_level;
G
Guy Schalnat 已提交
1388 1389
}

1390
void PNGAPI
G
Guy Schalnat 已提交
1391
png_set_compression_strategy(png_structp png_ptr, int strategy)
G
Guy Schalnat 已提交
1392
{
A
Andreas Dilger 已提交
1393
   png_debug(1, "in png_set_compression_strategy\n");
1394 1395
   if (png_ptr == NULL)
      return;
G
Guy Schalnat 已提交
1396
   png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_STRATEGY;
G
Guy Schalnat 已提交
1397 1398 1399
   png_ptr->zlib_strategy = strategy;
}

1400
void PNGAPI
G
Guy Schalnat 已提交
1401
png_set_compression_window_bits(png_structp png_ptr, int window_bits)
G
Guy Schalnat 已提交
1402
{
1403 1404
   if (png_ptr == NULL)
      return;
G
Guy Schalnat 已提交
1405 1406
   if (window_bits > 15)
      png_warning(png_ptr, "Only compression windows <= 32k supported by PNG");
1407 1408
   else if (window_bits < 8)
      png_warning(png_ptr, "Only compression windows >= 256 supported by PNG");
1409 1410 1411 1412 1413 1414 1415 1416
#ifndef WBITS_8_OK
   /* avoid libpng bug with 256-byte windows */
   if (window_bits == 8)
     {
       png_warning(png_ptr, "Compression window is being reset to 512");
       window_bits=9;
     }
#endif
G
Guy Schalnat 已提交
1417
   png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_WINDOW_BITS;
G
Guy Schalnat 已提交
1418 1419 1420
   png_ptr->zlib_window_bits = window_bits;
}

1421
void PNGAPI
G
Guy Schalnat 已提交
1422
png_set_compression_method(png_structp png_ptr, int method)
G
Guy Schalnat 已提交
1423
{
A
Andreas Dilger 已提交
1424
   png_debug(1, "in png_set_compression_method\n");
1425 1426
   if (png_ptr == NULL)
      return;
G
Guy Schalnat 已提交
1427 1428 1429
   if (method != 8)
      png_warning(png_ptr, "Only compression method 8 is supported by PNG");
   png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_METHOD;
G
Guy Schalnat 已提交
1430
   png_ptr->zlib_method = method;
G
Guy Schalnat 已提交
1431 1432
}

1433
void PNGAPI
1434 1435
png_set_write_status_fn(png_structp png_ptr, png_write_status_ptr write_row_fn)
{
1436 1437
   if (png_ptr == NULL)
      return;
1438 1439 1440 1441
   png_ptr->write_row_fn = write_row_fn;
}

#if defined(PNG_WRITE_USER_TRANSFORM_SUPPORTED)
1442
void PNGAPI
1443 1444 1445 1446
png_set_write_user_transform_fn(png_structp png_ptr, png_user_transform_ptr
   write_user_transform_fn)
{
   png_debug(1, "in png_set_write_user_transform_fn\n");
1447 1448
   if (png_ptr == NULL)
      return;
1449 1450 1451 1452
   png_ptr->transformations |= PNG_USER_TRANSFORM;
   png_ptr->write_user_transform_fn = write_user_transform_fn;
}
#endif
1453 1454 1455


#if defined(PNG_INFO_IMAGE_SUPPORTED)
1456 1457
void PNGAPI
png_write_png(png_structp png_ptr, png_infop info_ptr,
1458
              int transforms, voidp params)
1459
{
1460 1461
   if (png_ptr == NULL || info_ptr == NULL)
      return;
1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533
#if defined(PNG_WRITE_INVERT_ALPHA_SUPPORTED)
   /* invert the alpha channel from opacity to transparency */
   if (transforms & PNG_TRANSFORM_INVERT_ALPHA)
       png_set_invert_alpha(png_ptr);
#endif

   /* Write the file header information. */
   png_write_info(png_ptr, info_ptr);

   /* ------ these transformations don't touch the info structure ------- */

#if defined(PNG_WRITE_INVERT_SUPPORTED)
   /* invert monochrome pixels */
   if (transforms & PNG_TRANSFORM_INVERT_MONO)
       png_set_invert_mono(png_ptr);
#endif

#if defined(PNG_WRITE_SHIFT_SUPPORTED)
   /* Shift the pixels up to a legal bit depth and fill in
    * as appropriate to correctly scale the image.
    */
   if ((transforms & PNG_TRANSFORM_SHIFT)
               && (info_ptr->valid & PNG_INFO_sBIT))
       png_set_shift(png_ptr, &info_ptr->sig_bit);
#endif

#if defined(PNG_WRITE_PACK_SUPPORTED)
   /* pack pixels into bytes */
   if (transforms & PNG_TRANSFORM_PACKING)
       png_set_packing(png_ptr);
#endif

#if defined(PNG_WRITE_SWAP_ALPHA_SUPPORTED)
   /* swap location of alpha bytes from ARGB to RGBA */
   if (transforms & PNG_TRANSFORM_SWAP_ALPHA)
       png_set_swap_alpha(png_ptr);
#endif

#if defined(PNG_WRITE_FILLER_SUPPORTED)
   /* Get rid of filler (OR ALPHA) bytes, pack XRGB/RGBX/ARGB/RGBA into
    * RGB (4 channels -> 3 channels). The second parameter is not used.
    */
   if (transforms & PNG_TRANSFORM_STRIP_FILLER)
       png_set_filler(png_ptr, 0, PNG_FILLER_BEFORE);
#endif

#if defined(PNG_WRITE_BGR_SUPPORTED)
   /* flip BGR pixels to RGB */
   if (transforms & PNG_TRANSFORM_BGR)
       png_set_bgr(png_ptr);
#endif

#if defined(PNG_WRITE_SWAP_SUPPORTED)
   /* swap bytes of 16-bit files to most significant byte first */
   if (transforms & PNG_TRANSFORM_SWAP_ENDIAN)
       png_set_swap(png_ptr);
#endif

#if defined(PNG_WRITE_PACKSWAP_SUPPORTED)
   /* swap bits of 1, 2, 4 bit packed pixel formats */
   if (transforms & PNG_TRANSFORM_PACKSWAP)
       png_set_packswap(png_ptr);
#endif

   /* ----------------------- end of transformations ------------------- */

   /* write the bits */
   if (info_ptr->valid & PNG_INFO_IDAT)
       png_write_image(png_ptr, info_ptr->row_pointers);

   /* It is REQUIRED to call this to finish writing the rest of the file */
   png_write_end(png_ptr, info_ptr);
1534

1535 1536
   transforms = transforms; /* quiet compiler warnings */
   params = params;
1537 1538
}
#endif
1539
#endif /* PNG_WRITE_SUPPORTED */