example.c 38.2 KB
Newer Older
1

2 3
#if 0 /* in case someone actually tries to compile this */

4
/* example.c - an example of using libpng
5
 * Last changed in libpng 1.6.0 [(PENDING RELEASE)]
6
 * Maintained 1998-2011 Glenn Randers-Pehrson
7 8
 * Maintained 1996, 1997 Andreas Dilger)
 * Written 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)
9 10 11
 * To the extent possible under law, the authors have waived
 * all copyright and related or neighboring rights to this file.
 * This work is published from: United States.
12
 */
G
Guy Schalnat 已提交
13

A
Andreas Dilger 已提交
14
/* This is an example of how to use libpng to read and write PNG files.
15
 * The file libpng-manual.txt is much more verbose then this.  If you have not
16
 * read it, do so first.  This was designed to be a starting point of an
17 18
 * implementation.  This is not officially part of libpng, is hereby placed
 * in the public domain, and therefore does not require a copyright notice.
19 20 21 22
 *
 * This file does not currently compile, because it is missing certain
 * parts, like allocating memory to hold an image.  You will have to
 * supply these parts to get it to compile.  For an example of a minimal
23 24
 * working PNG reader/writer, see pngtest.c, included in this distribution;
 * see also the programs in the contrib directory.
25
 */
G
Guy Schalnat 已提交
26

27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
/* The simple, but restricted, approach to reading a PNG file or data stream
 * just requires two function calls, as in the following complete program.
 * Writing a file just needs one function call, so long as the data has an
 * appropriate layout.
 *
 * The following code reads PNG image data from a file and writes it, in a
 * potentially new format, to a new file.  While this code will compile there is
 * minimal (insufficient) error checking; for a more realistic version look at
 * contrib/examples/pngtopng.c
 */
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <png.h>
42
#include <zlib.h>
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100

