example.c 39.4 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.15 [November 20, 2014]
6
 * Maintained 1998-2014 Glenn Randers-Pehrson
7 8 9 10 11
 * Maintained 1996, 1997 Andreas Dilger)
 * Written 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)
 * 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 42 43 44 45 46 47 48 49 50 51 52 53 54
/* 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>
#include <zlib.h>

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));
      image.version = PNG_IMAGE_VERSION;

      /* The first argument is the file to read: */
55
      if (png_image_begin_read_from_file(&image, argv[1]) != 0)
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
      {
         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 fourth 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.
          *
          * The final argument is a pointer to a buffer for the colormap;
          * colormaps have exactly the same format as a row of image pixels (so
          * you choose what format to make the colormap by setting
          * image.format).  A colormap is only returned if
          * PNG_FORMAT_FLAG_COLORMAP is also set in image.format, so in this
          * case NULL is passed as the final argument.  If you do want to force
          * all images into an index/color-mapped format then you can use:
          *
          *    PNG_IMAGE_COLORMAP_SIZE(image)
          *
          * to find the maximum size of the colormap in bytes.
          */
         if (buffer != NULL &&
            png_image_finish_read(&image, NULL/*background*/, buffer,
100
               0/*row_stride*/, NULL/*colormap*/) != 0)
101 102 103 104 105 106 107
         {
            /* 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*/,
108
               buffer, 0/*row_stride*/, NULL/*colormap*/) != 0)
109 110 111 112 113 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 182 183 184 185 186 187 188 189 190
            {
               /* The image has been written successfully. */
               exit(0);
            }
         }

         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);
      }

      /* 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
191
 * the current libpng implementation of conversion to 16-bit is also
192 193 194
 * 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.
 */
195

196
/* If your program needs more information from the PNG data it reads, or if you
197
 * need to do more complex transformations, or minimize transformations, on the
198 199 200 201 202 203 204 205 206 207 208 209 210 211 212
 * 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 已提交
213

214 215 216 217 218 219 220
 /* 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
221
#  define png_jmpbuf(png_ptr) ((png_ptr)->png_jmpbuf)
222 223
#endif

224 225 226 227 228
/* 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.
229 230 231 232 233 234 235 236 237 238 239 240
 *
 * 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
241
 * to pass the bytes to png_sig_cmp() or even skip that if you know
242 243
 * you have a PNG file, and call png_set_sig_bytes().
 */
A
Andreas Dilger 已提交
244 245
#define PNG_BYTES_TO_CHECK 4
int check_if_png(char *file_name, FILE **fp)
G
Guy Schalnat 已提交
246
{
A
Andreas Dilger 已提交
247
   char buf[PNG_BYTES_TO_CHECK];
G
Guy Schalnat 已提交
248

A
Andreas Dilger 已提交
249
   /* Open the prospective PNG file. */
250
   if ((*fp = fopen(file_name, "rb")) == NULL)
G
Guy Schalnat 已提交
251 252
      return 0;

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

257 258 259
   /* Compare the first PNG_BYTES_TO_CHECK bytes of the signature.
      Return nonzero (true) if they match */

260
   return(!png_sig_cmp(buf, (png_size_t)0, PNG_BYTES_TO_CHECK));
G
Guy Schalnat 已提交
261 262
}

A
Andreas Dilger 已提交
263
/* Read a PNG file.  You may want to return an error code if the read
264 265 266 267 268
 * 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).
 */
