blob: 0eb5f3c966ff6bff1f1fa306cf6d85c2d0519c16 [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-Pehrson895a9c92008-07-25 08:51:18 -05004 * Last changed in libpng 1.4.0 [July 25, 2008]
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06005 * For conditions of distribution and use, see copyright notice in png.h
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05006 * Copyright (c) 1998-2008 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)
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -050035 if ((png_ptr->mode&PNG_HAVE_PNG_SIGNATURE)&&(png_ptr->mng_features_permitted))
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -060036 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -050037 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-Pehrson145f5c82008-07-10 09:13:13 -0500115 if (up->size == 0)
116 png_warning(png_ptr, "Writing zero-length unknown chunk");
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600117 png_write_chunk(png_ptr, up->name, up->data, up->size);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600118 }
119 }
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600120 }
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500121#endif
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600122 png_ptr->mode |= PNG_WROTE_INFO_BEFORE_PLTE;
123 }
124}
125
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500126void PNGAPI
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600127png_write_info(png_structp png_ptr, png_infop info_ptr)
128{
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600129#if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_sPLT_SUPPORTED)
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600130 int i;
131#endif
132
133 png_debug(1, "in png_write_info\n");
134
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600135 if (png_ptr == NULL || info_ptr == NULL)
136 return;
137
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600138 png_write_info_before_PLTE(png_ptr, info_ptr);
139
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600140 if (info_ptr->valid & PNG_INFO_PLTE)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500141 png_write_PLTE(png_ptr, info_ptr->palette,
142 (png_uint_32)info_ptr->num_palette);
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600143 else if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600144 png_error(png_ptr, "Valid palette required for paletted images");
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600145
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500146#if defined(PNG_WRITE_tRNS_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600147 if (info_ptr->valid & PNG_INFO_tRNS)
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600148 {
149#if defined(PNG_WRITE_INVERT_ALPHA_SUPPORTED)
150 /* invert the alpha channel (in tRNS) */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600151 if ((png_ptr->transformations & PNG_INVERT_ALPHA) &&
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600152 info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
153 {
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600154 int j;
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600155 for (j=0; j<(int)info_ptr->num_trans; j++)
Glenn Randers-Pehrson860ab2b1999-10-14 07:43:10 -0500156 info_ptr->trans[j] = (png_byte)(255 - info_ptr->trans[j]);
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600157 }
158#endif
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600159 png_write_tRNS(png_ptr, info_ptr->trans, &(info_ptr->trans_values),
160 info_ptr->num_trans, info_ptr->color_type);
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600161 }
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500162#endif
163#if defined(PNG_WRITE_bKGD_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600164 if (info_ptr->valid & PNG_INFO_bKGD)
165 png_write_bKGD(png_ptr, &(info_ptr->background), info_ptr->color_type);
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500166#endif
167#if defined(PNG_WRITE_hIST_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600168 if (info_ptr->valid & PNG_INFO_hIST)
169 png_write_hIST(png_ptr, info_ptr->hist, info_ptr->num_palette);
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500170#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500171#if defined(PNG_WRITE_oFFs_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600172 if (info_ptr->valid & PNG_INFO_oFFs)
173 png_write_oFFs(png_ptr, info_ptr->x_offset, info_ptr->y_offset,
174 info_ptr->offset_unit_type);
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500175#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500176#if defined(PNG_WRITE_pCAL_SUPPORTED)
177 if (info_ptr->valid & PNG_INFO_pCAL)
178 png_write_pCAL(png_ptr, info_ptr->pcal_purpose, info_ptr->pcal_X0,
179 info_ptr->pcal_X1, info_ptr->pcal_type, info_ptr->pcal_nparams,
180 info_ptr->pcal_units, info_ptr->pcal_params);
181#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600182#if defined(PNG_WRITE_sCAL_SUPPORTED)
183 if (info_ptr->valid & PNG_INFO_sCAL)
Glenn Randers-Pehrson68ea2432000-04-01 21:10:05 -0600184#if defined(PNG_FLOATING_POINT_SUPPORTED) && !defined(PNG_NO_STDIO)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600185 png_write_sCAL(png_ptr, (int)info_ptr->scal_unit,
186 info_ptr->scal_pixel_width, info_ptr->scal_pixel_height);
187#else
188#ifdef PNG_FIXED_POINT_SUPPORTED
189 png_write_sCAL_s(png_ptr, (int)info_ptr->scal_unit,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600190 info_ptr->scal_s_width, info_ptr->scal_s_height);
Glenn Randers-Pehrson68ea2432000-04-01 21:10:05 -0600191#else
192 png_warning(png_ptr,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500193 "png_write_sCAL not supported; sCAL chunk not written");
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600194#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600195#endif
196#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500197#if defined(PNG_WRITE_pHYs_SUPPORTED)
198 if (info_ptr->valid & PNG_INFO_pHYs)
199 png_write_pHYs(png_ptr, info_ptr->x_pixels_per_unit,
200 info_ptr->y_pixels_per_unit, info_ptr->phys_unit_type);
201#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500202#if defined(PNG_WRITE_tIME_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600203 if (info_ptr->valid & PNG_INFO_tIME)
Guy Schalnate5a37791996-06-05 15:50:50 -0500204 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600205 png_write_tIME(png_ptr, &(info_ptr->mod_time));
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600206 png_ptr->mode |= PNG_WROTE_tIME;
Guy Schalnate5a37791996-06-05 15:50:50 -0500207 }
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500208#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600209#if defined(PNG_WRITE_sPLT_SUPPORTED)
210 if (info_ptr->valid & PNG_INFO_sPLT)
211 for (i = 0; i < (int)info_ptr->splt_palettes_num; i++)
212 png_write_sPLT(png_ptr, info_ptr->splt_palettes + i);
213#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600214#if defined(PNG_WRITE_TEXT_SUPPORTED)
Guy Schalnate5a37791996-06-05 15:50:50 -0500215 /* Check to see if we need to write text chunks */
Andreas Dilger47a0c421997-05-16 02:46:07 -0500216 for (i = 0; i < info_ptr->num_text; i++)
Guy Schalnat0d580581995-07-20 02:43:20 -0500217 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500218 png_debug2(2, "Writing header text chunk %d, type %d\n", i,
219 info_ptr->text[i].compression);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600220 /* an internationalized chunk? */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600221 if (info_ptr->text[i].compression > 0)
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600222 {
223#if defined(PNG_WRITE_iTXt_SUPPORTED)
224 /* write international chunk */
225 png_write_iTXt(png_ptr,
226 info_ptr->text[i].compression,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600227 info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600228 info_ptr->text[i].lang,
229 info_ptr->text[i].lang_key,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600230 info_ptr->text[i].text);
231#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600232 png_warning(png_ptr, "Unable to write international text");
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600233#endif
234 /* Mark this chunk as written */
235 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
236 }
Andreas Dilger47a0c421997-05-16 02:46:07 -0500237 /* If we want a compressed text chunk */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600238 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_zTXt)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500239 {
240#if defined(PNG_WRITE_zTXt_SUPPORTED)
241 /* write compressed chunk */
242 png_write_zTXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600243 info_ptr->text[i].text, 0,
Andreas Dilger47a0c421997-05-16 02:46:07 -0500244 info_ptr->text[i].compression);
245#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600246 png_warning(png_ptr, "Unable to write compressed text");
Andreas Dilger47a0c421997-05-16 02:46:07 -0500247#endif
248 /* Mark this chunk as written */
249 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
250 }
251 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
252 {
253#if defined(PNG_WRITE_tEXt_SUPPORTED)
254 /* write uncompressed chunk */
255 png_write_tEXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600256 info_ptr->text[i].text,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600257 0);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500258#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600259 png_warning(png_ptr, "Unable to write uncompressed text");
Andreas Dilger47a0c421997-05-16 02:46:07 -0500260#endif
261 /* Mark this chunk as written */
262 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
263 }
264 }
265#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600266#if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
267 if (info_ptr->unknown_chunks_num)
268 {
269 png_unknown_chunk *up;
270
271 png_debug(5, "writing extra chunks\n");
272
273 for (up = info_ptr->unknown_chunks;
274 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
275 up++)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600276 {
277 int keep=png_handle_as_unknown(png_ptr, up->name);
Glenn Randers-Pehrsondff799e2004-08-07 21:42:49 -0500278 if (keep != PNG_HANDLE_CHUNK_NEVER &&
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600279 up->location && (up->location & PNG_HAVE_PLTE) &&
280 !(up->location & PNG_HAVE_IDAT) &&
Glenn Randers-Pehrsondff799e2004-08-07 21:42:49 -0500281 ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS ||
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600282 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
283 {
284 png_write_chunk(png_ptr, up->name, up->data, up->size);
285 }
286 }
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600287 }
288#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500289}
Guy Schalnat0d580581995-07-20 02:43:20 -0500290
Andreas Dilger47a0c421997-05-16 02:46:07 -0500291/* Writes the end of the PNG file. If you don't want to write comments or
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600292 * time information, you can pass NULL for info. If you already wrote these
293 * in png_write_info(), do not write them again here. If you have long
294 * comments, I suggest writing them here, and compressing them.
295 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500296void PNGAPI
Andreas Dilger47a0c421997-05-16 02:46:07 -0500297png_write_end(png_structp png_ptr, png_infop info_ptr)
298{
299 png_debug(1, "in png_write_end\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600300 if (png_ptr == NULL)
301 return;
Andreas Dilger47a0c421997-05-16 02:46:07 -0500302 if (!(png_ptr->mode & PNG_HAVE_IDAT))
303 png_error(png_ptr, "No IDATs written into file");
304
305 /* see if user wants us to write information chunks */
306 if (info_ptr != NULL)
307 {
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600308#if defined(PNG_WRITE_TEXT_SUPPORTED)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500309 int i; /* local index variable */
Glenn Randers-Pehrson70e3f541998-01-03 22:40:55 -0600310#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500311#if defined(PNG_WRITE_tIME_SUPPORTED)
312 /* check to see if user has supplied a time chunk */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600313 if ((info_ptr->valid & PNG_INFO_tIME) &&
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600314 !(png_ptr->mode & PNG_WROTE_tIME))
Andreas Dilger47a0c421997-05-16 02:46:07 -0500315 png_write_tIME(png_ptr, &(info_ptr->mod_time));
316#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600317#if defined(PNG_WRITE_TEXT_SUPPORTED)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500318 /* loop through comment chunks */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600319 for (i = 0; i < info_ptr->num_text; i++)
Guy Schalnat0d580581995-07-20 02:43:20 -0500320 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500321 png_debug2(2, "Writing trailer text chunk %d, type %d\n", i,
322 info_ptr->text[i].compression);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600323 /* an internationalized chunk? */
324 if (info_ptr->text[i].compression > 0)
325 {
326#if defined(PNG_WRITE_iTXt_SUPPORTED)
327 /* write international chunk */
328 png_write_iTXt(png_ptr,
329 info_ptr->text[i].compression,
330 info_ptr->text[i].key,
331 info_ptr->text[i].lang,
332 info_ptr->text[i].lang_key,
333 info_ptr->text[i].text);
334#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600335 png_warning(png_ptr, "Unable to write international text");
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600336#endif
337 /* Mark this chunk as written */
338 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
339 }
340 else if (info_ptr->text[i].compression >= PNG_TEXT_COMPRESSION_zTXt)
Guy Schalnat0d580581995-07-20 02:43:20 -0500341 {
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500342#if defined(PNG_WRITE_zTXt_SUPPORTED)
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600343 /* write compressed chunk */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600344 png_write_zTXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600345 info_ptr->text[i].text, 0,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600346 info_ptr->text[i].compression);
Guy Schalnate5a37791996-06-05 15:50:50 -0500347#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600348 png_warning(png_ptr, "Unable to write compressed text");
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500349#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500350 /* Mark this chunk as written */
351 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
Guy Schalnat0d580581995-07-20 02:43:20 -0500352 }
Andreas Dilger47a0c421997-05-16 02:46:07 -0500353 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
Guy Schalnat0d580581995-07-20 02:43:20 -0500354 {
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500355#if defined(PNG_WRITE_tEXt_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500356 /* write uncompressed chunk */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600357 png_write_tEXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600358 info_ptr->text[i].text, 0);
Guy Schalnate5a37791996-06-05 15:50:50 -0500359#else
Glenn Randers-Pehrsonc3d51c12006-03-02 07:23:18 -0600360 png_warning(png_ptr, "Unable to write uncompressed text");
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500361#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500362
Andreas Dilger47a0c421997-05-16 02:46:07 -0500363 /* Mark this chunk as written */
364 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
Guy Schalnat0d580581995-07-20 02:43:20 -0500365 }
366 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600367#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600368#if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
369 if (info_ptr->unknown_chunks_num)
370 {
371 png_unknown_chunk *up;
372
373 png_debug(5, "writing extra chunks\n");
374
375 for (up = info_ptr->unknown_chunks;
376 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
377 up++)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600378 {
379 int keep=png_handle_as_unknown(png_ptr, up->name);
Glenn Randers-Pehrsondff799e2004-08-07 21:42:49 -0500380 if (keep != PNG_HANDLE_CHUNK_NEVER &&
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600381 up->location && (up->location & PNG_AFTER_IDAT) &&
Glenn Randers-Pehrsondff799e2004-08-07 21:42:49 -0500382 ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS ||
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600383 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
384 {
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600385 png_write_chunk(png_ptr, up->name, up->data, up->size);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600386 }
387 }
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600388 }
389#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500390 }
Guy Schalnate5a37791996-06-05 15:50:50 -0500391
392 png_ptr->mode |= PNG_AFTER_IDAT;
393
Andreas Dilger47a0c421997-05-16 02:46:07 -0500394 /* write end of PNG file */
Guy Schalnat0d580581995-07-20 02:43:20 -0500395 png_write_IEND(png_ptr);
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500396 /* This flush, added in libpng-1.0.8, removed from libpng-1.0.9beta03,
397 * and restored again in libpng-1.2.30, may cause some applications that
398 * do not set png_ptr->output_flush_fn to crash. If your application
399 * experiences this problem, please report the event to
400 * png-mng-implement at lists.sf.net .
401 */
Glenn Randers-Pehrson32fc5ce2000-07-24 06:34:14 -0500402 png_flush(png_ptr);
Guy Schalnat0d580581995-07-20 02:43:20 -0500403}
404
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500405#if defined(PNG_WRITE_tIME_SUPPORTED)
Glenn Randers-Pehrson316f97a2000-07-08 13:19:41 -0500406/* "time.h" functions are not supported on WindowsCE */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500407void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600408png_convert_from_struct_tm(png_timep ptime, struct tm FAR * ttime)
Guy Schalnat0d580581995-07-20 02:43:20 -0500409{
Andreas Dilger47a0c421997-05-16 02:46:07 -0500410 png_debug(1, "in png_convert_from_struct_tm\n");
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600411 ptime->year = (png_uint_16)(1900 + ttime->tm_year);
412 ptime->month = (png_byte)(ttime->tm_mon + 1);
413 ptime->day = (png_byte)ttime->tm_mday;
414 ptime->hour = (png_byte)ttime->tm_hour;
415 ptime->minute = (png_byte)ttime->tm_min;
416 ptime->second = (png_byte)ttime->tm_sec;
Guy Schalnat0d580581995-07-20 02:43:20 -0500417}
418
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500419void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600420png_convert_from_time_t(png_timep ptime, time_t ttime)
Guy Schalnat0d580581995-07-20 02:43:20 -0500421{
422 struct tm *tbuf;
423
Andreas Dilger47a0c421997-05-16 02:46:07 -0500424 png_debug(1, "in png_convert_from_time_t\n");
Guy Schalnat0d580581995-07-20 02:43:20 -0500425 tbuf = gmtime(&ttime);
426 png_convert_from_struct_tm(ptime, tbuf);
427}
Guy Schalnat6d764711995-12-19 03:22:19 -0600428#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500429
Andreas Dilger47a0c421997-05-16 02:46:07 -0500430/* Initialize png_ptr structure, and allocate any memory needed */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500431png_structp PNGAPI
Glenn Randers-Pehrsond0dce401998-05-09 10:02:29 -0500432png_create_write_struct(png_const_charp user_png_ver, png_voidp error_ptr,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600433 png_error_ptr error_fn, png_error_ptr warn_fn)
Guy Schalnat0d580581995-07-20 02:43:20 -0500434{
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500435#ifdef PNG_USER_MEM_SUPPORTED
436 return (png_create_write_struct_2(user_png_ver, error_ptr, error_fn,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500437 warn_fn, NULL, NULL, NULL));
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500438}
439
440/* Alternate initialize png_ptr structure, and allocate any memory needed */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500441png_structp PNGAPI
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500442png_create_write_struct_2(png_const_charp user_png_ver, png_voidp error_ptr,
443 png_error_ptr error_fn, png_error_ptr warn_fn, png_voidp mem_ptr,
444 png_malloc_ptr malloc_fn, png_free_ptr free_fn)
445{
446#endif /* PNG_USER_MEM_SUPPORTED */
Guy Schalnate5a37791996-06-05 15:50:50 -0500447 png_structp png_ptr;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600448#ifdef PNG_SETJMP_SUPPORTED
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600449#ifdef USE_FAR_KEYWORD
450 jmp_buf jmpbuf;
451#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600452#endif
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500453 int i;
Andreas Dilger47a0c421997-05-16 02:46:07 -0500454 png_debug(1, "in png_create_write_struct\n");
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500455#ifdef PNG_USER_MEM_SUPPORTED
Glenn Randers-Pehrson9c0f0942002-02-21 23:14:23 -0600456 png_ptr = (png_structp)png_create_struct_2(PNG_STRUCT_PNG,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500457 (png_malloc_ptr)malloc_fn, (png_voidp)mem_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500458#else
Glenn Randers-Pehrson9c0f0942002-02-21 23:14:23 -0600459 png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500460#endif /* PNG_USER_MEM_SUPPORTED */
Glenn Randers-Pehrson9c0f0942002-02-21 23:14:23 -0600461 if (png_ptr == NULL)
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500462 return (NULL);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600463
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);
Guy Schalnate5a37791996-06-05 15:50:50 -0500478 png_destroy_struct(png_ptr);
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500479 return (NULL);
Guy Schalnate5a37791996-06-05 15:50:50 -0500480 }
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600481#ifdef USE_FAR_KEYWORD
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500482 png_memcpy(png_ptr->jmpbuf, jmpbuf, png_sizeof(jmp_buf));
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600483#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600484#endif
485
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500486#ifdef PNG_USER_MEM_SUPPORTED
487 png_set_mem_fn(png_ptr, mem_ptr, malloc_fn, free_fn);
488#endif /* PNG_USER_MEM_SUPPORTED */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600489 png_set_error_fn(png_ptr, error_ptr, error_fn, warn_fn);
Guy Schalnate5a37791996-06-05 15:50:50 -0500490
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500491 if (user_png_ver)
Guy Schalnate5a37791996-06-05 15:50:50 -0500492 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500493 i=0;
494 do
495 {
496 if (user_png_ver[i] != png_libpng_ver[i])
497 png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH;
498 } while (png_libpng_ver[i++]);
499 }
Guy Schalnat0d580581995-07-20 02:43:20 -0500500
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500501 if (png_ptr->flags & PNG_FLAG_LIBRARY_MISMATCH)
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500502 {
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500503 /* Libpng 0.90 and later are binary incompatible with libpng 0.89, so
504 * we must recompile any applications that use any older library version.
505 * For versions after libpng 1.0, we will be compatible, so we need
506 * only check the first digit.
507 */
508 if (user_png_ver == NULL || user_png_ver[0] != png_libpng_ver[0] ||
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500509 (user_png_ver[0] == '1' && user_png_ver[2] != png_libpng_ver[2]) ||
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500510 (user_png_ver[0] == '0' && user_png_ver[2] < '9'))
511 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500512#if !defined(PNG_NO_STDIO) && !defined(_WIN32_WCE)
Glenn Randers-Pehrsonb1828932001-06-23 08:03:17 -0500513 char msg[80];
514 if (user_png_ver)
515 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500516 png_snprintf(msg, 80,
517 "Application was compiled with png.h from libpng-%.20s",
Glenn Randers-Pehrsonb1828932001-06-23 08:03:17 -0500518 user_png_ver);
519 png_warning(png_ptr, msg);
520 }
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500521 png_snprintf(msg, 80,
522 "Application is running with png.c from libpng-%.20s",
Glenn Randers-Pehrsonb1828932001-06-23 08:03:17 -0500523 png_libpng_ver);
524 png_warning(png_ptr, msg);
525#endif
526#ifdef PNG_ERROR_NUMBERS_SUPPORTED
527 png_ptr->flags=0;
528#endif
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500529 png_error(png_ptr,
530 "Incompatible libpng version in application and library");
531 }
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500532 }
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500533
Guy Schalnat0d580581995-07-20 02:43:20 -0500534 /* initialize zbuf - compression buffer */
535 png_ptr->zbuf_size = PNG_ZBUF_SIZE;
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500536 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr,
537 (png_uint_32)png_ptr->zbuf_size);
Guy Schalnate5a37791996-06-05 15:50:50 -0500538
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500539 png_set_write_fn(png_ptr, NULL, NULL, NULL);
Guy Schalnate5a37791996-06-05 15:50:50 -0500540
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600541#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
542 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500543 1, NULL, NULL);
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600544#endif
545
Glenn Randers-Pehrson03008a02002-04-27 10:11:25 -0500546#ifdef PNG_SETJMP_SUPPORTED
547/* Applications that neglect to set up their own setjmp() and then encounter
548 a png_error() will longjmp here. Since the jmpbuf is then meaningless we
549 abort instead of returning. */
550#ifdef USE_FAR_KEYWORD
551 if (setjmp(jmpbuf))
552 PNG_ABORT();
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500553 png_memcpy(png_ptr->jmpbuf, jmpbuf, png_sizeof(jmp_buf));
Glenn Randers-Pehrson03008a02002-04-27 10:11:25 -0500554#else
555 if (setjmp(png_ptr->jmpbuf))
556 PNG_ABORT();
557#endif
558#endif
559 return (png_ptr);
Guy Schalnate5a37791996-06-05 15:50:50 -0500560}
561
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500562/* Initialize png_ptr structure, and allocate any memory needed */
563
564
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500565void PNGAPI
566png_write_init_3(png_structpp ptr_ptr, png_const_charp user_png_ver,
567 png_size_t png_struct_size)
568{
569 png_structp png_ptr=*ptr_ptr;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600570#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500571 jmp_buf tmp_jmp; /* to save current jump buffer */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600572#endif
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600573
Glenn Randers-Pehrson4deeb792006-03-22 16:21:59 -0600574 int i = 0;
575
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600576 if (png_ptr == NULL)
577 return;
578
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500579 do
580 {
581 if (user_png_ver[i] != png_libpng_ver[i])
582 {
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500583#ifdef PNG_LEGACY_SUPPORTED
584 png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH;
585#else
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500586 png_ptr->warning_fn=NULL;
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500587 png_warning(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500588 "Application uses deprecated png_write_init() and should be recompiled.");
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500589 break;
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500590#endif
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500591 }
592 } while (png_libpng_ver[i++]);
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500593
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500594 png_debug(1, "in png_write_init_3\n");
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500595
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600596#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500597 /* save jump buffer and error functions */
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500598 png_memcpy(tmp_jmp, png_ptr->jmpbuf, png_sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600599#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500600
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500601 if (png_sizeof(png_struct) > png_struct_size)
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500602 {
603 png_destroy_struct(png_ptr);
604 png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG);
605 *ptr_ptr = png_ptr;
606 }
607
Guy Schalnate5a37791996-06-05 15:50:50 -0500608 /* reset all variables to 0 */
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500609 png_memset(png_ptr, 0, png_sizeof(png_struct));
Guy Schalnate5a37791996-06-05 15:50:50 -0500610
Glenn Randers-Pehrson272489d2004-08-04 06:34:52 -0500611 /* added at libpng-1.2.6 */
612#ifdef PNG_SET_USER_LIMITS_SUPPORTED
613 png_ptr->user_width_max=PNG_USER_WIDTH_MAX;
614 png_ptr->user_height_max=PNG_USER_HEIGHT_MAX;
615#endif
616
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600617#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500618 /* restore jump buffer */
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500619 png_memcpy(png_ptr->jmpbuf, tmp_jmp, png_sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600620#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500621
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500622 png_set_write_fn(png_ptr, NULL, NULL, NULL);
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500623
Guy Schalnate5a37791996-06-05 15:50:50 -0500624 /* initialize zbuf - compression buffer */
625 png_ptr->zbuf_size = PNG_ZBUF_SIZE;
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500626 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr,
627 (png_uint_32)png_ptr->zbuf_size);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500628
629#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
630 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500631 1, NULL, NULL);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500632#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500633}
634
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600635/* Write a few rows of image data. If the image is interlaced,
636 * either you will have to write the 7 sub images, or, if you
637 * have called png_set_interlace_handling(), you will have to
638 * "write" the image seven times.
639 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500640void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600641png_write_rows(png_structp png_ptr, png_bytepp row,
Guy Schalnat0d580581995-07-20 02:43:20 -0500642 png_uint_32 num_rows)
643{
644 png_uint_32 i; /* row counter */
Guy Schalnat6d764711995-12-19 03:22:19 -0600645 png_bytepp rp; /* row pointer */
Guy Schalnat0d580581995-07-20 02:43:20 -0500646
Andreas Dilger47a0c421997-05-16 02:46:07 -0500647 png_debug(1, "in png_write_rows\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600648
649 if (png_ptr == NULL)
650 return;
651
Guy Schalnat0d580581995-07-20 02:43:20 -0500652 /* loop through the rows */
653 for (i = 0, rp = row; i < num_rows; i++, rp++)
654 {
655 png_write_row(png_ptr, *rp);
656 }
657}
658
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600659/* Write the image. You only need to call this function once, even
660 * if you are writing an interlaced image.
661 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500662void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600663png_write_image(png_structp png_ptr, png_bytepp image)
Guy Schalnat0d580581995-07-20 02:43:20 -0500664{
665 png_uint_32 i; /* row index */
666 int pass, num_pass; /* pass variables */
Guy Schalnat6d764711995-12-19 03:22:19 -0600667 png_bytepp rp; /* points to current row */
Guy Schalnat0d580581995-07-20 02:43:20 -0500668
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600669 if (png_ptr == NULL)
670 return;
671
Andreas Dilger47a0c421997-05-16 02:46:07 -0500672 png_debug(1, "in png_write_image\n");
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600673#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500674 /* intialize interlace handling. If image is not interlaced,
675 this will set pass to 1 */
676 num_pass = png_set_interlace_handling(png_ptr);
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600677#else
678 num_pass = 1;
679#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500680 /* loop through passes */
681 for (pass = 0; pass < num_pass; pass++)
682 {
683 /* loop through image */
684 for (i = 0, rp = image; i < png_ptr->height; i++, rp++)
685 {
686 png_write_row(png_ptr, *rp);
687 }
688 }
689}
690
Guy Schalnate5a37791996-06-05 15:50:50 -0500691/* called by user to write a row of image data */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500692void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600693png_write_row(png_structp png_ptr, png_bytep row)
Guy Schalnat0d580581995-07-20 02:43:20 -0500694{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600695 if (png_ptr == NULL)
696 return;
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600697 png_debug2(1, "in png_write_row (row %ld, pass %d)\n",
698 png_ptr->row_number, png_ptr->pass);
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600699
Guy Schalnat0d580581995-07-20 02:43:20 -0500700 /* initialize transformations and other stuff if first time */
Guy Schalnat6d764711995-12-19 03:22:19 -0600701 if (png_ptr->row_number == 0 && png_ptr->pass == 0)
Guy Schalnat0d580581995-07-20 02:43:20 -0500702 {
Glenn Randers-Pehrson5cded0b2001-11-07 07:10:08 -0600703 /* make sure we wrote the header info */
704 if (!(png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE))
705 png_error(png_ptr,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500706 "png_write_info was never called before png_write_row");
Glenn Randers-Pehrson5cded0b2001-11-07 07:10:08 -0600707
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500708 /* check for transforms that have been set but were defined out */
709#if !defined(PNG_WRITE_INVERT_SUPPORTED) && defined(PNG_READ_INVERT_SUPPORTED)
710 if (png_ptr->transformations & PNG_INVERT_MONO)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500711 png_warning(png_ptr, "PNG_WRITE_INVERT_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500712#endif
713#if !defined(PNG_WRITE_FILLER_SUPPORTED) && defined(PNG_READ_FILLER_SUPPORTED)
714 if (png_ptr->transformations & PNG_FILLER)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500715 png_warning(png_ptr, "PNG_WRITE_FILLER_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500716#endif
717#if !defined(PNG_WRITE_PACKSWAP_SUPPORTED) && defined(PNG_READ_PACKSWAP_SUPPORTED)
718 if (png_ptr->transformations & PNG_PACKSWAP)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500719 png_warning(png_ptr, "PNG_WRITE_PACKSWAP_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500720#endif
721#if !defined(PNG_WRITE_PACK_SUPPORTED) && defined(PNG_READ_PACK_SUPPORTED)
722 if (png_ptr->transformations & PNG_PACK)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500723 png_warning(png_ptr, "PNG_WRITE_PACK_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500724#endif
725#if !defined(PNG_WRITE_SHIFT_SUPPORTED) && defined(PNG_READ_SHIFT_SUPPORTED)
726 if (png_ptr->transformations & PNG_SHIFT)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500727 png_warning(png_ptr, "PNG_WRITE_SHIFT_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500728#endif
729#if !defined(PNG_WRITE_BGR_SUPPORTED) && defined(PNG_READ_BGR_SUPPORTED)
730 if (png_ptr->transformations & PNG_BGR)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500731 png_warning(png_ptr, "PNG_WRITE_BGR_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500732#endif
733#if !defined(PNG_WRITE_SWAP_SUPPORTED) && defined(PNG_READ_SWAP_SUPPORTED)
734 if (png_ptr->transformations & PNG_SWAP_BYTES)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500735 png_warning(png_ptr, "PNG_WRITE_SWAP_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500736#endif
737
Guy Schalnat0d580581995-07-20 02:43:20 -0500738 png_write_start_row(png_ptr);
739 }
740
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500741#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500742 /* if interlaced and not interested in row, return */
743 if (png_ptr->interlaced && (png_ptr->transformations & PNG_INTERLACE))
744 {
745 switch (png_ptr->pass)
746 {
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600747 case 0:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600748 if (png_ptr->row_number & 0x07)
Guy Schalnat0d580581995-07-20 02:43:20 -0500749 {
750 png_write_finish_row(png_ptr);
751 return;
752 }
753 break;
754 case 1:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600755 if ((png_ptr->row_number & 0x07) || png_ptr->width < 5)
Guy Schalnat0d580581995-07-20 02:43:20 -0500756 {
757 png_write_finish_row(png_ptr);
758 return;
759 }
760 break;
761 case 2:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600762 if ((png_ptr->row_number & 0x07) != 4)
Guy Schalnat0d580581995-07-20 02:43:20 -0500763 {
764 png_write_finish_row(png_ptr);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600765 return;
Guy Schalnat0d580581995-07-20 02:43:20 -0500766 }
767 break;
768 case 3:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600769 if ((png_ptr->row_number & 0x03) || png_ptr->width < 3)
Guy Schalnat0d580581995-07-20 02:43:20 -0500770 {
771 png_write_finish_row(png_ptr);
772 return;
773 }
774 break;
775 case 4:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600776 if ((png_ptr->row_number & 0x03) != 2)
Guy Schalnat0d580581995-07-20 02:43:20 -0500777 {
778 png_write_finish_row(png_ptr);
779 return;
780 }
781 break;
782 case 5:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600783 if ((png_ptr->row_number & 0x01) || png_ptr->width < 2)
Guy Schalnat0d580581995-07-20 02:43:20 -0500784 {
785 png_write_finish_row(png_ptr);
786 return;
787 }
788 break;
789 case 6:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600790 if (!(png_ptr->row_number & 0x01))
Guy Schalnat0d580581995-07-20 02:43:20 -0500791 {
792 png_write_finish_row(png_ptr);
793 return;
794 }
795 break;
796 }
797 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600798#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500799
800 /* set up row info for transformations */
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600801 png_ptr->row_info.color_type = png_ptr->color_type;
Guy Schalnat0d580581995-07-20 02:43:20 -0500802 png_ptr->row_info.width = png_ptr->usr_width;
803 png_ptr->row_info.channels = png_ptr->usr_channels;
804 png_ptr->row_info.bit_depth = png_ptr->usr_bit_depth;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600805 png_ptr->row_info.pixel_depth = (png_byte)(png_ptr->row_info.bit_depth *
806 png_ptr->row_info.channels);
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600807
Glenn Randers-Pehrson272489d2004-08-04 06:34:52 -0500808 png_ptr->row_info.rowbytes = PNG_ROWBYTES(png_ptr->row_info.pixel_depth,
809 png_ptr->row_info.width);
Guy Schalnat0d580581995-07-20 02:43:20 -0500810
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600811 png_debug1(3, "row_info->color_type = %d\n", png_ptr->row_info.color_type);
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500812 png_debug1(3, "row_info->width = %lu\n", png_ptr->row_info.width);
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600813 png_debug1(3, "row_info->channels = %d\n", png_ptr->row_info.channels);
814 png_debug1(3, "row_info->bit_depth = %d\n", png_ptr->row_info.bit_depth);
815 png_debug1(3, "row_info->pixel_depth = %d\n", png_ptr->row_info.pixel_depth);
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500816 png_debug1(3, "row_info->rowbytes = %lu\n", png_ptr->row_info.rowbytes);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500817
818 /* Copy user's row into buffer, leaving room for filter byte. */
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500819 png_memcpy(png_ptr->row_buf + 1, row, png_ptr->row_info.rowbytes);
Guy Schalnat0d580581995-07-20 02:43:20 -0500820
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500821#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500822 /* handle interlacing */
823 if (png_ptr->interlaced && png_ptr->pass < 6 &&
824 (png_ptr->transformations & PNG_INTERLACE))
825 {
826 png_do_write_interlace(&(png_ptr->row_info),
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600827 png_ptr->row_buf + 1, png_ptr->pass);
Guy Schalnat0d580581995-07-20 02:43:20 -0500828 /* this should always get caught above, but still ... */
829 if (!(png_ptr->row_info.width))
830 {
831 png_write_finish_row(png_ptr);
832 return;
833 }
834 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600835#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500836
837 /* handle other transformations */
838 if (png_ptr->transformations)
839 png_do_write_transformations(png_ptr);
840
Glenn Randers-Pehrson2ad31ae2000-12-15 08:54:42 -0600841#if defined(PNG_MNG_FEATURES_SUPPORTED)
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -0600842 /* Write filter_method 64 (intrapixel differencing) only if
843 * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and
844 * 2. Libpng did not write a PNG signature (this filter_method is only
845 * used in PNG datastreams that are embedded in MNG datastreams) and
846 * 3. The application called png_permit_mng_features with a mask that
847 * included PNG_FLAG_MNG_FILTER_64 and
848 * 4. The filter_method is 64 and
849 * 5. The color_type is RGB or RGBA
850 */
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500851 if ((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
Glenn Randers-Pehrson2ad31ae2000-12-15 08:54:42 -0600852 (png_ptr->filter_type == PNG_INTRAPIXEL_DIFFERENCING))
853 {
854 /* Intrapixel differencing */
855 png_do_write_intrapixel(&(png_ptr->row_info), png_ptr->row_buf + 1);
856 }
857#endif
858
Andreas Dilger47a0c421997-05-16 02:46:07 -0500859 /* Find a filter if necessary, filter the row and write it out. */
Guy Schalnate5a37791996-06-05 15:50:50 -0500860 png_write_find_filter(png_ptr, &(png_ptr->row_info));
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -0600861
862 if (png_ptr->write_row_fn != NULL)
863 (*(png_ptr->write_row_fn))(png_ptr, png_ptr->row_number, png_ptr->pass);
Guy Schalnat0d580581995-07-20 02:43:20 -0500864}
865
Guy Schalnat0f716451995-11-28 11:22:13 -0600866#if defined(PNG_WRITE_FLUSH_SUPPORTED)
867/* Set the automatic flush interval or 0 to turn flushing off */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500868void PNGAPI
Guy Schalnat4ee97b01996-01-16 01:51:56 -0600869png_set_flush(png_structp png_ptr, int nrows)
Guy Schalnat0f716451995-11-28 11:22:13 -0600870{
Andreas Dilger47a0c421997-05-16 02:46:07 -0500871 png_debug(1, "in png_set_flush\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600872 if (png_ptr == NULL)
873 return;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600874 png_ptr->flush_dist = (nrows < 0 ? 0 : nrows);
Guy Schalnat0f716451995-11-28 11:22:13 -0600875}
876
877/* flush the current output buffers now */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500878void PNGAPI
Guy Schalnat4ee97b01996-01-16 01:51:56 -0600879png_write_flush(png_structp png_ptr)
Guy Schalnat0f716451995-11-28 11:22:13 -0600880{
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600881 int wrote_IDAT;
Guy Schalnat0f716451995-11-28 11:22:13 -0600882
Andreas Dilger47a0c421997-05-16 02:46:07 -0500883 png_debug(1, "in png_write_flush\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600884 if (png_ptr == NULL)
885 return;
Guy Schalnate5a37791996-06-05 15:50:50 -0500886 /* We have already written out all of the data */
887 if (png_ptr->row_number >= png_ptr->num_rows)
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600888 return;
Guy Schalnat0f716451995-11-28 11:22:13 -0600889
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600890 do
891 {
892 int ret;
Guy Schalnat0f716451995-11-28 11:22:13 -0600893
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600894 /* compress the data */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600895 ret = deflate(&png_ptr->zstream, Z_SYNC_FLUSH);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600896 wrote_IDAT = 0;
Guy Schalnat0f716451995-11-28 11:22:13 -0600897
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600898 /* check for compression errors */
899 if (ret != Z_OK)
900 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500901 if (png_ptr->zstream.msg != NULL)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600902 png_error(png_ptr, png_ptr->zstream.msg);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600903 else
904 png_error(png_ptr, "zlib error");
905 }
Guy Schalnat0f716451995-11-28 11:22:13 -0600906
Andreas Dilger47a0c421997-05-16 02:46:07 -0500907 if (!(png_ptr->zstream.avail_out))
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600908 {
909 /* write the IDAT and reset the zlib output buffer */
910 png_write_IDAT(png_ptr, png_ptr->zbuf,
911 png_ptr->zbuf_size);
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600912 png_ptr->zstream.next_out = png_ptr->zbuf;
913 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600914 wrote_IDAT = 1;
915 }
916 } while(wrote_IDAT == 1);
Guy Schalnat0f716451995-11-28 11:22:13 -0600917
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600918 /* If there is any data left to be output, write it into a new IDAT */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600919 if (png_ptr->zbuf_size != png_ptr->zstream.avail_out)
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600920 {
921 /* write the IDAT and reset the zlib output buffer */
922 png_write_IDAT(png_ptr, png_ptr->zbuf,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600923 png_ptr->zbuf_size - png_ptr->zstream.avail_out);
924 png_ptr->zstream.next_out = png_ptr->zbuf;
925 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600926 }
927 png_ptr->flush_rows = 0;
928 png_flush(png_ptr);
Guy Schalnat0f716451995-11-28 11:22:13 -0600929}
930#endif /* PNG_WRITE_FLUSH_SUPPORTED */
931
Guy Schalnate5a37791996-06-05 15:50:50 -0500932/* free all memory used by the write */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500933void PNGAPI
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600934png_destroy_write_struct(png_structpp png_ptr_ptr, png_infopp info_ptr_ptr)
Guy Schalnate5a37791996-06-05 15:50:50 -0500935{
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600936 png_structp png_ptr = NULL;
937 png_infop info_ptr = NULL;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500938#ifdef PNG_USER_MEM_SUPPORTED
939 png_free_ptr free_fn = NULL;
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500940 png_voidp mem_ptr = NULL;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500941#endif
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600942
Andreas Dilger47a0c421997-05-16 02:46:07 -0500943 png_debug(1, "in png_destroy_write_struct\n");
944 if (png_ptr_ptr != NULL)
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500945 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600946 png_ptr = *png_ptr_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500947#ifdef PNG_USER_MEM_SUPPORTED
948 free_fn = png_ptr->free_fn;
Glenn Randers-Pehrsonfcbd7872002-04-07 16:35:38 -0500949 mem_ptr = png_ptr->mem_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500950#endif
951 }
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600952
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500953#ifdef PNG_USER_MEM_SUPPORTED
954 if (png_ptr != NULL)
955 {
956 free_fn = png_ptr->free_fn;
957 mem_ptr = png_ptr->mem_ptr;
958 }
959#endif
960
Andreas Dilger47a0c421997-05-16 02:46:07 -0500961 if (info_ptr_ptr != NULL)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600962 info_ptr = *info_ptr_ptr;
963
Andreas Dilger47a0c421997-05-16 02:46:07 -0500964 if (info_ptr != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500965 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500966 if (png_ptr != NULL)
967 {
968 png_free_data(png_ptr, info_ptr, PNG_FREE_ALL, -1);
Glenn Randers-Pehrsona77ef622000-02-18 13:48:52 -0600969
Glenn Randers-Pehrson6942d532000-05-01 09:31:54 -0500970#if defined(PNG_UNKNOWN_CHUNKS_SUPPORTED)
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500971 if (png_ptr->num_chunk_list)
972 {
973 png_free(png_ptr, png_ptr->chunk_list);
Glenn Randers-Pehrson895a9c92008-07-25 08:51:18 -0500974 png_ptr->num_chunk_list = 0;
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500975 }
Glenn Randers-Pehrson6942d532000-05-01 09:31:54 -0500976#endif
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500977 }
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600978
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500979#ifdef PNG_USER_MEM_SUPPORTED
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500980 png_destroy_struct_2((png_voidp)info_ptr, (png_free_ptr)free_fn,
981 (png_voidp)mem_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500982#else
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600983 png_destroy_struct((png_voidp)info_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500984#endif
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500985 *info_ptr_ptr = NULL;
Guy Schalnate5a37791996-06-05 15:50:50 -0500986 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600987
Andreas Dilger47a0c421997-05-16 02:46:07 -0500988 if (png_ptr != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500989 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600990 png_write_destroy(png_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500991#ifdef PNG_USER_MEM_SUPPORTED
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500992 png_destroy_struct_2((png_voidp)png_ptr, (png_free_ptr)free_fn,
993 (png_voidp)mem_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500994#else
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600995 png_destroy_struct((png_voidp)png_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500996#endif
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500997 *png_ptr_ptr = NULL;
Guy Schalnate5a37791996-06-05 15:50:50 -0500998 }
999}
1000
1001
Andreas Dilger47a0c421997-05-16 02:46:07 -05001002/* Free any memory used in png_ptr struct (old method) */
Glenn Randers-Pehrsonf64a06f2001-04-11 07:38:00 -05001003void /* PRIVATE */
Guy Schalnat6d764711995-12-19 03:22:19 -06001004png_write_destroy(png_structp png_ptr)
Guy Schalnat0d580581995-07-20 02:43:20 -05001005{
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001006#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnatb2e01bd1996-01-26 01:38:47 -06001007 jmp_buf tmp_jmp; /* save jump buffer */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001008#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001009 png_error_ptr error_fn;
1010 png_error_ptr warning_fn;
1011 png_voidp error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -05001012#ifdef PNG_USER_MEM_SUPPORTED
1013 png_free_ptr free_fn;
1014#endif
Guy Schalnat0d580581995-07-20 02:43:20 -05001015
Andreas Dilger47a0c421997-05-16 02:46:07 -05001016 png_debug(1, "in png_write_destroy\n");
Guy Schalnat0d580581995-07-20 02:43:20 -05001017 /* free any memory zlib uses */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -06001018 deflateEnd(&png_ptr->zstream);
Guy Schalnate5a37791996-06-05 15:50:50 -05001019
Guy Schalnat0d580581995-07-20 02:43:20 -05001020 /* free our memory. png_free checks NULL for us. */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -06001021 png_free(png_ptr, png_ptr->zbuf);
1022 png_free(png_ptr, png_ptr->row_buf);
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001023#ifndef PNG_NO_WRITE_FILTER
Andreas Dilger02ad0ef1997-01-17 01:34:35 -06001024 png_free(png_ptr, png_ptr->prev_row);
1025 png_free(png_ptr, png_ptr->sub_row);
1026 png_free(png_ptr, png_ptr->up_row);
1027 png_free(png_ptr, png_ptr->avg_row);
1028 png_free(png_ptr, png_ptr->paeth_row);
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001029#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001030
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -06001031#if defined(PNG_TIME_RFC1123_SUPPORTED)
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06001032 png_free(png_ptr, png_ptr->time_buffer);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001033#endif
1034
Andreas Dilger47a0c421997-05-16 02:46:07 -05001035#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
1036 png_free(png_ptr, png_ptr->prev_filters);
1037 png_free(png_ptr, png_ptr->filter_weights);
1038 png_free(png_ptr, png_ptr->inv_filter_weights);
1039 png_free(png_ptr, png_ptr->filter_costs);
1040 png_free(png_ptr, png_ptr->inv_filter_costs);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001041#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001042
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001043#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnat0d580581995-07-20 02:43:20 -05001044 /* reset structure */
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001045 png_memcpy(tmp_jmp, png_ptr->jmpbuf, png_sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001046#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001047
1048 error_fn = png_ptr->error_fn;
1049 warning_fn = png_ptr->warning_fn;
1050 error_ptr = png_ptr->error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -05001051#ifdef PNG_USER_MEM_SUPPORTED
1052 free_fn = png_ptr->free_fn;
1053#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001054
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001055 png_memset(png_ptr, 0, png_sizeof(png_struct));
Guy Schalnate5a37791996-06-05 15:50:50 -05001056
1057 png_ptr->error_fn = error_fn;
1058 png_ptr->warning_fn = warning_fn;
1059 png_ptr->error_ptr = error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -05001060#ifdef PNG_USER_MEM_SUPPORTED
1061 png_ptr->free_fn = free_fn;
1062#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001063
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001064#ifdef PNG_SETJMP_SUPPORTED
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001065 png_memcpy(png_ptr->jmpbuf, tmp_jmp, png_sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001066#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001067}
Guy Schalnate5a37791996-06-05 15:50:50 -05001068
Andreas Dilger47a0c421997-05-16 02:46:07 -05001069/* Allow the application to select one or more row filters to use. */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001070void PNGAPI
Guy Schalnate5a37791996-06-05 15:50:50 -05001071png_set_filter(png_structp png_ptr, int method, int filters)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001072{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001073 png_debug(1, "in png_set_filter\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001074 if (png_ptr == NULL)
1075 return;
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -06001076#if defined(PNG_MNG_FEATURES_SUPPORTED)
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001077 if ((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -06001078 (method == PNG_INTRAPIXEL_DIFFERENCING))
1079 method = PNG_FILTER_TYPE_BASE;
1080#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -05001081 if (method == PNG_FILTER_TYPE_BASE)
Guy Schalnate5a37791996-06-05 15:50:50 -05001082 {
1083 switch (filters & (PNG_ALL_FILTERS | 0x07))
1084 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001085#ifndef PNG_NO_WRITE_FILTER
Guy Schalnate5a37791996-06-05 15:50:50 -05001086 case 5:
1087 case 6:
Andreas Dilger47a0c421997-05-16 02:46:07 -05001088 case 7: png_warning(png_ptr, "Unknown row filter for method 0");
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001089#endif /* PNG_NO_WRITE_FILTER */
1090 case PNG_FILTER_VALUE_NONE:
1091 png_ptr->do_filter=PNG_FILTER_NONE; break;
1092#ifndef PNG_NO_WRITE_FILTER
1093 case PNG_FILTER_VALUE_SUB:
1094 png_ptr->do_filter=PNG_FILTER_SUB; break;
1095 case PNG_FILTER_VALUE_UP:
1096 png_ptr->do_filter=PNG_FILTER_UP; break;
1097 case PNG_FILTER_VALUE_AVG:
1098 png_ptr->do_filter=PNG_FILTER_AVG; break;
1099 case PNG_FILTER_VALUE_PAETH:
1100 png_ptr->do_filter=PNG_FILTER_PAETH; break;
Guy Schalnate5a37791996-06-05 15:50:50 -05001101 default: png_ptr->do_filter = (png_byte)filters; break;
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001102#else
1103 default: png_warning(png_ptr, "Unknown row filter for method 0");
1104#endif /* PNG_NO_WRITE_FILTER */
Guy Schalnate5a37791996-06-05 15:50:50 -05001105 }
1106
Andreas Dilger47a0c421997-05-16 02:46:07 -05001107 /* If we have allocated the row_buf, this means we have already started
1108 * with the image and we should have allocated all of the filter buffers
1109 * that have been selected. If prev_row isn't already allocated, then
1110 * it is too late to start using the filters that need it, since we
1111 * will be missing the data in the previous row. If an application
1112 * wants to start and stop using particular filters during compression,
1113 * it should start out with all of the filters, and then add and
1114 * remove them after the start of compression.
Guy Schalnate5a37791996-06-05 15:50:50 -05001115 */
Andreas Dilger47a0c421997-05-16 02:46:07 -05001116 if (png_ptr->row_buf != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001117 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001118#ifndef PNG_NO_WRITE_FILTER
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001119 if ((png_ptr->do_filter & PNG_FILTER_SUB) && png_ptr->sub_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001120 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001121 png_ptr->sub_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001122 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001123 png_ptr->sub_row[0] = PNG_FILTER_VALUE_SUB;
Guy Schalnate5a37791996-06-05 15:50:50 -05001124 }
1125
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001126 if ((png_ptr->do_filter & PNG_FILTER_UP) && png_ptr->up_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001127 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001128 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001129 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001130 png_warning(png_ptr, "Can't add Up filter after starting");
Guy Schalnate5a37791996-06-05 15:50:50 -05001131 png_ptr->do_filter &= ~PNG_FILTER_UP;
1132 }
1133 else
1134 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001135 png_ptr->up_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001136 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001137 png_ptr->up_row[0] = PNG_FILTER_VALUE_UP;
Guy Schalnate5a37791996-06-05 15:50:50 -05001138 }
1139 }
1140
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001141 if ((png_ptr->do_filter & PNG_FILTER_AVG) && png_ptr->avg_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001142 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001143 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001144 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001145 png_warning(png_ptr, "Can't add Average filter after starting");
Guy Schalnate5a37791996-06-05 15:50:50 -05001146 png_ptr->do_filter &= ~PNG_FILTER_AVG;
1147 }
1148 else
1149 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001150 png_ptr->avg_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001151 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001152 png_ptr->avg_row[0] = PNG_FILTER_VALUE_AVG;
Guy Schalnate5a37791996-06-05 15:50:50 -05001153 }
1154 }
1155
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001156 if ((png_ptr->do_filter & PNG_FILTER_PAETH) &&
Andreas Dilger47a0c421997-05-16 02:46:07 -05001157 png_ptr->paeth_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001158 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001159 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001160 {
1161 png_warning(png_ptr, "Can't add Paeth filter after starting");
Glenn Randers-Pehrson860ab2b1999-10-14 07:43:10 -05001162 png_ptr->do_filter &= (png_byte)(~PNG_FILTER_PAETH);
Guy Schalnate5a37791996-06-05 15:50:50 -05001163 }
1164 else
1165 {
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -06001166 png_ptr->paeth_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001167 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001168 png_ptr->paeth_row[0] = PNG_FILTER_VALUE_PAETH;
Guy Schalnate5a37791996-06-05 15:50:50 -05001169 }
1170 }
1171
1172 if (png_ptr->do_filter == PNG_NO_FILTERS)
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001173#endif /* PNG_NO_WRITE_FILTER */
Guy Schalnate5a37791996-06-05 15:50:50 -05001174 png_ptr->do_filter = PNG_FILTER_NONE;
1175 }
1176 }
1177 else
Andreas Dilger47a0c421997-05-16 02:46:07 -05001178 png_error(png_ptr, "Unknown custom filter method");
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001179}
1180
Andreas Dilger47a0c421997-05-16 02:46:07 -05001181/* This allows us to influence the way in which libpng chooses the "best"
1182 * filter for the current scanline. While the "minimum-sum-of-absolute-
1183 * differences metric is relatively fast and effective, there is some
1184 * question as to whether it can be improved upon by trying to keep the
1185 * filtered data going to zlib more consistent, hopefully resulting in
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06001186 * better compression.
1187 */
Andreas Dilger47a0c421997-05-16 02:46:07 -05001188#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED) /* GRR 970116 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001189void PNGAPI
Andreas Dilger47a0c421997-05-16 02:46:07 -05001190png_set_filter_heuristics(png_structp png_ptr, int heuristic_method,
1191 int num_weights, png_doublep filter_weights,
1192 png_doublep filter_costs)
1193{
1194 int i;
1195
1196 png_debug(1, "in png_set_filter_heuristics\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001197 if (png_ptr == NULL)
1198 return;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001199 if (heuristic_method >= PNG_FILTER_HEURISTIC_LAST)
1200 {
1201 png_warning(png_ptr, "Unknown filter heuristic method");
1202 return;
1203 }
1204
1205 if (heuristic_method == PNG_FILTER_HEURISTIC_DEFAULT)
1206 {
1207 heuristic_method = PNG_FILTER_HEURISTIC_UNWEIGHTED;
1208 }
1209
1210 if (num_weights < 0 || filter_weights == NULL ||
1211 heuristic_method == PNG_FILTER_HEURISTIC_UNWEIGHTED)
1212 {
1213 num_weights = 0;
1214 }
1215
Glenn Randers-Pehrson860ab2b1999-10-14 07:43:10 -05001216 png_ptr->num_prev_filters = (png_byte)num_weights;
1217 png_ptr->heuristic_method = (png_byte)heuristic_method;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001218
1219 if (num_weights > 0)
1220 {
1221 if (png_ptr->prev_filters == NULL)
1222 {
1223 png_ptr->prev_filters = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001224 (png_uint_32)(png_sizeof(png_byte) * num_weights));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001225
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001226 /* To make sure that the weighting starts out fairly */
1227 for (i = 0; i < num_weights; i++)
Andreas Dilger47a0c421997-05-16 02:46:07 -05001228 {
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001229 png_ptr->prev_filters[i] = 255;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001230 }
1231 }
1232
1233 if (png_ptr->filter_weights == NULL)
1234 {
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001235 png_ptr->filter_weights = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001236 (png_uint_32)(png_sizeof(png_uint_16) * num_weights));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001237
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001238 png_ptr->inv_filter_weights = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001239 (png_uint_32)(png_sizeof(png_uint_16) * num_weights));
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001240 for (i = 0; i < num_weights; i++)
Andreas Dilger47a0c421997-05-16 02:46:07 -05001241 {
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001242 png_ptr->inv_filter_weights[i] =
1243 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001244 }
1245 }
1246
1247 for (i = 0; i < num_weights; i++)
1248 {
1249 if (filter_weights[i] < 0.0)
1250 {
1251 png_ptr->inv_filter_weights[i] =
1252 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
1253 }
1254 else
1255 {
1256 png_ptr->inv_filter_weights[i] =
1257 (png_uint_16)((double)PNG_WEIGHT_FACTOR*filter_weights[i]+0.5);
1258 png_ptr->filter_weights[i] =
1259 (png_uint_16)((double)PNG_WEIGHT_FACTOR/filter_weights[i]+0.5);
1260 }
1261 }
1262 }
1263
1264 /* If, in the future, there are other filter methods, this would
1265 * need to be based on png_ptr->filter.
1266 */
1267 if (png_ptr->filter_costs == NULL)
1268 {
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001269 png_ptr->filter_costs = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001270 (png_uint_32)(png_sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001271
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001272 png_ptr->inv_filter_costs = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001273 (png_uint_32)(png_sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001274
1275 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
1276 {
1277 png_ptr->inv_filter_costs[i] =
1278 png_ptr->filter_costs[i] = PNG_COST_FACTOR;
1279 }
1280 }
1281
1282 /* Here is where we set the relative costs of the different filters. We
1283 * should take the desired compression level into account when setting
1284 * the costs, so that Paeth, for instance, has a high relative cost at low
1285 * compression levels, while it has a lower relative cost at higher
1286 * compression settings. The filter types are in order of increasing
1287 * relative cost, so it would be possible to do this with an algorithm.
1288 */
1289 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
1290 {
1291 if (filter_costs == NULL || filter_costs[i] < 0.0)
1292 {
1293 png_ptr->inv_filter_costs[i] =
1294 png_ptr->filter_costs[i] = PNG_COST_FACTOR;
1295 }
1296 else if (filter_costs[i] >= 1.0)
1297 {
1298 png_ptr->inv_filter_costs[i] =
1299 (png_uint_16)((double)PNG_COST_FACTOR / filter_costs[i] + 0.5);
1300 png_ptr->filter_costs[i] =
1301 (png_uint_16)((double)PNG_COST_FACTOR * filter_costs[i] + 0.5);
1302 }
1303 }
1304}
1305#endif /* PNG_WRITE_WEIGHTED_FILTER_SUPPORTED */
1306
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001307void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001308png_set_compression_level(png_structp png_ptr, int level)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001309{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001310 png_debug(1, "in png_set_compression_level\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001311 if (png_ptr == NULL)
1312 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001313 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_LEVEL;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001314 png_ptr->zlib_level = level;
1315}
1316
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001317void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001318png_set_compression_mem_level(png_structp png_ptr, int mem_level)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001319{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001320 png_debug(1, "in png_set_compression_mem_level\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001321 if (png_ptr == NULL)
1322 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001323 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_MEM_LEVEL;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -06001324 png_ptr->zlib_mem_level = mem_level;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001325}
1326
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001327void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001328png_set_compression_strategy(png_structp png_ptr, int strategy)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001329{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001330 png_debug(1, "in png_set_compression_strategy\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001331 if (png_ptr == NULL)
1332 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001333 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_STRATEGY;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001334 png_ptr->zlib_strategy = strategy;
1335}
1336
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001337void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001338png_set_compression_window_bits(png_structp png_ptr, int window_bits)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001339{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001340 if (png_ptr == NULL)
1341 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001342 if (window_bits > 15)
1343 png_warning(png_ptr, "Only compression windows <= 32k supported by PNG");
Glenn Randers-Pehrson4393a9a1999-09-17 12:27:26 -05001344 else if (window_bits < 8)
1345 png_warning(png_ptr, "Only compression windows >= 256 supported by PNG");
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001346#ifndef WBITS_8_OK
1347 /* avoid libpng bug with 256-byte windows */
1348 if (window_bits == 8)
1349 {
1350 png_warning(png_ptr, "Compression window is being reset to 512");
1351 window_bits=9;
1352 }
1353#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001354 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_WINDOW_BITS;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001355 png_ptr->zlib_window_bits = window_bits;
1356}
1357
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001358void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001359png_set_compression_method(png_structp png_ptr, int method)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001360{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001361 png_debug(1, "in png_set_compression_method\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001362 if (png_ptr == NULL)
1363 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001364 if (method != 8)
1365 png_warning(png_ptr, "Only compression method 8 is supported by PNG");
1366 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_METHOD;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001367 png_ptr->zlib_method = method;
Guy Schalnat0d580581995-07-20 02:43:20 -05001368}
1369
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001370void PNGAPI
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001371png_set_write_status_fn(png_structp png_ptr, png_write_status_ptr write_row_fn)
1372{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001373 if (png_ptr == NULL)
1374 return;
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001375 png_ptr->write_row_fn = write_row_fn;
1376}
1377
1378#if defined(PNG_WRITE_USER_TRANSFORM_SUPPORTED)
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001379void PNGAPI
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001380png_set_write_user_transform_fn(png_structp png_ptr, png_user_transform_ptr
1381 write_user_transform_fn)
1382{
1383 png_debug(1, "in png_set_write_user_transform_fn\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001384 if (png_ptr == NULL)
1385 return;
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001386 png_ptr->transformations |= PNG_USER_TRANSFORM;
1387 png_ptr->write_user_transform_fn = write_user_transform_fn;
1388}
1389#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001390
1391
1392#if defined(PNG_INFO_IMAGE_SUPPORTED)
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001393void PNGAPI
1394png_write_png(png_structp png_ptr, png_infop info_ptr,
Glenn Randers-Pehrson1ef65b62000-05-12 06:19:53 -05001395 int transforms, voidp params)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001396{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001397 if (png_ptr == NULL || info_ptr == NULL)
1398 return;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001399#if defined(PNG_WRITE_INVERT_ALPHA_SUPPORTED)
1400 /* invert the alpha channel from opacity to transparency */
1401 if (transforms & PNG_TRANSFORM_INVERT_ALPHA)
1402 png_set_invert_alpha(png_ptr);
1403#endif
1404
1405 /* Write the file header information. */
1406 png_write_info(png_ptr, info_ptr);
1407
1408 /* ------ these transformations don't touch the info structure ------- */
1409
1410#if defined(PNG_WRITE_INVERT_SUPPORTED)
1411 /* invert monochrome pixels */
1412 if (transforms & PNG_TRANSFORM_INVERT_MONO)
1413 png_set_invert_mono(png_ptr);
1414#endif
1415
1416#if defined(PNG_WRITE_SHIFT_SUPPORTED)
1417 /* Shift the pixels up to a legal bit depth and fill in
1418 * as appropriate to correctly scale the image.
1419 */
1420 if ((transforms & PNG_TRANSFORM_SHIFT)
1421 && (info_ptr->valid & PNG_INFO_sBIT))
1422 png_set_shift(png_ptr, &info_ptr->sig_bit);
1423#endif
1424
1425#if defined(PNG_WRITE_PACK_SUPPORTED)
1426 /* pack pixels into bytes */
1427 if (transforms & PNG_TRANSFORM_PACKING)
1428 png_set_packing(png_ptr);
1429#endif
1430
1431#if defined(PNG_WRITE_SWAP_ALPHA_SUPPORTED)
1432 /* swap location of alpha bytes from ARGB to RGBA */
1433 if (transforms & PNG_TRANSFORM_SWAP_ALPHA)
1434 png_set_swap_alpha(png_ptr);
1435#endif
1436
1437#if defined(PNG_WRITE_FILLER_SUPPORTED)
1438 /* Get rid of filler (OR ALPHA) bytes, pack XRGB/RGBX/ARGB/RGBA into
1439 * RGB (4 channels -> 3 channels). The second parameter is not used.
1440 */
1441 if (transforms & PNG_TRANSFORM_STRIP_FILLER)
1442 png_set_filler(png_ptr, 0, PNG_FILLER_BEFORE);
1443#endif
1444
1445#if defined(PNG_WRITE_BGR_SUPPORTED)
1446 /* flip BGR pixels to RGB */
1447 if (transforms & PNG_TRANSFORM_BGR)
1448 png_set_bgr(png_ptr);
1449#endif
1450
1451#if defined(PNG_WRITE_SWAP_SUPPORTED)
1452 /* swap bytes of 16-bit files to most significant byte first */
1453 if (transforms & PNG_TRANSFORM_SWAP_ENDIAN)
1454 png_set_swap(png_ptr);
1455#endif
1456
1457#if defined(PNG_WRITE_PACKSWAP_SUPPORTED)
1458 /* swap bits of 1, 2, 4 bit packed pixel formats */
1459 if (transforms & PNG_TRANSFORM_PACKSWAP)
1460 png_set_packswap(png_ptr);
1461#endif
1462
1463 /* ----------------------- end of transformations ------------------- */
1464
1465 /* write the bits */
1466 if (info_ptr->valid & PNG_INFO_IDAT)
1467 png_write_image(png_ptr, info_ptr->row_pointers);
1468
1469 /* It is REQUIRED to call this to finish writing the rest of the file */
1470 png_write_end(png_ptr, info_ptr);
Glenn Randers-Pehrson520a7642000-03-21 05:13:06 -06001471
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001472 transforms = transforms; /* quiet compiler warnings */
1473 params = params;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001474}
1475#endif
Glenn Randers-Pehrson19095602001-03-14 07:08:39 -06001476#endif /* PNG_WRITE_SUPPORTED */