int main(int argc, const char **argv)
{
   if (argc == 3)
   {
      png_image image; /* The control structure used by libpng */

      /* Initialize the 'png_image' structure. */
      memset(&image, 0, sizeof image);

      /* The first argument is the file to read: */
      if (png_image_begin_read_from_file(&image, argv[1]))
      {
         png_bytep buffer;

         /* Set the format in which to read the PNG file; this code chooses a
          * simple sRGB format with a non-associated alpha channel, adequate to
          * store most images.
          */
         image.format = PNG_FORMAT_RGBA;

         /* Now allocate enough memory to hold the image in this format; the
          * PNG_IMAGE_SIZE macro uses the information about the image (width,
          * height and format) stored in 'image'.
          */
         buffer = malloc(PNG_IMAGE_SIZE(image));

         /* If enough memory was available read the image in the desired format
          * then write the result out to the new file.  'background' is not
          * necessary when reading the image because the alpha channel is
          * preserved; if it were to be removed, for example if we requested
          * PNG_FORMAT_RGB, then either a solid background color would have to
          * be supplied or the output buffer would have to be initialized to the
          * actual background of the image.
          *
          * The final argument to png_image_finish_read is the 'row_stride' -
          * this is the number of components allocated for the image in each
          * row.  It has to be at least as big as the value returned by
          * PNG_IMAGE_ROW_STRIDE, but if you just allocate space for the
          * default, minimum, size using PNG_IMAGE_SIZE as above you can pass
          * zero.
          */
         if (buffer != NULL &&
            png_image_finish_read(&image, NULL/*background*/, buffer,
               0/*row_stride*/))
         {
            /* Now write the image out to the second argument.  In the write
             * call 'convert_to_8bit' allows 16-bit data to be squashed down to
             * 8 bits; this isn't necessary here because the original read was
             * to the 8-bit format.
             */
            if (png_image_write_to_file(&image, argv[2], 0/*convert_to_8bit*/,
               buffer, 0/*row_stride*/))
            {
               /* The image has been written successfully. */
               exit(0);
            }
         }
101 102 103 104 105 106 107 108 109 110 111 112 113

         else
         {
            /* Calling png_free_image is optional unless the simplified API was
             * not run to completion.  In this case if there wasn't enough
             * memory for 'buffer' we didn't complete the read, so we must free
             * the image:
             */
            if (buffer == NULL)
               png_free_image(&image);

            else
               free(buffer);
114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181
      }

      /* Something went wrong reading or writing the image.  libpng stores a
       * textual message in the 'png_image' structure:
       */
      fprintf(stderr, "pngtopng: error: %s\n", image.message);
      exit (1);
   }

   fprintf(stderr, "pngtopng: usage: pngtopng input-file output-file\n");
   exit(1);
}

/* That's it ;-)  Of course you probably want to do more with PNG files than
 * just converting them all to 32-bit RGBA PNG files; you can do that between
 * the call to png_image_finish_read and png_image_write_to_file.  You can also
 * ask for the image data to be presented in a number of different formats.  You
 * do this by simply changing the 'format' parameter set before allocating the
 * buffer.
 *
 * The format parameter consists of five flags that define various aspects of
 * the image, you can simply add these together to get the format or you can use
 * one of the predefined macros from png.h (as above):
 *
 * PNG_FORMAT_FLAG_COLOR: if set the image will have three color components per
 *    pixel (red, green and blue), if not set the image will just have one
 *    luminance (grayscale) component.
 *
 * PNG_FORMAT_FLAG_ALPHA: if set each pixel in the image will have an additional
 *    alpha value; a linear value that describes the degree the image pixel
 *    covers (overwrites) the contents of the existing pixel on the display.
 *
 * PNG_FORMAT_FLAG_LINEAR: if set the components of each pixel will be returned
 *    as a series of 16-bit linear values, if not set the components will be
 *    returned as a series of 8-bit values encoded according to the 'sRGB'
 *    standard.  The 8-bit format is the normal format for images intended for
 *    direct display, because almost all display devices do the inverse of the
 *    sRGB transformation to the data they receive.  The 16-bit format is more
 *    common for scientific data and image data that must be further processed;
 *    because it is linear simple math can be done on the component values.
 *    Regardless of the setting of this flag the alpha channel is always linear,
 *    although it will be 8 bits or 16 bits wide as specified by the flag.
 *
 * PNG_FORMAT_FLAG_BGR: if set the components of a color pixel will be returned
 *    in the order blue, then green, then red.  If not set the pixel components
 *    are in the order red, then green, then blue.
 *
 * PNG_FORMAT_FLAG_AFIRST: if set the alpha channel (if present) precedes the
 *    color or grayscale components.  If not set the alpha channel follows the
 *    components.
 *
 * You do not have to read directly from a file.  You can read from memory or,
 * on systems that support it, from a <stdio.h> FILE*.  This is controlled by
 * the particular png_image_read_from_ function you call at the start.  Likewise
 * on write you can write to a FILE* if your system supports it.  Check the
 * macro PNG_STDIO_SUPPORTED to see if stdio support has been included in your
 * libpng build.
 *
 * If you read 16-bit (PNG_FORMAT_FLAG_LINEAR) data you may need to write it in
 * the 8-bit format for display.  You do this by setting the convert_to_8bit
 * flag to 'true'.
 *
 * Don't repeatedly convert between the 8-bit and 16-bit forms.  There is
 * significant data loss when 16-bit data is converted to the 8-bit encoding and
 * the current libpng implementation of convertion to 16-bit is also
 * significantly lossy.  The latter will be fixed in the future, but the former
 * is unavoidable - the 8-bit format just doesn't have enough resolution.
 */
182

183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199
/* If your program needs more information from the PNG data it reads, or if you
 * need to do more complex transformations, or minimise transformations, on the
 * data you read, then you must use one of the several lower level libpng
 * interfaces.
 *
 * All these interfaces require that you do your own error handling - your
 * program must be able to arrange for control to return to your own code any
 * time libpng encounters a problem.  There are several ways to do this, but the
 * standard way is to use the ANSI-C (C90) <setjmp.h> interface to establish a
 * return point within your own code.  You must do this if you do not use the
 * simplified interface (above).
 *
 * The first step is to include the header files you need, including the libpng
 * header file.  Include any standard headers and feature test macros your
 * program requires before including png.h:
 */
#include <png.h>
G
Guy Schalnat 已提交
200

201 202 203 204 205 206 207
 /* The png_jmpbuf() macro, used in error handling, became available in
  * libpng version 1.0.6.  If you want to be able to run your code with older
  * versions of libpng, you must define the macro yourself (but only if it
  * is not already defined by libpng!).
  */

#ifndef png_jmpbuf
208
#  define png_jmpbuf(png_ptr) ((png_ptr)->png_jmpbuf)
209 210
#endif

211 212 213 214 215
/* Check to see if a file is a PNG file using png_sig_cmp().  png_sig_cmp()
 * returns zero if the image is a PNG and nonzero if it isn't a PNG.
 *
 * The function check_if_png() shown here, but not used, returns nonzero (true)
 * if the file can be opened and is a PNG, 0 (false) otherwise.
216 217 218 219 220 221 222 223 224 225 226 227
 *
 * If this call is successful, and you are going to keep the file open,
 * you should call png_set_sig_bytes(png_ptr, PNG_BYTES_TO_CHECK); once
 * you have created the png_ptr, so that libpng knows your application
 * has read that many bytes from the start of the file.  Make sure you
 * don't call png_set_sig_bytes() with more than 8 bytes read or give it
 * an incorrect number of bytes read, or you will either have read too
 * many bytes (your fault), or you are telling libpng to read the wrong
 * number of magic bytes (also your fault).
 *
 * Many applications already read the first 2 or 4 bytes from the start
 * of the image to determine the file type, so it would be easiest just
228
 * to pass the bytes to png_sig_cmp() or even skip that if you know
229 230
 * you have a PNG file, and call png_set_sig_bytes().
 */
A
Andreas Dilger 已提交
231 232
#define PNG_BYTES_TO_CHECK 4
int check_if_png(char *file_name, FILE **fp)
G
Guy Schalnat 已提交
233
{
A
Andreas Dilger 已提交
234
   char buf[PNG_BYTES_TO_CHECK];
G
Guy Schalnat 已提交
235

A
Andreas Dilger 已提交
236
   /* Open the prospective PNG file. */
237
   if ((*fp = fopen(file_name, "rb")) == NULL)
G
Guy Schalnat 已提交
238 239
      return 0;

240
   /* Read in some of the signature bytes */
A
Andreas Dilger 已提交
241
   if (fread(buf, 1, PNG_BYTES_TO_CHECK, *fp) != PNG_BYTES_TO_CHECK)
G
Guy Schalnat 已提交
242 243
      return 0;

244 245 246
   /* Compare the first PNG_BYTES_TO_CHECK bytes of the signature.
      Return nonzero (true) if they match */

247
   return(!png_sig_cmp(buf, (png_size_t)0, PNG_BYTES_TO_CHECK));
G
Guy Schalnat 已提交
248 249
}

A
Andreas Dilger 已提交
250
/* Read a PNG file.  You may want to return an error code if the read
251 252 253 254 255
 * fails (depending upon the failure).  There are two "prototypes" given
 * here - one where we are given the filename, and we need to open the
 * file, and the other where we are given an open file (possibly with
 * some or all of the magic bytes read - see comments above).
 */
256
#ifdef open_file /* prototype 1 */
A
Andreas Dilger 已提交
257
void read_png(char *file_name)  /* We need to open the file */
G
Guy Schalnat 已提交
258
{
G
Guy Schalnat 已提交
259
   png_structp png_ptr;
G
Guy Schalnat 已提交
260
   png_infop info_ptr;
A
Andreas Dilger 已提交
261 262 263
   unsigned int sig_read = 0;
   png_uint_32 width, height;
   int bit_depth, color_type, interlace_type;
A
Andreas Dilger 已提交
264
   FILE *fp;
G
Guy Schalnat 已提交
265

A
Andreas Dilger 已提交
266
   if ((fp = fopen(file_name, "rb")) == NULL)
267
      return (ERROR);
268

269
#else no_open_file /* prototype 2 */
270
void read_png(FILE *fp, unsigned int sig_read)  /* File is already open */
A
Andreas Dilger 已提交
271 272 273
{
   png_structp png_ptr;
   png_infop info_ptr;
A
Andreas Dilger 已提交
274 275
   png_uint_32 width, height;
   int bit_depth, color_type, interlace_type;
276
#endif no_open_file /* Only use one prototype! */
G
Guy Schalnat 已提交
277

G
Guy Schalnat 已提交
278
   /* Create and initialize the png_struct with the desired error handler
A
Andreas Dilger 已提交
279 280 281 282 283
    * functions.  If you want to use the default stderr and longjump method,
    * you can supply NULL for the last three parameters.  We also supply the
    * the compiler header file version, so that we know if the application
    * was compiled with a compatible version of the library.  REQUIRED
    */
G
Guy Schalnat 已提交
284
   png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING,
285
      png_voidp user_error_ptr, user_error_fn, user_warning_fn);
G
Guy Schalnat 已提交
286

A
Andreas Dilger 已提交
287
   if (png_ptr == NULL)
G
Guy Schalnat 已提交
288 289
   {
      fclose(fp);
290
      return (ERROR);
G
Guy Schalnat 已提交
291 292
   }

A
Andreas Dilger 已提交
293
   /* Allocate/initialize the memory for image information.  REQUIRED. */
294
   info_ptr = png_create_info_struct(png_ptr);
A
Andreas Dilger 已提交
295
   if (info_ptr == NULL)
G
Guy Schalnat 已提交
296 297
   {
      fclose(fp);
298
      png_destroy_read_struct(&png_ptr, NULL, NULL);
299
      return (ERROR);
G
Guy Schalnat 已提交
300 301
   }

A
Andreas Dilger 已提交
302 303 304 305
   /* Set error handling if you are using the setjmp/longjmp method (this is
    * the normal method of doing things with libpng).  REQUIRED unless you
    * set up your own error handlers in the png_create_read_struct() earlier.
    */
306 307

   if (setjmp(png_jmpbuf(png_ptr)))
G
Guy Schalnat 已提交
308
   {
G
Guy Schalnat 已提交
309
      /* Free all of the memory associated with the png_ptr and info_ptr */
310
      png_destroy_read_struct(&png_ptr, &info_ptr, NULL);
G
Guy Schalnat 已提交
311 312
      fclose(fp);
      /* If we get here, we had a problem reading the file */
313
      return (ERROR);
G
Guy Schalnat 已提交
314 315
   }

A
Andreas Dilger 已提交
316
   /* One of the following I/O initialization methods is REQUIRED */
317
#ifdef streams /* PNG file I/O method 1 */
A
Andreas Dilger 已提交
318
   /* Set up the input control if you are using standard C streams */
G
Guy Schalnat 已提交
319 320
   png_init_io(png_ptr, fp);

321
#else no_streams /* PNG file I/O method 2 */
A
Andreas Dilger 已提交
322
   /* If you are using replacement read functions, instead of calling
323 324
    * png_init_io() here you would call:
    */
G
Guy Schalnat 已提交
325 326
   png_set_read_fn(png_ptr, (void *)user_io_ptr, user_read_fn);
   /* where user_io_ptr is a structure you want available to the callbacks */
327
#endif no_streams /* Use only one I/O method! */
G
Guy Schalnat 已提交
328

A
Andreas Dilger 已提交
329
   /* If we have already read some of the signature */
330
   png_set_sig_bytes(png_ptr, sig_read);
A
Andreas Dilger 已提交
331

332 333 334 335 336
#ifdef hilevel
   /*
    * If you have enough memory to read in the entire image at once,
    * and you need to specify only transforms that can be controlled
    * with one of the PNG_TRANSFORM_* bits (this presently excludes
337
    * quantizing, filling, setting background, and doing gamma
338 339 340
    * adjustment), then you can read the entire image (including
    * pixels) into the info structure with this call:
    */
341
   png_read_png(png_ptr, info_ptr, png_transforms, NULL);
342

343 344 345
#else
   /* OK, you're doing it the hard way, with the lower-level functions */

A
Andreas Dilger 已提交
346 347 348
   /* The call to png_read_info() gives us all of the information from the
    * PNG file before the first IDAT (image data chunk).  REQUIRED
    */
G
Guy Schalnat 已提交
349
   png_read_info(png_ptr, info_ptr);
G
Guy Schalnat 已提交
350

A
Andreas Dilger 已提交
351
   png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type,
352
       &interlace_type, NULL, NULL);
A
Andreas Dilger 已提交
353

354 355 356 357 358
   /* Set up the data transformations you want.  Note that these are all
    * optional.  Only call them if you want/need them.  Many of the
    * transformations only work on specific types of images, and many
    * are mutually exclusive.
    */
A
Andreas Dilger 已提交
359

360 361 362 363
   /* Tell libpng to strip 16 bit/color files down to 8 bits/color.
    * Use accurate scaling if it's available, otherwise just chop off the
    * low byte.
    */
364
#ifdef PNG_READ_SCALE_16_TO_8_SUPPORTED
365
    png_set_scale_16(png_ptr);
366
#else
367
   png_set_strip_16(png_ptr);
368
#endif
A
Andreas Dilger 已提交
369

370
   /* Strip alpha bytes from the input data without combining with the
371 372
    * background (not recommended).
    */
A
Andreas Dilger 已提交
373 374
   png_set_strip_alpha(png_ptr);

375
   /* Extract multiple pixels with bit depths of 1, 2, and 4 from a single
A
Andreas Dilger 已提交
376 377 378 379
    * byte into separate bytes (useful for paletted and grayscale images).
    */
   png_set_packing(png_ptr);

380
   /* Change the order of packed pixels to least significant bit first
A
Andreas Dilger 已提交
381 382
    * (not useful if you are using png_set_packing). */
   png_set_packswap(png_ptr);
G
Guy Schalnat 已提交
383

384
   /* Expand paletted colors into true RGB triplets */
A
Andreas Dilger 已提交
385
   if (color_type == PNG_COLOR_TYPE_PALETTE)
386
      png_set_palette_to_rgb(png_ptr);
G
Guy Schalnat 已提交
387

388
   /* Expand grayscale images to the full 8 bits from 1, 2, or 4 bits/pixel */
A
Andreas Dilger 已提交
389
   if (color_type == PNG_COLOR_TYPE_GRAY && bit_depth < 8)
390
      png_set_expand_gray_1_2_4_to_8(png_ptr);
G
Guy Schalnat 已提交
391

392 393 394
   /* Expand paletted or RGB images with transparency to full alpha channels
    * so the data will be available as RGBA quartets.
    */
A
Andreas Dilger 已提交
395
   if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS))