269
#ifdef open_file /* prototype 1 */
A
Andreas Dilger 已提交
270
void read_png(char *file_name)  /* We need to open the file */
G
Guy Schalnat 已提交
271
{
G
Guy Schalnat 已提交
272
   png_structp png_ptr;
G
Guy Schalnat 已提交
273
   png_infop info_ptr;
A
Andreas Dilger 已提交
274 275 276
   unsigned int sig_read = 0;
   png_uint_32 width, height;
   int bit_depth, color_type, interlace_type;
A
Andreas Dilger 已提交
277
   FILE *fp;
G
Guy Schalnat 已提交
278

A
Andreas Dilger 已提交
279
   if ((fp = fopen(file_name, "rb")) == NULL)
280
      return (ERROR);
281

282
#else no_open_file /* prototype 2 */
283
void read_png(FILE *fp, unsigned int sig_read)  /* File is already open */
A
Andreas Dilger 已提交
284 285 286
{
   png_structp png_ptr;
   png_infop info_ptr;
A
Andreas Dilger 已提交
287 288
   png_uint_32 width, height;
   int bit_depth, color_type, interlace_type;
289
#endif no_open_file /* Only use one prototype! */
G
Guy Schalnat 已提交
290

G
Guy Schalnat 已提交
291
   /* Create and initialize the png_struct with the desired error handler
A
Andreas Dilger 已提交
292 293 294 295 296
    * 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 已提交
297
   png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING,
298
      png_voidp user_error_ptr, user_error_fn, user_warning_fn);
G
Guy Schalnat 已提交
299

A
Andreas Dilger 已提交
300
   if (png_ptr == NULL)
G
Guy Schalnat 已提交
301 302
   {
      fclose(fp);
303
      return (ERROR);
G
Guy Schalnat 已提交
304 305
   }

A
Andreas Dilger 已提交
306
   /* Allocate/initialize the memory for image information.  REQUIRED. */
307
   info_ptr = png_create_info_struct(png_ptr);
A
Andreas Dilger 已提交
308
   if (info_ptr == NULL)
G
Guy Schalnat 已提交
309 310
   {
      fclose(fp);
311
      png_destroy_read_struct(&png_ptr, NULL, NULL);
312
      return (ERROR);
G
Guy Schalnat 已提交
313 314
   }

A
Andreas Dilger 已提交
315 316 317 318
   /* 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.
    */
319 320

   if (setjmp(png_jmpbuf(png_ptr)))
G
Guy Schalnat 已提交
321
   {
G
Guy Schalnat 已提交
322
      /* Free all of the memory associated with the png_ptr and info_ptr */
323
      png_destroy_read_struct(&png_ptr, &info_ptr, NULL);
G
Guy Schalnat 已提交
324 325
      fclose(fp);
      /* If we get here, we had a problem reading the file */
326
      return (ERROR);
G
Guy Schalnat 已提交
327 328
   }

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

334
#else no_streams /* PNG file I/O method 2 */
A
Andreas Dilger 已提交
335
   /* If you are using replacement read functions, instead of calling
336 337
    * png_init_io() here you would call:
    */
G
Guy Schalnat 已提交
338 339
   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 */
340
#endif no_streams /* Use only one I/O method! */
G
Guy Schalnat 已提交
341

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

345 346 347 348 349
#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
350
    * quantizing, filling, setting background, and doing gamma
351 352 353
    * adjustment), then you can read the entire image (including
    * pixels) into the info structure with this call:
    */
354
   png_read_png(png_ptr, info_ptr, png_transforms, NULL);
355

356 357 358
#else
   /* OK, you're doing it the hard way, with the lower-level functions */

A
Andreas Dilger 已提交
359 360 361
   /* 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 已提交
362
   png_read_info(png_ptr, info_ptr);
G
Guy Schalnat 已提交
363

A
Andreas Dilger 已提交
364
   png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type,
365
       &interlace_type, NULL, NULL);
A
Andreas Dilger 已提交
366

367 368 369 370 371
   /* 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 已提交
372

373 374 375 376
   /* 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.
    */
377
#ifdef PNG_READ_SCALE_16_TO_8_SUPPORTED
378
    png_set_scale_16(png_ptr);
379
#else
380
   png_set_strip_16(png_ptr);
381
#endif
A
Andreas Dilger 已提交
382

383
   /* Strip alpha bytes from the input data without combining with the
384 385
    * background (not recommended).
    */
A
Andreas Dilger 已提交
386 387
   png_set_strip_alpha(png_ptr);

388
   /* Extract multiple pixels with bit depths of 1, 2, and 4 from a single
A
Andreas Dilger 已提交
389 390 391 392
    * byte into separate bytes (useful for paletted and grayscale images).
    */
   png_set_packing(png_ptr);

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

397
   /* Expand paletted colors into true RGB triplets */
A
Andreas Dilger 已提交
398
   if (color_type == PNG_COLOR_TYPE_PALETTE)
399
      png_set_palette_to_rgb(png_ptr);
G
Guy Schalnat 已提交
400

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

405 406 407
   /* Expand paletted or RGB images with transparency to full alpha channels
    * so the data will be available as RGBA quartets.
    */
408
   if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS) != 0)
