/* pngwutil.c - utilities to write a PNG file

 *

 * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)

 * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)

 *

 * This code is released under the libpng license.

 * For conditions of distribution and use, see the disclaimer

 * and license in png.h

 */

#include "png.h"

#ifdef PNG_WRITE_SUPPORTED

/* Place a 32-bit number into a buffer in PNG byte order.  We work

 * with unsigned numbers for convenience, although one supported

 * ancillary chunk uses signed (two's complement) numbers.

 */

void PNGAPI

png_save_uint_32(png_bytep buf, png_uint_32 i)

{

   buf[0] = (png_byte)((i >> 24) & 0xff);

   buf[1] = (png_byte)((i >> 16) & 0xff);

   buf[2] = (png_byte)((i >> 8) & 0xff);

   buf[3] = (png_byte)(i & 0xff);

}

/* The png_save_int_32 function assumes integers are stored in two's

 * complement format.  If this isn't the case, then this routine needs to

 * be modified to write data in two's complement format.

 */

void PNGAPI

png_save_int_32(png_bytep buf, png_int_32 i)

{

   buf[0] = (png_byte)((i >> 24) & 0xff);

   buf[1] = (png_byte)((i >> 16) & 0xff);

   buf[2] = (png_byte)((i >> 8) & 0xff);

   buf[3] = (png_byte)(i & 0xff);

}

/* Place a 16-bit number into a buffer in PNG byte order.

 * The parameter is declared unsigned int, not png_uint_16,

 * just to avoid potential problems on pre-ANSI C compilers.

 */

void PNGAPI

png_save_uint_16(png_bytep buf, unsigned int i)

{

   buf[0] = (png_byte)((i >> 8) & 0xff);

   buf[1] = (png_byte)(i & 0xff);

}

/* Simple function to write the signature.  If we have already written

 * the magic bytes of the signature, or more likely, the PNG stream is

 * being embedded into another stream and doesn't need its own signature,

 * we should call png_set_sig_bytes() to tell libpng how many of the

 * bytes have already been written.

 */

png_write_sig(png_structp png_ptr)

{

   png_byte png_signature[8] = {137, 80, 78, 71, 13, 10, 26, 10};

   /* Write the rest of the 8 byte signature */

   png_write_data(png_ptr, &png_signature[png_ptr->sig_bytes],

      (png_size_t)(8 - png_ptr->sig_bytes));

   if (png_ptr->sig_bytes < 3)

      png_ptr->mode |= PNG_HAVE_PNG_SIGNATURE;

}

/* Write a PNG chunk all at once.  The type is an array of ASCII characters

 * representing the chunk name.  The array must be at least 4 bytes in

 * length, and does not need to be null terminated.  To be safe, pass the

 * pre-defined chunk names here, and if you need a new one, define it

 * where the others are defined.  The length is the length of the data.

 * All the data must be present.  If that is not possible, use the

 * png_write_chunk_start(), png_write_chunk_data(), and png_write_chunk_end()

 * functions instead.

 */

void PNGAPI

png_write_chunk(png_structp png_ptr, png_bytep chunk_name,

   png_bytep data, png_size_t length)

{

   if (png_ptr == NULL)

      return;

   png_write_chunk_start(png_ptr, chunk_name, (png_uint_32)length);

   png_write_chunk_data(png_ptr, data, (png_size_t)length);

   png_write_chunk_end(png_ptr);

}

/* Write the start of a PNG chunk.  The type is the chunk type.

 * The total_length is the sum of the lengths of all the data you will be

 * passing in png_write_chunk_data().

 */

void PNGAPI

png_write_chunk_start(png_structp png_ptr, png_bytep chunk_name,

   png_uint_32 length)

{

   png_byte buf[8];

   png_debug2(0, "Writing %s chunk, length = %lu", chunk_name,

      (unsigned long)length);

   if (png_ptr == NULL)

      return;

   /* Write the length and the chunk name */

   png_save_uint_32(buf, length);

   png_memcpy(buf + 4, chunk_name, 4);

   png_write_data(png_ptr, buf, (png_size_t)8);

   /* Put the chunk name into png_ptr->chunk_name */

   png_memcpy(png_ptr->chunk_name, chunk_name, 4);

   /* Reset the crc and run it over the chunk name */

   png_reset_crc(png_ptr);

}

