blob: 4bfa066366d6442148aeaa4d30122e71a7cba80e [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-Pehrson145f5c82008-07-10 09:13:13 -05004 * Last changed in libpng 1.4.0 [July 10, 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);
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-Pehrson145f5c82008-07-10 09:13:13 -0500483 png_memcpy(png_ptr->jmpbuf, jmpbuf, png_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-Pehrson145f5c82008-07-10 09:13:13 -0500492 if (user_png_ver)
Guy Schalnate5a37791996-06-05 15:50:50 -0500493 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500494 i=0;
495 do
496 {
497 if (user_png_ver[i] != png_libpng_ver[i])
498 png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH;
499 } while (png_libpng_ver[i++]);
500 }
Guy Schalnat0d580581995-07-20 02:43:20 -0500501
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500502 if (png_ptr->flags & PNG_FLAG_LIBRARY_MISMATCH)
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500503 {
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500504 /* Libpng 0.90 and later are binary incompatible with libpng 0.89, so
505 * we must recompile any applications that use any older library version.
506 * For versions after libpng 1.0, we will be compatible, so we need
507 * only check the first digit.
508 */
509 if (user_png_ver == NULL || user_png_ver[0] != png_libpng_ver[0] ||
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500510 (user_png_ver[0] == '1' && user_png_ver[2] != png_libpng_ver[2]) ||
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500511 (user_png_ver[0] == '0' && user_png_ver[2] < '9'))
512 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500513#if !defined(PNG_NO_STDIO) && !defined(_WIN32_WCE)
Glenn Randers-Pehrsonb1828932001-06-23 08:03:17 -0500514 char msg[80];
515 if (user_png_ver)
516 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500517 png_snprintf(msg, 80,
518 "Application was compiled with png.h from libpng-%.20s",
Glenn Randers-Pehrsonb1828932001-06-23 08:03:17 -0500519 user_png_ver);
520 png_warning(png_ptr, msg);
521 }
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500522 png_snprintf(msg, 80,
523 "Application is running with png.c from libpng-%.20s",
Glenn Randers-Pehrsonb1828932001-06-23 08:03:17 -0500524 png_libpng_ver);
525 png_warning(png_ptr, msg);
526#endif
527#ifdef PNG_ERROR_NUMBERS_SUPPORTED
528 png_ptr->flags=0;
529#endif
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500530 png_error(png_ptr,
531 "Incompatible libpng version in application and library");
532 }
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500533 }
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500534
Guy Schalnat0d580581995-07-20 02:43:20 -0500535 /* initialize zbuf - compression buffer */
536 png_ptr->zbuf_size = PNG_ZBUF_SIZE;
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500537 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr,
538 (png_uint_32)png_ptr->zbuf_size);
Guy Schalnate5a37791996-06-05 15:50:50 -0500539
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500540 png_set_write_fn(png_ptr, NULL, NULL, NULL);
Guy Schalnate5a37791996-06-05 15:50:50 -0500541
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600542#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
543 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500544 1, NULL, NULL);
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600545#endif
546
Glenn Randers-Pehrson03008a02002-04-27 10:11:25 -0500547#ifdef PNG_SETJMP_SUPPORTED
548/* Applications that neglect to set up their own setjmp() and then encounter
549 a png_error() will longjmp here. Since the jmpbuf is then meaningless we
550 abort instead of returning. */
551#ifdef USE_FAR_KEYWORD
552 if (setjmp(jmpbuf))
553 PNG_ABORT();
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500554 png_memcpy(png_ptr->jmpbuf, jmpbuf, png_sizeof(jmp_buf));
Glenn Randers-Pehrson03008a02002-04-27 10:11:25 -0500555#else
556 if (setjmp(png_ptr->jmpbuf))
557 PNG_ABORT();
558#endif
559#endif
560 return (png_ptr);
Guy Schalnate5a37791996-06-05 15:50:50 -0500561}
562
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500563/* Initialize png_ptr structure, and allocate any memory needed */
564
565
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500566void PNGAPI
567png_write_init_3(png_structpp ptr_ptr, png_const_charp user_png_ver,
568 png_size_t png_struct_size)
569{
570 png_structp png_ptr=*ptr_ptr;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600571#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500572 jmp_buf tmp_jmp; /* to save current jump buffer */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600573#endif
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600574
Glenn Randers-Pehrson4deeb792006-03-22 16:21:59 -0600575 int i = 0;
576
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600577 if (png_ptr == NULL)
578 return;
579
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500580 do
581 {
582 if (user_png_ver[i] != png_libpng_ver[i])
583 {
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500584#ifdef PNG_LEGACY_SUPPORTED
585 png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH;
586#else
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500587 png_ptr->warning_fn=NULL;
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500588 png_warning(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500589 "Application uses deprecated png_write_init() and should be recompiled.");
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500590 break;
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500591#endif
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500592 }
593 } while (png_libpng_ver[i++]);
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500594
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500595 png_debug(1, "in png_write_init_3\n");
Glenn Randers-Pehrson98c9d732000-05-03 21:06:11 -0500596
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600597#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500598 /* save jump buffer and error functions */
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500599 png_memcpy(tmp_jmp, png_ptr->jmpbuf, png_sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600600#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500601
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500602 if (png_sizeof(png_struct) > png_struct_size)
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500603 {
604 png_destroy_struct(png_ptr);
605 png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG);
606 *ptr_ptr = png_ptr;
607 }
608
Guy Schalnate5a37791996-06-05 15:50:50 -0500609 /* reset all variables to 0 */
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500610 png_memset(png_ptr, 0, png_sizeof(png_struct));
Guy Schalnate5a37791996-06-05 15:50:50 -0500611
Glenn Randers-Pehrson272489d2004-08-04 06:34:52 -0500612 /* added at libpng-1.2.6 */
613#ifdef PNG_SET_USER_LIMITS_SUPPORTED
614 png_ptr->user_width_max=PNG_USER_WIDTH_MAX;
615 png_ptr->user_height_max=PNG_USER_HEIGHT_MAX;
616#endif
617
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600618#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500619 /* restore jump buffer */
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500620 png_memcpy(png_ptr->jmpbuf, tmp_jmp, png_sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600621#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500622
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500623 png_set_write_fn(png_ptr, NULL, NULL, NULL);
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500624
Guy Schalnate5a37791996-06-05 15:50:50 -0500625 /* initialize zbuf - compression buffer */
626 png_ptr->zbuf_size = PNG_ZBUF_SIZE;
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500627 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr,
628 (png_uint_32)png_ptr->zbuf_size);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500629
630#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
631 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500632 1, NULL, NULL);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500633#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500634}
635
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600636/* Write a few rows of image data. If the image is interlaced,
637 * either you will have to write the 7 sub images, or, if you
638 * have called png_set_interlace_handling(), you will have to
639 * "write" the image seven times.
640 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500641void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600642png_write_rows(png_structp png_ptr, png_bytepp row,
Guy Schalnat0d580581995-07-20 02:43:20 -0500643 png_uint_32 num_rows)
644{
645 png_uint_32 i; /* row counter */
Guy Schalnat6d764711995-12-19 03:22:19 -0600646 png_bytepp rp; /* row pointer */
Guy Schalnat0d580581995-07-20 02:43:20 -0500647
Andreas Dilger47a0c421997-05-16 02:46:07 -0500648 png_debug(1, "in png_write_rows\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600649
650 if (png_ptr == NULL)
651 return;
652
Guy Schalnat0d580581995-07-20 02:43:20 -0500653 /* loop through the rows */
654 for (i = 0, rp = row; i < num_rows; i++, rp++)
655 {
656 png_write_row(png_ptr, *rp);
657 }
658}
659
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600660/* Write the image. You only need to call this function once, even
661 * if you are writing an interlaced image.
662 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500663void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600664png_write_image(png_structp png_ptr, png_bytepp image)
Guy Schalnat0d580581995-07-20 02:43:20 -0500665{
666 png_uint_32 i; /* row index */
667 int pass, num_pass; /* pass variables */
Guy Schalnat6d764711995-12-19 03:22:19 -0600668 png_bytepp rp; /* points to current row */
Guy Schalnat0d580581995-07-20 02:43:20 -0500669
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600670 if (png_ptr == NULL)
671 return;
672
Andreas Dilger47a0c421997-05-16 02:46:07 -0500673 png_debug(1, "in png_write_image\n");
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600674#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500675 /* intialize interlace handling. If image is not interlaced,
676 this will set pass to 1 */
677 num_pass = png_set_interlace_handling(png_ptr);
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600678#else
679 num_pass = 1;
680#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500681 /* loop through passes */
682 for (pass = 0; pass < num_pass; pass++)
683 {
684 /* loop through image */
685 for (i = 0, rp = image; i < png_ptr->height; i++, rp++)
686 {
687 png_write_row(png_ptr, *rp);
688 }
689 }
690}
691
Guy Schalnate5a37791996-06-05 15:50:50 -0500692/* called by user to write a row of image data */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500693void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -0600694png_write_row(png_structp png_ptr, png_bytep row)
Guy Schalnat0d580581995-07-20 02:43:20 -0500695{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600696 if (png_ptr == NULL)
697 return;
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600698 png_debug2(1, "in png_write_row (row %ld, pass %d)\n",
699 png_ptr->row_number, png_ptr->pass);
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600700
Guy Schalnat0d580581995-07-20 02:43:20 -0500701 /* initialize transformations and other stuff if first time */
Guy Schalnat6d764711995-12-19 03:22:19 -0600702 if (png_ptr->row_number == 0 && png_ptr->pass == 0)
Guy Schalnat0d580581995-07-20 02:43:20 -0500703 {
Glenn Randers-Pehrson5cded0b2001-11-07 07:10:08 -0600704 /* make sure we wrote the header info */
705 if (!(png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE))
706 png_error(png_ptr,
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500707 "png_write_info was never called before png_write_row");
Glenn Randers-Pehrson5cded0b2001-11-07 07:10:08 -0600708
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500709 /* check for transforms that have been set but were defined out */
710#if !defined(PNG_WRITE_INVERT_SUPPORTED) && defined(PNG_READ_INVERT_SUPPORTED)
711 if (png_ptr->transformations & PNG_INVERT_MONO)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500712 png_warning(png_ptr, "PNG_WRITE_INVERT_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500713#endif
714#if !defined(PNG_WRITE_FILLER_SUPPORTED) && defined(PNG_READ_FILLER_SUPPORTED)
715 if (png_ptr->transformations & PNG_FILLER)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500716 png_warning(png_ptr, "PNG_WRITE_FILLER_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500717#endif
718#if !defined(PNG_WRITE_PACKSWAP_SUPPORTED) && defined(PNG_READ_PACKSWAP_SUPPORTED)
719 if (png_ptr->transformations & PNG_PACKSWAP)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500720 png_warning(png_ptr, "PNG_WRITE_PACKSWAP_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500721#endif
722#if !defined(PNG_WRITE_PACK_SUPPORTED) && defined(PNG_READ_PACK_SUPPORTED)
723 if (png_ptr->transformations & PNG_PACK)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500724 png_warning(png_ptr, "PNG_WRITE_PACK_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500725#endif
726#if !defined(PNG_WRITE_SHIFT_SUPPORTED) && defined(PNG_READ_SHIFT_SUPPORTED)
727 if (png_ptr->transformations & PNG_SHIFT)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500728 png_warning(png_ptr, "PNG_WRITE_SHIFT_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500729#endif
730#if !defined(PNG_WRITE_BGR_SUPPORTED) && defined(PNG_READ_BGR_SUPPORTED)
731 if (png_ptr->transformations & PNG_BGR)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500732 png_warning(png_ptr, "PNG_WRITE_BGR_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500733#endif
734#if !defined(PNG_WRITE_SWAP_SUPPORTED) && defined(PNG_READ_SWAP_SUPPORTED)
735 if (png_ptr->transformations & PNG_SWAP_BYTES)
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500736 png_warning(png_ptr, "PNG_WRITE_SWAP_SUPPORTED is not defined");
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500737#endif
738
Guy Schalnat0d580581995-07-20 02:43:20 -0500739 png_write_start_row(png_ptr);
740 }
741
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500742#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500743 /* if interlaced and not interested in row, return */
744 if (png_ptr->interlaced && (png_ptr->transformations & PNG_INTERLACE))
745 {
746 switch (png_ptr->pass)
747 {
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600748 case 0:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600749 if (png_ptr->row_number & 0x07)
Guy Schalnat0d580581995-07-20 02:43:20 -0500750 {
751 png_write_finish_row(png_ptr);
752 return;
753 }
754 break;
755 case 1:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600756 if ((png_ptr->row_number & 0x07) || png_ptr->width < 5)
Guy Schalnat0d580581995-07-20 02:43:20 -0500757 {
758 png_write_finish_row(png_ptr);
759 return;
760 }
761 break;
762 case 2:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600763 if ((png_ptr->row_number & 0x07) != 4)
Guy Schalnat0d580581995-07-20 02:43:20 -0500764 {
765 png_write_finish_row(png_ptr);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600766 return;
Guy Schalnat0d580581995-07-20 02:43:20 -0500767 }
768 break;
769 case 3:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600770 if ((png_ptr->row_number & 0x03) || png_ptr->width < 3)
Guy Schalnat0d580581995-07-20 02:43:20 -0500771 {
772 png_write_finish_row(png_ptr);
773 return;
774 }
775 break;
776 case 4:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600777 if ((png_ptr->row_number & 0x03) != 2)
Guy Schalnat0d580581995-07-20 02:43:20 -0500778 {
779 png_write_finish_row(png_ptr);
780 return;
781 }
782 break;
783 case 5:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600784 if ((png_ptr->row_number & 0x01) || png_ptr->width < 2)
Guy Schalnat0d580581995-07-20 02:43:20 -0500785 {
786 png_write_finish_row(png_ptr);
787 return;
788 }
789 break;
790 case 6:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600791 if (!(png_ptr->row_number & 0x01))
Guy Schalnat0d580581995-07-20 02:43:20 -0500792 {
793 png_write_finish_row(png_ptr);
794 return;
795 }
796 break;
797 }
798 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600799#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500800
801 /* set up row info for transformations */
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600802 png_ptr->row_info.color_type = png_ptr->color_type;
Guy Schalnat0d580581995-07-20 02:43:20 -0500803 png_ptr->row_info.width = png_ptr->usr_width;
804 png_ptr->row_info.channels = png_ptr->usr_channels;
805 png_ptr->row_info.bit_depth = png_ptr->usr_bit_depth;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600806 png_ptr->row_info.pixel_depth = (png_byte)(png_ptr->row_info.bit_depth *
807 png_ptr->row_info.channels);
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600808
Glenn Randers-Pehrson272489d2004-08-04 06:34:52 -0500809 png_ptr->row_info.rowbytes = PNG_ROWBYTES(png_ptr->row_info.pixel_depth,
810 png_ptr->row_info.width);
Guy Schalnat0d580581995-07-20 02:43:20 -0500811
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600812 png_debug1(3, "row_info->color_type = %d\n", png_ptr->row_info.color_type);
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500813 png_debug1(3, "row_info->width = %lu\n", png_ptr->row_info.width);
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600814 png_debug1(3, "row_info->channels = %d\n", png_ptr->row_info.channels);
815 png_debug1(3, "row_info->bit_depth = %d\n", png_ptr->row_info.bit_depth);
816 png_debug1(3, "row_info->pixel_depth = %d\n", png_ptr->row_info.pixel_depth);
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500817 png_debug1(3, "row_info->rowbytes = %lu\n", png_ptr->row_info.rowbytes);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500818
819 /* Copy user's row into buffer, leaving room for filter byte. */
Glenn Randers-Pehrsonbeb572e2006-08-19 13:59:24 -0500820 png_memcpy(png_ptr->row_buf + 1, row, png_ptr->row_info.rowbytes);
Guy Schalnat0d580581995-07-20 02:43:20 -0500821
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500822#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500823 /* handle interlacing */
824 if (png_ptr->interlaced && png_ptr->pass < 6 &&
825 (png_ptr->transformations & PNG_INTERLACE))
826 {
827 png_do_write_interlace(&(png_ptr->row_info),
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600828 png_ptr->row_buf + 1, png_ptr->pass);
Guy Schalnat0d580581995-07-20 02:43:20 -0500829 /* this should always get caught above, but still ... */
830 if (!(png_ptr->row_info.width))
831 {
832 png_write_finish_row(png_ptr);
833 return;
834 }
835 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600836#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500837
838 /* handle other transformations */
839 if (png_ptr->transformations)
840 png_do_write_transformations(png_ptr);
841
Glenn Randers-Pehrson2ad31ae2000-12-15 08:54:42 -0600842#if defined(PNG_MNG_FEATURES_SUPPORTED)
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -0600843 /* Write filter_method 64 (intrapixel differencing) only if
844 * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and
845 * 2. Libpng did not write a PNG signature (this filter_method is only
846 * used in PNG datastreams that are embedded in MNG datastreams) and
847 * 3. The application called png_permit_mng_features with a mask that
848 * included PNG_FLAG_MNG_FILTER_64 and
849 * 4. The filter_method is 64 and
850 * 5. The color_type is RGB or RGBA
851 */
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500852 if ((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
Glenn Randers-Pehrson2ad31ae2000-12-15 08:54:42 -0600853 (png_ptr->filter_type == PNG_INTRAPIXEL_DIFFERENCING))
854 {
855 /* Intrapixel differencing */
856 png_do_write_intrapixel(&(png_ptr->row_info), png_ptr->row_buf + 1);
857 }
858#endif
859
Andreas Dilger47a0c421997-05-16 02:46:07 -0500860 /* Find a filter if necessary, filter the row and write it out. */
Guy Schalnate5a37791996-06-05 15:50:50 -0500861 png_write_find_filter(png_ptr, &(png_ptr->row_info));
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -0600862
863 if (png_ptr->write_row_fn != NULL)
864 (*(png_ptr->write_row_fn))(png_ptr, png_ptr->row_number, png_ptr->pass);
Guy Schalnat0d580581995-07-20 02:43:20 -0500865}
866
Guy Schalnat0f716451995-11-28 11:22:13 -0600867#if defined(PNG_WRITE_FLUSH_SUPPORTED)
868/* Set the automatic flush interval or 0 to turn flushing off */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500869void PNGAPI
Guy Schalnat4ee97b01996-01-16 01:51:56 -0600870png_set_flush(png_structp png_ptr, int nrows)
Guy Schalnat0f716451995-11-28 11:22:13 -0600871{
Andreas Dilger47a0c421997-05-16 02:46:07 -0500872 png_debug(1, "in png_set_flush\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600873 if (png_ptr == NULL)
874 return;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600875 png_ptr->flush_dist = (nrows < 0 ? 0 : nrows);
Guy Schalnat0f716451995-11-28 11:22:13 -0600876}
877
878/* flush the current output buffers now */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500879void PNGAPI
Guy Schalnat4ee97b01996-01-16 01:51:56 -0600880png_write_flush(png_structp png_ptr)
Guy Schalnat0f716451995-11-28 11:22:13 -0600881{
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600882 int wrote_IDAT;
Guy Schalnat0f716451995-11-28 11:22:13 -0600883
Andreas Dilger47a0c421997-05-16 02:46:07 -0500884 png_debug(1, "in png_write_flush\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -0600885 if (png_ptr == NULL)
886 return;
Guy Schalnate5a37791996-06-05 15:50:50 -0500887 /* We have already written out all of the data */
888 if (png_ptr->row_number >= png_ptr->num_rows)
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600889 return;
Guy Schalnat0f716451995-11-28 11:22:13 -0600890
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600891 do
892 {
893 int ret;
Guy Schalnat0f716451995-11-28 11:22:13 -0600894
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600895 /* compress the data */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600896 ret = deflate(&png_ptr->zstream, Z_SYNC_FLUSH);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600897 wrote_IDAT = 0;
Guy Schalnat0f716451995-11-28 11:22:13 -0600898
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600899 /* check for compression errors */
900 if (ret != Z_OK)
901 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500902 if (png_ptr->zstream.msg != NULL)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600903 png_error(png_ptr, png_ptr->zstream.msg);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600904 else
905 png_error(png_ptr, "zlib error");
906 }
Guy Schalnat0f716451995-11-28 11:22:13 -0600907
Andreas Dilger47a0c421997-05-16 02:46:07 -0500908 if (!(png_ptr->zstream.avail_out))
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600909 {
910 /* write the IDAT and reset the zlib output buffer */
911 png_write_IDAT(png_ptr, png_ptr->zbuf,
912 png_ptr->zbuf_size);
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600913 png_ptr->zstream.next_out = png_ptr->zbuf;
914 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600915 wrote_IDAT = 1;
916 }
917 } while(wrote_IDAT == 1);
Guy Schalnat0f716451995-11-28 11:22:13 -0600918
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600919 /* If there is any data left to be output, write it into a new IDAT */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600920 if (png_ptr->zbuf_size != png_ptr->zstream.avail_out)
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600921 {
922 /* write the IDAT and reset the zlib output buffer */
923 png_write_IDAT(png_ptr, png_ptr->zbuf,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600924 png_ptr->zbuf_size - png_ptr->zstream.avail_out);
925 png_ptr->zstream.next_out = png_ptr->zbuf;
926 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600927 }
928 png_ptr->flush_rows = 0;
929 png_flush(png_ptr);
Guy Schalnat0f716451995-11-28 11:22:13 -0600930}
931#endif /* PNG_WRITE_FLUSH_SUPPORTED */
932
Guy Schalnate5a37791996-06-05 15:50:50 -0500933/* free all memory used by the write */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -0500934void PNGAPI
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600935png_destroy_write_struct(png_structpp png_ptr_ptr, png_infopp info_ptr_ptr)
Guy Schalnate5a37791996-06-05 15:50:50 -0500936{
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600937 png_structp png_ptr = NULL;
938 png_infop info_ptr = NULL;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500939#ifdef PNG_USER_MEM_SUPPORTED
940 png_free_ptr free_fn = NULL;
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500941 png_voidp mem_ptr = NULL;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500942#endif
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600943
Andreas Dilger47a0c421997-05-16 02:46:07 -0500944 png_debug(1, "in png_destroy_write_struct\n");
945 if (png_ptr_ptr != NULL)
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500946 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600947 png_ptr = *png_ptr_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500948#ifdef PNG_USER_MEM_SUPPORTED
949 free_fn = png_ptr->free_fn;
Glenn Randers-Pehrsonfcbd7872002-04-07 16:35:38 -0500950 mem_ptr = png_ptr->mem_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500951#endif
952 }
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600953
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500954#ifdef PNG_USER_MEM_SUPPORTED
955 if (png_ptr != NULL)
956 {
957 free_fn = png_ptr->free_fn;
958 mem_ptr = png_ptr->mem_ptr;
959 }
960#endif
961
Andreas Dilger47a0c421997-05-16 02:46:07 -0500962 if (info_ptr_ptr != NULL)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600963 info_ptr = *info_ptr_ptr;
964
Andreas Dilger47a0c421997-05-16 02:46:07 -0500965 if (info_ptr != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500966 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500967 if (png_ptr != NULL)
968 {
969 png_free_data(png_ptr, info_ptr, PNG_FREE_ALL, -1);
Glenn Randers-Pehrsona77ef622000-02-18 13:48:52 -0600970
Glenn Randers-Pehrson6942d532000-05-01 09:31:54 -0500971#if defined(PNG_UNKNOWN_CHUNKS_SUPPORTED)
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500972 if (png_ptr->num_chunk_list)
973 {
974 png_free(png_ptr, png_ptr->chunk_list);
975 png_ptr->chunk_list=NULL;
976 png_ptr->num_chunk_list=0;
977 }
Glenn Randers-Pehrson6942d532000-05-01 09:31:54 -0500978#endif
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -0500979 }
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600980
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500981#ifdef PNG_USER_MEM_SUPPORTED
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500982 png_destroy_struct_2((png_voidp)info_ptr, (png_free_ptr)free_fn,
983 (png_voidp)mem_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500984#else
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600985 png_destroy_struct((png_voidp)info_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500986#endif
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500987 *info_ptr_ptr = NULL;
Guy Schalnate5a37791996-06-05 15:50:50 -0500988 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600989
Andreas Dilger47a0c421997-05-16 02:46:07 -0500990 if (png_ptr != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500991 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600992 png_write_destroy(png_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500993#ifdef PNG_USER_MEM_SUPPORTED
Glenn Randers-Pehrson8b6a8892001-05-18 04:54:50 -0500994 png_destroy_struct_2((png_voidp)png_ptr, (png_free_ptr)free_fn,
995 (png_voidp)mem_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500996#else
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600997 png_destroy_struct((png_voidp)png_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500998#endif
Glenn Randers-Pehrson3f549252001-10-27 07:35:13 -0500999 *png_ptr_ptr = NULL;
Guy Schalnate5a37791996-06-05 15:50:50 -05001000 }
1001}
1002
1003
Andreas Dilger47a0c421997-05-16 02:46:07 -05001004/* Free any memory used in png_ptr struct (old method) */
Glenn Randers-Pehrsonf64a06f2001-04-11 07:38:00 -05001005void /* PRIVATE */
Guy Schalnat6d764711995-12-19 03:22:19 -06001006png_write_destroy(png_structp png_ptr)
Guy Schalnat0d580581995-07-20 02:43:20 -05001007{
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001008#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnatb2e01bd1996-01-26 01:38:47 -06001009 jmp_buf tmp_jmp; /* save jump buffer */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001010#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001011 png_error_ptr error_fn;
1012 png_error_ptr warning_fn;
1013 png_voidp error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -05001014#ifdef PNG_USER_MEM_SUPPORTED
1015 png_free_ptr free_fn;
1016#endif
Guy Schalnat0d580581995-07-20 02:43:20 -05001017
Andreas Dilger47a0c421997-05-16 02:46:07 -05001018 png_debug(1, "in png_write_destroy\n");
Guy Schalnat0d580581995-07-20 02:43:20 -05001019 /* free any memory zlib uses */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -06001020 deflateEnd(&png_ptr->zstream);
Guy Schalnate5a37791996-06-05 15:50:50 -05001021
Guy Schalnat0d580581995-07-20 02:43:20 -05001022 /* free our memory. png_free checks NULL for us. */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -06001023 png_free(png_ptr, png_ptr->zbuf);
1024 png_free(png_ptr, png_ptr->row_buf);
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001025#ifndef PNG_NO_WRITE_FILTER
Andreas Dilger02ad0ef1997-01-17 01:34:35 -06001026 png_free(png_ptr, png_ptr->prev_row);
1027 png_free(png_ptr, png_ptr->sub_row);
1028 png_free(png_ptr, png_ptr->up_row);
1029 png_free(png_ptr, png_ptr->avg_row);
1030 png_free(png_ptr, png_ptr->paeth_row);
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001031#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001032
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -06001033#if defined(PNG_TIME_RFC1123_SUPPORTED)
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06001034 png_free(png_ptr, png_ptr->time_buffer);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001035#endif
1036
Andreas Dilger47a0c421997-05-16 02:46:07 -05001037#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
1038 png_free(png_ptr, png_ptr->prev_filters);
1039 png_free(png_ptr, png_ptr->filter_weights);
1040 png_free(png_ptr, png_ptr->inv_filter_weights);
1041 png_free(png_ptr, png_ptr->filter_costs);
1042 png_free(png_ptr, png_ptr->inv_filter_costs);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001043#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001044
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001045#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnat0d580581995-07-20 02:43:20 -05001046 /* reset structure */
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001047 png_memcpy(tmp_jmp, png_ptr->jmpbuf, png_sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001048#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001049
1050 error_fn = png_ptr->error_fn;
1051 warning_fn = png_ptr->warning_fn;
1052 error_ptr = png_ptr->error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -05001053#ifdef PNG_USER_MEM_SUPPORTED
1054 free_fn = png_ptr->free_fn;
1055#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001056
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001057 png_memset(png_ptr, 0, png_sizeof(png_struct));
Guy Schalnate5a37791996-06-05 15:50:50 -05001058
1059 png_ptr->error_fn = error_fn;
1060 png_ptr->warning_fn = warning_fn;
1061 png_ptr->error_ptr = error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -05001062#ifdef PNG_USER_MEM_SUPPORTED
1063 png_ptr->free_fn = free_fn;
1064#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001065
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001066#ifdef PNG_SETJMP_SUPPORTED
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001067 png_memcpy(png_ptr->jmpbuf, tmp_jmp, png_sizeof(jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001068#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001069}
Guy Schalnate5a37791996-06-05 15:50:50 -05001070
Andreas Dilger47a0c421997-05-16 02:46:07 -05001071/* Allow the application to select one or more row filters to use. */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001072void PNGAPI
Guy Schalnate5a37791996-06-05 15:50:50 -05001073png_set_filter(png_structp png_ptr, int method, int filters)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001074{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001075 png_debug(1, "in png_set_filter\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001076 if (png_ptr == NULL)
1077 return;
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -06001078#if defined(PNG_MNG_FEATURES_SUPPORTED)
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001079 if ((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
Glenn Randers-Pehrson408b4212000-12-18 09:33:57 -06001080 (method == PNG_INTRAPIXEL_DIFFERENCING))
1081 method = PNG_FILTER_TYPE_BASE;
1082#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -05001083 if (method == PNG_FILTER_TYPE_BASE)
Guy Schalnate5a37791996-06-05 15:50:50 -05001084 {
1085 switch (filters & (PNG_ALL_FILTERS | 0x07))
1086 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001087#ifndef PNG_NO_WRITE_FILTER
Guy Schalnate5a37791996-06-05 15:50:50 -05001088 case 5:
1089 case 6:
Andreas Dilger47a0c421997-05-16 02:46:07 -05001090 case 7: png_warning(png_ptr, "Unknown row filter for method 0");
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001091#endif /* PNG_NO_WRITE_FILTER */
1092 case PNG_FILTER_VALUE_NONE:
1093 png_ptr->do_filter=PNG_FILTER_NONE; break;
1094#ifndef PNG_NO_WRITE_FILTER
1095 case PNG_FILTER_VALUE_SUB:
1096 png_ptr->do_filter=PNG_FILTER_SUB; break;
1097 case PNG_FILTER_VALUE_UP:
1098 png_ptr->do_filter=PNG_FILTER_UP; break;
1099 case PNG_FILTER_VALUE_AVG:
1100 png_ptr->do_filter=PNG_FILTER_AVG; break;
1101 case PNG_FILTER_VALUE_PAETH:
1102 png_ptr->do_filter=PNG_FILTER_PAETH; break;
Guy Schalnate5a37791996-06-05 15:50:50 -05001103 default: png_ptr->do_filter = (png_byte)filters; break;
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001104#else
1105 default: png_warning(png_ptr, "Unknown row filter for method 0");
1106#endif /* PNG_NO_WRITE_FILTER */
Guy Schalnate5a37791996-06-05 15:50:50 -05001107 }
1108
Andreas Dilger47a0c421997-05-16 02:46:07 -05001109 /* If we have allocated the row_buf, this means we have already started
1110 * with the image and we should have allocated all of the filter buffers
1111 * that have been selected. If prev_row isn't already allocated, then
1112 * it is too late to start using the filters that need it, since we
1113 * will be missing the data in the previous row. If an application
1114 * wants to start and stop using particular filters during compression,
1115 * it should start out with all of the filters, and then add and
1116 * remove them after the start of compression.
Guy Schalnate5a37791996-06-05 15:50:50 -05001117 */
Andreas Dilger47a0c421997-05-16 02:46:07 -05001118 if (png_ptr->row_buf != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001119 {
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001120#ifndef PNG_NO_WRITE_FILTER
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001121 if ((png_ptr->do_filter & PNG_FILTER_SUB) && png_ptr->sub_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001122 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001123 png_ptr->sub_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->sub_row[0] = PNG_FILTER_VALUE_SUB;
Guy Schalnate5a37791996-06-05 15:50:50 -05001126 }
1127
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001128 if ((png_ptr->do_filter & PNG_FILTER_UP) && png_ptr->up_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001129 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001130 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001131 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001132 png_warning(png_ptr, "Can't add Up filter after starting");
Guy Schalnate5a37791996-06-05 15:50:50 -05001133 png_ptr->do_filter &= ~PNG_FILTER_UP;
1134 }
1135 else
1136 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001137 png_ptr->up_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001138 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001139 png_ptr->up_row[0] = PNG_FILTER_VALUE_UP;
Guy Schalnate5a37791996-06-05 15:50:50 -05001140 }
1141 }
1142
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001143 if ((png_ptr->do_filter & PNG_FILTER_AVG) && png_ptr->avg_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001144 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001145 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001146 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001147 png_warning(png_ptr, "Can't add Average filter after starting");
Guy Schalnate5a37791996-06-05 15:50:50 -05001148 png_ptr->do_filter &= ~PNG_FILTER_AVG;
1149 }
1150 else
1151 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001152 png_ptr->avg_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001153 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001154 png_ptr->avg_row[0] = PNG_FILTER_VALUE_AVG;
Guy Schalnate5a37791996-06-05 15:50:50 -05001155 }
1156 }
1157
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001158 if ((png_ptr->do_filter & PNG_FILTER_PAETH) &&
Andreas Dilger47a0c421997-05-16 02:46:07 -05001159 png_ptr->paeth_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001160 {
Andreas Dilger47a0c421997-05-16 02:46:07 -05001161 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -05001162 {
1163 png_warning(png_ptr, "Can't add Paeth filter after starting");
Glenn Randers-Pehrson860ab2b1999-10-14 07:43:10 -05001164 png_ptr->do_filter &= (png_byte)(~PNG_FILTER_PAETH);
Guy Schalnate5a37791996-06-05 15:50:50 -05001165 }
1166 else
1167 {
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -06001168 png_ptr->paeth_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001169 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001170 png_ptr->paeth_row[0] = PNG_FILTER_VALUE_PAETH;
Guy Schalnate5a37791996-06-05 15:50:50 -05001171 }
1172 }
1173
1174 if (png_ptr->do_filter == PNG_NO_FILTERS)
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001175#endif /* PNG_NO_WRITE_FILTER */
Guy Schalnate5a37791996-06-05 15:50:50 -05001176 png_ptr->do_filter = PNG_FILTER_NONE;
1177 }
1178 }
1179 else
Andreas Dilger47a0c421997-05-16 02:46:07 -05001180 png_error(png_ptr, "Unknown custom filter method");
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001181}
1182
Andreas Dilger47a0c421997-05-16 02:46:07 -05001183/* This allows us to influence the way in which libpng chooses the "best"
1184 * filter for the current scanline. While the "minimum-sum-of-absolute-
1185 * differences metric is relatively fast and effective, there is some
1186 * question as to whether it can be improved upon by trying to keep the
1187 * filtered data going to zlib more consistent, hopefully resulting in
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06001188 * better compression.
1189 */
Andreas Dilger47a0c421997-05-16 02:46:07 -05001190#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED) /* GRR 970116 */
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001191void PNGAPI
Andreas Dilger47a0c421997-05-16 02:46:07 -05001192png_set_filter_heuristics(png_structp png_ptr, int heuristic_method,
1193 int num_weights, png_doublep filter_weights,
1194 png_doublep filter_costs)
1195{
1196 int i;
1197
1198 png_debug(1, "in png_set_filter_heuristics\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001199 if (png_ptr == NULL)
1200 return;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001201 if (heuristic_method >= PNG_FILTER_HEURISTIC_LAST)
1202 {
1203 png_warning(png_ptr, "Unknown filter heuristic method");
1204 return;
1205 }
1206
1207 if (heuristic_method == PNG_FILTER_HEURISTIC_DEFAULT)
1208 {
1209 heuristic_method = PNG_FILTER_HEURISTIC_UNWEIGHTED;
1210 }
1211
1212 if (num_weights < 0 || filter_weights == NULL ||
1213 heuristic_method == PNG_FILTER_HEURISTIC_UNWEIGHTED)
1214 {
1215 num_weights = 0;
1216 }
1217
Glenn Randers-Pehrson860ab2b1999-10-14 07:43:10 -05001218 png_ptr->num_prev_filters = (png_byte)num_weights;
1219 png_ptr->heuristic_method = (png_byte)heuristic_method;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001220
1221 if (num_weights > 0)
1222 {
1223 if (png_ptr->prev_filters == NULL)
1224 {
1225 png_ptr->prev_filters = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001226 (png_uint_32)(png_sizeof(png_byte) * num_weights));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001227
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001228 /* To make sure that the weighting starts out fairly */
1229 for (i = 0; i < num_weights; i++)
Andreas Dilger47a0c421997-05-16 02:46:07 -05001230 {
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001231 png_ptr->prev_filters[i] = 255;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001232 }
1233 }
1234
1235 if (png_ptr->filter_weights == NULL)
1236 {
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001237 png_ptr->filter_weights = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001238 (png_uint_32)(png_sizeof(png_uint_16) * num_weights));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001239
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001240 png_ptr->inv_filter_weights = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001241 (png_uint_32)(png_sizeof(png_uint_16) * num_weights));
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001242 for (i = 0; i < num_weights; i++)
Andreas Dilger47a0c421997-05-16 02:46:07 -05001243 {
Glenn Randers-Pehrson82ae3832001-04-20 10:32:10 -05001244 png_ptr->inv_filter_weights[i] =
1245 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001246 }
1247 }
1248
1249 for (i = 0; i < num_weights; i++)
1250 {
1251 if (filter_weights[i] < 0.0)
1252 {
1253 png_ptr->inv_filter_weights[i] =
1254 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
1255 }
1256 else
1257 {
1258 png_ptr->inv_filter_weights[i] =
1259 (png_uint_16)((double)PNG_WEIGHT_FACTOR*filter_weights[i]+0.5);
1260 png_ptr->filter_weights[i] =
1261 (png_uint_16)((double)PNG_WEIGHT_FACTOR/filter_weights[i]+0.5);
1262 }
1263 }
1264 }
1265
1266 /* If, in the future, there are other filter methods, this would
1267 * need to be based on png_ptr->filter.
1268 */
1269 if (png_ptr->filter_costs == NULL)
1270 {
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001271 png_ptr->filter_costs = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001272 (png_uint_32)(png_sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001273
Glenn Randers-Pehrson87c6bc92001-04-03 22:43:19 -05001274 png_ptr->inv_filter_costs = (png_uint_16p)png_malloc(png_ptr,
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001275 (png_uint_32)(png_sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001276
1277 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
1278 {
1279 png_ptr->inv_filter_costs[i] =
1280 png_ptr->filter_costs[i] = PNG_COST_FACTOR;
1281 }
1282 }
1283
1284 /* Here is where we set the relative costs of the different filters. We
1285 * should take the desired compression level into account when setting
1286 * the costs, so that Paeth, for instance, has a high relative cost at low
1287 * compression levels, while it has a lower relative cost at higher
1288 * compression settings. The filter types are in order of increasing
1289 * relative cost, so it would be possible to do this with an algorithm.
1290 */
1291 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
1292 {
1293 if (filter_costs == NULL || filter_costs[i] < 0.0)
1294 {
1295 png_ptr->inv_filter_costs[i] =
1296 png_ptr->filter_costs[i] = PNG_COST_FACTOR;
1297 }
1298 else if (filter_costs[i] >= 1.0)
1299 {
1300 png_ptr->inv_filter_costs[i] =
1301 (png_uint_16)((double)PNG_COST_FACTOR / filter_costs[i] + 0.5);
1302 png_ptr->filter_costs[i] =
1303 (png_uint_16)((double)PNG_COST_FACTOR * filter_costs[i] + 0.5);
1304 }
1305 }
1306}
1307#endif /* PNG_WRITE_WEIGHTED_FILTER_SUPPORTED */
1308
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001309void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001310png_set_compression_level(png_structp png_ptr, int level)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001311{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001312 png_debug(1, "in png_set_compression_level\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001313 if (png_ptr == NULL)
1314 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001315 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_LEVEL;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001316 png_ptr->zlib_level = level;
1317}
1318
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001319void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001320png_set_compression_mem_level(png_structp png_ptr, int mem_level)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001321{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001322 png_debug(1, "in png_set_compression_mem_level\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001323 if (png_ptr == NULL)
1324 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001325 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_MEM_LEVEL;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -06001326 png_ptr->zlib_mem_level = mem_level;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001327}
1328
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001329void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001330png_set_compression_strategy(png_structp png_ptr, int strategy)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001331{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001332 png_debug(1, "in png_set_compression_strategy\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001333 if (png_ptr == NULL)
1334 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001335 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_STRATEGY;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001336 png_ptr->zlib_strategy = strategy;
1337}
1338
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001339void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001340png_set_compression_window_bits(png_structp png_ptr, int window_bits)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001341{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001342 if (png_ptr == NULL)
1343 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001344 if (window_bits > 15)
1345 png_warning(png_ptr, "Only compression windows <= 32k supported by PNG");
Glenn Randers-Pehrson4393a9a1999-09-17 12:27:26 -05001346 else if (window_bits < 8)
1347 png_warning(png_ptr, "Only compression windows >= 256 supported by PNG");
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001348#ifndef WBITS_8_OK
1349 /* avoid libpng bug with 256-byte windows */
1350 if (window_bits == 8)
1351 {
1352 png_warning(png_ptr, "Compression window is being reset to 512");
1353 window_bits=9;
1354 }
1355#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001356 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_WINDOW_BITS;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001357 png_ptr->zlib_window_bits = window_bits;
1358}
1359
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001360void PNGAPI
Guy Schalnat6d764711995-12-19 03:22:19 -06001361png_set_compression_method(png_structp png_ptr, int method)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001362{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001363 png_debug(1, "in png_set_compression_method\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001364 if (png_ptr == NULL)
1365 return;
Guy Schalnate5a37791996-06-05 15:50:50 -05001366 if (method != 8)
1367 png_warning(png_ptr, "Only compression method 8 is supported by PNG");
1368 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_METHOD;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001369 png_ptr->zlib_method = method;
Guy Schalnat0d580581995-07-20 02:43:20 -05001370}
1371
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001372void PNGAPI
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001373png_set_write_status_fn(png_structp png_ptr, png_write_status_ptr write_row_fn)
1374{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001375 if (png_ptr == NULL)
1376 return;
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001377 png_ptr->write_row_fn = write_row_fn;
1378}
1379
1380#if defined(PNG_WRITE_USER_TRANSFORM_SUPPORTED)
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001381void PNGAPI
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001382png_set_write_user_transform_fn(png_structp png_ptr, png_user_transform_ptr
1383 write_user_transform_fn)
1384{
1385 png_debug(1, "in png_set_write_user_transform_fn\n");
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001386 if (png_ptr == NULL)
1387 return;
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001388 png_ptr->transformations |= PNG_USER_TRANSFORM;
1389 png_ptr->write_user_transform_fn = write_user_transform_fn;
1390}
1391#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001392
1393
1394#if defined(PNG_INFO_IMAGE_SUPPORTED)
Glenn Randers-Pehrson75294572000-05-06 14:09:57 -05001395void PNGAPI
1396png_write_png(png_structp png_ptr, png_infop info_ptr,
Glenn Randers-Pehrson1ef65b62000-05-12 06:19:53 -05001397 int transforms, voidp params)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001398{
Glenn Randers-Pehrson170b70c2006-03-10 10:19:04 -06001399 if (png_ptr == NULL || info_ptr == NULL)
1400 return;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001401#if defined(PNG_WRITE_INVERT_ALPHA_SUPPORTED)
1402 /* invert the alpha channel from opacity to transparency */
1403 if (transforms & PNG_TRANSFORM_INVERT_ALPHA)
1404 png_set_invert_alpha(png_ptr);
1405#endif
1406
1407 /* Write the file header information. */
1408 png_write_info(png_ptr, info_ptr);
1409
1410 /* ------ these transformations don't touch the info structure ------- */
1411
1412#if defined(PNG_WRITE_INVERT_SUPPORTED)
1413 /* invert monochrome pixels */
1414 if (transforms & PNG_TRANSFORM_INVERT_MONO)
1415 png_set_invert_mono(png_ptr);
1416#endif
1417
1418#if defined(PNG_WRITE_SHIFT_SUPPORTED)
1419 /* Shift the pixels up to a legal bit depth and fill in
1420 * as appropriate to correctly scale the image.
1421 */
1422 if ((transforms & PNG_TRANSFORM_SHIFT)
1423 && (info_ptr->valid & PNG_INFO_sBIT))
1424 png_set_shift(png_ptr, &info_ptr->sig_bit);
1425#endif
1426
1427#if defined(PNG_WRITE_PACK_SUPPORTED)
1428 /* pack pixels into bytes */
1429 if (transforms & PNG_TRANSFORM_PACKING)
1430 png_set_packing(png_ptr);
1431#endif
1432
1433#if defined(PNG_WRITE_SWAP_ALPHA_SUPPORTED)
1434 /* swap location of alpha bytes from ARGB to RGBA */
1435 if (transforms & PNG_TRANSFORM_SWAP_ALPHA)
1436 png_set_swap_alpha(png_ptr);
1437#endif
1438
1439#if defined(PNG_WRITE_FILLER_SUPPORTED)
1440 /* Get rid of filler (OR ALPHA) bytes, pack XRGB/RGBX/ARGB/RGBA into
1441 * RGB (4 channels -> 3 channels). The second parameter is not used.
1442 */
1443 if (transforms & PNG_TRANSFORM_STRIP_FILLER)
1444 png_set_filler(png_ptr, 0, PNG_FILLER_BEFORE);
1445#endif
1446
1447#if defined(PNG_WRITE_BGR_SUPPORTED)
1448 /* flip BGR pixels to RGB */
1449 if (transforms & PNG_TRANSFORM_BGR)
1450 png_set_bgr(png_ptr);
1451#endif
1452
1453#if defined(PNG_WRITE_SWAP_SUPPORTED)
1454 /* swap bytes of 16-bit files to most significant byte first */
1455 if (transforms & PNG_TRANSFORM_SWAP_ENDIAN)
1456 png_set_swap(png_ptr);
1457#endif
1458
1459#if defined(PNG_WRITE_PACKSWAP_SUPPORTED)
1460 /* swap bits of 1, 2, 4 bit packed pixel formats */
1461 if (transforms & PNG_TRANSFORM_PACKSWAP)
1462 png_set_packswap(png_ptr);
1463#endif
1464
1465 /* ----------------------- end of transformations ------------------- */
1466
1467 /* write the bits */
1468 if (info_ptr->valid & PNG_INFO_IDAT)
1469 png_write_image(png_ptr, info_ptr->row_pointers);
1470
1471 /* It is REQUIRED to call this to finish writing the rest of the file */
1472 png_write_end(png_ptr, info_ptr);
Glenn Randers-Pehrson520a7642000-03-21 05:13:06 -06001473
Glenn Randers-Pehrson145f5c82008-07-10 09:13:13 -05001474 transforms = transforms; /* quiet compiler warnings */
1475 params = params;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001476}
1477#endif
Glenn Randers-Pehrson19095602001-03-14 07:08:39 -06001478#endif /* PNG_WRITE_SUPPORTED */