409
      png_set_tRNS_to_alpha(png_ptr);
G
Guy Schalnat 已提交
410

A
Andreas Dilger 已提交
411 412 413 414 415 416
   /* 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 已提交
417

418
   png_color_16 my_background, *image_background;
G
Guy Schalnat 已提交
419

420
   if (png_get_bKGD(png_ptr, info_ptr, &image_background) != 0)
421
      png_set_background(png_ptr, image_background,
G
Guy Schalnat 已提交
422
                         PNG_BACKGROUND_GAMMA_FILE, 1, 1.0);
G
Guy Schalnat 已提交
423 424
   else
      png_set_background(png_ptr, &my_background,
G
Guy Schalnat 已提交
425
                         PNG_BACKGROUND_GAMMA_SCREEN, 0, 1.0);
G
Guy Schalnat 已提交
426

427 428 429
   /* Some suggestions as to how to get a screen gamma value
    *
    * Note that screen gamma is the display_exponent, which includes
430 431
    * the CRT_exponent and any correction for viewing conditions
    */
A
Andreas Dilger 已提交
432 433 434 435 436
   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 */
437
   else if ((gamma_str = getenv("SCREEN_GAMMA")) != NULL)
A
Andreas Dilger 已提交
438 439 440 441
   {
      screen_gamma = atof(gamma_str);
   }
   /* If we don't have another value */
G
Guy Schalnat 已提交
442
   else
A
Andreas Dilger 已提交
443
   {
444 445 446
      screen_gamma = PNG_DEFAULT_sRGB;  /* A good guess for a PC monitor
                                           in a dimly lit room */
      screen_gamma = PNG_GAMMA_MAC_18 or 1.0; /* Good guesses for Mac systems */
A
Andreas Dilger 已提交
447
   }
G
Guy Schalnat 已提交
448

449
   /* Tell libpng to handle the gamma conversion for you.  The final call
A
Andreas Dilger 已提交
450 451 452 453
    * 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.
    */
454 455 456

   int intent;

457
   if (png_get_sRGB(png_ptr, info_ptr, &intent) != 0)
458
      png_set_gamma(png_ptr, screen_gamma, PNG_DEFAULT_sRGB);
459 460 461
   else
   {
      double image_gamma;
462
      if (png_get_gAMA(png_ptr, info_ptr, &image_gamma) != 0)
463 464
         png_set_gamma(png_ptr, screen_gamma, image_gamma);
      else
465 466
         png_set_gamma(png_ptr, screen_gamma, 0.45455);
   }
G
Guy Schalnat 已提交
467

468 469
#ifdef PNG_READ_QUANTIZE_SUPPORTED
   /* Quantize RGB files down to 8 bit palette or reduce palettes
470 471
    * to the number of colors available on your screen.
    */
472
   if ((color_type & PNG_COLOR_MASK_COLOR) != 0)
G
Guy Schalnat 已提交
473
   {
474
      int num_palette;
A
Andreas Dilger 已提交
475 476 477
      png_colorp palette;

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

483
         png_set_quantize(png_ptr, std_color_cube, MAX_SCREEN_COLORS,
484
            MAX_SCREEN_COLORS, NULL, 0);
A
Andreas Dilger 已提交
485 486
      }
      /* This reduces the image to the palette supplied in the file */