396
      png_set_tRNS_to_alpha(png_ptr);
G
Guy Schalnat 已提交
397

A
Andreas Dilger 已提交
398 399 400 401 402 403
   /* Set the background color to draw transparent and alpha images over.
    * It is possible to set the red, green, and blue components directly
    * for paletted images instead of supplying a palette index.  Note that
    * even if the PNG file supplies a background, you are not required to
    * use it - you should use the (solid) application background if it has one.
    */
G
Guy Schalnat 已提交
404

405
   png_color_16 my_background, *image_background;
G
Guy Schalnat 已提交
406

407 408
   if (png_get_bKGD(png_ptr, info_ptr, &image_background))
      png_set_background(png_ptr, image_background,
G
Guy Schalnat 已提交
409
                         PNG_BACKGROUND_GAMMA_FILE, 1, 1.0);
G
Guy Schalnat 已提交
410 411
   else
      png_set_background(png_ptr, &my_background,
G
Guy Schalnat 已提交
412
                         PNG_BACKGROUND_GAMMA_SCREEN, 0, 1.0);
G
Guy Schalnat 已提交
413

414 415 416
   /* Some suggestions as to how to get a screen gamma value
    *
    * Note that screen gamma is the display_exponent, which includes
417 418
    * the CRT_exponent and any correction for viewing conditions
    */
