blob: 09573e83f7238ab8ebe7c3e12716fce68c1b2ea5 [file] [log] [blame]
Glenn Randers-Pehrson5c6aeb21998-12-29 11:47:59 -06001
Andreas Dilger47a0c421997-05-16 02:46:07 -05002/* pngwrite.c - general routines to write a PNG file
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06003 *
Glenn Randers-Pehrsond60b8fa2006-04-20 21:31:14 -05004 * Last changed in libpng 1.4.0 April 20, 2006
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06005 * For conditions of distribution and use, see copyright notice in png.h
Glenn Randers-Pehrson9c3ab682006-02-20 22:09:05 -06006 * Copyright (c) 1998-2006 Glenn Randers-Pehrson
Glenn Randers-Pehrsond4366722000-06-04 14:29:29 -05007 * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)
8 * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06009 */
Guy Schalnat0d580581995-07-20 02:43:20 -050010
11/* get internal access to png.h */
Guy Schalnat0d580581995-07-20 02:43:20 -050012#include "png.h"
Glenn Randers-Pehrson19095602001-03-14 07:08:39 -060013#ifdef PNG_WRITE_SUPPORTED
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -050014#include "pngpriv.h"
Guy Schalnat0d580581995-07-20 02:43:20 -050015
Andreas Dilger47a0c421997-05-16 02:46:07 -050016/* Writes all the PNG information. This is the suggested way to use the
17 * library. If you have a new chunk to add, make a function to write it,
18 * and put it in the correct location here. If you want the chunk written
Glenn Randers-Pehrson345bc271998-06-14 14:43:31 -050019 * after the image data, put it in png_write_end(). I strongly encourage
Andreas Dilger47a0c421997-05-16 02:46:07 -050020 * you to supply a PNG_INFO_ flag, and check info_ptr->valid before writing
21 * the chunk, as that will keep the code from breaking if you want to just
22 * write a plain PNG file. If you have long comments, I suggest writing
23 * them in png_write_end(), and compressing them.
24 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -050025void PNGAPI
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -060026png_write_info_before_PLTE(png_structp png_ptr, png_infop info_ptr)
Guy Schalnat0d580581995-07-20 02:43:20 -050027{
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -060028 png_debug(1, "in png_write_info_before_PLTE\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -060029 if (png_ptr == NULL || info_ptr == NULL)
30 return;
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -060031 if (!(png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE))
32 {
Guy Schalnat0d580581995-07-20 02:43:20 -050033 png_write_sig(png_ptr); /* write PNG signature */
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -060034#if defined(PNG_MNG_FEATURES_SUPPORTED)
35 if((png_ptr->mode&PNG_HAVE_PNG_SIGNATURE)&&(png_ptr->mng_features_permitted))
36 {
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -060037 png_warning(png_ptr,"MNG features are not allowed in a PNG datastream");
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -060038 png_ptr->mng_features_permitted=0;
39 }
40#endif
Guy Schalnat0d580581995-07-20 02:43:20 -050041 /* write IHDR information. */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060042 png_write_IHDR(png_ptr, info_ptr->width, info_ptr->height,
43 info_ptr->bit_depth, info_ptr->color_type, info_ptr->compression_type,
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -060044 info_ptr->filter_type,
45#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
46 info_ptr->interlace_type);
47#else
48 0);
49#endif
Guy Schalnat0d580581995-07-20 02:43:20 -050050 /* the rest of these check to see if the valid field has the appropriate
51 flag set, and if it does, writes the chunk. */
Guy Schalnat51f0eb41995-09-26 05:22:39 -050052#if defined(PNG_WRITE_gAMA_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060053 if (info_ptr->valid & PNG_INFO_gAMA)
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060054 {
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060055# ifdef PNG_FLOATING_POINT_SUPPORTED
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060056 png_write_gAMA(png_ptr, info_ptr->gamma);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -060057#else
58#ifdef PNG_FIXED_POINT_SUPPORTED
59 png_write_gAMA_fixed(png_ptr, info_ptr->int_gamma);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060060# endif
61#endif
62 }
Guy Schalnat51f0eb41995-09-26 05:22:39 -050063#endif
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -060064#if defined(PNG_WRITE_sRGB_SUPPORTED)
65 if (info_ptr->valid & PNG_INFO_sRGB)
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -060066 png_write_sRGB(png_ptr, (int)info_ptr->srgb_intent);
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -060067#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060068#if defined(PNG_WRITE_iCCP_SUPPORTED)
69 if (info_ptr->valid & PNG_INFO_iCCP)
Glenn Randers-Pehrson76e5fd62000-12-28 07:50:05 -060070 png_write_iCCP(png_ptr, info_ptr->iccp_name, PNG_COMPRESSION_TYPE_BASE,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060071 info_ptr->iccp_profile, (int)info_ptr->iccp_proflen);
72#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -050073#if defined(PNG_WRITE_sBIT_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060074 if (info_ptr->valid & PNG_INFO_sBIT)
75 png_write_sBIT(png_ptr, &(info_ptr->sig_bit), info_ptr->color_type);
Guy Schalnat51f0eb41995-09-26 05:22:39 -050076#endif
77#if defined(PNG_WRITE_cHRM_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060078 if (info_ptr->valid & PNG_INFO_cHRM)
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060079 {
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -060080#ifdef PNG_FLOATING_POINT_SUPPORTED
Guy Schalnat0d580581995-07-20 02:43:20 -050081 png_write_cHRM(png_ptr,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060082 info_ptr->x_white, info_ptr->y_white,
83 info_ptr->x_red, info_ptr->y_red,
84 info_ptr->x_green, info_ptr->y_green,
85 info_ptr->x_blue, info_ptr->y_blue);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -060086#else
87# ifdef PNG_FIXED_POINT_SUPPORTED
88 png_write_cHRM_fixed(png_ptr,
89 info_ptr->int_x_white, info_ptr->int_y_white,
90 info_ptr->int_x_red, info_ptr->int_y_red,
91 info_ptr->int_x_green, info_ptr->int_y_green,
92 info_ptr->int_x_blue, info_ptr->int_y_blue);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060093# endif
94#endif
95 }
96#endif
97#if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
98 if (info_ptr->unknown_chunks_num)
99 {
100 png_unknown_chunk *up;
101
102 png_debug(5, "writing extra chunks\n");
103
104 for (up = info_ptr->unknown_chunks;
105 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
106 up++)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600107 {
108 int keep=png_handle_as_unknown(png_ptr, up->name);
Glenn Randers-Pehrsondff799e2004-08-07 21:42:49 -0500109 if (keep != PNG_HANDLE_CHUNK_NEVER &&
Glenn Randers-Pehrson5b5dcf82004-07-17 22:45:44 -0500110 up->location && !(up->location & PNG_HAVE_PLTE) &&
111 !(up->location & PNG_HAVE_IDAT) &&
Glenn Randers-Pehrsondff799e2004-08-07 21:42:49 -0500112 ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS ||
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600113 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
114 {
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600115 png_write_chunk(png_ptr, up->name, up->data, up->size);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600116 }
117 }
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600118 }
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500119#endif
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600120 png_ptr->mode |= PNG_WROTE_INFO_BEFORE_PLTE;
121 }
122}
123
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500124void PNGAPI
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600125png_write_info(png_structp png_ptr, png_infop info_ptr)
126{
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600127#if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_sPLT_SUPPORTED)
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600128 int i;
129#endif
130
131 png_debug(1, "in png_write_info\n");
132
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600133 if (png_ptr == NULL || info_ptr == NULL)
134 return;
135
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600136 png_write_info_before_PLTE(png_ptr, info_ptr);
137
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600138 if (info_ptr->valid & PNG_INFO_PLTE)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500139 png_write_PLTE(png_ptr, info_ptr->palette,
140 (png_uint_32)info_ptr->num_palette);
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600141 else if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600142 png_error(png_ptr, "Valid palette required for paletted images");
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600143
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500144#if defined(PNG_WRITE_tRNS_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600145 if (info_ptr->valid & PNG_INFO_tRNS)
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600146 {
147#if defined(PNG_WRITE_INVERT_ALPHA_SUPPORTED)
148 /* invert the alpha channel (in tRNS) */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600149 if ((png_ptr->transformations & PNG_INVERT_ALPHA) &&
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600150 info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
151 {
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600152 int j;
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600153 for (j=0; j<(int)info_ptr->num_trans; j++)
Glenn Randers-Pehrson860ab2b1999-10-14 07:43:10 -0500154 info_ptr->trans[j] = (png_byte)(255 - info_ptr->trans[j]);
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600155 }
156#endif
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600157 png_write_tRNS(png_ptr, info_ptr->trans, &(info_ptr->trans_values),
158 info_ptr->num_trans, info_ptr->color_type);
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600159 }
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500160#endif
161#if defined(PNG_WRITE_bKGD_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600162 if (info_ptr->valid & PNG_INFO_bKGD)
163 png_write_bKGD(png_ptr, &(info_ptr->background), info_ptr->color_type);
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500164#endif
165#if defined(PNG_WRITE_hIST_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600166 if (info_ptr->valid & PNG_INFO_hIST)
167 png_write_hIST(png_ptr, info_ptr->hist, info_ptr->num_palette);
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500168#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500169#if defined(PNG_WRITE_oFFs_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600170 if (info_ptr->valid & PNG_INFO_oFFs)
171 png_write_oFFs(png_ptr, info_ptr->x_offset, info_ptr->y_offset,
172 info_ptr->offset_unit_type);
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500173#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500174#if defined(PNG_WRITE_pCAL_SUPPORTED)
175 if (info_ptr->valid & PNG_INFO_pCAL)
176 png_write_pCAL(png_ptr, info_ptr->pcal_purpose, info_ptr->pcal_X0,
177 info_ptr->pcal_X1, info_ptr->pcal_type, info_ptr->pcal_nparams,
178 info_ptr->pcal_units, info_ptr->pcal_params);
179#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600180#if defined(PNG_WRITE_sCAL_SUPPORTED)
181 if (info_ptr->valid & PNG_INFO_sCAL)
Glenn Randers-Pehrson68ea2432000-04-01 21:10:05 -0600182#if defined(PNG_FLOATING_POINT_SUPPORTED) && !defined(PNG_NO_STDIO)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600183 png_write_sCAL(png_ptr, (int)info_ptr->scal_unit,
184 info_ptr->scal_pixel_width, info_ptr->scal_pixel_height);
185#else
186#ifdef PNG_FIXED_POINT_SUPPORTED
187 png_write_sCAL_s(png_ptr, (int)info_ptr->scal_unit,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600188 info_ptr->scal_s_width, info_ptr->scal_s_height);
Glenn Randers-Pehrson68ea2432000-04-01 21:10:05 -0600189#else
190 png_warning(png_ptr,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500191 "png_write_sCAL not supported; sCAL chunk not written");
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600192#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600193#endif
194#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500195#if defined(PNG_WRITE_pHYs_SUPPORTED)
196 if (info_ptr->valid & PNG_INFO_pHYs)
197 png_write_pHYs(png_ptr, info_ptr->x_pixels_per_unit,
198 info_ptr->y_pixels_per_unit, info_ptr->phys_unit_type);
199#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500200#if defined(PNG_WRITE_tIME_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600201 if (info_ptr->valid & PNG_INFO_tIME)
Guy Schalnate5a37791996-06-05 15:50:50 -0500202 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600203 png_write_tIME(png_ptr, &(info_ptr->mod_time));
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600204 png_ptr->mode |= PNG_WROTE_tIME;
Guy Schalnate5a37791996-06-05 15:50:50 -0500205 }
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500206#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600207#if defined(PNG_WRITE_sPLT_SUPPORTED)
208 if (info_ptr->valid & PNG_INFO_sPLT)
209 for (i = 0; i < (int)info_ptr->splt_palettes_num; i++)
210 png_write_sPLT(png_ptr, info_ptr->splt_palettes + i);
211#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600212#if defined(PNG_WRITE_TEXT_SUPPORTED)
Guy Schalnate5a37791996-06-05 15:50:50 -0500213 /* Check to see if we need to write text chunks */
Andreas Dilger47a0c421997-05-16 02:46:07 -0500214 for (i = 0; i < info_ptr->num_text; i++)
Guy Schalnat0d580581995-07-20 02:43:20 -0500215 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500216 png_debug2(2, "Writing header text chunk %d, type %d\n", i,
217 info_ptr->text[i].compression);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600218 /* an internationalized chunk? */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600219 if (info_ptr->text[i].compression > 0)
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600220 {
221#if defined(PNG_WRITE_iTXt_SUPPORTED)
222 /* write international chunk */
223 png_write_iTXt(png_ptr,
224 info_ptr->text[i].compression,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600225 info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600226 info_ptr->text[i].lang,
227 info_ptr->text[i].lang_key,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600228 info_ptr->text[i].text);
229#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600230 png_warning(png_ptr, "Unable to write international text");
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600231#endif
232 /* Mark this chunk as written */
233 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
234 }
Andreas Dilger47a0c421997-05-16 02:46:07 -0500235 /* If we want a compressed text chunk */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600236 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_zTXt)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500237 {
238#if defined(PNG_WRITE_zTXt_SUPPORTED)
239 /* write compressed chunk */
240 png_write_zTXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600241 info_ptr->text[i].text, 0,
Andreas Dilger47a0c421997-05-16 02:46:07 -0500242 info_ptr->text[i].compression);
243#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600244 png_warning(png_ptr, "Unable to write compressed text");
Andreas Dilger47a0c421997-05-16 02:46:07 -0500245#endif
246 /* Mark this chunk as written */
247 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
248 }
249 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
250 {
251#if defined(PNG_WRITE_tEXt_SUPPORTED)
252 /* write uncompressed chunk */
253 png_write_tEXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600254 info_ptr->text[i].text,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600255 0);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500256#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600257 png_warning(png_ptr, "Unable to write uncompressed text");
Andreas Dilger47a0c421997-05-16 02:46:07 -0500258#endif
259 /* Mark this chunk as written */
260 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
261 }
262 }
263#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600264#if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
265 if (info_ptr->unknown_chunks_num)
266 {
267 png_unknown_chunk *up;
268
269 png_debug(5, "writing extra chunks\n");
270
271 for (up = info_ptr->unknown_chunks;
272 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
273 up++)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600274 {
275 int keep=png_handle_as_unknown(png_ptr, up->name);
Glenn Randers-Pehrsondff799e2004-08-07 21:42:49 -0500276 if (keep != PNG_HANDLE_CHUNK_NEVER &&
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600277 up->location && (up->location & PNG_HAVE_PLTE) &&
278 !(up->location & PNG_HAVE_IDAT) &&
Glenn Randers-Pehrsondff799e2004-08-07 21:42:49 -0500279 ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS ||
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600280 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
281 {
282 png_write_chunk(png_ptr, up->name, up->data, up->size);
283 }
284 }
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600285 }
286#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500287}
Guy Schalnat0d580581995-07-20 02:43:20 -0500288
Andreas Dilger47a0c421997-05-16 02:46:07 -0500289/* Writes the end of the PNG file. If you don't want to write comments or
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600290 * time information, you can pass NULL for info. If you already wrote these
291 * in png_write_info(), do not write them again here. If you have long
292 * comments, I suggest writing them here, and compressing them.
293 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500294void PNGAPI
Andreas Dilger47a0c421997-05-16 02:46:07 -0500295png_write_end(png_structp png_ptr, png_infop info_ptr)
296{
297 png_debug(1, "in png_write_end\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600298 if (png_ptr == NULL)
299 return;
Andreas Dilger47a0c421997-05-16 02:46:07 -0500300 if (!(png_ptr->mode & PNG_HAVE_IDAT))
301 png_error(png_ptr, "No IDATs written into file");
302
303 /* see if user wants us to write information chunks */
304 if (info_ptr != NULL)
305 {
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600306#if defined(PNG_WRITE_TEXT_SUPPORTED)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500307 int i; /* local index variable */
Glenn Randers-Pehrson70e3f541998-01-03 22:40:55 -0600308#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500309#if defined(PNG_WRITE_tIME_SUPPORTED)
310 /* check to see if user has supplied a time chunk */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600311 if ((info_ptr->valid & PNG_INFO_tIME) &&
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600312 !(png_ptr->mode & PNG_WROTE_tIME))
Andreas Dilger47a0c421997-05-16 02:46:07 -0500313 png_write_tIME(png_ptr, &(info_ptr->mod_time));
314#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600315#if defined(PNG_WRITE_TEXT_SUPPORTED)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500316 /* loop through comment chunks */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600317 for (i = 0; i < info_ptr->num_text; i++)
Guy Schalnat0d580581995-07-20 02:43:20 -0500318 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500319 png_debug2(2, "Writing trailer text chunk %d, type %d\n", i,
320 info_ptr->text[i].compression);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600321 /* an internationalized chunk? */
322 if (info_ptr->text[i].compression > 0)
323 {
324#if defined(PNG_WRITE_iTXt_SUPPORTED)
325 /* write international chunk */
326 png_write_iTXt(png_ptr,
327 info_ptr->text[i].compression,
328 info_ptr->text[i].key,
329 info_ptr->text[i].lang,
330 info_ptr->text[i].lang_key,
331 info_ptr->text[i].text);
332#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600333 png_warning(png_ptr, "Unable to write international text");
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600334#endif
335 /* Mark this chunk as written */
336 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
337 }
338 else if (info_ptr->text[i].compression >= PNG_TEXT_COMPRESSION_zTXt)
Guy Schalnat0d580581995-07-20 02:43:20 -0500339 {
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500340#if defined(PNG_WRITE_zTXt_SUPPORTED)
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600341 /* write compressed chunk */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600342 png_write_zTXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600343 info_ptr->text[i].text, 0,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600344 info_ptr->text[i].compression);
Guy Schalnate5a37791996-06-05 15:50:50 -0500345#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600346 png_warning(png_ptr, "Unable to write compressed text");
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500347#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500348 /* Mark this chunk as written */
349 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
Guy Schalnat0d580581995-07-20 02:43:20 -0500350 }
Andreas Dilger47a0c421997-05-16 02:46:07 -0500351 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
Guy Schalnat0d580581995-07-20 02:43:20 -0500352 {
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500353#if defined(PNG_WRITE_tEXt_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500354 /* write uncompressed chunk */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600355 png_write_tEXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600356 info_ptr->text[i].text, 0);
Guy Schalnate5a37791996-06-05 15:50:50 -0500357#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600358 png_warning(png_ptr, "Unable to write uncompressed text");
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500359#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500360
Andreas Dilger47a0c421997-05-16 02:46:07 -0500361 /* Mark this chunk as written */
362 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
Guy Schalnat0d580581995-07-20 02:43:20 -0500363 }
364 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600365#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600366#if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
367 if (info_ptr->unknown_chunks_num)
368 {
369 png_unknown_chunk *up;
370
371 png_debug(5, "writing extra chunks\n");
372
373 for (up = info_ptr->unknown_chunks;
374 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
375 up++)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600376 {
377 int keep=png_handle_as_unknown(png_ptr, up->name);
Glenn Randers-Pehrsondff799e2004-08-07 21:42:49 -0500378 if (keep != PNG_HANDLE_CHUNK_NEVER &&
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600379 up->location && (up->location & PNG_AFTER_IDAT) &&
Glenn Randers-Pehrsondff799e2004-08-07 21:42:49 -0500380 ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS ||
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600381 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
382 {
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600383 png_write_chunk(png_ptr, up->name, up->data, up->size);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600384 }
385 }
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600386 }
387#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500388 }
Guy Schalnate5a37791996-06-05 15:50:50 -0500389
390 png_ptr->mode |= PNG_AFTER_IDAT;
391
Andreas Dilger47a0c421997-05-16 02:46:07 -0500392 /* write end of PNG file */
Guy Schalnat0d580581995-07-20 02:43:20 -0500393 png_write_IEND(png_ptr);
Glenn Randers-Pehrsond56aca72000-11-23 11:51:42 -0600394#if 0
395/* This flush, added in libpng-1.0.8, causes some applications to crash
396 because they do not set png_ptr->output_flush_fn */
Glenn Randers-Pehrson32fc5ce2000-07-24 06:34:14 -0500397 png_flush(png_ptr);
Glenn Randers-Pehrsond56aca72000-11-23 11:51:42 -0600398#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500399}
400
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500401#if defined(PNG_WRITE_tIME_SUPPORTED)
Glenn Randers-Pehrson316f97a2000-07-08 13:19:41 -0500402/* "time.h" functions are not supported on WindowsCE */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500403void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600404png_convert_from_struct_tm(png_timep ptime, struct tm FAR * ttime)
Guy Schalnat0d580581995-07-20 02:43:20 -0500405{
Andreas Dilger47a0c421997-05-16 02:46:07 -0500406 png_debug(1, "in png_convert_from_struct_tm\n");
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600407 ptime->year = (png_uint_16)(1900 + ttime->tm_year);
408 ptime->month = (png_byte)(ttime->tm_mon + 1);
409 ptime->day = (png_byte)ttime->tm_mday;
410 ptime->hour = (png_byte)ttime->tm_hour;
411 ptime->minute = (png_byte)ttime->tm_min;
412 ptime->second = (png_byte)ttime->tm_sec;
Guy Schalnat0d580581995-07-20 02:43:20 -0500413}
414
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500415void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600416png_convert_from_time_t(png_timep ptime, time_t ttime)
Guy Schalnat0d580581995-07-20 02:43:20 -0500417{
418 struct tm *tbuf;
419
Andreas Dilger47a0c421997-05-16 02:46:07 -0500420 png_debug(1, "in png_convert_from_time_t\n");
Guy Schalnat0d580581995-07-20 02:43:20 -0500421 tbuf = gmtime(&ttime);
422 png_convert_from_struct_tm(ptime, tbuf);
423}
Guy Schalnat6d764711995-12-19 03:22:19 -0600424#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500425
Andreas Dilger47a0c421997-05-16 02:46:07 -0500426/* Initialize png_ptr structure, and allocate any memory needed */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500427png_structp PNGAPI
Glenn Randers-Pehrsond0dce401998-05-09 10:02:29 -0500428png_create_write_struct(png_const_charp user_png_ver, png_voidp error_ptr,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600429 png_error_ptr error_fn, png_error_ptr warn_fn)
Guy Schalnat0d580581995-07-20 02:43:20 -0500430{
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500431#ifdef PNG_USER_MEM_SUPPORTED
432 return (png_create_write_struct_2(user_png_ver, error_ptr, error_fn,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500433 warn_fn, NULL, NULL, NULL));
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500434}
435
436/* Alternate initialize png_ptr structure, and allocate any memory needed */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500437png_structp PNGAPI
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500438png_create_write_struct_2(png_const_charp user_png_ver, png_voidp error_ptr,
439 png_error_ptr error_fn, png_error_ptr warn_fn, png_voidp mem_ptr,
440 png_malloc_ptr malloc_fn, png_free_ptr free_fn)
441{
442#endif /* PNG_USER_MEM_SUPPORTED */
Guy Schalnate5a37791996-06-05 15:50:50 -0500443 png_structp png_ptr;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600444#ifdef PNG_SETJMP_SUPPORTED
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600445#ifdef USE_FAR_KEYWORD
446 jmp_buf jmpbuf;
447#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600448#endif
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500449 int i;
Andreas Dilger47a0c421997-05-16 02:46:07 -0500450 png_debug(1, "in png_create_write_struct\n");
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500451#ifdef PNG_USER_MEM_SUPPORTED
Glenn Randers-Pehrson9c0f0942002-02-21 23:14:23 -0600452 png_ptr = (png_structp)png_create_struct_2(PNG_STRUCT_PNG,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500453 malloc_fn, mem_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500454#else
Glenn Randers-Pehrson9c0f0942002-02-21 23:14:23 -0600455 png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500456#endif /* PNG_USER_MEM_SUPPORTED */
Glenn Randers-Pehrson9c0f0942002-02-21 23:14:23 -0600457 if (png_ptr == NULL)
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500458 return (NULL);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600459
Glenn Randers-Pehrson17218292006-04-20 07:20:46 -0500460#ifdef PNG_MMX_CODE_SUPPORTED
Glenn Randers-Pehrson1fd5fb32001-05-06 05:34:26 -0500461 png_init_mmx_flags(png_ptr); /* 1.2.0 addition */
462#endif
463
Glenn Randers-Pehrson272489d2004-08-04 06:34:52 -0500464 /* added at libpng-1.2.6 */
465#ifdef PNG_SET_USER_LIMITS_SUPPORTED
466 png_ptr->user_width_max=PNG_USER_WIDTH_MAX;
467 png_ptr->user_height_max=PNG_USER_HEIGHT_MAX;
468#endif
469
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600470#ifdef PNG_SETJMP_SUPPORTED
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600471#ifdef USE_FAR_KEYWORD
472 if (setjmp(jmpbuf))
473#else
Guy Schalnate5a37791996-06-05 15:50:50 -0500474 if (setjmp(png_ptr->jmpbuf))
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600475#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500476 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600477 png_free(png_ptr, png_ptr->zbuf);
Glenn Randers-Pehrson4766a242000-07-17 06:17:09 -0500478 png_ptr->zbuf=NULL;
Guy Schalnate5a37791996-06-05 15:50:50 -0500479 png_destroy_struct(png_ptr);
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500480 return (NULL);
Guy Schalnate5a37791996-06-05 15:50:50 -0500481 }
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600482#ifdef USE_FAR_KEYWORD
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500483 png_memcpy(png_ptr->jmpbuf, jmpbuf, sizeof(jmp_buf));
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600484#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600485#endif
486
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500487#ifdef PNG_USER_MEM_SUPPORTED
488 png_set_mem_fn(png_ptr, mem_ptr, malloc_fn, free_fn);
489#endif /* PNG_USER_MEM_SUPPORTED */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600490 png_set_error_fn(png_ptr, error_ptr, error_fn, warn_fn);
Guy Schalnate5a37791996-06-05 15:50:50 -0500491
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500492 i=0;
493 do
Guy Schalnate5a37791996-06-05 15:50:50 -0500494 {
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500495 if(user_png_ver[i] != png_libpng_ver[i])
496 png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH;
497 } while (png_libpng_ver[i++]);
Guy Schalnat0d580581995-07-20 02:43:20 -0500498
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500499 if (png_ptr->flags & PNG_FLAG_LIBRARY_MISMATCH)
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500500 {
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500501 /* Libpng 0.90 and later are binary incompatible with libpng 0.89, so
502 * we must recompile any applications that use any older library version.
503 * For versions after libpng 1.0, we will be compatible, so we need
504 * only check the first digit.
505 */
506 if (user_png_ver == NULL || user_png_ver[0] != png_libpng_ver[0] ||
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500507 (user_png_ver[0] == '1' && user_png_ver[2] != png_libpng_ver[2]) ||
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500508 (user_png_ver[0] == '0' && user_png_ver[2] < '9'))
509 {
Glenn Randers-Pehrson17218292006-04-20 07:20:46 -0500510#ifndef PNG_NO_STDIO
Glenn Randers-Pehrsonb1828932001-06-23 08:03:17 -0500511 char msg[80];
512 if (user_png_ver)
513 {
Glenn Randers-Pehrson17218292006-04-20 07:20:46 -0500514 png_sprintf(msg, "Application was compiled with png.h from libpng-%.20s",
Glenn Randers-Pehrsonb1828932001-06-23 08:03:17 -0500515 user_png_ver);
516 png_warning(png_ptr, msg);
517 }
Glenn Randers-Pehrson17218292006-04-20 07:20:46 -0500518 png_sprintf(msg, "Application is running with png.c from libpng-%.20s",
Glenn Randers-Pehrsonb1828932001-06-23 08:03:17 -0500519 png_libpng_ver);
520 png_warning(png_ptr, msg);
521#endif
522#ifdef PNG_ERROR_NUMBERS_SUPPORTED
523 png_ptr->flags=0;
524#endif
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500525 png_error(png_ptr,
526 "Incompatible libpng version in application and library");
527 }
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500528 }
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500529
Guy Schalnat0d580581995-07-20 02:43:20 -0500530 /* initialize zbuf - compression buffer */
531 png_ptr->zbuf_size = PNG_ZBUF_SIZE;
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500532 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr, png_ptr->zbuf_size);
Guy Schalnate5a37791996-06-05 15:50:50 -0500533
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500534 png_set_write_fn(png_ptr, NULL, NULL, NULL);
Guy Schalnate5a37791996-06-05 15:50:50 -0500535
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600536#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
537 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500538 1, NULL, NULL);
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600539#endif
540
Glenn Randers-Pehrson03008a02002-04-27 10:11:25 -0500541#ifdef PNG_SETJMP_SUPPORTED
542/* Applications that neglect to set up their own setjmp() and then encounter
543 a png_error() will longjmp here. Since the jmpbuf is then meaningless we
544 abort instead of returning. */
545#ifdef USE_FAR_KEYWORD
546 if (setjmp(jmpbuf))
547 PNG_ABORT();
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500548 png_memcpy(png_ptr->jmpbuf, jmpbuf, sizeof(jmp_buf));
Glenn Randers-Pehrson03008a02002-04-27 10:11:25 -0500549#else
550 if (setjmp(png_ptr->jmpbuf))
551 PNG_ABORT();
552#endif
553#endif
554 return (png_ptr);
Guy Schalnate5a37791996-06-05 15:50:50 -0500555}
556
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500557void PNGAPI
558png_write_init_3(png_structpp ptr_ptr, png_const_charp user_png_ver,
559 png_size_t png_struct_size)
560{
561 png_structp png_ptr=*ptr_ptr;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600562#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500563 jmp_buf tmp_jmp; /* to save current jump buffer */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600564#endif
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600565
Glenn Randers-Pehrson4deeb792006-03-22 16:21:59 -0600566 int i = 0;
567
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600568 if (png_ptr == NULL)
569 return;
570
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500571 do
572 {
573 if (user_png_ver[i] != png_libpng_ver[i])
574 {
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500575#ifdef PNG_LEGACY_SUPPORTED
576 png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH;
577#else
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500578 png_ptr->warning_fn=NULL;
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500579 png_warning(png_ptr,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500580 "Application uses deprecated png_write_init() and should be recompiled");
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500581 break;
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500582#endif
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500583 }
584 } while (png_libpng_ver[i++]);
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500585
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500586 png_debug(1, "in png_write_init_3\n");
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500587
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600588#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500589 /* save jump buffer and error functions */
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500590 png_memcpy(tmp_jmp, png_ptr->jmpbuf, sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600591#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500592
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500593 if (sizeof(png_struct) > png_struct_size)
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500594 {
595 png_destroy_struct(png_ptr);
596 png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG);
597 *ptr_ptr = png_ptr;
598 }
599
Guy Schalnate5a37791996-06-05 15:50:50 -0500600 /* reset all variables to 0 */
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500601 png_memset(png_ptr, 0, sizeof(png_struct));
Guy Schalnate5a37791996-06-05 15:50:50 -0500602
Glenn Randers-Pehrson272489d2004-08-04 06:34:52 -0500603 /* added at libpng-1.2.6 */
604#ifdef PNG_SET_USER_LIMITS_SUPPORTED
605 png_ptr->user_width_max=PNG_USER_WIDTH_MAX;
606 png_ptr->user_height_max=PNG_USER_HEIGHT_MAX;
607#endif
608
Glenn Randers-Pehrson17218292006-04-20 07:20:46 -0500609#ifdef PNG_MMX_CODE_SUPPORTED
Glenn Randers-Pehrson1fd5fb32001-05-06 05:34:26 -0500610 png_init_mmx_flags(png_ptr); /* 1.2.0 addition */
611#endif
612
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600613#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500614 /* restore jump buffer */
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500615 png_memcpy(png_ptr->jmpbuf, tmp_jmp, sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600616#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500617
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500618 png_set_write_fn(png_ptr, NULL, NULL, NULL);
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500619
Guy Schalnate5a37791996-06-05 15:50:50 -0500620 /* initialize zbuf - compression buffer */
621 png_ptr->zbuf_size = PNG_ZBUF_SIZE;
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500622 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr, png_ptr->zbuf_size);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500623
624#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
625 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500626 1, NULL, NULL);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500627#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500628}
629
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600630/* Write a few rows of image data. If the image is interlaced,
631 * either you will have to write the 7 sub images, or, if you
632 * have called png_set_interlace_handling(), you will have to
633 * "write" the image seven times.
634 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500635void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600636png_write_rows(png_structp png_ptr, png_bytepp row,
Guy Schalnat0d580581995-07-20 02:43:20 -0500637 png_uint_32 num_rows)
638{
639 png_uint_32 i; /* row counter */
Guy Schalnat6d764711995-12-19 03:22:19 -0600640 png_bytepp rp; /* row pointer */
Guy Schalnat0d580581995-07-20 02:43:20 -0500641
Andreas Dilger47a0c421997-05-16 02:46:07 -0500642 png_debug(1, "in png_write_rows\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600643
644 if (png_ptr == NULL)
645 return;
646
Guy Schalnat0d580581995-07-20 02:43:20 -0500647 /* loop through the rows */
648 for (i = 0, rp = row; i < num_rows; i++, rp++)
649 {
650 png_write_row(png_ptr, *rp);
651 }
652}
653
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600654/* Write the image. You only need to call this function once, even
655 * if you are writing an interlaced image.
656 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500657void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600658png_write_image(png_structp png_ptr, png_bytepp image)
Guy Schalnat0d580581995-07-20 02:43:20 -0500659{
660 png_uint_32 i; /* row index */
661 int pass, num_pass; /* pass variables */
Guy Schalnat6d764711995-12-19 03:22:19 -0600662 png_bytepp rp; /* points to current row */
Guy Schalnat0d580581995-07-20 02:43:20 -0500663
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600664 if (png_ptr == NULL)
665 return;
666
Andreas Dilger47a0c421997-05-16 02:46:07 -0500667 png_debug(1, "in png_write_image\n");
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600668#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500669 /* intialize interlace handling. If image is not interlaced,
670 this will set pass to 1 */
671 num_pass = png_set_interlace_handling(png_ptr);
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600672#else
673 num_pass = 1;
674#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500675 /* loop through passes */
676 for (pass = 0; pass < num_pass; pass++)
677 {
678 /* loop through image */
679 for (i = 0, rp = image; i < png_ptr->height; i++, rp++)
680 {
681 png_write_row(png_ptr, *rp);
682 }
683 }
684}
685
Guy Schalnate5a37791996-06-05 15:50:50 -0500686/* called by user to write a row of image data */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500687void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600688png_write_row(png_structp png_ptr, png_bytep row)
Guy Schalnat0d580581995-07-20 02:43:20 -0500689{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600690 if (png_ptr == NULL)
691 return;
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600692 png_debug2(1, "in png_write_row (row %ld, pass %d)\n",
693 png_ptr->row_number, png_ptr->pass);
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600694
Guy Schalnat0d580581995-07-20 02:43:20 -0500695 /* initialize transformations and other stuff if first time */
Guy Schalnat6d764711995-12-19 03:22:19 -0600696 if (png_ptr->row_number == 0 && png_ptr->pass == 0)
Guy Schalnat0d580581995-07-20 02:43:20 -0500697 {
Glenn Randers-Pehrson5cded0b2001-11-07 07:10:08 -0600698 /* make sure we wrote the header info */
699 if (!(png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE))
700 png_error(png_ptr,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500701 "png_write_info was never called before png_write_row");
Glenn Randers-Pehrson5cded0b2001-11-07 07:10:08 -0600702
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500703 /* check for transforms that have been set but were defined out */
704#if !defined(PNG_WRITE_INVERT_SUPPORTED) && defined(PNG_READ_INVERT_SUPPORTED)
705 if (png_ptr->transformations & PNG_INVERT_MONO)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500706 png_warning(png_ptr, "PNG_WRITE_INVERT_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500707#endif
708#if !defined(PNG_WRITE_FILLER_SUPPORTED) && defined(PNG_READ_FILLER_SUPPORTED)
709 if (png_ptr->transformations & PNG_FILLER)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500710 png_warning(png_ptr, "PNG_WRITE_FILLER_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500711#endif
712#if !defined(PNG_WRITE_PACKSWAP_SUPPORTED) && defined(PNG_READ_PACKSWAP_SUPPORTED)
713 if (png_ptr->transformations & PNG_PACKSWAP)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500714 png_warning(png_ptr, "PNG_WRITE_PACKSWAP_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500715#endif
716#if !defined(PNG_WRITE_PACK_SUPPORTED) && defined(PNG_READ_PACK_SUPPORTED)
717 if (png_ptr->transformations & PNG_PACK)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500718 png_warning(png_ptr, "PNG_WRITE_PACK_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500719#endif
720#if !defined(PNG_WRITE_SHIFT_SUPPORTED) && defined(PNG_READ_SHIFT_SUPPORTED)
721 if (png_ptr->transformations & PNG_SHIFT)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500722 png_warning(png_ptr, "PNG_WRITE_SHIFT_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500723#endif
724#if !defined(PNG_WRITE_BGR_SUPPORTED) && defined(PNG_READ_BGR_SUPPORTED)
725 if (png_ptr->transformations & PNG_BGR)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500726 png_warning(png_ptr, "PNG_WRITE_BGR_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500727#endif
728#if !defined(PNG_WRITE_SWAP_SUPPORTED) && defined(PNG_READ_SWAP_SUPPORTED)
729 if (png_ptr->transformations & PNG_SWAP_BYTES)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500730 png_warning(png_ptr, "PNG_WRITE_SWAP_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500731#endif
732
Guy Schalnat0d580581995-07-20 02:43:20 -0500733 png_write_start_row(png_ptr);
734 }
735
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500736#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500737 /* if interlaced and not interested in row, return */
738 if (png_ptr->interlaced && (png_ptr->transformations & PNG_INTERLACE))
739 {
740 switch (png_ptr->pass)
741 {
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600742 case 0:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600743 if (png_ptr->row_number & 0x07)
Guy Schalnat0d580581995-07-20 02:43:20 -0500744 {
745 png_write_finish_row(png_ptr);
746 return;
747 }
748 break;
749 case 1:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600750 if ((png_ptr->row_number & 0x07) || png_ptr->width < 5)
Guy Schalnat0d580581995-07-20 02:43:20 -0500751 {
752 png_write_finish_row(png_ptr);
753 return;
754 }
755 break;
756 case 2:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600757 if ((png_ptr->row_number & 0x07) != 4)
Guy Schalnat0d580581995-07-20 02:43:20 -0500758 {
759 png_write_finish_row(png_ptr);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600760 return;
Guy Schalnat0d580581995-07-20 02:43:20 -0500761 }
762 break;
763 case 3:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600764 if ((png_ptr->row_number & 0x03) || png_ptr->width < 3)
Guy Schalnat0d580581995-07-20 02:43:20 -0500765 {
766 png_write_finish_row(png_ptr);
767 return;
768 }
769 break;
770 case 4:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600771 if ((png_ptr->row_number & 0x03) != 2)
Guy Schalnat0d580581995-07-20 02:43:20 -0500772 {
773 png_write_finish_row(png_ptr);
774 return;
775 }
776 break;
777 case 5:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600778 if ((png_ptr->row_number & 0x01) || png_ptr->width < 2)
Guy Schalnat0d580581995-07-20 02:43:20 -0500779 {
780 png_write_finish_row(png_ptr);
781 return;
782 }
783 break;
784 case 6:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600785 if (!(png_ptr->row_number & 0x01))
Guy Schalnat0d580581995-07-20 02:43:20 -0500786 {
787 png_write_finish_row(png_ptr);
788 return;
789 }
790 break;
791 }
792 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600793#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500794
795 /* set up row info for transformations */
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600796 png_ptr->row_info.color_type = png_ptr->color_type;
Guy Schalnat0d580581995-07-20 02:43:20 -0500797 png_ptr->row_info.width = png_ptr->usr_width;
798 png_ptr->row_info.channels = png_ptr->usr_channels;
799 png_ptr->row_info.bit_depth = png_ptr->usr_bit_depth;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600800 png_ptr->row_info.pixel_depth = (png_byte)(png_ptr->row_info.bit_depth *
801 png_ptr->row_info.channels);
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600802
Glenn Randers-Pehrson272489d2004-08-04 06:34:52 -0500803 png_ptr->row_info.rowbytes = PNG_ROWBYTES(png_ptr->row_info.pixel_depth,
804 png_ptr->row_info.width);
Guy Schalnat0d580581995-07-20 02:43:20 -0500805
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600806 png_debug1(3, "row_info->color_type = %d\n", png_ptr->row_info.color_type);
Glenn Randers-Pehrson6bc53be2006-06-16 07:52:03 -0500807 png_debug1(3, "row_info->width = %lu\n",
808 (unsigned long) png_ptr->row_info.width);
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600809 png_debug1(3, "row_info->channels = %d\n", png_ptr->row_info.channels);
810 png_debug1(3, "row_info->bit_depth = %d\n", png_ptr->row_info.bit_depth);
811 png_debug1(3, "row_info->pixel_depth = %d\n", png_ptr->row_info.pixel_depth);
Glenn Randers-Pehrson6bc53be2006-06-16 07:52:03 -0500812 png_debug1(3, "row_info->rowbytes = %lu\n",
813 (unsigned long) png_ptr->row_info.rowbytes);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500814
815 /* Copy user's row into buffer, leaving room for filter byte. */
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500816 png_memcpy(png_ptr->row_buf + 1, row, png_ptr->row_info.rowbytes);
Guy Schalnat0d580581995-07-20 02:43:20 -0500817
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500818#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500819 /* handle interlacing */
820 if (png_ptr->interlaced && png_ptr->pass < 6 &&
821 (png_ptr->transformations & PNG_INTERLACE))
822 {
823 png_do_write_interlace(&(png_ptr->row_info),
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600824 png_ptr->row_buf + 1, png_ptr->pass);
Guy Schalnat0d580581995-07-20 02:43:20 -0500825 /* this should always get caught above, but still ... */
826 if (!(png_ptr->row_info.width))
827 {
828 png_write_finish_row(png_ptr);
829 return;
830 }
831 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600832#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500833
834 /* handle other transformations */
835 if (png_ptr->transformations)
836 png_do_write_transformations(png_ptr);
837
Glenn Randers-Pehrson2ad31ae2000-12-15 08:54:42 -0600838#if defined(PNG_MNG_FEATURES_SUPPORTED)
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -0600839 /* Write filter_method 64 (intrapixel differencing) only if
840 * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and
841 * 2. Libpng did not write a PNG signature (this filter_method is only
842 * used in PNG datastreams that are embedded in MNG datastreams) and
843 * 3. The application called png_permit_mng_features with a mask that
844 * included PNG_FLAG_MNG_FILTER_64 and
845 * 4. The filter_method is 64 and
846 * 5. The color_type is RGB or RGBA
847 */
Glenn Randers-Pehrson2ad31ae2000-12-15 08:54:42 -0600848 if((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
849 (png_ptr->filter_type == PNG_INTRAPIXEL_DIFFERENCING))
850 {
851 /* Intrapixel differencing */
852 png_do_write_intrapixel(&(png_ptr->row_info), png_ptr->row_buf + 1);
853 }
854#endif
855
Andreas Dilger47a0c421997-05-16 02:46:07 -0500856 /* Find a filter if necessary, filter the row and write it out. */
Guy Schalnate5a37791996-06-05 15:50:50 -0500857 png_write_find_filter(png_ptr, &(png_ptr->row_info));
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -0600858
859 if (png_ptr->write_row_fn != NULL)
860 (*(png_ptr->write_row_fn))(png_ptr, png_ptr->row_number, png_ptr->pass);
Guy Schalnat0d580581995-07-20 02:43:20 -0500861}
862
Guy Schalnat0f716451995-11-28 11:22:13 -0600863#if defined(PNG_WRITE_FLUSH_SUPPORTED)
864/* Set the automatic flush interval or 0 to turn flushing off */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500865void PNGAPI
Guy Schalnat4ee97b01996-01-16 01:51:56 -0600866png_set_flush(png_structp png_ptr, int nrows)
Guy Schalnat0f716451995-11-28 11:22:13 -0600867{
Andreas Dilger47a0c421997-05-16 02:46:07 -0500868 png_debug(1, "in png_set_flush\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600869 if (png_ptr == NULL)
870 return;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600871 png_ptr->flush_dist = (nrows < 0 ? 0 : nrows);
Guy Schalnat0f716451995-11-28 11:22:13 -0600872}
873
874/* flush the current output buffers now */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500875void PNGAPI
Guy Schalnat4ee97b01996-01-16 01:51:56 -0600876png_write_flush(png_structp png_ptr)
Guy Schalnat0f716451995-11-28 11:22:13 -0600877{
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600878 int wrote_IDAT;
Guy Schalnat0f716451995-11-28 11:22:13 -0600879
Andreas Dilger47a0c421997-05-16 02:46:07 -0500880 png_debug(1, "in png_write_flush\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600881 if (png_ptr == NULL)
882 return;
Guy Schalnate5a37791996-06-05 15:50:50 -0500883 /* We have already written out all of the data */
884 if (png_ptr->row_number >= png_ptr->num_rows)
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600885 return;
Guy Schalnat0f716451995-11-28 11:22:13 -0600886
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600887 do
888 {
889 int ret;
Guy Schalnat0f716451995-11-28 11:22:13 -0600890
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600891 /* compress the data */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600892 ret = deflate(&png_ptr->zstream, Z_SYNC_FLUSH);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600893 wrote_IDAT = 0;
Guy Schalnat0f716451995-11-28 11:22:13 -0600894
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600895 /* check for compression errors */
896 if (ret != Z_OK)
897 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500898 if (png_ptr->zstream.msg != NULL)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600899 png_error(png_ptr, png_ptr->zstream.msg);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600900 else
901 png_error(png_ptr, "zlib error");
902 }
Guy Schalnat0f716451995-11-28 11:22:13 -0600903
Andreas Dilger47a0c421997-05-16 02:46:07 -0500904 if (!(png_ptr->zstream.avail_out))
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600905 {
906 /* write the IDAT and reset the zlib output buffer */
907 png_write_IDAT(png_ptr, png_ptr->zbuf,
908 png_ptr->zbuf_size);
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600909 png_ptr->zstream.next_out = png_ptr->zbuf;
910 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600911 wrote_IDAT = 1;
912 }
913 } while(wrote_IDAT == 1);
Guy Schalnat0f716451995-11-28 11:22:13 -0600914
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600915 /* If there is any data left to be output, write it into a new IDAT */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600916 if (png_ptr->zbuf_size != png_ptr->zstream.avail_out)
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600917 {
918 /* write the IDAT and reset the zlib output buffer */
919 png_write_IDAT(png_ptr, png_ptr->zbuf,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600920 png_ptr->zbuf_size - png_ptr->zstream.avail_out);
921 png_ptr->zstream.next_out = png_ptr->zbuf;
922 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600923 }
924 png_ptr->flush_rows = 0;
925 png_flush(png_ptr);
Guy Schalnat0f716451995-11-28 11:22:13 -0600926}
927#endif /* PNG_WRITE_FLUSH_SUPPORTED */
928
Guy Schalnate5a37791996-06-05 15:50:50 -0500929/* free all memory used by the write */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500930void PNGAPI
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600931png_destroy_write_struct(png_structpp png_ptr_ptr, png_infopp info_ptr_ptr)
Guy Schalnate5a37791996-06-05 15:50:50 -0500932{
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600933 png_structp png_ptr = NULL;
934 png_infop info_ptr = NULL;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500935#ifdef PNG_USER_MEM_SUPPORTED
936 png_free_ptr free_fn = NULL;
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500937 png_voidp mem_ptr = NULL;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500938#endif
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600939
Andreas Dilger47a0c421997-05-16 02:46:07 -0500940 png_debug(1, "in png_destroy_write_struct\n");
941 if (png_ptr_ptr != NULL)
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500942 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600943 png_ptr = *png_ptr_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500944#ifdef PNG_USER_MEM_SUPPORTED
945 free_fn = png_ptr->free_fn;
Glenn Randers-Pehrsonfcbd7872002-04-07 16:35:38 -0500946 mem_ptr = png_ptr->mem_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500947#endif
948 }
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600949
Andreas Dilger47a0c421997-05-16 02:46:07 -0500950 if (info_ptr_ptr != NULL)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600951 info_ptr = *info_ptr_ptr;
952
Andreas Dilger47a0c421997-05-16 02:46:07 -0500953 if (info_ptr != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500954 {
Glenn Randers-Pehrsona77ef622000-02-18 13:48:52 -0600955 png_free_data(png_ptr, info_ptr, PNG_FREE_ALL, -1);
956
Glenn Randers-Pehrson6942d532000-05-01 09:31:54 -0500957#if defined(PNG_UNKNOWN_CHUNKS_SUPPORTED)
Glenn Randers-Pehrsona77ef622000-02-18 13:48:52 -0600958 if (png_ptr->num_chunk_list)
959 {
960 png_free(png_ptr, png_ptr->chunk_list);
Glenn Randers-Pehrson4766a242000-07-17 06:17:09 -0500961 png_ptr->chunk_list=NULL;
Glenn Randers-Pehrsona77ef622000-02-18 13:48:52 -0600962 png_ptr->num_chunk_list=0;
963 }
Glenn Randers-Pehrson6942d532000-05-01 09:31:54 -0500964#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600965
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500966#ifdef PNG_USER_MEM_SUPPORTED
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500967 png_destroy_struct_2((png_voidp)info_ptr, (png_free_ptr)free_fn,
968 (png_voidp)mem_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500969#else
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600970 png_destroy_struct((png_voidp)info_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500971#endif
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500972 *info_ptr_ptr = NULL;
Guy Schalnate5a37791996-06-05 15:50:50 -0500973 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600974
Andreas Dilger47a0c421997-05-16 02:46:07 -0500975 if (png_ptr != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500976 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600977 png_write_destroy(png_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500978#ifdef PNG_USER_MEM_SUPPORTED
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500979 png_destroy_struct_2((png_voidp)png_ptr, (png_free_ptr)free_fn,
980 (png_voidp)mem_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500981#else
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600982 png_destroy_struct((png_voidp)png_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500983#endif
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500984 *png_ptr_ptr = NULL;
Guy Schalnate5a37791996-06-05 15:50:50 -0500985 }
986}
987
988
Andreas Dilger47a0c421997-05-16 02:46:07 -0500989/* Free any memory used in png_ptr struct (old method) */
Glenn Randers-Pehrsonf64a06f2001-04-11 07:38:00 -0500990void /* PRIVATE */
Guy Schalnat6d764711995-12-19 03:22:19 -0600991png_write_destroy(png_structp png_ptr)
Guy Schalnat0d580581995-07-20 02:43:20 -0500992{
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600993#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600994 jmp_buf tmp_jmp; /* save jump buffer */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600995#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500996 png_error_ptr error_fn;
997 png_error_ptr warning_fn;
998 png_voidp error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500999#ifdef PNG_USER_MEM_SUPPORTED
1000 png_free_ptr free_fn;
1001#endif
Guy Schalnat0d580581995-07-20 02:43:20 -05001002
Andreas Dilger47a0c421997-05-16 02:46:07 -05001003 png_debug(1, "in png_write_destroy\n");
Guy Schalnat0d580581995-07-20 02:43:20 -05001004 /* free any memory zlib uses */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -06001005 deflateEnd(&png_ptr->zstream);
Guy Schalnate5a37791996-06-05 15:50:50 -05001006
Guy Schalnat0d580581995-07-20 02:43:20 -05001007 /* free our memory. png_free checks NULL for us. */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -06001008 png_free(png_ptr, png_ptr->zbuf);
1009 png_free(png_ptr, png_ptr->row_buf);
1010 png_free(png_ptr, png_ptr->prev_row);
1011 png_free(png_ptr, png_ptr->sub_row);
1012 png_free(png_ptr, png_ptr->up_row);
1013 png_free(png_ptr, png_ptr->avg_row);
1014 png_free(png_ptr, png_ptr->paeth_row);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001015
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -06001016#if defined(PNG_TIME_RFC1123_SUPPORTED)
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06001017 png_free(png_ptr, png_ptr->time_buffer);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001018#endif
1019
Andreas Dilger47a0c421997-05-16 02:46:07 -05001020#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
1021 png_free(png_ptr, png_ptr->prev_filters);
1022 png_free(png_ptr, png_ptr->filter_weights);
1023 png_free(png_ptr, png_ptr->inv_filter_weights);
1024 png_free(png_ptr, png_ptr->filter_costs);
1025 png_free(png_ptr, png_ptr->inv_filter_costs);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001026#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001027
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001028#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnat0d580581995-07-20 02:43:20 -05001029 /* reset structure */
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -05001030 png_memcpy(tmp_jmp, png_ptr->jmpbuf, sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001031#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001032
1033 error_fn = png_ptr->error_fn;
1034 warning_fn = png_ptr->warning_fn;
1035 error_ptr = png_ptr->error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -05001036#ifdef PNG_USER_MEM_SUPPORTED
1037 free_fn = png_ptr->free_fn;
1038#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001039
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -05001040 png_memset(png_ptr, 0, sizeof(png_struct));
Guy Schalnate5a37791996-06-05 15:50:50 -05001041
1042 png_ptr->error_fn = error_fn;
1043 png_ptr->warning_fn = warning_fn;
1044 png_ptr->error_ptr = error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -05001045#ifdef PNG_USER_MEM_SUPPORTED
1046 png_ptr->free_fn = free_fn;
1047#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001048
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001049#ifdef PNG_SETJMP_SUPPORTED
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -05001050 png_memcpy(png_ptr->jmpbuf, tmp_jmp, sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001051#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001052}
Guy Schalnate5a37791996-06-05 15:50:50 -05001053
Andreas Dilger47a0c421997-05-16 02:46:07 -05001054/* Allow the application to select one or more row filters to use. */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001055void PNGAPI
Guy Schalnate5a37791996-06-05 15:50:50 -05001056png_set_filter(png_structp png_ptr, int method, int filters)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001057{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001058 png_debug(1, "in png_set_filter\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001059 if (png_ptr == NULL)
1060 return;
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -06001061#if defined(PNG_MNG_FEATURES_SUPPORTED)
Glenn Randers-Pehrsonf05f8032000-12-23 14:27:39 -06001062 if((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -06001063 (method == PNG_INTRAPIXEL_DIFFERENCING))
1064 method = PNG_FILTER_TYPE_BASE;
1065#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -05001066 if (method == PNG_FILTER_TYPE_BASE)
Guy Schalnate5a37791996-06-05 15:50:50 -05001067 {
1068 switch (filters & (PNG_ALL_FILTERS | 0x07))
1069 {
1070 case 5:
1071 case 6:
Andreas Dilger47a0c421997-05-16 02:46:07 -05001072 case 7: png_warning(png_ptr, "Unknown row filter for method 0");
1073 case PNG_FILTER_VALUE_NONE: png_ptr->do_filter=PNG_FILTER_NONE; break;
1074 case PNG_FILTER_VALUE_SUB: png_ptr->do_filter=PNG_FILTER_SUB; break;
1075 case PNG_FILTER_VALUE_UP: png_ptr->do_filter=PNG_FILTER_UP; break;
1076 case PNG_FILTER_VALUE_AVG: png_ptr->do_filter=PNG_FILTER_AVG; break;
1077 case PNG_FILTER_VALUE_PAETH: png_ptr->do_filter=PNG_FILTER_PAETH;break;
Guy Schalnate5a37791996-06-05 15:50:50 -05001078 default: png_ptr->do_filter = (png_byte)filters; break;
1079 }
1080
Andreas Dilger47a0c421997-05-16 02:46:07 -05001081 /* If we have allocated the row_buf, this means we have already started
1082 * with the image and we should have allocated all of the filter buffers
1083 * that have been selected. If prev_row isn't already allocated, then
1084 * it is too late to start using the filters that need it, since we
1085 * will be missing the data in the previous row. If an application
1086 * wants to start and stop using particular filters during compression,
1087 * it should start out with all of the filters, and then add and
1088 * remove them after the start of compression.
Guy Schalnate5a37791996-06-05 15:50:50 -05001089 */
Andreas Dilger47a0c421997-05-16 02:46:07 -05001090 if (png_ptr->row_buf != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001091 {
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001092 if ((png_ptr->do_filter & PNG_FILTER_SUB) && png_ptr->sub_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001093 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001094 png_ptr->sub_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001095 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001096 png_ptr->sub_row[0] = PNG_FILTER_VALUE_SUB;
Guy Schalnate5a37791996-06-05 15:50:50 -05001097 }
1098
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001099 if ((png_ptr->do_filter & PNG_FILTER_UP) && png_ptr->up_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001100 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001101 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001102 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001103 png_warning(png_ptr, "Can't add Up filter after starting");
Guy Schalnate5a37791996-06-05 15:50:50 -05001104 png_ptr->do_filter &= ~PNG_FILTER_UP;
1105 }
1106 else
1107 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001108 png_ptr->up_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001109 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001110 png_ptr->up_row[0] = PNG_FILTER_VALUE_UP;
Guy Schalnate5a37791996-06-05 15:50:50 -05001111 }
1112 }
1113
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001114 if ((png_ptr->do_filter & PNG_FILTER_AVG) && png_ptr->avg_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001115 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001116 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001117 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001118 png_warning(png_ptr, "Can't add Average filter after starting");
Guy Schalnate5a37791996-06-05 15:50:50 -05001119 png_ptr->do_filter &= ~PNG_FILTER_AVG;
1120 }
1121 else
1122 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001123 png_ptr->avg_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001124 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001125 png_ptr->avg_row[0] = PNG_FILTER_VALUE_AVG;
Guy Schalnate5a37791996-06-05 15:50:50 -05001126 }
1127 }
1128
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001129 if ((png_ptr->do_filter & PNG_FILTER_PAETH) &&
Andreas Dilger47a0c421997-05-16 02:46:07 -05001130 png_ptr->paeth_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001131 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001132 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001133 {
1134 png_warning(png_ptr, "Can't add Paeth filter after starting");
Glenn Randers-Pehrson860ab2b1999-10-14 07:43:10 -05001135 png_ptr->do_filter &= (png_byte)(~PNG_FILTER_PAETH);
Guy Schalnate5a37791996-06-05 15:50:50 -05001136 }
1137 else
1138 {
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -06001139 png_ptr->paeth_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001140 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001141 png_ptr->paeth_row[0] = PNG_FILTER_VALUE_PAETH;
Guy Schalnate5a37791996-06-05 15:50:50 -05001142 }
1143 }
1144
1145 if (png_ptr->do_filter == PNG_NO_FILTERS)
1146 png_ptr->do_filter = PNG_FILTER_NONE;
1147 }
1148 }
1149 else
Andreas Dilger47a0c421997-05-16 02:46:07 -05001150 png_error(png_ptr, "Unknown custom filter method");
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001151}
1152
Andreas Dilger47a0c421997-05-16 02:46:07 -05001153/* This allows us to influence the way in which libpng chooses the "best"
1154 * filter for the current scanline. While the "minimum-sum-of-absolute-
1155 * differences metric is relatively fast and effective, there is some
1156 * question as to whether it can be improved upon by trying to keep the
1157 * filtered data going to zlib more consistent, hopefully resulting in
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06001158 * better compression.
1159 */
Andreas Dilger47a0c421997-05-16 02:46:07 -05001160#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED) /* GRR 970116 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001161void PNGAPI
Andreas Dilger47a0c421997-05-16 02:46:07 -05001162png_set_filter_heuristics(png_structp png_ptr, int heuristic_method,
1163 int num_weights, png_doublep filter_weights,
1164 png_doublep filter_costs)
1165{
1166 int i;
1167
1168 png_debug(1, "in png_set_filter_heuristics\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001169 if (png_ptr == NULL)
1170 return;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001171 if (heuristic_method >= PNG_FILTER_HEURISTIC_LAST)
1172 {
1173 png_warning(png_ptr, "Unknown filter heuristic method");
1174 return;
1175 }
1176
1177 if (heuristic_method == PNG_FILTER_HEURISTIC_DEFAULT)
1178 {
1179 heuristic_method = PNG_FILTER_HEURISTIC_UNWEIGHTED;
1180 }
1181
1182 if (num_weights < 0 || filter_weights == NULL ||
1183 heuristic_method == PNG_FILTER_HEURISTIC_UNWEIGHTED)
1184 {
1185 num_weights = 0;
1186 }
1187
Glenn Randers-Pehrson860ab2b1999-10-14 07:43:10 -05001188 png_ptr->num_prev_filters = (png_byte)num_weights;
1189 png_ptr->heuristic_method = (png_byte)heuristic_method;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001190
1191 if (num_weights > 0)
1192 {
1193 if (png_ptr->prev_filters == NULL)
1194 {
1195 png_ptr->prev_filters = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -05001196 (png_size_t)(sizeof(png_byte) * num_weights));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001197
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001198 /* To make sure that the weighting starts out fairly */
1199 for (i = 0; i < num_weights; i++)
Andreas Dilger47a0c421997-05-16 02:46:07 -05001200 {
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001201 png_ptr->prev_filters[i] = 255;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001202 }
1203 }
1204
1205 if (png_ptr->filter_weights == NULL)
1206 {
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001207 png_ptr->filter_weights = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -05001208 (png_size_t)(sizeof(png_uint_16) * num_weights));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001209
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001210 png_ptr->inv_filter_weights = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -05001211 (png_size_t)(sizeof(png_uint_16) * num_weights));
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001212 for (i = 0; i < num_weights; i++)
Andreas Dilger47a0c421997-05-16 02:46:07 -05001213 {
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001214 png_ptr->inv_filter_weights[i] =
1215 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001216 }
1217 }
1218
1219 for (i = 0; i < num_weights; i++)
1220 {
1221 if (filter_weights[i] < 0.0)
1222 {
1223 png_ptr->inv_filter_weights[i] =
1224 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
1225 }
1226 else
1227 {
1228 png_ptr->inv_filter_weights[i] =
1229 (png_uint_16)((double)PNG_WEIGHT_FACTOR*filter_weights[i]+0.5);
1230 png_ptr->filter_weights[i] =
1231 (png_uint_16)((double)PNG_WEIGHT_FACTOR/filter_weights[i]+0.5);
1232 }
1233 }
1234 }
1235
1236 /* If, in the future, there are other filter methods, this would
1237 * need to be based on png_ptr->filter.
1238 */
1239 if (png_ptr->filter_costs == NULL)
1240 {
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001241 png_ptr->filter_costs = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -05001242 (sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001243
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001244 png_ptr->inv_filter_costs = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -05001245 (sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001246
1247 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
1248 {
1249 png_ptr->inv_filter_costs[i] =
1250 png_ptr->filter_costs[i] = PNG_COST_FACTOR;
1251 }
1252 }
1253
1254 /* Here is where we set the relative costs of the different filters. We
1255 * should take the desired compression level into account when setting
1256 * the costs, so that Paeth, for instance, has a high relative cost at low
1257 * compression levels, while it has a lower relative cost at higher
1258 * compression settings. The filter types are in order of increasing
1259 * relative cost, so it would be possible to do this with an algorithm.
1260 */
1261 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
1262 {
1263 if (filter_costs == NULL || filter_costs[i] < 0.0)
1264 {
1265 png_ptr->inv_filter_costs[i] =
1266 png_ptr->filter_costs[i] = PNG_COST_FACTOR;
1267 }
1268 else if (filter_costs[i] >= 1.0)
1269 {
1270 png_ptr->inv_filter_costs[i] =
1271 (png_uint_16)((double)PNG_COST_FACTOR / filter_costs[i] + 0.5);
1272 png_ptr->filter_costs[i] =
1273 (png_uint_16)((double)PNG_COST_FACTOR * filter_costs[i] + 0.5);
1274 }
1275 }
1276}
1277#endif /* PNG_WRITE_WEIGHTED_FILTER_SUPPORTED */
1278
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001279void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001280png_set_compression_level(png_structp png_ptr, int level)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001281{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001282 png_debug(1, "in png_set_compression_level\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001283 if (png_ptr == NULL)
1284 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001285 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_LEVEL;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001286 png_ptr->zlib_level = level;
1287}
1288
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001289void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001290png_set_compression_mem_level(png_structp png_ptr, int mem_level)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001291{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001292 png_debug(1, "in png_set_compression_mem_level\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001293 if (png_ptr == NULL)
1294 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001295 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_MEM_LEVEL;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -06001296 png_ptr->zlib_mem_level = mem_level;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001297}
1298
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001299void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001300png_set_compression_strategy(png_structp png_ptr, int strategy)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001301{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001302 png_debug(1, "in png_set_compression_strategy\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001303 if (png_ptr == NULL)
1304 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001305 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_STRATEGY;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001306 png_ptr->zlib_strategy = strategy;
1307}
1308
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001309void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001310png_set_compression_window_bits(png_structp png_ptr, int window_bits)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001311{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001312 if (png_ptr == NULL)
1313 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001314 if (window_bits > 15)
1315 png_warning(png_ptr, "Only compression windows <= 32k supported by PNG");
Glenn Randers-Pehrson4393a9a1999-09-17 12:27:26 -05001316 else if (window_bits < 8)
1317 png_warning(png_ptr, "Only compression windows >= 256 supported by PNG");
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001318#ifndef WBITS_8_OK
1319 /* avoid libpng bug with 256-byte windows */
1320 if (window_bits == 8)
1321 {
1322 png_warning(png_ptr, "Compression window is being reset to 512");
1323 window_bits=9;
1324 }
1325#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001326 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_WINDOW_BITS;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001327 png_ptr->zlib_window_bits = window_bits;
1328}
1329
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001330void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001331png_set_compression_method(png_structp png_ptr, int method)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001332{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001333 png_debug(1, "in png_set_compression_method\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001334 if (png_ptr == NULL)
1335 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001336 if (method != 8)
1337 png_warning(png_ptr, "Only compression method 8 is supported by PNG");
1338 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_METHOD;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001339 png_ptr->zlib_method = method;
Guy Schalnat0d580581995-07-20 02:43:20 -05001340}
1341
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001342void PNGAPI
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001343png_set_write_status_fn(png_structp png_ptr, png_write_status_ptr write_row_fn)
1344{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001345 if (png_ptr == NULL)
1346 return;
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001347 png_ptr->write_row_fn = write_row_fn;
1348}
1349
1350#if defined(PNG_WRITE_USER_TRANSFORM_SUPPORTED)
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001351void PNGAPI
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001352png_set_write_user_transform_fn(png_structp png_ptr, png_user_transform_ptr
1353 write_user_transform_fn)
1354{
1355 png_debug(1, "in png_set_write_user_transform_fn\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001356 if (png_ptr == NULL)
1357 return;
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001358 png_ptr->transformations |= PNG_USER_TRANSFORM;
1359 png_ptr->write_user_transform_fn = write_user_transform_fn;
1360}
1361#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001362
1363
1364#if defined(PNG_INFO_IMAGE_SUPPORTED)
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001365void PNGAPI
1366png_write_png(png_structp png_ptr, png_infop info_ptr,
Glenn Randers-Pehrson1ef65b62000-05-12 06:19:53 -05001367 int transforms, voidp params)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001368{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001369 if (png_ptr == NULL || info_ptr == NULL)
1370 return;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001371#if defined(PNG_WRITE_INVERT_ALPHA_SUPPORTED)
1372 /* invert the alpha channel from opacity to transparency */
1373 if (transforms & PNG_TRANSFORM_INVERT_ALPHA)
1374 png_set_invert_alpha(png_ptr);
1375#endif
1376
1377 /* Write the file header information. */
1378 png_write_info(png_ptr, info_ptr);
1379
1380 /* ------ these transformations don't touch the info structure ------- */
1381
1382#if defined(PNG_WRITE_INVERT_SUPPORTED)
1383 /* invert monochrome pixels */
1384 if (transforms & PNG_TRANSFORM_INVERT_MONO)
1385 png_set_invert_mono(png_ptr);
1386#endif
1387
1388#if defined(PNG_WRITE_SHIFT_SUPPORTED)
1389 /* Shift the pixels up to a legal bit depth and fill in
1390 * as appropriate to correctly scale the image.
1391 */
1392 if ((transforms & PNG_TRANSFORM_SHIFT)
1393 && (info_ptr->valid & PNG_INFO_sBIT))
1394 png_set_shift(png_ptr, &info_ptr->sig_bit);
1395#endif
1396
1397#if defined(PNG_WRITE_PACK_SUPPORTED)
1398 /* pack pixels into bytes */
1399 if (transforms & PNG_TRANSFORM_PACKING)
1400 png_set_packing(png_ptr);
1401#endif
1402
1403#if defined(PNG_WRITE_SWAP_ALPHA_SUPPORTED)
1404 /* swap location of alpha bytes from ARGB to RGBA */
1405 if (transforms & PNG_TRANSFORM_SWAP_ALPHA)
1406 png_set_swap_alpha(png_ptr);
1407#endif
1408
1409#if defined(PNG_WRITE_FILLER_SUPPORTED)
1410 /* Get rid of filler (OR ALPHA) bytes, pack XRGB/RGBX/ARGB/RGBA into
1411 * RGB (4 channels -> 3 channels). The second parameter is not used.
1412 */
1413 if (transforms & PNG_TRANSFORM_STRIP_FILLER)
1414 png_set_filler(png_ptr, 0, PNG_FILLER_BEFORE);
1415#endif
1416
1417#if defined(PNG_WRITE_BGR_SUPPORTED)
1418 /* flip BGR pixels to RGB */
1419 if (transforms & PNG_TRANSFORM_BGR)
1420 png_set_bgr(png_ptr);
1421#endif
1422
1423#if defined(PNG_WRITE_SWAP_SUPPORTED)
1424 /* swap bytes of 16-bit files to most significant byte first */
1425 if (transforms & PNG_TRANSFORM_SWAP_ENDIAN)
1426 png_set_swap(png_ptr);
1427#endif
1428
1429#if defined(PNG_WRITE_PACKSWAP_SUPPORTED)
1430 /* swap bits of 1, 2, 4 bit packed pixel formats */
1431 if (transforms & PNG_TRANSFORM_PACKSWAP)
1432 png_set_packswap(png_ptr);
1433#endif
1434
1435 /* ----------------------- end of transformations ------------------- */
1436
1437 /* write the bits */
1438 if (info_ptr->valid & PNG_INFO_IDAT)
1439 png_write_image(png_ptr, info_ptr->row_pointers);
1440
1441 /* It is REQUIRED to call this to finish writing the rest of the file */
1442 png_write_end(png_ptr, info_ptr);
Glenn Randers-Pehrson520a7642000-03-21 05:13:06 -06001443
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001444 if(transforms == 0 || params == NULL)
Glenn Randers-Pehrson520a7642000-03-21 05:13:06 -06001445 /* quiet compiler warnings */ return;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001446}
1447#endif
Glenn Randers-Pehrson19095602001-03-14 07:08:39 -06001448#endif /* PNG_WRITE_SUPPORTED */