487
      else if (png_get_PLTE(png_ptr, info_ptr, &palette, &num_palette) != 0)
A
Andreas Dilger 已提交
488
      {
489
         png_uint_16p histogram = NULL;
A
Andreas Dilger 已提交
490 491 492

         png_get_hIST(png_ptr, info_ptr, &histogram);

493
         png_set_quantize(png_ptr, palette, num_palette,
A
Andreas Dilger 已提交
494
                        max_screen_colors, histogram, 0);
G
Guy Schalnat 已提交
495
      }
G
Guy Schalnat 已提交
496
   }
497
#endif /* READ_QUANTIZE */
G
Guy Schalnat 已提交
498

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

A
Andreas Dilger 已提交
502 503 504 505
   /* 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:
    */
506
   if (png_get_valid(png_ptr, info_ptr, PNG_INFO_sBIT) != 0)
A
Andreas Dilger 已提交
507
   {
508
      png_color_8p sig_bit_p;
A
Andreas Dilger 已提交
509

510 511
      png_get_sBIT(png_ptr, info_ptr, &sig_bit_p);
      png_set_shift(png_ptr, sig_bit_p);
A
Andreas Dilger 已提交
512
   }
G
Guy Schalnat 已提交
513

514
   /* Flip the RGB pixels to BGR (or RGBA to BGRA) */
515
   if ((color_type & PNG_COLOR_MASK_COLOR) != 0)
516
      png_set_bgr(png_ptr);
G
Guy Schalnat 已提交
517

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

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

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

527
#ifdef PNG_READ_INTERLACING_SUPPORTED
A
Andreas Dilger 已提交
528 529
   /* Turn on interlace handling.  REQUIRED if you are not using
    * png_read_image().  To see how to handle interlacing passes,
530
    * see the png_read_row() method below:
A
Andreas Dilger 已提交
531
    */
G
Guy Schalnat 已提交
532
   number_passes = png_set_interlace_handling(png_ptr);
533 534
#else
   number_passes = 1;
535
#endif /* READ_INTERLACING */
536

G
Guy Schalnat 已提交
537

538
   /* Optional call to gamma correct and add the background to the palette
A
Andreas Dilger 已提交
539 540 541
    * 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 已提交
542 543
   png_read_update_info(png_ptr, info_ptr);

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

546
   /* The easiest way to read the image: */
G
Guy Schalnat 已提交
547
   png_bytep row_pointers[height];
G
Guy Schalnat 已提交
548

549 550 551 552
   /* Clear the pointer array */
   for (row = 0; row < height; row++)
      row_pointers[row] = NULL;

G
Guy Schalnat 已提交
553
   for (row = 0; row < height; row++)
554 555
      row_pointers[row] = png_malloc(png_ptr, png_get_rowbytes(png_ptr,
         info_ptr));
G
Guy Schalnat 已提交
556

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

561
#else no_entire /* Read the image one or more scanlines at a time */
562
   /* The other way to read images - deal with interlacing: */
G
Guy Schalnat 已提交
563 564 565

   for (pass = 0; pass < number_passes; pass++)
   {
566
#ifdef single /* Read the image a single row at a time */
G
Guy Schalnat 已提交
567 568
      for (y = 0; y < height; y++)
      {
569
         png_read_rows(png_ptr, &row_pointers[y], NULL, 1);
G
Guy Schalnat 已提交
570 571
      }

572
#else no_single /* Read the image several rows at a time */
A
Andreas Dilger 已提交
573 574
      for (y = 0; y < height; y += number_of_rows)
      {
575
#ifdef sparkle /* Read the image using the "sparkle" effect. */
576 577
         png_read_rows(png_ptr, &row_pointers[y], NULL,
            number_of_rows);
578
#else no_sparkle /* Read the image using the "rectangle" effect */
579 580
         png_read_rows(png_ptr, NULL, &row_pointers[y],
            number_of_rows);
581
#endif no_sparkle /* Use only one of these two methods */
A
Andreas Dilger 已提交
582
      }
583

584 585
      /* 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 已提交
586
   }
587
#endif no_entire /* Use only one of these two methods */
G
Guy Schalnat 已提交
588

589
   /* Read rest of file, and get additional chunks in info_ptr - REQUIRED */
G
Guy Schalnat 已提交
590
   png_read_end(png_ptr, info_ptr);
591 592 593
#endif hilevel

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

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

598
   /* Close the file */
G
Guy Schalnat 已提交
599 600
   fclose(fp);

601
   /* That's it */
602
   return (OK);
G
Guy Schalnat 已提交
603 604
}