A
Andreas Dilger 已提交
419 420 421 422 423
   if (/* We have a user-defined screen gamma value */)
   {
      screen_gamma = user-defined screen_gamma;
   }
   /* This is one way that applications share the same screen gamma value */
424
   else if ((gamma_str = getenv("SCREEN_GAMMA")) != NULL)
A
Andreas Dilger 已提交
425 426 427 428
   {
      screen_gamma = atof(gamma_str);
   }
   /* If we don't have another value */
G
Guy Schalnat 已提交
429
   else
A
Andreas Dilger 已提交
430
   {
431
      screen_gamma = 2.2;  /* A good guess for a PC monitor in a dimly
432
                              lit room */
A
Andreas Dilger 已提交
433 434
      screen_gamma = 1.7 or 1.0;  /* A good guess for Mac systems */
   }
G
Guy Schalnat 已提交
435

436
   /* Tell libpng to handle the gamma conversion for you.  The final call
A
Andreas Dilger 已提交
437 438 439 440
    * is a good guess for PC generated images, but it should be configurable
    * by the user at run time by the user.  It is strongly suggested that
    * your application support gamma correction.
    */
441 442 443

   int intent;

444
   if (png_get_sRGB(png_ptr, info_ptr, &intent))
445
      png_set_gamma(png_ptr, screen_gamma, 0.45455);
446 447 448
   else
   {
      double image_gamma;
449
      if (png_get_gAMA(png_ptr, info_ptr, &image_gamma))
450 451
         png_set_gamma(png_ptr, screen_gamma, image_gamma);
      else
452 453
         png_set_gamma(png_ptr, screen_gamma, 0.45455);
   }
G
Guy Schalnat 已提交
454

455 456
#ifdef PNG_READ_QUANTIZE_SUPPORTED
   /* Quantize RGB files down to 8 bit palette or reduce palettes
457 458
    * to the number of colors available on your screen.
    */
A
Andreas Dilger 已提交
459
   if (color_type & PNG_COLOR_MASK_COLOR)
G
Guy Schalnat 已提交
460
   {
461
      int num_palette;
A
Andreas Dilger 已提交
462 463 464
      png_colorp palette;

      /* This reduces the image to the application supplied palette */
465
      if (/* We have our own palette */)
G
Guy Schalnat 已提交
466
      {
467
         /* An array of colors to which the image should be quantized */
A
Andreas Dilger 已提交
468
         png_color std_color_cube[MAX_SCREEN_COLORS];
G
Guy Schalnat 已提交
469

470
         png_set_quantize(png_ptr, std_color_cube, MAX_SCREEN_COLORS,
471
            MAX_SCREEN_COLORS, NULL, 0);
A
Andreas Dilger 已提交
472 473
      }
      /* This reduces the image to the palette supplied in the file */
474
      else if (png_get_PLTE(png_ptr, info_ptr, &palette, &num_palette))
A
Andreas Dilger 已提交
475
      {
476
         png_uint_16p histogram = NULL;
A
Andreas Dilger 已提交
477 478 479

         png_get_hIST(png_ptr, info_ptr, &histogram);

480
         png_set_quantize(png_ptr, palette, num_palette,
A
Andreas Dilger 已提交
481
                        max_screen_colors, histogram, 0);
G
Guy Schalnat 已提交
482
      }
G
Guy Schalnat 已提交
483
   }
484
#endif /* PNG_READ_QUANTIZE_SUPPORTED */
G
Guy Schalnat 已提交
485

486
   /* Invert monochrome files to have 0 as white and 1 as black */
487
   png_set_invert_mono(png_ptr);
G
Guy Schalnat 已提交
488

A
Andreas Dilger 已提交
489 490 491 492 493 494
   /* If you want to shift the pixel values from the range [0,255] or
    * [0,65535] to the original [0,7] or [0,31], or whatever range the
    * colors were originally in:
    */
   if (png_get_valid(png_ptr, info_ptr, PNG_INFO_sBIT))
   {
495
      png_color_8p sig_bit_p;
A
Andreas Dilger 已提交
496

497 498
      png_get_sBIT(png_ptr, info_ptr, &sig_bit_p);
      png_set_shift(png_ptr, sig_bit_p);
A
Andreas Dilger 已提交
499
   }
G
Guy Schalnat 已提交
500

501
   /* Flip the RGB pixels to BGR (or RGBA to BGRA) */
502 503
   if (color_type & PNG_COLOR_MASK_COLOR)
      png_set_bgr(png_ptr);
G
Guy Schalnat 已提交
504

505
   /* Swap the RGBA or GA data to ARGB or AG (or BGRA to ABGR) */
A
Andreas Dilger 已提交
506
   png_set_swap_alpha(png_ptr);
G
Guy Schalnat 已提交
507

508
   /* Swap bytes of 16 bit files to least significant byte first */
A
Andreas Dilger 已提交
509
   png_set_swap(png_ptr);
G
Guy Schalnat 已提交
510

A
Andreas Dilger 已提交
511 512
   /* Add filler (or alpha) byte (before/after each RGB triplet) */
   png_set_filler(png_ptr, 0xff, PNG_FILLER_AFTER);
G
Guy Schalnat 已提交
513

514
#ifdef PNG_READ_INTERLACING_SUPPORTED
A
Andreas Dilger 已提交
515 516
   /* Turn on interlace handling.  REQUIRED if you are not using
    * png_read_image().  To see how to handle interlacing passes,
517
    * see the png_read_row() method below:
A
Andreas Dilger 已提交
518
    */
G
Guy Schalnat 已提交
519
   number_passes = png_set_interlace_handling(png_ptr);
520 521 522 523
#else
   number_passes = 1;