/* Write the data of a PNG chunk started with png_write_chunk_start().

 * Note that multiple calls to this function are allowed, and that the

 * sum of the lengths from these calls *must* add up to the total_length

 * given to png_write_chunk_start().

 */

void PNGAPI

png_write_chunk_data(png_structp png_ptr, png_bytep data, png_size_t length)

{

   /* Write the data, and run the CRC over it */

   if (png_ptr == NULL)

      return;

   if (data != NULL && length > 0)

   {

      png_write_data(png_ptr, data, length);

      /* Update the CRC after writing the data,

       * in case that the user I/O routine alters it.

       */

      png_calculate_crc(png_ptr, data, length);

   }

}

/* Finish a chunk started with png_write_chunk_start(). */

void PNGAPI

png_write_chunk_end(png_structp png_ptr)

{

   png_byte buf[4];

   if (png_ptr == NULL) return;

   /* Write the crc in a single operation */

   png_save_uint_32(buf, png_ptr->crc);

   png_write_data(png_ptr, buf, (png_size_t)4);

}

#if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_iCCP_SUPPORTED)

/* This pair of functions encapsulates the operation of (a) compressing a

 * text string, and (b) issuing it later as a series of chunk data writes.

 * The compression_state structure is shared context for these functions

 * set up by the caller in order to make the whole mess thread-safe.

 */

typedef struct

{

   char *input;   /* The uncompressed input data */

   int input_len;   /* Its length */

   int num_output_ptr; /* Number of output pointers used */

   int max_output_ptr; /* Size of output_ptr */

   png_charpp output_ptr; /* Array of pointers to output */

} compression_state;

/* Compress given text into storage in the png_ptr structure */

static int /* PRIVATE */

png_text_compress(png_structp png_ptr,

        png_charp text, png_size_t text_len, int compression,

        compression_state *comp)