605
/* Progressively read a file */
G
Guy Schalnat 已提交
606 607

int
G
Guy Schalnat 已提交
608
initialize_png_reader(png_structp *png_ptr, png_infop *info_ptr)
G
Guy Schalnat 已提交
609
{
G
Guy Schalnat 已提交
610
   /* Create and initialize the png_struct with the desired error handler
A
Andreas Dilger 已提交
611 612 613 614
    * 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 已提交
615 616
    */
   *png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING,
617
       png_voidp user_error_ptr, user_error_fn, user_warning_fn);
G
Guy Schalnat 已提交
618

A
Andreas Dilger 已提交
619
   if (*png_ptr == NULL)
G
Guy Schalnat 已提交
620
   {
G
Guy Schalnat 已提交
621
      *info_ptr = NULL;
622
      return (ERROR);
G
Guy Schalnat 已提交
623 624
   }

G
Guy Schalnat 已提交
625 626
   *info_ptr = png_create_info_struct(png_ptr);

A
Andreas Dilger 已提交
627
   if (*info_ptr == NULL)
G
Guy Schalnat 已提交
628
   {
629
      png_destroy_read_struct(png_ptr, info_ptr, NULL);
630
      return (ERROR);
G
Guy Schalnat 已提交
631 632
   }

633
   if (setjmp(png_jmpbuf((*png_ptr))))
G
Guy Schalnat 已提交
634
   {
635
      png_destroy_read_struct(png_ptr, info_ptr, NULL);
636
      return (ERROR);
G
Guy Schalnat 已提交
637
   }
G
Guy Schalnat 已提交
638

639
   /* This one's new.  You will need to provide all three
A
Andreas Dilger 已提交
640
    * function callbacks, even if you aren't using them all.
641 642 643
    * 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 已提交
644 645 646 647 648 649 650
    * 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 已提交
651
   png_set_progressive_read_fn(*png_ptr, (void *)stream_data,
G
Guy Schalnat 已提交
652 653
      info_callback, row_callback, end_callback);

654
   return (OK);
G
Guy Schalnat 已提交
655 656 657
}

int
G
Guy Schalnat 已提交
658 659
process_data(png_structp *png_ptr, png_infop *info_ptr,
   png_bytep buffer, png_uint_32 length)
G
Guy Schalnat 已提交
660
{
661
   if (setjmp(png_jmpbuf((*png_ptr))))
G
Guy Schalnat 已提交
662
   {
G
Guy Schalnat 已提交
663
      /* Free the png_ptr and info_ptr memory on error */
664
      png_destroy_read_struct(png_ptr, info_ptr, NULL);
665
      return (ERROR);
G
Guy Schalnat 已提交
666 667
   }

A
Andreas Dilger 已提交
668 669
   /* This one's new also.  Simply give it chunks of data as
    * they arrive from the data stream (in order, of course).
670
    * On segmented machines, don't give it any more than 64K.
A
Andreas Dilger 已提交
671 672 673 674 675 676 677
    * 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 已提交
678
   png_process_data(*png_ptr, *info_ptr, buffer, length);
679
   return (OK);
G
Guy Schalnat 已提交
680 681 682 683
}

info_callback(png_structp png_ptr, png_infop info)
{
684 685 686 687 688 689 690
   /* 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 已提交
691 692 693
}

row_callback(png_structp png_ptr, png_bytep new_row,
G
Guy Schalnat 已提交
694
   png_uint_32 row_num, int pass)
G
Guy Schalnat 已提交
695
{
696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712
   /*
    * 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:
    */