#endif /* PNG_READ_INTERLACING_SUPPORTED */

G
Guy Schalnat 已提交
524

525
   /* Optional call to gamma correct and add the background to the palette
A
Andreas Dilger 已提交
526 527 528
    * and update info structure.  REQUIRED if you are expecting libpng to
    * update the palette for you (ie you selected such a transform above).
    */
G
Guy Schalnat 已提交
529 530
   png_read_update_info(png_ptr, info_ptr);

531
   /* Allocate the memory to hold the image using the fields of info_ptr. */
G
Guy Schalnat 已提交
532

533
   /* The easiest way to read the image: */
G
Guy Schalnat 已提交
534
   png_bytep row_pointers[height];
G
Guy Schalnat 已提交
535

536 537 538 539
   /* Clear the pointer array */
   for (row = 0; row < height; row++)
      row_pointers[row] = NULL;

G
Guy Schalnat 已提交
540
   for (row = 0; row < height; row++)
541 542
      row_pointers[row] = png_malloc(png_ptr, png_get_rowbytes(png_ptr,
         info_ptr));
G
Guy Schalnat 已提交
543

A
Andreas Dilger 已提交
544
   /* Now it's time to read the image.  One of these methods is REQUIRED */
545
#ifdef entire /* Read the entire image in one go */
G
Guy Schalnat 已提交
546 547
   png_read_image(png_ptr, row_pointers);

548
#else no_entire /* Read the image one or more scanlines at a time */
549
   /* The other way to read images - deal with interlacing: */
G
Guy Schalnat 已提交
550 551 552

   for (pass = 0; pass < number_passes; pass++)
   {
553
#ifdef single /* Read the image a single row at a time */
G
Guy Schalnat 已提交
554 555
      for (y = 0; y < height; y++)
      {
556
         png_read_rows(png_ptr, &row_pointers[y], NULL, 1);
G
Guy Schalnat 已提交
557 558
      }

559
#else no_single /* Read the image several rows at a time */
A
Andreas Dilger 已提交
560 561
      for (y = 0; y < height; y += number_of_rows)
      {
562
#ifdef sparkle /* Read the image using the "sparkle" effect. */
563 564
         png_read_rows(png_ptr, &row_pointers[y], NULL,
            number_of_rows);
565
#else no_sparkle /* Read the image using the "rectangle" effect */
566 567
         png_read_rows(png_ptr, NULL, &row_pointers[y],
            number_of_rows);
568
#endif no_sparkle /* Use only one of these two methods */
A
Andreas Dilger 已提交
569
      }
570

571 572
      /* If you want to display the image after every pass, do so here */
#endif no_single /* Use only one of these two methods */
G
Guy Schalnat 已提交
573
   }
574
#endif no_entire /* Use only one of these two methods */
G
Guy Schalnat 已提交
575

576
   /* Read rest of file, and get additional chunks in info_ptr - REQUIRED */
G
Guy Schalnat 已提交
577
   png_read_end(png_ptr, info_ptr);
578 579 580
#endif hilevel

   /* At this point you have read the entire image */
G
Guy Schalnat 已提交
581

582
   /* Clean up after the read, and free any memory allocated - REQUIRED */
583
   png_destroy_read_struct(&png_ptr, &info_ptr, NULL);
G
Guy Schalnat 已提交
584

585
   /* Close the file */
G
Guy Schalnat 已提交
586 587
   fclose(fp);

588
   /* That's it */
589
   return (OK);
G
Guy Schalnat 已提交
590 591
}

592
/* Progressively read a file */
G
Guy Schalnat 已提交
593 594

int
G
Guy Schalnat 已提交
595
initialize_png_reader(png_structp *png_ptr, png_infop *info_ptr)
G
Guy Schalnat 已提交
596
{
G
Guy Schalnat 已提交
597
   /* Create and initialize the png_struct with the desired error handler
A
Andreas Dilger 已提交
598 599 600 601
    * functions.  If you want to use the default stderr and longjump method,
    * you can supply NULL for the last three parameters.  We also check that
    * the library version is compatible in case we are using dynamically
    * linked libraries.
G
Guy Schalnat 已提交
602 603
    */
   *png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING,
604
       png_voidp user_error_ptr, user_error_fn, user_warning_fn);
G
Guy Schalnat 已提交
605

A
Andreas Dilger 已提交
606
   if (*png_ptr == NULL)
G
Guy Schalnat 已提交
607
   {
G
Guy Schalnat 已提交
608
      *info_ptr = NULL;
609
      return (ERROR);
G
Guy Schalnat 已提交
610 611
   }

G
Guy Schalnat 已提交
612 613
   *info_ptr = png_create_info_struct(png_ptr);

A
Andreas Dilger 已提交
614
   if (*info_ptr == NULL)
G
Guy Schalnat 已提交
615
   {
616
      png_destroy_read_struct(png_ptr, info_ptr, NULL);
617
      return (ERROR);
G
Guy Schalnat 已提交
618 619
   }

620
   if (setjmp(png_jmpbuf((*png_ptr))))
G
Guy Schalnat 已提交
621
   {
622
      png_destroy_read_struct(png_ptr, info_ptr, NULL);
623
      return (ERROR);
G
Guy Schalnat 已提交
624
   }
G
Guy Schalnat 已提交
625

626
   /* This one's new.  You will need to provide all three
A
Andreas Dilger 已提交
627
    * function callbacks, even if you aren't using them all.
628 629 630
    * If you aren't using all functions, you can specify NULL
    * parameters.  Even when all three functions are NULL,
    * you need to call png_set_progressive_read_fn().
A
Andreas Dilger 已提交
631 632 633 634 635 636 637
    * These functions shouldn't be dependent on global or
    * static variables if you are decoding several images
    * simultaneously.  You should store stream specific data
    * in a separate struct, given as the second parameter,
    * and retrieve the pointer from inside the callbacks using
    * the function png_get_progressive_ptr(png_ptr).
    */
G
Guy Schalnat 已提交
638
   png_set_progressive_read_fn(*png_ptr, (void *)stream_data,
G
Guy Schalnat 已提交
639 640
      info_callback, row_callback, end_callback);

641
   return (OK);
G
Guy Schalnat 已提交
642 643 644
}

int
G
Guy Schalnat 已提交
645 646
process_data(png_structp *png_ptr, png_infop *info_ptr,
   png_bytep buffer, png_uint_32 length)