{

   int ret;

   comp->num_output_ptr = 0;

   comp->max_output_ptr = 0;

   comp->output_ptr = NULL;

   comp->input = NULL;

   comp->input_len = 0;

   /* We may just want to pass the text right through */

   if (compression == PNG_TEXT_COMPRESSION_NONE)

   {

       comp->input = text;

       comp->input_len = text_len;

       return((int)text_len);

   }

   if (compression >= PNG_TEXT_COMPRESSION_LAST)

   {

      char msg[50];

      png_snprintf(msg, 50, "Unknown compression type %d", compression);

      png_warning(png_ptr, msg);

#else

      png_warning(png_ptr, "Unknown compression type");

#endif

   }

   /* We can't write the chunk until we find out how much data we have,

    * which means we need to run the compressor first and save the

    * output.  This shouldn't be a problem, as the vast majority of

    * comments should be reasonable, but we will set up an array of

    * malloc'd pointers to be sure.

    *

    * If we knew the application was well behaved, we could simplify this

    * greatly by assuming we can always malloc an output buffer large

    * enough to hold the compressed text ((1001 * text_len / 1000) + 12)

    * and malloc this directly.  The only time this would be a bad idea is

    * if we can't malloc more than 64K and we have 64K of random input

    * data, or if the input string is incredibly large (although this

    * wouldn't cause a failure, just a slowdown due to swapping).

    */

   /* Set up the compression buffers */

   png_ptr->zstream.avail_in = (uInt)text_len;

   png_ptr->zstream.next_in = (Bytef *)text;

   png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;

   png_ptr->zstream.next_out = (Bytef *)png_ptr->zbuf;

   /* This is the same compression loop as in png_write_row() */

   do

   {

      /* Compress the data */

      ret = deflate(&png_ptr->zstream, Z_NO_FLUSH);

      if (ret != Z_OK)

      {

         /* Error */

         if (png_ptr->zstream.msg != NULL)

            png_error(png_ptr, png_ptr->zstream.msg);

         else

            png_error(png_ptr, "zlib error");

      }

      /* Check to see if we need more room */

      if (!(png_ptr->zstream.avail_out))

      {

         /* Make sure the output array has room */

         if (comp->num_output_ptr >= comp->max_output_ptr)

         {

            int old_max;

            old_max = comp->max_output_ptr;

            comp->max_output_ptr = comp->num_output_ptr + 4;

            if (comp->output_ptr != NULL)

            {

               png_charpp old_ptr;

               old_ptr = comp->output_ptr;

               comp->output_ptr = (png_charpp)png_malloc(png_ptr,

                  (comp->max_output_ptr * png_sizeof(png_charpp)));

               png_memcpy(comp->output_ptr, old_ptr, old_max

                  * png_sizeof(png_charp));

               png_free(png_ptr, old_ptr);

            }

            else

               comp->output_ptr = (png_charpp)png_malloc(png_ptr,

                  (comp->max_output_ptr * png_sizeof(png_charp)));

         }

         /* Save the data */

         comp->output_ptr[comp->num_output_ptr] =

            (png_charp)png_malloc(png_ptr,

         png_memcpy(comp->output_ptr[comp->num_output_ptr], png_ptr->zbuf,

            png_ptr->zbuf_size);

         comp->num_output_ptr++;

         /* and reset the buffer */

         png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;

         png_ptr->zstream.next_out = png_ptr->zbuf;

      }

   /* Continue until we don't have any more to compress */

   } while (png_ptr->zstream.avail_in);

   /* Finish the compression */

   do

   {

      /* Tell zlib we are finished */

      ret = deflate(&png_ptr->zstream, Z_FINISH);

      if (ret == Z_OK)

      {

         /* Check to see if we need more room */

         if (!(png_ptr->zstream.avail_out))

         {

            /* Check to make sure our output array has room */

            if (comp->num_output_ptr >= comp->max_output_ptr)

            {

               int old_max;

               old_max = comp->max_output_ptr;

               comp->max_output_ptr = comp->num_output_ptr + 4;

               if (comp->output_ptr != NULL)

               {

                  png_charpp old_ptr;

                  old_ptr = comp->output_ptr;

                  /* This could be optimized to realloc() */

                  comp->output_ptr = (png_charpp)png_malloc(png_ptr,

                     png_sizeof(png_charp)));

                  png_memcpy(comp->output_ptr, old_ptr,

                     old_max * png_sizeof(png_charp));

                  png_free(png_ptr, old_ptr);

               }

               else

                  comp->output_ptr = (png_charpp)png_malloc(png_ptr,

                     png_sizeof(png_charp)));

            }

            /* Save the data */

            comp->output_ptr[comp->num_output_ptr] =

               (png_charp)png_malloc(png_ptr,

            png_memcpy(comp->output_ptr[comp->num_output_ptr], png_ptr->zbuf,

               png_ptr->zbuf_size);

            comp->num_output_ptr++;

            /* and reset the buffer pointers */

            png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;

            png_ptr->zstream.next_out = png_ptr->zbuf;

         }

      }

      else if (ret != Z_STREAM_END)

      {

         /* We got an error */

         if (png_ptr->zstream.msg != NULL)

            png_error(png_ptr, png_ptr->zstream.msg);

         else

            png_error(png_ptr, "zlib error");

      }

   } while (ret != Z_STREAM_END);

   /* Text length is number of buffers plus last buffer */

   text_len = png_ptr->zbuf_size * comp->num_output_ptr;

   if (png_ptr->zstream.avail_out < png_ptr->zbuf_size)

      text_len += png_ptr->zbuf_size - (png_size_t)png_ptr->zstream.avail_out;

   return((int)text_len);

}

/* Ship the compressed text out via chunk writes */

static void /* PRIVATE */

png_write_compressed_data_out(png_structp png_ptr, compression_state *comp)

{

   int i;

   /* Handle the no-compression case */

   if (comp->input)

   {

      png_write_chunk_data(png_ptr, (png_bytep)comp->input,

                            (png_size_t)comp->input_len);

      return;

   }

   /* Write saved output buffers, if any */

   for (i = 0; i < comp->num_output_ptr; i++)

   {

      png_write_chunk_data(png_ptr, (png_bytep)comp->output_ptr[i],

         (png_size_t)png_ptr->zbuf_size);

      png_free(png_ptr, comp->output_ptr[i]);

   }

   if (comp->max_output_ptr != 0)

      png_free(png_ptr, comp->output_ptr);

   /* Write anything left in zbuf */

   if (png_ptr->zstream.avail_out < (png_uint_32)png_ptr->zbuf_size)

      png_write_chunk_data(png_ptr, png_ptr->zbuf,

         (png_size_t)(png_ptr->zbuf_size - png_ptr->zstream.avail_out));

   /* Reset zlib for another zTXt/iTXt or image data */

   deflateReset(&png_ptr->zstream);

   png_ptr->zstream.data_type = Z_BINARY;

}

