blob: e28caf3d0df8f79747f81343015b21165c783f37 [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-Pehrson6942d532000-05-01 09:31:54 -05004 * libpng 1.0.6i - May 1, 2000
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06005 * For conditions of distribution and use, see copyright notice in png.h
6 * Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.
7 * Copyright (c) 1996, 1997 Andreas Dilger
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06008 * Copyright (c) 1998, 1999, 2000 Glenn Randers-Pehrson
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06009 */
Guy Schalnat0d580581995-07-20 02:43:20 -050010
11/* get internal access to png.h */
12#define PNG_INTERNAL
13#include "png.h"
14
Andreas Dilger47a0c421997-05-16 02:46:07 -050015/* Writes all the PNG information. This is the suggested way to use the
16 * library. If you have a new chunk to add, make a function to write it,
17 * and put it in the correct location here. If you want the chunk written
Glenn Randers-Pehrson345bc271998-06-14 14:43:31 -050018 * after the image data, put it in png_write_end(). I strongly encourage
Andreas Dilger47a0c421997-05-16 02:46:07 -050019 * you to supply a PNG_INFO_ flag, and check info_ptr->valid before writing
20 * the chunk, as that will keep the code from breaking if you want to just
21 * write a plain PNG file. If you have long comments, I suggest writing
22 * them in png_write_end(), and compressing them.
23 */
Guy Schalnat0d580581995-07-20 02:43:20 -050024void
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -060025png_write_info_before_PLTE(png_structp png_ptr, png_infop info_ptr)
Guy Schalnat0d580581995-07-20 02:43:20 -050026{
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -060027 png_debug(1, "in png_write_info_before_PLTE\n");
28 if (!(png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE))
29 {
Guy Schalnat0d580581995-07-20 02:43:20 -050030 png_write_sig(png_ptr); /* write PNG signature */
31 /* write IHDR information. */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060032 png_write_IHDR(png_ptr, info_ptr->width, info_ptr->height,
33 info_ptr->bit_depth, info_ptr->color_type, info_ptr->compression_type,
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -060034 info_ptr->filter_type,
35#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
36 info_ptr->interlace_type);
37#else
38 0);
39#endif
Guy Schalnat0d580581995-07-20 02:43:20 -050040 /* the rest of these check to see if the valid field has the appropriate
41 flag set, and if it does, writes the chunk. */
Guy Schalnat51f0eb41995-09-26 05:22:39 -050042#if defined(PNG_WRITE_gAMA_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060043 if (info_ptr->valid & PNG_INFO_gAMA)
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060044 {
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060045# ifdef PNG_FLOATING_POINT_SUPPORTED
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060046 png_write_gAMA(png_ptr, info_ptr->gamma);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -060047#else
48#ifdef PNG_FIXED_POINT_SUPPORTED
49 png_write_gAMA_fixed(png_ptr, info_ptr->int_gamma);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060050# endif
51#endif
52 }
Guy Schalnat51f0eb41995-09-26 05:22:39 -050053#endif
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -060054#if defined(PNG_WRITE_sRGB_SUPPORTED)
55 if (info_ptr->valid & PNG_INFO_sRGB)
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -060056 png_write_sRGB(png_ptr, (int)info_ptr->srgb_intent);
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -060057#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060058#if defined(PNG_WRITE_iCCP_SUPPORTED)
59 if (info_ptr->valid & PNG_INFO_iCCP)
60 png_write_iCCP(png_ptr, info_ptr->iccp_name, PNG_TEXT_COMPRESSION_NONE,
61 info_ptr->iccp_profile, (int)info_ptr->iccp_proflen);
62#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -050063#if defined(PNG_WRITE_sBIT_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060064 if (info_ptr->valid & PNG_INFO_sBIT)
65 png_write_sBIT(png_ptr, &(info_ptr->sig_bit), info_ptr->color_type);
Guy Schalnat51f0eb41995-09-26 05:22:39 -050066#endif
67#if defined(PNG_WRITE_cHRM_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060068 if (info_ptr->valid & PNG_INFO_cHRM)
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060069 {
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -060070#ifdef PNG_FLOATING_POINT_SUPPORTED
Guy Schalnat0d580581995-07-20 02:43:20 -050071 png_write_cHRM(png_ptr,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -060072 info_ptr->x_white, info_ptr->y_white,
73 info_ptr->x_red, info_ptr->y_red,
74 info_ptr->x_green, info_ptr->y_green,
75 info_ptr->x_blue, info_ptr->y_blue);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -060076#else
77# ifdef PNG_FIXED_POINT_SUPPORTED
78 png_write_cHRM_fixed(png_ptr,
79 info_ptr->int_x_white, info_ptr->int_y_white,
80 info_ptr->int_x_red, info_ptr->int_y_red,
81 info_ptr->int_x_green, info_ptr->int_y_green,
82 info_ptr->int_x_blue, info_ptr->int_y_blue);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -060083# endif
84#endif
85 }
86#endif
87#if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
88 if (info_ptr->unknown_chunks_num)
89 {
90 png_unknown_chunk *up;
91
92 png_debug(5, "writing extra chunks\n");
93
94 for (up = info_ptr->unknown_chunks;
95 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
96 up++)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -060097 {
98 int keep=png_handle_as_unknown(png_ptr, up->name);
99 if (keep != HANDLE_CHUNK_NEVER &&
100 up->location && (!(up->location & PNG_HAVE_PLTE)) &&
101 ((up->name[3] & 0x20) || keep == HANDLE_CHUNK_ALWAYS ||
102 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
103 {
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600104 png_write_chunk(png_ptr, up->name, up->data, up->size);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600105 }
106 }
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600107 }
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500108#endif
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600109 png_ptr->mode |= PNG_WROTE_INFO_BEFORE_PLTE;
110 }
111}
112
113void
114png_write_info(png_structp png_ptr, png_infop info_ptr)
115{
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600116#if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_sPLT_SUPPORTED)
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600117 int i;
118#endif
119
120 png_debug(1, "in png_write_info\n");
121
122 png_write_info_before_PLTE(png_ptr, info_ptr);
123
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600124 if (info_ptr->valid & PNG_INFO_PLTE)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500125 png_write_PLTE(png_ptr, info_ptr->palette,
126 (png_uint_32)info_ptr->num_palette);
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600127 else if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
Guy Schalnate5a37791996-06-05 15:50:50 -0500128 png_error(png_ptr, "Valid palette required for paletted images\n");
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600129
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500130#if defined(PNG_WRITE_tRNS_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600131 if (info_ptr->valid & PNG_INFO_tRNS)
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600132 {
133#if defined(PNG_WRITE_INVERT_ALPHA_SUPPORTED)
134 /* invert the alpha channel (in tRNS) */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600135 if ((png_ptr->transformations & PNG_INVERT_ALPHA) &&
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600136 info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
137 {
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600138 int j;
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600139 for (j=0; j<(int)info_ptr->num_trans; j++)
Glenn Randers-Pehrson860ab2b1999-10-14 07:43:10 -0500140 info_ptr->trans[j] = (png_byte)(255 - info_ptr->trans[j]);
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600141 }
142#endif
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600143 png_write_tRNS(png_ptr, info_ptr->trans, &(info_ptr->trans_values),
144 info_ptr->num_trans, info_ptr->color_type);
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600145 }
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500146#endif
147#if defined(PNG_WRITE_bKGD_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600148 if (info_ptr->valid & PNG_INFO_bKGD)
149 png_write_bKGD(png_ptr, &(info_ptr->background), info_ptr->color_type);
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500150#endif
151#if defined(PNG_WRITE_hIST_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600152 if (info_ptr->valid & PNG_INFO_hIST)
153 png_write_hIST(png_ptr, info_ptr->hist, info_ptr->num_palette);
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500154#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500155#if defined(PNG_WRITE_oFFs_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600156 if (info_ptr->valid & PNG_INFO_oFFs)
157 png_write_oFFs(png_ptr, info_ptr->x_offset, info_ptr->y_offset,
158 info_ptr->offset_unit_type);
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500159#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500160#if defined(PNG_WRITE_pCAL_SUPPORTED)
161 if (info_ptr->valid & PNG_INFO_pCAL)
162 png_write_pCAL(png_ptr, info_ptr->pcal_purpose, info_ptr->pcal_X0,
163 info_ptr->pcal_X1, info_ptr->pcal_type, info_ptr->pcal_nparams,
164 info_ptr->pcal_units, info_ptr->pcal_params);
165#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600166#if defined(PNG_WRITE_sCAL_SUPPORTED)
167 if (info_ptr->valid & PNG_INFO_sCAL)
Glenn Randers-Pehrson68ea2432000-04-01 21:10:05 -0600168#if defined(PNG_FLOATING_POINT_SUPPORTED) && !defined(PNG_NO_STDIO)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600169 png_write_sCAL(png_ptr, (int)info_ptr->scal_unit,
170 info_ptr->scal_pixel_width, info_ptr->scal_pixel_height);
171#else
172#ifdef PNG_FIXED_POINT_SUPPORTED
173 png_write_sCAL_s(png_ptr, (int)info_ptr->scal_unit,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600174 info_ptr->scal_s_width, info_ptr->scal_s_height);
Glenn Randers-Pehrson68ea2432000-04-01 21:10:05 -0600175#else
176 png_warning(png_ptr,
177 "png_write_sCAL not supported; sCAL chunk not written.\n");
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600178#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600179#endif
180#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500181#if defined(PNG_WRITE_pHYs_SUPPORTED)
182 if (info_ptr->valid & PNG_INFO_pHYs)
183 png_write_pHYs(png_ptr, info_ptr->x_pixels_per_unit,
184 info_ptr->y_pixels_per_unit, info_ptr->phys_unit_type);
185#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500186#if defined(PNG_WRITE_tIME_SUPPORTED)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600187 if (info_ptr->valid & PNG_INFO_tIME)
Guy Schalnate5a37791996-06-05 15:50:50 -0500188 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600189 png_write_tIME(png_ptr, &(info_ptr->mod_time));
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600190 png_ptr->mode |= PNG_WROTE_tIME;
Guy Schalnate5a37791996-06-05 15:50:50 -0500191 }
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500192#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600193#if defined(PNG_WRITE_sPLT_SUPPORTED)
194 if (info_ptr->valid & PNG_INFO_sPLT)
195 for (i = 0; i < (int)info_ptr->splt_palettes_num; i++)
196 png_write_sPLT(png_ptr, info_ptr->splt_palettes + i);
197#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600198#if defined(PNG_WRITE_TEXT_SUPPORTED)
Guy Schalnate5a37791996-06-05 15:50:50 -0500199 /* Check to see if we need to write text chunks */
Andreas Dilger47a0c421997-05-16 02:46:07 -0500200 for (i = 0; i < info_ptr->num_text; i++)
Guy Schalnat0d580581995-07-20 02:43:20 -0500201 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500202 png_debug2(2, "Writing header text chunk %d, type %d\n", i,
203 info_ptr->text[i].compression);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600204 /* an internationalized chunk? */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600205 if (info_ptr->text[i].compression > 0)
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600206 {
207#if defined(PNG_WRITE_iTXt_SUPPORTED)
208 /* write international chunk */
209 png_write_iTXt(png_ptr,
210 info_ptr->text[i].compression,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600211 info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600212 info_ptr->text[i].lang,
213 info_ptr->text[i].lang_key,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600214 info_ptr->text[i].text);
215#else
216 png_warning(png_ptr, "Unable to write international text\n");
217#endif
218 /* Mark this chunk as written */
219 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
220 }
Andreas Dilger47a0c421997-05-16 02:46:07 -0500221 /* If we want a compressed text chunk */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600222 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_zTXt)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500223 {
224#if defined(PNG_WRITE_zTXt_SUPPORTED)
225 /* write compressed chunk */
226 png_write_zTXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600227 info_ptr->text[i].text, 0,
Andreas Dilger47a0c421997-05-16 02:46:07 -0500228 info_ptr->text[i].compression);
229#else
230 png_warning(png_ptr, "Unable to write compressed text\n");
231#endif
232 /* Mark this chunk as written */
233 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
234 }
235 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
236 {
237#if defined(PNG_WRITE_tEXt_SUPPORTED)
238 /* write uncompressed chunk */
239 png_write_tEXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600240 info_ptr->text[i].text,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600241 0);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500242#else
243 png_warning(png_ptr, "Unable to write uncompressed text\n");
244#endif
245 /* Mark this chunk as written */
246 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
247 }
248 }
249#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600250#if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
251 if (info_ptr->unknown_chunks_num)
252 {
253 png_unknown_chunk *up;
254
255 png_debug(5, "writing extra chunks\n");
256
257 for (up = info_ptr->unknown_chunks;
258 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
259 up++)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600260 {
261 int keep=png_handle_as_unknown(png_ptr, up->name);
262 if (keep != HANDLE_CHUNK_NEVER &&
263 up->location && (up->location & PNG_HAVE_PLTE) &&
264 !(up->location & PNG_HAVE_IDAT) &&
265 ((up->name[3] & 0x20) || keep == HANDLE_CHUNK_ALWAYS ||
266 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
267 {
268 png_write_chunk(png_ptr, up->name, up->data, up->size);
269 }
270 }
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600271 }
272#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500273}
Guy Schalnat0d580581995-07-20 02:43:20 -0500274
Andreas Dilger47a0c421997-05-16 02:46:07 -0500275/* Writes the end of the PNG file. If you don't want to write comments or
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600276 * time information, you can pass NULL for info. If you already wrote these
277 * in png_write_info(), do not write them again here. If you have long
278 * comments, I suggest writing them here, and compressing them.
279 */
Andreas Dilger47a0c421997-05-16 02:46:07 -0500280void
281png_write_end(png_structp png_ptr, png_infop info_ptr)
282{
283 png_debug(1, "in png_write_end\n");
284 if (!(png_ptr->mode & PNG_HAVE_IDAT))
285 png_error(png_ptr, "No IDATs written into file");
286
287 /* see if user wants us to write information chunks */
288 if (info_ptr != NULL)
289 {
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600290#if defined(PNG_WRITE_TEXT_SUPPORTED)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500291 int i; /* local index variable */
Glenn Randers-Pehrson70e3f541998-01-03 22:40:55 -0600292#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500293#if defined(PNG_WRITE_tIME_SUPPORTED)
294 /* check to see if user has supplied a time chunk */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600295 if ((info_ptr->valid & PNG_INFO_tIME) &&
Glenn Randers-Pehrson5379b241999-11-27 10:22:33 -0600296 !(png_ptr->mode & PNG_WROTE_tIME))
Andreas Dilger47a0c421997-05-16 02:46:07 -0500297 png_write_tIME(png_ptr, &(info_ptr->mod_time));
298#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600299#if defined(PNG_WRITE_TEXT_SUPPORTED)
Andreas Dilger47a0c421997-05-16 02:46:07 -0500300 /* loop through comment chunks */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600301 for (i = 0; i < info_ptr->num_text; i++)
Guy Schalnat0d580581995-07-20 02:43:20 -0500302 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500303 png_debug2(2, "Writing trailer text chunk %d, type %d\n", i,
304 info_ptr->text[i].compression);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600305 /* an internationalized chunk? */
306 if (info_ptr->text[i].compression > 0)
307 {
308#if defined(PNG_WRITE_iTXt_SUPPORTED)
309 /* write international chunk */
310 png_write_iTXt(png_ptr,
311 info_ptr->text[i].compression,
312 info_ptr->text[i].key,
313 info_ptr->text[i].lang,
314 info_ptr->text[i].lang_key,
315 info_ptr->text[i].text);
316#else
317 png_warning(png_ptr, "Unable to write international text\n");
318#endif
319 /* Mark this chunk as written */
320 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
321 }
322 else if (info_ptr->text[i].compression >= PNG_TEXT_COMPRESSION_zTXt)
Guy Schalnat0d580581995-07-20 02:43:20 -0500323 {
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500324#if defined(PNG_WRITE_zTXt_SUPPORTED)
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600325 /* write compressed chunk */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600326 png_write_zTXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600327 info_ptr->text[i].text, 0,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600328 info_ptr->text[i].compression);
Guy Schalnate5a37791996-06-05 15:50:50 -0500329#else
330 png_warning(png_ptr, "Unable to write compressed text\n");
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500331#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500332 /* Mark this chunk as written */
333 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
Guy Schalnat0d580581995-07-20 02:43:20 -0500334 }
Andreas Dilger47a0c421997-05-16 02:46:07 -0500335 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
Guy Schalnat0d580581995-07-20 02:43:20 -0500336 {
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500337#if defined(PNG_WRITE_tEXt_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500338 /* write uncompressed chunk */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600339 png_write_tEXt(png_ptr, info_ptr->text[i].key,
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600340 info_ptr->text[i].text, 0);
Guy Schalnate5a37791996-06-05 15:50:50 -0500341#else
342 png_warning(png_ptr, "Unable to write uncompressed text\n");
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500343#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500344
Andreas Dilger47a0c421997-05-16 02:46:07 -0500345 /* Mark this chunk as written */
346 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
Guy Schalnat0d580581995-07-20 02:43:20 -0500347 }
348 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600349#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600350#if defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED)
351 if (info_ptr->unknown_chunks_num)
352 {
353 png_unknown_chunk *up;
354
355 png_debug(5, "writing extra chunks\n");
356
357 for (up = info_ptr->unknown_chunks;
358 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
359 up++)
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600360 {
361 int keep=png_handle_as_unknown(png_ptr, up->name);
362 if (keep != HANDLE_CHUNK_NEVER &&
363 up->location && (up->location & PNG_AFTER_IDAT) &&
364 ((up->name[3] & 0x20) || keep == HANDLE_CHUNK_ALWAYS ||
365 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS)))
366 {
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600367 png_write_chunk(png_ptr, up->name, up->data, up->size);
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600368 }
369 }
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600370 }
371#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500372 }
Guy Schalnate5a37791996-06-05 15:50:50 -0500373
374 png_ptr->mode |= PNG_AFTER_IDAT;
375
Andreas Dilger47a0c421997-05-16 02:46:07 -0500376 /* write end of PNG file */
Guy Schalnat0d580581995-07-20 02:43:20 -0500377 png_write_IEND(png_ptr);
378}
379
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500380#if defined(PNG_WRITE_tIME_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500381void
Guy Schalnat6d764711995-12-19 03:22:19 -0600382png_convert_from_struct_tm(png_timep ptime, struct tm FAR * ttime)
Guy Schalnat0d580581995-07-20 02:43:20 -0500383{
Andreas Dilger47a0c421997-05-16 02:46:07 -0500384 png_debug(1, "in png_convert_from_struct_tm\n");
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600385 ptime->year = (png_uint_16)(1900 + ttime->tm_year);
386 ptime->month = (png_byte)(ttime->tm_mon + 1);
387 ptime->day = (png_byte)ttime->tm_mday;
388 ptime->hour = (png_byte)ttime->tm_hour;
389 ptime->minute = (png_byte)ttime->tm_min;
390 ptime->second = (png_byte)ttime->tm_sec;
Guy Schalnat0d580581995-07-20 02:43:20 -0500391}
392
393void
Guy Schalnat6d764711995-12-19 03:22:19 -0600394png_convert_from_time_t(png_timep ptime, time_t ttime)
Guy Schalnat0d580581995-07-20 02:43:20 -0500395{
396 struct tm *tbuf;
397
Andreas Dilger47a0c421997-05-16 02:46:07 -0500398 png_debug(1, "in png_convert_from_time_t\n");
Guy Schalnat0d580581995-07-20 02:43:20 -0500399 tbuf = gmtime(&ttime);
400 png_convert_from_struct_tm(ptime, tbuf);
401}
Guy Schalnat6d764711995-12-19 03:22:19 -0600402#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500403
Andreas Dilger47a0c421997-05-16 02:46:07 -0500404/* Initialize png_ptr structure, and allocate any memory needed */
Guy Schalnate5a37791996-06-05 15:50:50 -0500405png_structp
Glenn Randers-Pehrsond0dce401998-05-09 10:02:29 -0500406png_create_write_struct(png_const_charp user_png_ver, png_voidp error_ptr,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600407 png_error_ptr error_fn, png_error_ptr warn_fn)
Guy Schalnat0d580581995-07-20 02:43:20 -0500408{
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500409#ifdef PNG_USER_MEM_SUPPORTED
410 return (png_create_write_struct_2(user_png_ver, error_ptr, error_fn,
411 warn_fn, NULL, NULL, NULL));
412}
413
414/* Alternate initialize png_ptr structure, and allocate any memory needed */
415png_structp
416png_create_write_struct_2(png_const_charp user_png_ver, png_voidp error_ptr,
417 png_error_ptr error_fn, png_error_ptr warn_fn, png_voidp mem_ptr,
418 png_malloc_ptr malloc_fn, png_free_ptr free_fn)
419{
420#endif /* PNG_USER_MEM_SUPPORTED */
Guy Schalnate5a37791996-06-05 15:50:50 -0500421 png_structp png_ptr;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600422#ifdef PNG_SETJMP_SUPPORTED
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600423#ifdef USE_FAR_KEYWORD
424 jmp_buf jmpbuf;
425#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600426#endif
Andreas Dilger47a0c421997-05-16 02:46:07 -0500427 png_debug(1, "in png_create_write_struct\n");
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500428#ifdef PNG_USER_MEM_SUPPORTED
429 if ((png_ptr = (png_structp)png_create_struct_2(PNG_STRUCT_PNG,
430 (png_malloc_ptr)malloc_fn)) == NULL)
431#else
Guy Schalnate5a37791996-06-05 15:50:50 -0500432 if ((png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG)) == NULL)
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500433#endif /* PNG_USER_MEM_SUPPORTED */
Guy Schalnate5a37791996-06-05 15:50:50 -0500434 {
Glenn Randers-Pehrsonb2120021998-01-31 20:07:59 -0600435 return ((png_structp)NULL);
Guy Schalnate5a37791996-06-05 15:50:50 -0500436 }
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600437
438#ifdef PNG_SETJMP_SUPPORTED
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600439#ifdef USE_FAR_KEYWORD
440 if (setjmp(jmpbuf))
441#else
Guy Schalnate5a37791996-06-05 15:50:50 -0500442 if (setjmp(png_ptr->jmpbuf))
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600443#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500444 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600445 png_free(png_ptr, png_ptr->zbuf);
Guy Schalnate5a37791996-06-05 15:50:50 -0500446 png_destroy_struct(png_ptr);
Glenn Randers-Pehrsonb2120021998-01-31 20:07:59 -0600447 return ((png_structp)NULL);
Guy Schalnate5a37791996-06-05 15:50:50 -0500448 }
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600449#ifdef USE_FAR_KEYWORD
450 png_memcpy(png_ptr->jmpbuf,jmpbuf,sizeof(jmp_buf));
451#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600452#endif
453
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500454#ifdef PNG_USER_MEM_SUPPORTED
455 png_set_mem_fn(png_ptr, mem_ptr, malloc_fn, free_fn);
456#endif /* PNG_USER_MEM_SUPPORTED */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600457 png_set_error_fn(png_ptr, error_ptr, error_fn, warn_fn);
Guy Schalnate5a37791996-06-05 15:50:50 -0500458
Andreas Dilger47a0c421997-05-16 02:46:07 -0500459 /* Libpng 0.90 and later are binary incompatible with libpng 0.89, so
460 * we must recompile any applications that use any older library version.
461 * For versions after libpng 1.0, we will be compatible, so we need
462 * only check the first digit.
463 */
464 if (user_png_ver == NULL || user_png_ver[0] != png_libpng_ver[0] ||
465 (png_libpng_ver[0] == '0' && user_png_ver[2] < '9'))
Guy Schalnate5a37791996-06-05 15:50:50 -0500466 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500467 png_error(png_ptr,
468 "Incompatible libpng version in application and library");
Guy Schalnate5a37791996-06-05 15:50:50 -0500469 }
Guy Schalnat0d580581995-07-20 02:43:20 -0500470
471 /* initialize zbuf - compression buffer */
472 png_ptr->zbuf_size = PNG_ZBUF_SIZE;
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600473 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr,
474 (png_uint_32)png_ptr->zbuf_size);
Guy Schalnate5a37791996-06-05 15:50:50 -0500475
476 png_set_write_fn(png_ptr, NULL, NULL, NULL);
477
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600478#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
479 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
480 1, NULL, NULL);
481#endif
482
Glenn Randers-Pehrson6942d532000-05-01 09:31:54 -0500483 png_ptr->mode |= PNG_CREATED_WRITE_STRUCT;
484
Glenn Randers-Pehrsonb2120021998-01-31 20:07:59 -0600485 return ((png_structp)png_ptr);
Guy Schalnate5a37791996-06-05 15:50:50 -0500486}
487
Andreas Dilger47a0c421997-05-16 02:46:07 -0500488/* Initialize png_ptr structure, and allocate any memory needed */
Guy Schalnate5a37791996-06-05 15:50:50 -0500489void
490png_write_init(png_structp png_ptr)
491{
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600492#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500493 jmp_buf tmp_jmp; /* to save current jump buffer */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600494#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500495
Andreas Dilger47a0c421997-05-16 02:46:07 -0500496 png_debug(1, "in png_write_init\n");
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600497#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500498 /* save jump buffer and error functions */
499 png_memcpy(tmp_jmp, png_ptr->jmpbuf, sizeof (jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600500#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500501
502 /* reset all variables to 0 */
503 png_memset(png_ptr, 0, sizeof (png_struct));
504
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600505#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnate5a37791996-06-05 15:50:50 -0500506 /* restore jump buffer */
507 png_memcpy(png_ptr->jmpbuf, tmp_jmp, sizeof (jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600508#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500509
Glenn Randers-Pehrson6942d532000-05-01 09:31:54 -0500510#ifndef PNG_LEGACY_SUPPORTED
511 if(!(png_ptr->mode & PNG_CREATED_WRITE_STRUCT))
512 {
513 png_ptr->error_fn=NULL;
514 png_error(png_ptr,
515 "Write struct not properly created; use a legacy-supporting libpng.");
516 }
517#endif
518
Guy Schalnate5a37791996-06-05 15:50:50 -0500519 /* initialize zbuf - compression buffer */
520 png_ptr->zbuf_size = PNG_ZBUF_SIZE;
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600521 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr,
522 (png_uint_32)png_ptr->zbuf_size);
Guy Schalnate5a37791996-06-05 15:50:50 -0500523 png_set_write_fn(png_ptr, NULL, NULL, NULL);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500524
525#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
526 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT,
527 1, NULL, NULL);
528#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500529}
530
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600531/* Write a few rows of image data. If the image is interlaced,
532 * either you will have to write the 7 sub images, or, if you
533 * have called png_set_interlace_handling(), you will have to
534 * "write" the image seven times.
535 */
Guy Schalnat0d580581995-07-20 02:43:20 -0500536void
Guy Schalnat6d764711995-12-19 03:22:19 -0600537png_write_rows(png_structp png_ptr, png_bytepp row,
Guy Schalnat0d580581995-07-20 02:43:20 -0500538 png_uint_32 num_rows)
539{
540 png_uint_32 i; /* row counter */
Guy Schalnat6d764711995-12-19 03:22:19 -0600541 png_bytepp rp; /* row pointer */
Guy Schalnat0d580581995-07-20 02:43:20 -0500542
Andreas Dilger47a0c421997-05-16 02:46:07 -0500543 png_debug(1, "in png_write_rows\n");
Guy Schalnat0d580581995-07-20 02:43:20 -0500544 /* loop through the rows */
545 for (i = 0, rp = row; i < num_rows; i++, rp++)
546 {
547 png_write_row(png_ptr, *rp);
548 }
549}
550
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600551/* Write the image. You only need to call this function once, even
552 * if you are writing an interlaced image.
553 */
Guy Schalnat0d580581995-07-20 02:43:20 -0500554void
Guy Schalnat6d764711995-12-19 03:22:19 -0600555png_write_image(png_structp png_ptr, png_bytepp image)
Guy Schalnat0d580581995-07-20 02:43:20 -0500556{
557 png_uint_32 i; /* row index */
558 int pass, num_pass; /* pass variables */
Guy Schalnat6d764711995-12-19 03:22:19 -0600559 png_bytepp rp; /* points to current row */
Guy Schalnat0d580581995-07-20 02:43:20 -0500560
Andreas Dilger47a0c421997-05-16 02:46:07 -0500561 png_debug(1, "in png_write_image\n");
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600562#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500563 /* intialize interlace handling. If image is not interlaced,
564 this will set pass to 1 */
565 num_pass = png_set_interlace_handling(png_ptr);
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600566#else
567 num_pass = 1;
568#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500569 /* loop through passes */
570 for (pass = 0; pass < num_pass; pass++)
571 {
572 /* loop through image */
573 for (i = 0, rp = image; i < png_ptr->height; i++, rp++)
574 {
575 png_write_row(png_ptr, *rp);
576 }
577 }
578}
579
Guy Schalnate5a37791996-06-05 15:50:50 -0500580/* called by user to write a row of image data */
Guy Schalnat0d580581995-07-20 02:43:20 -0500581void
Guy Schalnat6d764711995-12-19 03:22:19 -0600582png_write_row(png_structp png_ptr, png_bytep row)
Guy Schalnat0d580581995-07-20 02:43:20 -0500583{
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600584 png_debug2(1, "in png_write_row (row %ld, pass %d)\n",
585 png_ptr->row_number, png_ptr->pass);
Guy Schalnat0d580581995-07-20 02:43:20 -0500586 /* initialize transformations and other stuff if first time */
Guy Schalnat6d764711995-12-19 03:22:19 -0600587 if (png_ptr->row_number == 0 && png_ptr->pass == 0)
Guy Schalnat0d580581995-07-20 02:43:20 -0500588 {
Glenn Randers-Pehrson1d963611998-05-02 12:52:25 -0500589 /* check for transforms that have been set but were defined out */
590#if !defined(PNG_WRITE_INVERT_SUPPORTED) && defined(PNG_READ_INVERT_SUPPORTED)
591 if (png_ptr->transformations & PNG_INVERT_MONO)
592 png_warning(png_ptr, "PNG_WRITE_INVERT_SUPPORTED is not defined.");
593#endif
594#if !defined(PNG_WRITE_FILLER_SUPPORTED) && defined(PNG_READ_FILLER_SUPPORTED)
595 if (png_ptr->transformations & PNG_FILLER)
596 png_warning(png_ptr, "PNG_WRITE_FILLER_SUPPORTED is not defined.");
597#endif
598#if !defined(PNG_WRITE_PACKSWAP_SUPPORTED) && defined(PNG_READ_PACKSWAP_SUPPORTED)
599 if (png_ptr->transformations & PNG_PACKSWAP)
600 png_warning(png_ptr, "PNG_WRITE_PACKSWAP_SUPPORTED is not defined.");
601#endif
602#if !defined(PNG_WRITE_PACK_SUPPORTED) && defined(PNG_READ_PACK_SUPPORTED)
603 if (png_ptr->transformations & PNG_PACK)
604 png_warning(png_ptr, "PNG_WRITE_PACK_SUPPORTED is not defined.");
605#endif
606#if !defined(PNG_WRITE_SHIFT_SUPPORTED) && defined(PNG_READ_SHIFT_SUPPORTED)
607 if (png_ptr->transformations & PNG_SHIFT)
608 png_warning(png_ptr, "PNG_WRITE_SHIFT_SUPPORTED is not defined.");
609#endif
610#if !defined(PNG_WRITE_BGR_SUPPORTED) && defined(PNG_READ_BGR_SUPPORTED)
611 if (png_ptr->transformations & PNG_BGR)
612 png_warning(png_ptr, "PNG_WRITE_BGR_SUPPORTED is not defined.");
613#endif
614#if !defined(PNG_WRITE_SWAP_SUPPORTED) && defined(PNG_READ_SWAP_SUPPORTED)
615 if (png_ptr->transformations & PNG_SWAP_BYTES)
616 png_warning(png_ptr, "PNG_WRITE_SWAP_SUPPORTED is not defined.");
617#endif
618
Guy Schalnat0d580581995-07-20 02:43:20 -0500619 png_write_start_row(png_ptr);
620 }
621
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500622#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500623 /* if interlaced and not interested in row, return */
624 if (png_ptr->interlaced && (png_ptr->transformations & PNG_INTERLACE))
625 {
626 switch (png_ptr->pass)
627 {
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600628 case 0:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600629 if (png_ptr->row_number & 0x07)
Guy Schalnat0d580581995-07-20 02:43:20 -0500630 {
631 png_write_finish_row(png_ptr);
632 return;
633 }
634 break;
635 case 1:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600636 if ((png_ptr->row_number & 0x07) || png_ptr->width < 5)
Guy Schalnat0d580581995-07-20 02:43:20 -0500637 {
638 png_write_finish_row(png_ptr);
639 return;
640 }
641 break;
642 case 2:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600643 if ((png_ptr->row_number & 0x07) != 4)
Guy Schalnat0d580581995-07-20 02:43:20 -0500644 {
645 png_write_finish_row(png_ptr);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600646 return;
Guy Schalnat0d580581995-07-20 02:43:20 -0500647 }
648 break;
649 case 3:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600650 if ((png_ptr->row_number & 0x03) || png_ptr->width < 3)
Guy Schalnat0d580581995-07-20 02:43:20 -0500651 {
652 png_write_finish_row(png_ptr);
653 return;
654 }
655 break;
656 case 4:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600657 if ((png_ptr->row_number & 0x03) != 2)
Guy Schalnat0d580581995-07-20 02:43:20 -0500658 {
659 png_write_finish_row(png_ptr);
660 return;
661 }
662 break;
663 case 5:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600664 if ((png_ptr->row_number & 0x01) || png_ptr->width < 2)
Guy Schalnat0d580581995-07-20 02:43:20 -0500665 {
666 png_write_finish_row(png_ptr);
667 return;
668 }
669 break;
670 case 6:
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600671 if (!(png_ptr->row_number & 0x01))
Guy Schalnat0d580581995-07-20 02:43:20 -0500672 {
673 png_write_finish_row(png_ptr);
674 return;
675 }
676 break;
677 }
678 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600679#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500680
681 /* set up row info for transformations */
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600682 png_ptr->row_info.color_type = png_ptr->color_type;
Guy Schalnat0d580581995-07-20 02:43:20 -0500683 png_ptr->row_info.width = png_ptr->usr_width;
684 png_ptr->row_info.channels = png_ptr->usr_channels;
685 png_ptr->row_info.bit_depth = png_ptr->usr_bit_depth;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600686 png_ptr->row_info.pixel_depth = (png_byte)(png_ptr->row_info.bit_depth *
687 png_ptr->row_info.channels);
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600688
Guy Schalnat0d580581995-07-20 02:43:20 -0500689 png_ptr->row_info.rowbytes = ((png_ptr->row_info.width *
690 (png_uint_32)png_ptr->row_info.pixel_depth + 7) >> 3);
691
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600692 png_debug1(3, "row_info->color_type = %d\n", png_ptr->row_info.color_type);
693 png_debug1(3, "row_info->width = %d\n", png_ptr->row_info.width);
694 png_debug1(3, "row_info->channels = %d\n", png_ptr->row_info.channels);
695 png_debug1(3, "row_info->bit_depth = %d\n", png_ptr->row_info.bit_depth);
696 png_debug1(3, "row_info->pixel_depth = %d\n", png_ptr->row_info.pixel_depth);
697 png_debug1(3, "row_info->rowbytes = %d\n", png_ptr->row_info.rowbytes);
Andreas Dilger47a0c421997-05-16 02:46:07 -0500698
699 /* Copy user's row into buffer, leaving room for filter byte. */
Glenn Randers-Pehrsona357b991998-02-08 20:56:40 -0600700 png_memcpy_check(png_ptr, png_ptr->row_buf + 1, row,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600701 png_ptr->row_info.rowbytes);
Guy Schalnat0d580581995-07-20 02:43:20 -0500702
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500703#if defined(PNG_WRITE_INTERLACING_SUPPORTED)
Guy Schalnat0d580581995-07-20 02:43:20 -0500704 /* handle interlacing */
705 if (png_ptr->interlaced && png_ptr->pass < 6 &&
706 (png_ptr->transformations & PNG_INTERLACE))
707 {
708 png_do_write_interlace(&(png_ptr->row_info),
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600709 png_ptr->row_buf + 1, png_ptr->pass);
Guy Schalnat0d580581995-07-20 02:43:20 -0500710 /* this should always get caught above, but still ... */
711 if (!(png_ptr->row_info.width))
712 {
713 png_write_finish_row(png_ptr);
714 return;
715 }
716 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600717#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500718
719 /* handle other transformations */
720 if (png_ptr->transformations)
721 png_do_write_transformations(png_ptr);
722
Andreas Dilger47a0c421997-05-16 02:46:07 -0500723 /* Find a filter if necessary, filter the row and write it out. */
Guy Schalnate5a37791996-06-05 15:50:50 -0500724 png_write_find_filter(png_ptr, &(png_ptr->row_info));
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -0600725
726 if (png_ptr->write_row_fn != NULL)
727 (*(png_ptr->write_row_fn))(png_ptr, png_ptr->row_number, png_ptr->pass);
Guy Schalnat0d580581995-07-20 02:43:20 -0500728}
729
Guy Schalnat0f716451995-11-28 11:22:13 -0600730#if defined(PNG_WRITE_FLUSH_SUPPORTED)
731/* Set the automatic flush interval or 0 to turn flushing off */
732void
Guy Schalnat4ee97b01996-01-16 01:51:56 -0600733png_set_flush(png_structp png_ptr, int nrows)
Guy Schalnat0f716451995-11-28 11:22:13 -0600734{
Andreas Dilger47a0c421997-05-16 02:46:07 -0500735 png_debug(1, "in png_set_flush\n");
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600736 png_ptr->flush_dist = (nrows < 0 ? 0 : nrows);
Guy Schalnat0f716451995-11-28 11:22:13 -0600737}
738
739/* flush the current output buffers now */
740void
Guy Schalnat4ee97b01996-01-16 01:51:56 -0600741png_write_flush(png_structp png_ptr)
Guy Schalnat0f716451995-11-28 11:22:13 -0600742{
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600743 int wrote_IDAT;
Guy Schalnat0f716451995-11-28 11:22:13 -0600744
Andreas Dilger47a0c421997-05-16 02:46:07 -0500745 png_debug(1, "in png_write_flush\n");
Guy Schalnate5a37791996-06-05 15:50:50 -0500746 /* We have already written out all of the data */
747 if (png_ptr->row_number >= png_ptr->num_rows)
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600748 return;
Guy Schalnat0f716451995-11-28 11:22:13 -0600749
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600750 do
751 {
752 int ret;
Guy Schalnat0f716451995-11-28 11:22:13 -0600753
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600754 /* compress the data */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600755 ret = deflate(&png_ptr->zstream, Z_SYNC_FLUSH);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600756 wrote_IDAT = 0;
Guy Schalnat0f716451995-11-28 11:22:13 -0600757
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600758 /* check for compression errors */
759 if (ret != Z_OK)
760 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500761 if (png_ptr->zstream.msg != NULL)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600762 png_error(png_ptr, png_ptr->zstream.msg);
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600763 else
764 png_error(png_ptr, "zlib error");
765 }
Guy Schalnat0f716451995-11-28 11:22:13 -0600766
Andreas Dilger47a0c421997-05-16 02:46:07 -0500767 if (!(png_ptr->zstream.avail_out))
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600768 {
769 /* write the IDAT and reset the zlib output buffer */
770 png_write_IDAT(png_ptr, png_ptr->zbuf,
771 png_ptr->zbuf_size);
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600772 png_ptr->zstream.next_out = png_ptr->zbuf;
773 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600774 wrote_IDAT = 1;
775 }
776 } while(wrote_IDAT == 1);
Guy Schalnat0f716451995-11-28 11:22:13 -0600777
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600778 /* If there is any data left to be output, write it into a new IDAT */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600779 if (png_ptr->zbuf_size != png_ptr->zstream.avail_out)
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600780 {
781 /* write the IDAT and reset the zlib output buffer */
782 png_write_IDAT(png_ptr, png_ptr->zbuf,
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600783 png_ptr->zbuf_size - png_ptr->zstream.avail_out);
784 png_ptr->zstream.next_out = png_ptr->zbuf;
785 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600786 }
787 png_ptr->flush_rows = 0;
788 png_flush(png_ptr);
Guy Schalnat0f716451995-11-28 11:22:13 -0600789}
790#endif /* PNG_WRITE_FLUSH_SUPPORTED */
791
Guy Schalnate5a37791996-06-05 15:50:50 -0500792/* free all memory used by the write */
793void
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600794png_destroy_write_struct(png_structpp png_ptr_ptr, png_infopp info_ptr_ptr)
Guy Schalnate5a37791996-06-05 15:50:50 -0500795{
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600796 png_structp png_ptr = NULL;
797 png_infop info_ptr = NULL;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500798#ifdef PNG_USER_MEM_SUPPORTED
799 png_free_ptr free_fn = NULL;
800#endif
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600801
Andreas Dilger47a0c421997-05-16 02:46:07 -0500802 png_debug(1, "in png_destroy_write_struct\n");
803 if (png_ptr_ptr != NULL)
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500804 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600805 png_ptr = *png_ptr_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500806#ifdef PNG_USER_MEM_SUPPORTED
807 free_fn = png_ptr->free_fn;
808#endif
809 }
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600810
Andreas Dilger47a0c421997-05-16 02:46:07 -0500811 if (info_ptr_ptr != NULL)
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600812 info_ptr = *info_ptr_ptr;
813
Andreas Dilger47a0c421997-05-16 02:46:07 -0500814 if (info_ptr != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500815 {
Glenn Randers-Pehrsona77ef622000-02-18 13:48:52 -0600816 png_free_data(png_ptr, info_ptr, PNG_FREE_ALL, -1);
817
Glenn Randers-Pehrson6942d532000-05-01 09:31:54 -0500818#if defined(PNG_UNKNOWN_CHUNKS_SUPPORTED)
Glenn Randers-Pehrsona77ef622000-02-18 13:48:52 -0600819 if (png_ptr->num_chunk_list)
820 {
821 png_free(png_ptr, png_ptr->chunk_list);
822 png_ptr->num_chunk_list=0;
823 }
Glenn Randers-Pehrson6942d532000-05-01 09:31:54 -0500824#endif
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600825
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500826#ifdef PNG_USER_MEM_SUPPORTED
827 png_destroy_struct_2((png_voidp)info_ptr, free_fn);
828#else
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600829 png_destroy_struct((png_voidp)info_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500830#endif
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600831 *info_ptr_ptr = (png_infop)NULL;
Guy Schalnate5a37791996-06-05 15:50:50 -0500832 }
Guy Schalnat6d764711995-12-19 03:22:19 -0600833
Andreas Dilger47a0c421997-05-16 02:46:07 -0500834 if (png_ptr != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500835 {
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600836 png_write_destroy(png_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500837#ifdef PNG_USER_MEM_SUPPORTED
838 png_destroy_struct_2((png_voidp)png_ptr, free_fn);
839#else
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600840 png_destroy_struct((png_voidp)png_ptr);
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500841#endif
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600842 *png_ptr_ptr = (png_structp)NULL;
Guy Schalnate5a37791996-06-05 15:50:50 -0500843 }
844}
845
846
Andreas Dilger47a0c421997-05-16 02:46:07 -0500847/* Free any memory used in png_ptr struct (old method) */
Guy Schalnat0d580581995-07-20 02:43:20 -0500848void
Guy Schalnat6d764711995-12-19 03:22:19 -0600849png_write_destroy(png_structp png_ptr)
Guy Schalnat0d580581995-07-20 02:43:20 -0500850{
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600851#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnatb2e01bd1996-01-26 01:38:47 -0600852 jmp_buf tmp_jmp; /* save jump buffer */
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600853#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500854 png_error_ptr error_fn;
855 png_error_ptr warning_fn;
856 png_voidp error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500857#ifdef PNG_USER_MEM_SUPPORTED
858 png_free_ptr free_fn;
859#endif
Guy Schalnat0d580581995-07-20 02:43:20 -0500860
Andreas Dilger47a0c421997-05-16 02:46:07 -0500861 png_debug(1, "in png_write_destroy\n");
Guy Schalnat0d580581995-07-20 02:43:20 -0500862 /* free any memory zlib uses */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600863 deflateEnd(&png_ptr->zstream);
Guy Schalnate5a37791996-06-05 15:50:50 -0500864
Guy Schalnat0d580581995-07-20 02:43:20 -0500865 /* free our memory. png_free checks NULL for us. */
Andreas Dilger02ad0ef1997-01-17 01:34:35 -0600866 png_free(png_ptr, png_ptr->zbuf);
867 png_free(png_ptr, png_ptr->row_buf);
868 png_free(png_ptr, png_ptr->prev_row);
869 png_free(png_ptr, png_ptr->sub_row);
870 png_free(png_ptr, png_ptr->up_row);
871 png_free(png_ptr, png_ptr->avg_row);
872 png_free(png_ptr, png_ptr->paeth_row);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600873
Glenn Randers-Pehrsonc4a2ae61998-01-16 22:06:18 -0600874#if defined(PNG_TIME_RFC1123_SUPPORTED)
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -0600875 png_free(png_ptr, png_ptr->time_buffer);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600876#endif
877
Andreas Dilger47a0c421997-05-16 02:46:07 -0500878#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED)
879 png_free(png_ptr, png_ptr->prev_filters);
880 png_free(png_ptr, png_ptr->filter_weights);
881 png_free(png_ptr, png_ptr->inv_filter_weights);
882 png_free(png_ptr, png_ptr->filter_costs);
883 png_free(png_ptr, png_ptr->inv_filter_costs);
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -0600884#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500885
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600886#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnat0d580581995-07-20 02:43:20 -0500887 /* reset structure */
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500888 png_memcpy(tmp_jmp, png_ptr->jmpbuf, sizeof (jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600889#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500890
891 error_fn = png_ptr->error_fn;
892 warning_fn = png_ptr->warning_fn;
893 error_ptr = png_ptr->error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500894#ifdef PNG_USER_MEM_SUPPORTED
895 free_fn = png_ptr->free_fn;
896#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500897
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500898 png_memset(png_ptr, 0, sizeof (png_struct));
Guy Schalnate5a37791996-06-05 15:50:50 -0500899
900 png_ptr->error_fn = error_fn;
901 png_ptr->warning_fn = warning_fn;
902 png_ptr->error_ptr = error_ptr;
Glenn Randers-Pehrsonf7d1a171998-06-06 15:31:35 -0500903#ifdef PNG_USER_MEM_SUPPORTED
904 png_ptr->free_fn = free_fn;
905#endif
Guy Schalnate5a37791996-06-05 15:50:50 -0500906
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600907#ifdef PNG_SETJMP_SUPPORTED
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500908 png_memcpy(png_ptr->jmpbuf, tmp_jmp, sizeof (jmp_buf));
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600909#endif
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500910}
Guy Schalnate5a37791996-06-05 15:50:50 -0500911
Andreas Dilger47a0c421997-05-16 02:46:07 -0500912/* Allow the application to select one or more row filters to use. */
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500913void
Guy Schalnate5a37791996-06-05 15:50:50 -0500914png_set_filter(png_structp png_ptr, int method, int filters)
Guy Schalnat51f0eb41995-09-26 05:22:39 -0500915{
Andreas Dilger47a0c421997-05-16 02:46:07 -0500916 png_debug(1, "in png_set_filter\n");
917 /* We allow 'method' only for future expansion of the base filter method. */
918 if (method == PNG_FILTER_TYPE_BASE)
Guy Schalnate5a37791996-06-05 15:50:50 -0500919 {
920 switch (filters & (PNG_ALL_FILTERS | 0x07))
921 {
922 case 5:
923 case 6:
Andreas Dilger47a0c421997-05-16 02:46:07 -0500924 case 7: png_warning(png_ptr, "Unknown row filter for method 0");
925 case PNG_FILTER_VALUE_NONE: png_ptr->do_filter=PNG_FILTER_NONE; break;
926 case PNG_FILTER_VALUE_SUB: png_ptr->do_filter=PNG_FILTER_SUB; break;
927 case PNG_FILTER_VALUE_UP: png_ptr->do_filter=PNG_FILTER_UP; break;
928 case PNG_FILTER_VALUE_AVG: png_ptr->do_filter=PNG_FILTER_AVG; break;
929 case PNG_FILTER_VALUE_PAETH: png_ptr->do_filter=PNG_FILTER_PAETH;break;
Guy Schalnate5a37791996-06-05 15:50:50 -0500930 default: png_ptr->do_filter = (png_byte)filters; break;
931 }
932
Andreas Dilger47a0c421997-05-16 02:46:07 -0500933 /* If we have allocated the row_buf, this means we have already started
934 * with the image and we should have allocated all of the filter buffers
935 * that have been selected. If prev_row isn't already allocated, then
936 * it is too late to start using the filters that need it, since we
937 * will be missing the data in the previous row. If an application
938 * wants to start and stop using particular filters during compression,
939 * it should start out with all of the filters, and then add and
940 * remove them after the start of compression.
Guy Schalnate5a37791996-06-05 15:50:50 -0500941 */
Andreas Dilger47a0c421997-05-16 02:46:07 -0500942 if (png_ptr->row_buf != NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500943 {
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600944 if ((png_ptr->do_filter & PNG_FILTER_SUB) && png_ptr->sub_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500945 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500946 png_ptr->sub_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600947 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -0500948 png_ptr->sub_row[0] = PNG_FILTER_VALUE_SUB;
Guy Schalnate5a37791996-06-05 15:50:50 -0500949 }
950
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600951 if ((png_ptr->do_filter & PNG_FILTER_UP) && png_ptr->up_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500952 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500953 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500954 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500955 png_warning(png_ptr, "Can't add Up filter after starting");
Guy Schalnate5a37791996-06-05 15:50:50 -0500956 png_ptr->do_filter &= ~PNG_FILTER_UP;
957 }
958 else
959 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500960 png_ptr->up_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600961 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -0500962 png_ptr->up_row[0] = PNG_FILTER_VALUE_UP;
Guy Schalnate5a37791996-06-05 15:50:50 -0500963 }
964 }
965
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600966 if ((png_ptr->do_filter & PNG_FILTER_AVG) && png_ptr->avg_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500967 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500968 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500969 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500970 png_warning(png_ptr, "Can't add Average filter after starting");
Guy Schalnate5a37791996-06-05 15:50:50 -0500971 png_ptr->do_filter &= ~PNG_FILTER_AVG;
972 }
973 else
974 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500975 png_ptr->avg_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600976 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -0500977 png_ptr->avg_row[0] = PNG_FILTER_VALUE_AVG;
Guy Schalnate5a37791996-06-05 15:50:50 -0500978 }
979 }
980
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -0600981 if ((png_ptr->do_filter & PNG_FILTER_PAETH) &&
Andreas Dilger47a0c421997-05-16 02:46:07 -0500982 png_ptr->paeth_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500983 {
Andreas Dilger47a0c421997-05-16 02:46:07 -0500984 if (png_ptr->prev_row == NULL)
Guy Schalnate5a37791996-06-05 15:50:50 -0500985 {
986 png_warning(png_ptr, "Can't add Paeth filter after starting");
Glenn Randers-Pehrson860ab2b1999-10-14 07:43:10 -0500987 png_ptr->do_filter &= (png_byte)(~PNG_FILTER_PAETH);
Guy Schalnate5a37791996-06-05 15:50:50 -0500988 }
989 else
990 {
Glenn Randers-Pehrson46f61e21998-01-30 21:45:12 -0600991 png_ptr->paeth_row = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -0600992 (png_ptr->rowbytes + 1));
Andreas Dilger47a0c421997-05-16 02:46:07 -0500993 png_ptr->paeth_row[0] = PNG_FILTER_VALUE_PAETH;
Guy Schalnate5a37791996-06-05 15:50:50 -0500994 }
995 }
996
997 if (png_ptr->do_filter == PNG_NO_FILTERS)
998 png_ptr->do_filter = PNG_FILTER_NONE;
999 }
1000 }
1001 else
Andreas Dilger47a0c421997-05-16 02:46:07 -05001002 png_error(png_ptr, "Unknown custom filter method");
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001003}
1004
Andreas Dilger47a0c421997-05-16 02:46:07 -05001005/* This allows us to influence the way in which libpng chooses the "best"
1006 * filter for the current scanline. While the "minimum-sum-of-absolute-
1007 * differences metric is relatively fast and effective, there is some
1008 * question as to whether it can be improved upon by trying to keep the
1009 * filtered data going to zlib more consistent, hopefully resulting in
Glenn Randers-Pehrsonb6ce43d1998-01-01 07:13:13 -06001010 * better compression.
1011 */
Andreas Dilger47a0c421997-05-16 02:46:07 -05001012#if defined(PNG_WRITE_WEIGHTED_FILTER_SUPPORTED) /* GRR 970116 */
1013void
1014png_set_filter_heuristics(png_structp png_ptr, int heuristic_method,
1015 int num_weights, png_doublep filter_weights,
1016 png_doublep filter_costs)
1017{
1018 int i;
1019
1020 png_debug(1, "in png_set_filter_heuristics\n");
1021 if (heuristic_method >= PNG_FILTER_HEURISTIC_LAST)
1022 {
1023 png_warning(png_ptr, "Unknown filter heuristic method");
1024 return;
1025 }
1026
1027 if (heuristic_method == PNG_FILTER_HEURISTIC_DEFAULT)
1028 {
1029 heuristic_method = PNG_FILTER_HEURISTIC_UNWEIGHTED;
1030 }
1031
1032 if (num_weights < 0 || filter_weights == NULL ||
1033 heuristic_method == PNG_FILTER_HEURISTIC_UNWEIGHTED)
1034 {
1035 num_weights = 0;
1036 }
1037
Glenn Randers-Pehrson860ab2b1999-10-14 07:43:10 -05001038 png_ptr->num_prev_filters = (png_byte)num_weights;
1039 png_ptr->heuristic_method = (png_byte)heuristic_method;
Andreas Dilger47a0c421997-05-16 02:46:07 -05001040
1041 if (num_weights > 0)
1042 {
1043 if (png_ptr->prev_filters == NULL)
1044 {
1045 png_ptr->prev_filters = (png_bytep)png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001046 (png_uint_32)(sizeof(png_byte) * num_weights));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001047
1048 /* To make sure that the weighting starts out fairly */
1049 for (i = 0; i < num_weights; i++)
1050 {
1051 png_ptr->prev_filters[i] = 255;
1052 }
1053 }
1054
1055 if (png_ptr->filter_weights == NULL)
1056 {
Glenn Randers-Pehrson983ec161998-03-07 11:24:03 -06001057 png_ptr->filter_weights = (png_uint_16p) png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001058 (png_uint_32)(sizeof(png_uint_16) * num_weights));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001059
Glenn Randers-Pehrson983ec161998-03-07 11:24:03 -06001060 png_ptr->inv_filter_weights = (png_uint_16p) png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001061 (png_uint_32)(sizeof(png_uint_16) * num_weights));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001062
1063 for (i = 0; i < num_weights; i++)
1064 {
1065 png_ptr->inv_filter_weights[i] =
1066 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
1067 }
1068 }
1069
1070 for (i = 0; i < num_weights; i++)
1071 {
1072 if (filter_weights[i] < 0.0)
1073 {
1074 png_ptr->inv_filter_weights[i] =
1075 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR;
1076 }
1077 else
1078 {
1079 png_ptr->inv_filter_weights[i] =
1080 (png_uint_16)((double)PNG_WEIGHT_FACTOR*filter_weights[i]+0.5);
1081 png_ptr->filter_weights[i] =
1082 (png_uint_16)((double)PNG_WEIGHT_FACTOR/filter_weights[i]+0.5);
1083 }
1084 }
1085 }
1086
1087 /* If, in the future, there are other filter methods, this would
1088 * need to be based on png_ptr->filter.
1089 */
1090 if (png_ptr->filter_costs == NULL)
1091 {
Glenn Randers-Pehrson983ec161998-03-07 11:24:03 -06001092 png_ptr->filter_costs = (png_uint_16p) png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001093 (png_uint_32)(sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001094
Glenn Randers-Pehrson983ec161998-03-07 11:24:03 -06001095 png_ptr->inv_filter_costs = (png_uint_16p) png_malloc(png_ptr,
Glenn Randers-Pehrson0f881d61998-02-07 10:20:57 -06001096 (png_uint_32)(sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST));
Andreas Dilger47a0c421997-05-16 02:46:07 -05001097
1098 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
1099 {
1100 png_ptr->inv_filter_costs[i] =
1101 png_ptr->filter_costs[i] = PNG_COST_FACTOR;
1102 }
1103 }
1104
1105 /* Here is where we set the relative costs of the different filters. We
1106 * should take the desired compression level into account when setting
1107 * the costs, so that Paeth, for instance, has a high relative cost at low
1108 * compression levels, while it has a lower relative cost at higher
1109 * compression settings. The filter types are in order of increasing
1110 * relative cost, so it would be possible to do this with an algorithm.
1111 */
1112 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++)
1113 {
1114 if (filter_costs == NULL || filter_costs[i] < 0.0)
1115 {
1116 png_ptr->inv_filter_costs[i] =
1117 png_ptr->filter_costs[i] = PNG_COST_FACTOR;
1118 }
1119 else if (filter_costs[i] >= 1.0)
1120 {
1121 png_ptr->inv_filter_costs[i] =
1122 (png_uint_16)((double)PNG_COST_FACTOR / filter_costs[i] + 0.5);
1123 png_ptr->filter_costs[i] =
1124 (png_uint_16)((double)PNG_COST_FACTOR * filter_costs[i] + 0.5);
1125 }
1126 }
1127}
1128#endif /* PNG_WRITE_WEIGHTED_FILTER_SUPPORTED */
1129
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001130void
Guy Schalnat6d764711995-12-19 03:22:19 -06001131png_set_compression_level(png_structp png_ptr, int level)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001132{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001133 png_debug(1, "in png_set_compression_level\n");
Guy Schalnate5a37791996-06-05 15:50:50 -05001134 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_LEVEL;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001135 png_ptr->zlib_level = level;
1136}
1137
1138void
Guy Schalnat6d764711995-12-19 03:22:19 -06001139png_set_compression_mem_level(png_structp png_ptr, int mem_level)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001140{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001141 png_debug(1, "in png_set_compression_mem_level\n");
Guy Schalnate5a37791996-06-05 15:50:50 -05001142 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_MEM_LEVEL;
Guy Schalnatb2e01bd1996-01-26 01:38:47 -06001143 png_ptr->zlib_mem_level = mem_level;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001144}
1145
1146void
Guy Schalnat6d764711995-12-19 03:22:19 -06001147png_set_compression_strategy(png_structp png_ptr, int strategy)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001148{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001149 png_debug(1, "in png_set_compression_strategy\n");
Guy Schalnate5a37791996-06-05 15:50:50 -05001150 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_STRATEGY;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001151 png_ptr->zlib_strategy = strategy;
1152}
1153
1154void
Guy Schalnat6d764711995-12-19 03:22:19 -06001155png_set_compression_window_bits(png_structp png_ptr, int window_bits)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001156{
Guy Schalnate5a37791996-06-05 15:50:50 -05001157 if (window_bits > 15)
1158 png_warning(png_ptr, "Only compression windows <= 32k supported by PNG");
Glenn Randers-Pehrson4393a9a1999-09-17 12:27:26 -05001159 else if (window_bits < 8)
1160 png_warning(png_ptr, "Only compression windows >= 256 supported by PNG");
Glenn Randers-Pehrson166c5a31999-12-10 09:43:02 -06001161#ifndef WBITS_8_OK
1162 /* avoid libpng bug with 256-byte windows */
1163 if (window_bits == 8)
1164 {
1165 png_warning(png_ptr, "Compression window is being reset to 512");
1166 window_bits=9;
1167 }
1168#endif
Guy Schalnate5a37791996-06-05 15:50:50 -05001169 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_WINDOW_BITS;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001170 png_ptr->zlib_window_bits = window_bits;
1171}
1172
1173void
Guy Schalnat6d764711995-12-19 03:22:19 -06001174png_set_compression_method(png_structp png_ptr, int method)
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001175{
Andreas Dilger47a0c421997-05-16 02:46:07 -05001176 png_debug(1, "in png_set_compression_method\n");
Guy Schalnate5a37791996-06-05 15:50:50 -05001177 if (method != 8)
1178 png_warning(png_ptr, "Only compression method 8 is supported by PNG");
1179 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_METHOD;
Guy Schalnat51f0eb41995-09-26 05:22:39 -05001180 png_ptr->zlib_method = method;
Guy Schalnat0d580581995-07-20 02:43:20 -05001181}
1182
Glenn Randers-Pehrson08a33431998-03-07 06:06:55 -06001183void
1184png_set_write_status_fn(png_structp png_ptr, png_write_status_ptr write_row_fn)
1185{
1186 png_ptr->write_row_fn = write_row_fn;
1187}
1188
1189#if defined(PNG_WRITE_USER_TRANSFORM_SUPPORTED)
1190void
1191png_set_write_user_transform_fn(png_structp png_ptr, png_user_transform_ptr
1192 write_user_transform_fn)
1193{
1194 png_debug(1, "in png_set_write_user_transform_fn\n");
1195 png_ptr->transformations |= PNG_USER_TRANSFORM;
1196 png_ptr->write_user_transform_fn = write_user_transform_fn;
1197}
1198#endif
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001199
1200
1201#if defined(PNG_INFO_IMAGE_SUPPORTED)
1202void png_write_png(png_structp png_ptr, png_infop info_ptr,
1203 int transforms,
1204 voidp params)
1205{
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001206#if defined(PNG_WRITE_INVERT_ALPHA_SUPPORTED)
1207 /* invert the alpha channel from opacity to transparency */
1208 if (transforms & PNG_TRANSFORM_INVERT_ALPHA)
1209 png_set_invert_alpha(png_ptr);
1210#endif
1211
1212 /* Write the file header information. */
1213 png_write_info(png_ptr, info_ptr);
1214
1215 /* ------ these transformations don't touch the info structure ------- */
1216
1217#if defined(PNG_WRITE_INVERT_SUPPORTED)
1218 /* invert monochrome pixels */
1219 if (transforms & PNG_TRANSFORM_INVERT_MONO)
1220 png_set_invert_mono(png_ptr);
1221#endif
1222
1223#if defined(PNG_WRITE_SHIFT_SUPPORTED)
1224 /* Shift the pixels up to a legal bit depth and fill in
1225 * as appropriate to correctly scale the image.
1226 */
1227 if ((transforms & PNG_TRANSFORM_SHIFT)
1228 && (info_ptr->valid & PNG_INFO_sBIT))
1229 png_set_shift(png_ptr, &info_ptr->sig_bit);
1230#endif
1231
1232#if defined(PNG_WRITE_PACK_SUPPORTED)
1233 /* pack pixels into bytes */
1234 if (transforms & PNG_TRANSFORM_PACKING)
1235 png_set_packing(png_ptr);
1236#endif
1237
1238#if defined(PNG_WRITE_SWAP_ALPHA_SUPPORTED)
1239 /* swap location of alpha bytes from ARGB to RGBA */
1240 if (transforms & PNG_TRANSFORM_SWAP_ALPHA)
1241 png_set_swap_alpha(png_ptr);
1242#endif
1243
1244#if defined(PNG_WRITE_FILLER_SUPPORTED)
1245 /* Get rid of filler (OR ALPHA) bytes, pack XRGB/RGBX/ARGB/RGBA into
1246 * RGB (4 channels -> 3 channels). The second parameter is not used.
1247 */
1248 if (transforms & PNG_TRANSFORM_STRIP_FILLER)
1249 png_set_filler(png_ptr, 0, PNG_FILLER_BEFORE);
1250#endif
1251
1252#if defined(PNG_WRITE_BGR_SUPPORTED)
1253 /* flip BGR pixels to RGB */
1254 if (transforms & PNG_TRANSFORM_BGR)
1255 png_set_bgr(png_ptr);
1256#endif
1257
1258#if defined(PNG_WRITE_SWAP_SUPPORTED)
1259 /* swap bytes of 16-bit files to most significant byte first */
1260 if (transforms & PNG_TRANSFORM_SWAP_ENDIAN)
1261 png_set_swap(png_ptr);
1262#endif
1263
1264#if defined(PNG_WRITE_PACKSWAP_SUPPORTED)
1265 /* swap bits of 1, 2, 4 bit packed pixel formats */
1266 if (transforms & PNG_TRANSFORM_PACKSWAP)
1267 png_set_packswap(png_ptr);
1268#endif
1269
1270 /* ----------------------- end of transformations ------------------- */
1271
1272 /* write the bits */
1273 if (info_ptr->valid & PNG_INFO_IDAT)
1274 png_write_image(png_ptr, info_ptr->row_pointers);
1275
1276 /* It is REQUIRED to call this to finish writing the rest of the file */
1277 png_write_end(png_ptr, info_ptr);
Glenn Randers-Pehrson520a7642000-03-21 05:13:06 -06001278
1279 if(transforms == 0 || params == (voidp)NULL)
1280 /* quiet compiler warnings */ return;
Glenn Randers-Pehrson61c32d92000-02-04 23:40:16 -06001281}
1282#endif