G
Guy Schalnat 已提交
647
{
648
   if (setjmp(png_jmpbuf((*png_ptr))))
G
Guy Schalnat 已提交
649
   {
G
Guy Schalnat 已提交
650
      /* Free the png_ptr and info_ptr memory on error */
651
      png_destroy_read_struct(png_ptr, info_ptr, NULL);
652
      return (ERROR);
G
Guy Schalnat 已提交
653 654
   }

A
Andreas Dilger 已提交
655 656
   /* This one's new also.  Simply give it chunks of data as
    * they arrive from the data stream (in order, of course).
657
    * On segmented machines, don't give it any more than 64K.
A
Andreas Dilger 已提交
658 659 660 661 662 663 664
    * The library seems to run fine with sizes of 4K, although
    * you can give it much less if necessary (I assume you can
    * give it chunks of 1 byte, but I haven't tried with less
    * than 256 bytes yet).  When this function returns, you may
    * want to display any rows that were generated in the row
    * callback, if you aren't already displaying them there.
    */
G
Guy Schalnat 已提交
665
   png_process_data(*png_ptr, *info_ptr, buffer, length);
666
   return (OK);
G
Guy Schalnat 已提交
667 668 669 670
}

info_callback(png_structp png_ptr, png_infop info)
{
671 672 673 674 675 676 677
   /* Do any setup here, including setting any of the transformations
    * mentioned in the Reading PNG files section.  For now, you _must_
    * call either png_start_read_image() or png_read_update_info()
    * after all the transformations are set (even if you don't set
    * any).  You may start getting rows before png_process_data()
    * returns, so this is your last chance to prepare for that.
    */
G
Guy Schalnat 已提交
678 679 680
}

row_callback(png_structp png_ptr, png_bytep new_row,
G
Guy Schalnat 已提交
681
   png_uint_32 row_num, int pass)
G
Guy Schalnat 已提交
682
{
683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699
   /*
    * This function is called for every row in the image.  If the
    * image is interlaced, and you turned on the interlace handler,
    * this function will be called for every row in every pass.
    *
    * In this function you will receive a pointer to new row data from
    * libpng called new_row that is to replace a corresponding row (of
    * the same data format) in a buffer allocated by your application.
    *
    * The new row data pointer "new_row" may be NULL, indicating there is
    * no new data to be replaced (in cases of interlace loading).
    *
    * If new_row is not NULL then you need to call
    * png_progressive_combine_row() to replace the corresponding row as
    * shown below:
    */

700 701 702 703 704
   /* Get pointer to corresponding row in our
    * PNG read buffer.
    */
   png_bytep old_row = ((png_bytep *)our_data)[row_num];

705
#ifdef PNG_READ_INTERLACING_SUPPORTED
706 707 708 709 710 711
   /* If both rows are allocated then copy the new row
    * data to the corresponding row data.
    */
   if ((old_row != NULL) && (new_row != NULL))
   png_progressive_combine_row(png_ptr, old_row, new_row);

712 713 714 715 716 717 718 719 720 721 722 723
   /*
    * The rows and passes are called in order, so you don't really
    * need the row_num and pass, but I'm supplying them because it
    * may make your life easier.
    *
    * For the non-NULL rows of interlaced images, you must call
    * png_progressive_combine_row() passing in the new row and the
    * old row, as demonstrated above.  You can call this function for
    * NULL rows (it will just return) and for non-interlaced images
    * (it just does the png_memcpy for you) if it will make the code
    * easier.  Thus, you can just do this for all cases:
    */
G
Guy Schalnat 已提交
724 725 726

   png_progressive_combine_row(png_ptr, old_row, new_row);

727 728 729 730 731 732 733
   /* where old_row is what was displayed for previous rows.  Note
    * that the first pass (pass == 0 really) will completely cover
    * the old row, so the rows do not have to be initialized.  After
    * the first pass (and only for interlaced images), you will have
    * to pass the current row as new_row, and the function will combine
    * the old row and the new row.
    */
734
#endif /* PNG_READ_INTERLACING_SUPPORTED */
G
Guy Schalnat 已提交
735 736 737 738
}

end_callback(png_structp png_ptr, png_infop info)
{
739 740 741 742 743 744 745 746 747
   /* This function is called when the whole image has been read,
    * including any chunks after the image (up to and including
    * the IEND).  You will usually have the same info chunk as you
    * had in the header, although some data may have been added
    * to the comments and time fields.
    *
    * Most people won't do much here, perhaps setting a flag that
    * marks the image as finished.
    */
G
Guy Schalnat 已提交
748 749
}

750
/* Write a png file */
751
void write_png(char *file_name /* , ... other image information ... */)
G
Guy Schalnat 已提交
752 753
{
   FILE *fp;
G
Guy Schalnat 已提交
754 755
   png_structp png_ptr;
   png_infop info_ptr;
756
   png_colorp palette;
G
Guy Schalnat 已提交
757

758
   /* Open the file */
G
Guy Schalnat 已提交
759
   fp = fopen(file_name, "wb");
A
Andreas Dilger 已提交
760
   if (fp == NULL)
761
      return (ERROR);
G
Guy Schalnat 已提交
762

G
Guy Schalnat 已提交
763
   /* Create and initialize the png_struct with the desired error handler
A
Andreas Dilger 已提交
764 765 766 767
    * functions.  If you want to use the default stderr and longjump method,
    * you can supply NULL for the last three parameters.  We also check that
    * the library version is compatible with the one used at compile time,
    * in case we are using dynamically linked libraries.  REQUIRED.
G
Guy Schalnat 已提交
768 769
    */
   png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING,
770
      png_voidp user_error_ptr, user_error_fn, user_warning_fn);
G
Guy Schalnat 已提交
771

A
Andreas Dilger 已提交
772
   if (png_ptr == NULL)
G
Guy Schalnat 已提交
773 774
   {
      fclose(fp);
775
      return (ERROR);
G
Guy Schalnat 已提交
776 777
   }

A
Andreas Dilger 已提交
778
   /* Allocate/initialize the image information data.  REQUIRED */
G
Guy Schalnat 已提交
779
   info_ptr = png_create_info_struct(png_ptr);
A
Andreas Dilger 已提交
780
   if (info_ptr == NULL)
G
Guy Schalnat 已提交
781 782
   {
      fclose(fp);
783
      png_destroy_write_struct(&png_ptr,  NULL);
784
      return (ERROR);
G
Guy Schalnat 已提交
785 786
   }

