git.cworth.org Git - apitrace/blob - libpng/pngwrite.c

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