#endif

/* Write the IHDR chunk, and update the png_struct with the necessary

 * information.  Note that the rest of this code depends upon this

 * information being correct.

 */

void /* PRIVATE */

png_write_IHDR(png_structp png_ptr, png_uint_32 width, png_uint_32 height,

   int bit_depth, int color_type, int compression_type, int filter_type,

   int interlace_type)

{

   PNG_IHDR;

   int ret;

   png_byte buf[13]; /* Buffer to store the IHDR info */

   png_debug(1, "in png_write_IHDR");

   /* Check that we have valid input data from the application info */

   switch (color_type)

   {

      case PNG_COLOR_TYPE_GRAY:

         switch (bit_depth)

         {

            case 1:

            case 2:

            case 4:

            case 8:

            case 16: png_ptr->channels = 1; break;

            default: png_error(png_ptr, "Invalid bit depth for grayscale image");

         }

         break;

      case PNG_COLOR_TYPE_RGB:

         if (bit_depth != 8 && bit_depth != 16)

            png_error(png_ptr, "Invalid bit depth for RGB image");

         png_ptr->channels = 3;

         break;

      case PNG_COLOR_TYPE_PALETTE:

         switch (bit_depth)

         {

            case 1:

            case 2:

            case 4:

            case 8: png_ptr->channels = 1; break;

            default: png_error(png_ptr, "Invalid bit depth for paletted image");

         }

         break;

      case PNG_COLOR_TYPE_GRAY_ALPHA:

         if (bit_depth != 8 && bit_depth != 16)

            png_error(png_ptr, "Invalid bit depth for grayscale+alpha image");

         png_ptr->channels = 2;

         break;

      case PNG_COLOR_TYPE_RGB_ALPHA:

         if (bit_depth != 8 && bit_depth != 16)

            png_error(png_ptr, "Invalid bit depth for RGBA image");

         png_ptr->channels = 4;

         break;

      default:

         png_error(png_ptr, "Invalid image color type specified");

   }

   if (compression_type != PNG_COMPRESSION_TYPE_BASE)

   {

      png_warning(png_ptr, "Invalid compression type specified");

      compression_type = PNG_COMPRESSION_TYPE_BASE;

   }

   /* Write filter_method 64 (intrapixel differencing) only if

    * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and

    * 2. Libpng did not write a PNG signature (this filter_method is only

    *    used in PNG datastreams that are embedded in MNG datastreams) and

    * 3. The application called png_permit_mng_features with a mask that

    *    included PNG_FLAG_MNG_FILTER_64 and

    * 4. The filter_method is 64 and

    * 5. The color_type is RGB or RGBA

    */

   if (

      !((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&

      ((png_ptr->mode&PNG_HAVE_PNG_SIGNATURE) == 0) &&

      (color_type == PNG_COLOR_TYPE_RGB ||

       color_type == PNG_COLOR_TYPE_RGB_ALPHA) &&

      (filter_type == PNG_INTRAPIXEL_DIFFERENCING)) &&

#endif

      filter_type != PNG_FILTER_TYPE_BASE)

   {

      png_warning(png_ptr, "Invalid filter type specified");

      filter_type = PNG_FILTER_TYPE_BASE;

   }

#ifdef PNG_WRITE_INTERLACING_SUPPORTED

   if (interlace_type != PNG_INTERLACE_NONE &&

      interlace_type != PNG_INTERLACE_ADAM7)

   {

      png_warning(png_ptr, "Invalid interlace type specified");

      interlace_type = PNG_INTERLACE_ADAM7;

   }

#else

   interlace_type=PNG_INTERLACE_NONE;

#endif

   /* Save the relevent information */

   png_ptr->bit_depth = (png_byte)bit_depth;

   png_ptr->color_type = (png_byte)color_type;

   png_ptr->interlaced = (png_byte)interlace_type;

   png_ptr->filter_type = (png_byte)filter_type;

#endif

   png_ptr->compression_type = (png_byte)compression_type;

   png_ptr->width = width;

   png_ptr->height = height;

   png_ptr->pixel_depth = (png_byte)(bit_depth * png_ptr->channels);

   png_ptr->rowbytes = PNG_ROWBYTES(png_ptr->pixel_depth, width);