713 714 715 716 717
   /* Get pointer to corresponding row in our
    * PNG read buffer.
    */
   png_bytep old_row = ((png_bytep *)our_data)[row_num];

718
#ifdef PNG_READ_INTERLACING_SUPPORTED
719 720 721 722 723 724
   /* 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);

725 726 727 728 729 730 731 732 733
   /*
    * 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
734
    * (it just does the memcpy for you) if it will make the code
735 736
    * easier.  Thus, you can just do this for all cases:
    */
G
Guy Schalnat 已提交
737 738 739

   png_progressive_combine_row(png_ptr, old_row, new_row);

740 741 742 743 744 745 746
   /* 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.
    */
747
#endif /* READ_INTERLACING */
G
Guy Schalnat 已提交
748 749 750 751
}

end_callback(png_structp png_ptr, png_infop info)
{
752 753 754 755 756 757 758 759 760
   /* 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 已提交
761 762
}

763
/* Write a png file */
764
void write_png(char *file_name /* , ... other image information ... */)
G
Guy Schalnat 已提交
765 766
{
   FILE *fp;
G
Guy Schalnat 已提交
767 768
   png_structp png_ptr;
   png_infop info_ptr;
769
   png_colorp palette;
G
Guy Schalnat 已提交
770

771
   /* Open the file */
G
Guy Schalnat 已提交
772
   fp = fopen(file_name, "wb");
A
Andreas Dilger 已提交
773
   if (fp == NULL)
774
      return (ERROR);
G
Guy Schalnat 已提交
775

G
Guy Schalnat 已提交
776
   /* Create and initialize the png_struct with the desired error handler
A
Andreas Dilger 已提交
777 778 779 780
    * 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 已提交
781 782
    */
   png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING,
783
      png_voidp user_error_ptr, user_error_fn, user_warning_fn);
G
Guy Schalnat 已提交
784

A
Andreas Dilger 已提交
785
   if (png_ptr == NULL)
G
Guy Schalnat 已提交
786 787
   {
      fclose(fp);
788
      return (ERROR);
G
Guy Schalnat 已提交
789 790
   }

A
Andreas Dilger 已提交
791
   /* Allocate/initialize the image information data.  REQUIRED */
G
Guy Schalnat 已提交
792
   info_ptr = png_create_info_struct(png_ptr);
A
Andreas Dilger 已提交
793
   if (info_ptr == NULL)
G
Guy Schalnat 已提交
794 795
   {
      fclose(fp);
796
      png_destroy_write_struct(&png_ptr,  NULL);
797
      return (ERROR);
G
Guy Schalnat 已提交
798 799
   }

A
Andreas Dilger 已提交
800
   /* Set error handling.  REQUIRED if you aren't supplying your own
801
    * error handling functions in the png_create_write_struct() call.
A
Andreas Dilger 已提交
802
    */
803
   if (setjmp(png_jmpbuf(png_ptr)))
G
Guy Schalnat 已提交
804
   {
805
      /* If we get here, we had a problem writing the file */
G
Guy Schalnat 已提交
806
      fclose(fp);
807
      png_destroy_write_struct(&png_ptr, &info_ptr);
808
      return (ERROR);
G
Guy Schalnat 已提交
809 810
   }

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

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

817
#else no_streams /* I/O initialization method 2 */
818
   /* If you are using replacement write functions, instead of calling
819 820
    * png_init_io() here you would call
    */
A
Andreas Dilger 已提交
821 822 823
   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 */
824
#endif no_streams /* Only use one initialization method */
A
Andreas Dilger 已提交
825

826 827
#ifdef hilevel
   /* This is the easy way.  Use it if you already have all the
828
    * image info living in the structure.  You could "|" many
829 830
    * PNG_TRANSFORM flags into the png_transforms integer here.
    */
831
   png_write_png(png_ptr, info_ptr, png_transforms, NULL);
832

833 834 835
#else
   /* This is the hard way */

A
Andreas Dilger 已提交
836 837 838 839 840 841 842 843 844 845
   /* 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 已提交
846

847
   /* Set the palette if there is one.  REQUIRED for indexed-color images */
848
   palette = (png_colorp)png_malloc(png_ptr, PNG_MAX_PALETTE_LENGTH
849
             * (sizeof (png_color)));
850
   /* ... Set palette colors ... */
851
   png_set_PLTE(png_ptr, info_ptr, palette, PNG_MAX_PALETTE_LENGTH);
852
   /* You must not free palette here, because png_set_PLTE only makes a link to
853 854 855
    * the palette that you malloced.  Wait until you are about to destroy
    * the png structure.
    */
G
Guy Schalnat 已提交
856

857
   /* Optional significant bit (sBIT) chunk */
858
   png_color_8 sig_bit;
859

860
   /* If we are dealing with a grayscale image then */
A
Andreas Dilger 已提交
861
   sig_bit.gray = true_bit_depth;
862

863
   /* Otherwise, if we are dealing with a color image then */
A
Andreas Dilger 已提交
864 865 866
   sig_bit.red = true_red_bit_depth;
   sig_bit.green = true_green_bit_depth;
   sig_bit.blue = true_blue_bit_depth;
867

868
   /* If the image has an alpha channel then */
A
Andreas Dilger 已提交
869
   sig_bit.alpha = true_alpha_bit_depth;
870

871
   png_set_sBIT(png_ptr, info_ptr, &sig_bit);
G
Guy Schalnat 已提交
872

873

A
Andreas Dilger 已提交
874
   /* Optional gamma chunk is strongly suggested if you have any guess
875 876
    * as to the correct gamma of the image.
    */
A
Andreas Dilger 已提交
877 878 879
   png_set_gAMA(png_ptr, info_ptr, gamma);

   /* Optionally write comments into the image */
880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911
   {
      png_text text_ptr[3];

      char key0[]="Title";
      char text0[]="Mona Lisa";
      text_ptr[0].key = key0;
      text_ptr[0].text = text0;
      text_ptr[0].compression = PNG_TEXT_COMPRESSION_NONE;
      text_ptr[0].itxt_length = 0;
      text_ptr[0].lang = NULL;
      text_ptr[0].lang_key = NULL;

      char key1[]="Author";
      char text1[]="Leonardo DaVinci";
      text_ptr[1].key = key1;
      text_ptr[1].text = text1;
      text_ptr[1].compression = PNG_TEXT_COMPRESSION_NONE;
      text_ptr[1].itxt_length = 0;
      text_ptr[1].lang = NULL;
      text_ptr[1].lang_key = NULL;

      char key2[]="Description";
      char text2[]="<long text>";
      text_ptr[2].key = key2;
      text_ptr[2].text = text2;
      text_ptr[2].compression = PNG_TEXT_COMPRESSION_zTXt;
      text_ptr[2].itxt_length = 0;
      text_ptr[2].lang = NULL;
      text_ptr[2].lang_key = NULL;

      png_set_text(write_ptr, write_info_ptr, text_ptr, 3);
   }
A
Andreas Dilger 已提交
912

913
   /* Other optional chunks like cHRM, bKGD, tRNS, tIME, oFFs, pHYs */
914

915
   /* Note that if sRGB is present the gAMA and cHRM chunks must be ignored
916
    * on read and, if your application chooses to write them, they must
917 918
    * be written in accordance with the sRGB profile
    */
A
Andreas Dilger 已提交
919 920

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

923 924 925 926 927 928
   /* 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);
929
    *
930
    * However, given the level of known- and unknown-chunk support in 1.2.0
931
    * and up, this should no longer be necessary.
932 933
    */

A
Andreas Dilger 已提交
934
   /* Once we write out the header, the compression type on the text
935
    * chunk gets changed to PNG_TEXT_COMPRESSION_NONE_WR or
A
Andreas Dilger 已提交
936 937 938 939
    * PNG_TEXT_COMPRESSION_zTXt_WR, so it doesn't get written out again
    * at the end.
    */

940
   /* Set up the transformations you want.  Note that these are
941 942
    * all optional.  Only call them if you want them.
    */
G
Guy Schalnat 已提交
943

944
   /* Invert monochrome pixels */
945
   png_set_invert_mono(png_ptr);
G
Guy Schalnat 已提交
946

A
Andreas Dilger 已提交
947
   /* Shift the pixels up to a legal bit depth and fill in
948 949
    * as appropriate to correctly scale the image.
    */
A
Andreas Dilger 已提交
950
   png_set_shift(png_ptr, &sig_bit);
G
Guy Schalnat 已提交
951

952
   /* Pack pixels into bytes */
G
Guy Schalnat 已提交
953 954
   png_set_packing(png_ptr);

955
   /* Swap location of alpha bytes from ARGB to RGBA */
A
Andreas Dilger 已提交
956 957 958
   png_set_swap_alpha(png_ptr);

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

963
   /* Flip BGR pixels to RGB */
G
Guy Schalnat 已提交
964 965
   png_set_bgr(png_ptr);

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

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

972
   /* Turn on interlace handling if you are not using png_write_image() */
973
   if (interlacing != 0)
G
Guy Schalnat 已提交
974
      number_passes = png_set_interlace_handling(png_ptr);
975

G
Guy Schalnat 已提交
976 977 978
   else
      number_passes = 1;

A
Andreas Dilger 已提交
979 980 981 982
   /* 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.
    */
983
   png_uint_32 k, height, width;
984 985 986 987

   /* In this example, "image" is a one-dimensional array of bytes */
   png_byte image[height*width*bytes_per_pixel];

988
   png_bytep row_pointers[height];
989

990
   if (height > PNG_UINT_32_MAX/(sizeof (png_bytep)))
991 992
     png_error (png_ptr, "Image is too tall to process in memory");

993
   /* Set up pointers into your "image" byte array */
994
   for (k = 0; k < height; k++)
995
     row_pointers[k] = image + k*width*bytes_per_pixel;
G
Guy Schalnat 已提交
996

A
Andreas Dilger 已提交
997
   /* One of the following output methods is REQUIRED */
998 999

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

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

1004 1005
#else no_entire /* Write out the image data by one or more scanlines */

A
Andreas Dilger 已提交
1006 1007 1008
   /* The number of passes is either 1 for non-interlaced images,
    * or 7 for interlaced images.
    */
G
Guy Schalnat 已提交
1009 1010 1011
   for (pass = 0; pass < number_passes; pass++)
   {
      /* Write a few rows at a time. */
1012
      png_write_rows(png_ptr, &row_pointers[first_row], number_of_rows);
G
Guy Schalnat 已提交
1013 1014 1015

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

A
Andreas Dilger 已提交
1020
   /* You can write optional chunks like tEXt, zTXt, and tIME at the end
1021
    * as well.  Shouldn't be necessary in 1.2.0 and up as all the public
1022 1023
    * 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 已提交
1024 1025
    */

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

1030
   /* If you png_malloced a palette, free it here (don't free info_ptr->palette,
1031 1032 1033 1034 1035
    * 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().
    */
1036
   png_free(png_ptr, palette);
1037
   palette = NULL;
1038 1039

   /* Similarly, if you png_malloced any data that you passed in with
1040 1041 1042
    * png_set_something(), such as a hist or trans array, free it here,
    * when you can be sure that libpng is through with it.
    */
1043
   png_free(png_ptr, trans);
1044
   trans = NULL;
1045 1046 1047 1048 1049
   /* 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 已提交
1050

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

1054
   /* Close the file */
G
Guy Schalnat 已提交
1055 1056
   fclose(fp);

1057
   /* That's it */
1058
   return (OK);
G
Guy Schalnat 已提交
1059 1060
}

1061
#endif /* if 0 */