A
Andreas Dilger 已提交
787
   /* Set error handling.  REQUIRED if you aren't supplying your own
788
    * error handling functions in the png_create_write_struct() call.
A
Andreas Dilger 已提交
789
    */
790
   if (setjmp(png_jmpbuf(png_ptr)))
G
Guy Schalnat 已提交
791
   {
792
      /* If we get here, we had a problem writing the file */
G
Guy Schalnat 已提交
793
      fclose(fp);
794
      png_destroy_write_struct(&png_ptr, &info_ptr);
795
      return (ERROR);
G
Guy Schalnat 已提交
796 797
   }

A
Andreas Dilger 已提交
798
   /* One of the following I/O initialization functions is REQUIRED */
799

800
#ifdef streams /* I/O initialization method 1 */
801
   /* Set up the output control if you are using standard C streams */
G
Guy Schalnat 已提交
802
   png_init_io(png_ptr, fp);
803

804
#else no_streams /* I/O initialization method 2 */
805
   /* If you are using replacement write functions, instead of calling
806 807
    * png_init_io() here you would call
    */
A
Andreas Dilger 已提交
808 809 810
   png_set_write_fn(png_ptr, (void *)user_io_ptr, user_write_fn,
      user_IO_flush_function);
   /* where user_io_ptr is a structure you want available to the callbacks */
811
#endif no_streams /* Only use one initialization method */
A
Andreas Dilger 已提交
812

813 814
#ifdef hilevel
   /* This is the easy way.  Use it if you already have all the
815
    * image info living in the structure.  You could "|" many
816 817
    * PNG_TRANSFORM flags into the png_transforms integer here.
    */
818
   png_write_png(png_ptr, info_ptr, png_transforms, NULL);
819

820 821 822
#else
   /* This is the hard way */

A
Andreas Dilger 已提交
823 824 825 826 827 828 829 830 831 832
   /* Set the image information here.  Width and height are up to 2^31,
    * bit_depth is one of 1, 2, 4, 8, or 16, but valid values also depend on
    * the color_type selected. color_type is one of PNG_COLOR_TYPE_GRAY,
    * PNG_COLOR_TYPE_GRAY_ALPHA, PNG_COLOR_TYPE_PALETTE, PNG_COLOR_TYPE_RGB,
    * or PNG_COLOR_TYPE_RGB_ALPHA.  interlace is either PNG_INTERLACE_NONE or
    * PNG_INTERLACE_ADAM7, and the compression_type and filter_type MUST
    * currently be PNG_COMPRESSION_TYPE_BASE and PNG_FILTER_TYPE_BASE. REQUIRED
    */
   png_set_IHDR(png_ptr, info_ptr, width, height, bit_depth, PNG_COLOR_TYPE_???,
      PNG_INTERLACE_????, PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
G
Guy Schalnat 已提交
833

834
   /* Set the palette if there is one.  REQUIRED for indexed-color images */
835
   palette = (png_colorp)png_malloc(png_ptr, PNG_MAX_PALETTE_LENGTH
836
             * png_sizeof(png_color));
837
   /* ... Set palette colors ... */
838
   png_set_PLTE(png_ptr, info_ptr, palette, PNG_MAX_PALETTE_LENGTH);
839
   /* You must not free palette here, because png_set_PLTE only makes a link to
840 841 842
    * the palette that you malloced.  Wait until you are about to destroy
    * the png structure.
    */
G
Guy Schalnat 已提交
843

844
   /* Optional significant bit (sBIT) chunk */
845
   png_color_8 sig_bit;
846

847
   /* If we are dealing with a grayscale image then */
A
Andreas Dilger 已提交
848
   sig_bit.gray = true_bit_depth;
849

850
   /* Otherwise, if we are dealing with a color image then */
A
Andreas Dilger 已提交
851 852 853
   sig_bit.red = true_red_bit_depth;
   sig_bit.green = true_green_bit_depth;
   sig_bit.blue = true_blue_bit_depth;
854

855
   /* If the image has an alpha channel then */
A
Andreas Dilger 已提交
856
   sig_bit.alpha = true_alpha_bit_depth;
857

858
   png_set_sBIT(png_ptr, info_ptr, &sig_bit);
G
Guy Schalnat 已提交
859

860

A
Andreas Dilger 已提交
861
   /* Optional gamma chunk is strongly suggested if you have any guess
862 863
    * as to the correct gamma of the image.
    */
A
Andreas Dilger 已提交
864 865 866 867 868 869
   png_set_gAMA(png_ptr, info_ptr, gamma);

   /* Optionally write comments into the image */
   text_ptr[0].key = "Title";
   text_ptr[0].text = "Mona Lisa";
   text_ptr[0].compression = PNG_TEXT_COMPRESSION_NONE;
870 871 872
   text_ptr[0].itxt_length = 0;
   text_ptr[0].lang = NULL;
   text_ptr[0].lang_key = NULL;
A
Andreas Dilger 已提交
873 874 875
   text_ptr[1].key = "Author";
   text_ptr[1].text = "Leonardo DaVinci";
   text_ptr[1].compression = PNG_TEXT_COMPRESSION_NONE;
876 877 878
   text_ptr[1].itxt_length = 0;
   text_ptr[1].lang = NULL;
   text_ptr[1].lang_key = NULL;
A
Andreas Dilger 已提交
879 880 881
   text_ptr[2].key = "Description";
   text_ptr[2].text = "<long text>";
   text_ptr[2].compression = PNG_TEXT_COMPRESSION_zTXt;
882
   text_ptr[2].itxt_length = 0;
883
   text_ptr[2].lang = NULL;
884
   text_ptr[2].lang_key = NULL;
885
   png_set_text(png_ptr, info_ptr, text_ptr, 3);
A
Andreas Dilger 已提交
886

887
   /* Other optional chunks like cHRM, bKGD, tRNS, tIME, oFFs, pHYs */
888

889
   /* Note that if sRGB is present the gAMA and cHRM chunks must be ignored
890
    * on read and, if your application chooses to write them, they must
891 892
    * be written in accordance with the sRGB profile
    */
A
Andreas Dilger 已提交
893 894

   /* Write the file header information.  REQUIRED */
G
Guy Schalnat 已提交
895 896
   png_write_info(png_ptr, info_ptr);

897 898 899 900 901 902
   /* If you want, you can write the info in two steps, in case you need to
    * write your private chunk ahead of PLTE:
    *
    *   png_write_info_before_PLTE(write_ptr, write_info_ptr);
    *   write_my_chunk();
    *   png_write_info(png_ptr, info_ptr);
903
    *
904
    * However, given the level of known- and unknown-chunk support in 1.2.0
905
    * and up, this should no longer be necessary.
906 907
    */

A
Andreas Dilger 已提交
908 909 910 911 912 913
   /* Once we write out the header, the compression type on the text
    * chunks gets changed to PNG_TEXT_COMPRESSION_NONE_WR or
    * PNG_TEXT_COMPRESSION_zTXt_WR, so it doesn't get written out again
    * at the end.
    */

914
   /* Set up the transformations you want.  Note that these are
915 916
    * all optional.  Only call them if you want them.
    */
G
Guy Schalnat 已提交
917

918
   /* Invert monochrome pixels */
919
   png_set_invert_mono(png_ptr);
G
Guy Schalnat 已提交
920

A
Andreas Dilger 已提交
921
   /* Shift the pixels up to a legal bit depth and fill in
922 923
    * as appropriate to correctly scale the image.
    */
A
Andreas Dilger 已提交
924
   png_set_shift(png_ptr, &sig_bit);
G
Guy Schalnat 已提交
925

926
   /* Pack pixels into bytes */
G
Guy Schalnat 已提交
927 928
   png_set_packing(png_ptr);

929
   /* Swap location of alpha bytes from ARGB to RGBA */
A
Andreas Dilger 已提交
930 931 932
   png_set_swap_alpha(png_ptr);

   /* Get rid of filler (OR ALPHA) bytes, pack XRGB/RGBX/ARGB/RGBA into
933 934
    * RGB (4 channels -> 3 channels). The second parameter is not used.
    */
A
Andreas Dilger 已提交
935 936
   png_set_filler(png_ptr, 0, PNG_FILLER_BEFORE);

937
   /* Flip BGR pixels to RGB */
G
Guy Schalnat 已提交
938 939
   png_set_bgr(png_ptr);

940
   /* Swap bytes of 16-bit files to most significant byte first */
G
Guy Schalnat 已提交
941 942
   png_set_swap(png_ptr);

943
   /* Swap bits of 1, 2, 4 bit packed pixel formats */
A
Andreas Dilger 已提交
944
   png_set_packswap(png_ptr);
G
Guy Schalnat 已提交
945

946
   /* Turn on interlace handling if you are not using png_write_image() */
G
Guy Schalnat 已提交
947 948
   if (interlacing)
      number_passes = png_set_interlace_handling(png_ptr);
949

G
Guy Schalnat 已提交
950 951 952
   else
      number_passes = 1;

A
Andreas Dilger 已提交
953 954 955 956
   /* The easiest way to write the image (you may have a different memory
    * layout, however, so choose what fits your needs best).  You need to
    * use the first method if you aren't handling interlacing yourself.
    */
957
   png_uint_32 k, height, width;
958
   png_byte image[height][width*bytes_per_pixel];
959
   png_bytep row_pointers[height];
960

961
   if (height > PNG_UINT_32_MAX/png_sizeof(png_bytep))
962 963
     png_error (png_ptr, "Image is too tall to process in memory");

964
   for (k = 0; k < height; k++)
965
     row_pointers[k] = image + k*width*bytes_per_pixel;
G
Guy Schalnat 已提交
966

A
Andreas Dilger 已提交
967
   /* One of the following output methods is REQUIRED */
968 969

#ifdef entire /* Write out the entire image data in one call */
G
Guy Schalnat 已提交
970 971
   png_write_image(png_ptr, row_pointers);

972
   /* The other way to write the image - deal with interlacing */
G
Guy Schalnat 已提交
973

974 975
#else no_entire /* Write out the image data by one or more scanlines */

A
Andreas Dilger 已提交
976 977 978
   /* The number of passes is either 1 for non-interlaced images,
    * or 7 for interlaced images.
    */
G
Guy Schalnat 已提交
979 980 981
   for (pass = 0; pass < number_passes; pass++)
   {
      /* Write a few rows at a time. */
982
      png_write_rows(png_ptr, &row_pointers[first_row], number_of_rows);
G
Guy Schalnat 已提交
983 984 985

      /* If you are only writing one row at a time, this works */
      for (y = 0; y < height; y++)
986
         png_write_rows(png_ptr, &row_pointers[y], 1);
G
Guy Schalnat 已提交
987
   }
988
#endif no_entire /* Use only one output method */
G
Guy Schalnat 已提交
989

A
Andreas Dilger 已提交
990
   /* You can write optional chunks like tEXt, zTXt, and tIME at the end
991
    * as well.  Shouldn't be necessary in 1.2.0 and up as all the public
992 993
    * chunks are supported and you can use png_set_unknown_chunks() to
    * register unknown chunks into the info structure to be written out.
G
Guy Schalnat 已提交
994 995
    */

A
Andreas Dilger 已提交
996
   /* It is REQUIRED to call this to finish writing the rest of the file */
G
Guy Schalnat 已提交
997
   png_write_end(png_ptr, info_ptr);
998
#endif hilevel
G
Guy Schalnat 已提交
999

1000
   /* If you png_malloced a palette, free it here (don't free info_ptr->palette,
1001 1002 1003 1004 1005
    * as recommended in versions 1.0.5m and earlier of this example; if
    * libpng mallocs info_ptr->palette, libpng will free it).  If you
    * allocated it with malloc() instead of png_malloc(), use free() instead
    * of png_free().
    */
1006
   png_free(png_ptr, palette);
1007
   palette = NULL;
1008 1009

   /* Similarly, if you png_malloced any data that you passed in with
1010 1011 1012
    * png_set_something(), such as a hist or trans array, free it here,
    * when you can be sure that libpng is through with it.
    */
1013
   png_free(png_ptr, trans);
1014
   trans = NULL;
1015 1016 1017 1018 1019
   /* Whenever you use png_free() it is a good idea to set the pointer to
    * NULL in case your application inadvertently tries to png_free() it
    * again.  When png_free() sees a NULL it returns without action, thus
    * avoiding the double-free security problem.
    */
G
Guy Schalnat 已提交
1020

1021
   /* Clean up after the write, and free any memory allocated */
1022
   png_destroy_write_struct(&png_ptr, &info_ptr);
G
Guy Schalnat 已提交
1023

1024
   /* Close the file */
G
Guy Schalnat 已提交
1025 1026
   fclose(fp);

1027
   /* That's it */
1028
   return (OK);
G
Guy Schalnat 已提交
1029 1030
}

1031
#endif /* if 0 */