1 /*-------------------------------------
2 * PNGFILE.C -- Image File Functions
3 *-------------------------------------
4 *
5 * Copyright 2000, Willem van Schaik.
6 *
7 * This code is released under the libpng license.
8 * For conditions of distribution and use, see the disclaimer
9 * and license in png.h
10 */
11
12 #include <windows.h>
13 #include <commdlg.h>
14 #include <stdio.h>
15 #include <stdlib.h>
16 #include <zlib.h>
17
18 #include "png.h"
19 #include "pngfile.h"
20 #include "cexcept.h"
21
22 define_exception_type(const char *);
23 extern struct exception_context the_exception_context[1];
24 struct exception_context the_exception_context[1];
25 png_const_charp msg;
26
27 static OPENFILENAME ofn;
28
29 static png_structp png_ptr = NULL;
30 static png_infop info_ptr = NULL;
31
32
33 /* cexcept interface */
34
35 static void
png_cexcept_error(png_structp png_ptr,png_const_charp msg)36 png_cexcept_error(png_structp png_ptr, png_const_charp msg)
37 {
38 if(png_ptr)
39 ;
40 #ifdef PNG_CONSOLE_IO_SUPPORTED
41 fprintf(stderr, "libpng error: %s\n", msg);
42 #endif
43 {
44 Throw msg;
45 }
46 }
47
48 /* Windows open-file functions */
49
PngFileInitialize(HWND hwnd)50 void PngFileInitialize (HWND hwnd)
51 {
52 static TCHAR szFilter[] = TEXT ("PNG Files (*.PNG)\0*.png\0")
53 TEXT ("All Files (*.*)\0*.*\0\0");
54
55 ofn.lStructSize = sizeof (OPENFILENAME);
56 ofn.hwndOwner = hwnd;
57 ofn.hInstance = NULL;
58 ofn.lpstrFilter = szFilter;
59 ofn.lpstrCustomFilter = NULL;
60 ofn.nMaxCustFilter = 0;
61 ofn.nFilterIndex = 0;
62 ofn.lpstrFile = NULL; /* Set in Open and Close functions */
63 ofn.nMaxFile = MAX_PATH;
64 ofn.lpstrFileTitle = NULL; /* Set in Open and Close functions */
65 ofn.nMaxFileTitle = MAX_PATH;
66 ofn.lpstrInitialDir = NULL;
67 ofn.lpstrTitle = NULL;
68 ofn.Flags = 0; /* Set in Open and Close functions */
69 ofn.nFileOffset = 0;
70 ofn.nFileExtension = 0;
71 ofn.lpstrDefExt = TEXT ("png");
72 ofn.lCustData = 0;
73 ofn.lpfnHook = NULL;
74 ofn.lpTemplateName = NULL;
75 }
76
PngFileOpenDlg(HWND hwnd,PTSTR pstrFileName,PTSTR pstrTitleName)77 BOOL PngFileOpenDlg (HWND hwnd, PTSTR pstrFileName, PTSTR pstrTitleName)
78 {
79 ofn.hwndOwner = hwnd;
80 ofn.lpstrFile = pstrFileName;
81 ofn.lpstrFileTitle = pstrTitleName;
82 ofn.Flags = OFN_HIDEREADONLY;
83
84 return GetOpenFileName (&ofn);
85 }
86
PngFileSaveDlg(HWND hwnd,PTSTR pstrFileName,PTSTR pstrTitleName)87 BOOL PngFileSaveDlg (HWND hwnd, PTSTR pstrFileName, PTSTR pstrTitleName)
88 {
89 ofn.hwndOwner = hwnd;
90 ofn.lpstrFile = pstrFileName;
91 ofn.lpstrFileTitle = pstrTitleName;
92 ofn.Flags = OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT;
93
94 return GetSaveFileName (&ofn);
95 }
96
97 /* PNG image handler functions */
98
PngLoadImage(PTSTR pstrFileName,png_byte ** ppbImageData,int * piWidth,int * piHeight,int * piChannels,png_color * pBkgColor)99 BOOL PngLoadImage (PTSTR pstrFileName, png_byte **ppbImageData,
100 int *piWidth, int *piHeight, int *piChannels, png_color *pBkgColor)
101 {
102 static FILE *pfFile;
103 png_byte pbSig[8];
104 int iBitDepth;
105 int iColorType;
106 double dGamma;
107 png_color_16 *pBackground;
108 png_uint_32 ulChannels;
109 png_uint_32 ulRowBytes;
110 png_byte *pbImageData = *ppbImageData;
111 static png_byte **ppbRowPointers = NULL;
112 int i;
113
114 /* open the PNG input file */
115
116 if (!pstrFileName)
117 {
118 *ppbImageData = pbImageData = NULL;
119 return FALSE;
120 }
121
122 if (!(pfFile = fopen(pstrFileName, "rb")))
123 {
124 *ppbImageData = pbImageData = NULL;
125 return FALSE;
126 }
127
128 /* first check the eight byte PNG signature */
129
130 fread(pbSig, 1, 8, pfFile);
131 if (png_sig_cmp(pbSig, 0, 8))
132 {
133 *ppbImageData = pbImageData = NULL;
134 return FALSE;
135 }
136
137 /* create the two png(-info) structures */
138
139 png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL,
140 (png_error_ptr)png_cexcept_error, (png_error_ptr)NULL);
141 if (!png_ptr)
142 {
143 *ppbImageData = pbImageData = NULL;
144 return FALSE;
145 }
146
147 info_ptr = png_create_info_struct(png_ptr);
148 if (!info_ptr)
149 {
150 png_destroy_read_struct(&png_ptr, NULL, NULL);
151 *ppbImageData = pbImageData = NULL;
152 return FALSE;
153 }
154
155 Try
156 {
157
158 /* initialize the png structure */
159
160 #ifdef PNG_STDIO_SUPPORTED
161 png_init_io(png_ptr, pfFile);
162 #else
163 png_set_read_fn(png_ptr, (png_voidp)pfFile, png_read_data);
164 #endif
165
166 png_set_sig_bytes(png_ptr, 8);
167
168 /* read all PNG info up to image data */
169
170 png_read_info(png_ptr, info_ptr);
171
172 /* get width, height, bit-depth and color-type */
173
174 png_get_IHDR(png_ptr, info_ptr, piWidth, piHeight, &iBitDepth,
175 &iColorType, NULL, NULL, NULL);
176
177 /* expand images of all color-type and bit-depth to 3x8-bit RGB */
178 /* let the library process alpha, transparency, background, etc. */
179
180 #ifdef PNG_READ_16_TO_8_SUPPORTED
181 if (iBitDepth == 16)
182 # ifdef PNG_READ_SCALE_16_TO_8_SUPPORTED
183 png_set_scale_16(png_ptr);
184 # else
185 png_set_strip_16(png_ptr);
186 # endif
187 #endif
188 if (iColorType == PNG_COLOR_TYPE_PALETTE)
189 png_set_expand(png_ptr);
190 if (iBitDepth < 8)
191 png_set_expand(png_ptr);
192 if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS))
193 png_set_expand(png_ptr);
194 if (iColorType == PNG_COLOR_TYPE_GRAY ||
195 iColorType == PNG_COLOR_TYPE_GRAY_ALPHA)
196 png_set_gray_to_rgb(png_ptr);
197
198 /* set the background color to draw transparent and alpha images over */
199 if (png_get_bKGD(png_ptr, info_ptr, &pBackground))
200 {
201 png_set_background(png_ptr, pBackground, PNG_BACKGROUND_GAMMA_FILE, 1, 1.0);
202 pBkgColor->red = (byte) pBackground->red;
203 pBkgColor->green = (byte) pBackground->green;
204 pBkgColor->blue = (byte) pBackground->blue;
205 }
206 else
207 {
208 pBkgColor = NULL;
209 }
210
211 /* if required set gamma conversion */
212 if (png_get_gAMA(png_ptr, info_ptr, &dGamma))
213 png_set_gamma(png_ptr, (double) 2.2, dGamma);
214
215 /* after the transformations are registered, update info_ptr data */
216
217 png_read_update_info(png_ptr, info_ptr);
218
219 /* get again width, height and the new bit-depth and color-type */
220
221 png_get_IHDR(png_ptr, info_ptr, piWidth, piHeight, &iBitDepth,
222 &iColorType, NULL, NULL, NULL);
223
224
225 /* row_bytes is the width x number of channels */
226
227 ulRowBytes = png_get_rowbytes(png_ptr, info_ptr);
228 ulChannels = png_get_channels(png_ptr, info_ptr);
229
230 *piChannels = ulChannels;
231
232 /* now we can allocate memory to store the image */
233
234 if (pbImageData)
235 {
236 free (pbImageData);
237 pbImageData = NULL;
238 }
239 if ((pbImageData = (png_byte *) malloc(ulRowBytes * (*piHeight)
240 * sizeof(png_byte))) == NULL)
241 {
242 png_error(png_ptr, "Visual PNG: out of memory");
243 }
244 *ppbImageData = pbImageData;
245
246 /* and allocate memory for an array of row-pointers */
247
248 if ((ppbRowPointers = (png_bytepp) malloc((*piHeight)
249 * sizeof(png_bytep))) == NULL)
250 {
251 png_error(png_ptr, "Visual PNG: out of memory");
252 }
253
254 /* set the individual row-pointers to point at the correct offsets */
255
256 for (i = 0; i < (*piHeight); i++)
257 ppbRowPointers[i] = pbImageData + i * ulRowBytes;
258
259 /* now we can go ahead and just read the whole image */
260
261 png_read_image(png_ptr, ppbRowPointers);
262
263 /* read the additional chunks in the PNG file (not really needed) */
264
265 png_read_end(png_ptr, NULL);
266
267 /* and we're done */
268
269 free (ppbRowPointers);
270 ppbRowPointers = NULL;
271
272 /* yepp, done */
273 }
274
275 Catch (msg)
276 {
277 png_destroy_read_struct(&png_ptr, &info_ptr, NULL);
278
279 *ppbImageData = pbImageData = NULL;
280
281 if(ppbRowPointers)
282 free (ppbRowPointers);
283
284 fclose(pfFile);
285
286 return FALSE;
287 }
288
289 fclose (pfFile);
290
291 return TRUE;
292 }
293
294
PngSaveImage(PTSTR pstrFileName,png_byte * pDiData,int iWidth,int iHeight,png_color bkgColor)295 BOOL PngSaveImage (PTSTR pstrFileName, png_byte *pDiData,
296 int iWidth, int iHeight, png_color bkgColor)
297 {
298 const int ciBitDepth = 8;
299 const int ciChannels = 3;
300
301 static FILE *pfFile;
302 png_uint_32 ulRowBytes;
303 static png_byte **ppbRowPointers = NULL;
304 int i;
305
306 /* open the PNG output file */
307
308 if (!pstrFileName)
309 return FALSE;
310
311 if (!(pfFile = fopen(pstrFileName, "wb")))
312 return FALSE;
313
314 /* prepare the standard PNG structures */
315
316 png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL,
317 (png_error_ptr)png_cexcept_error, (png_error_ptr)NULL);
318 if (!png_ptr)
319 {
320 fclose(pfFile);
321 return FALSE;
322 }
323
324 info_ptr = png_create_info_struct(png_ptr);
325 if (!info_ptr) {
326 fclose(pfFile);
327 png_destroy_write_struct(&png_ptr, (png_infopp) NULL);
328 return FALSE;
329 }
330
331 Try
332 {
333 /* initialize the png structure */
334
335 #ifdef PNG_STDIO_SUPPORTED
336 png_init_io(png_ptr, pfFile);
337 #else
338 png_set_write_fn(png_ptr, (png_voidp)pfFile, png_write_data, png_flush);
339 #endif
340
341 /* we're going to write a very simple 3x8-bit RGB image */
342
343 png_set_IHDR(png_ptr, info_ptr, iWidth, iHeight, ciBitDepth,
344 PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_BASE,
345 PNG_FILTER_TYPE_BASE);
346
347 /* write the file header information */
348
349 png_write_info(png_ptr, info_ptr);
350
351 /* swap the BGR pixels in the DiData structure to RGB */
352
353 png_set_bgr(png_ptr);
354
355 /* row_bytes is the width x number of channels */
356
357 ulRowBytes = iWidth * ciChannels;
358
359 /* we can allocate memory for an array of row-pointers */
360
361 if ((ppbRowPointers = (png_bytepp) malloc(iHeight * sizeof(png_bytep))) == NULL)
362 Throw "Visualpng: Out of memory";
363
364 /* set the individual row-pointers to point at the correct offsets */
365
366 for (i = 0; i < iHeight; i++)
367 ppbRowPointers[i] = pDiData + i * (((ulRowBytes + 3) >> 2) << 2);
368
369 /* write out the entire image data in one call */
370
371 png_write_image (png_ptr, ppbRowPointers);
372
373 /* write the additional chunks to the PNG file (not really needed) */
374
375 png_write_end(png_ptr, info_ptr);
376
377 /* and we're done */
378
379 free (ppbRowPointers);
380 ppbRowPointers = NULL;
381
382 /* clean up after the write, and free any memory allocated */
383
384 png_destroy_write_struct(&png_ptr, (png_infopp) NULL);
385
386 /* yepp, done */
387 }
388
389 Catch (msg)
390 {
391 png_destroy_write_struct(&png_ptr, (png_infopp) NULL);
392
393 if(ppbRowPointers)
394 free (ppbRowPointers);
395
396 fclose(pfFile);
397
398 return FALSE;
399 }
400
401 fclose (pfFile);
402
403 return TRUE;
404 }
405
406 #ifndef PNG_STDIO_SUPPORTED
407
408 static void
png_read_data(png_structp png_ptr,png_bytep data,png_size_t length)409 png_read_data(png_structp png_ptr, png_bytep data, png_size_t length)
410 {
411 png_size_t check;
412
413 /* fread() returns 0 on error, so it is OK to store this in a png_size_t
414 * instead of an int, which is what fread() actually returns.
415 */
416 check = (png_size_t)fread(data, (png_size_t)1, length,
417 (FILE *)png_ptr->io_ptr);
418
419 if (check != length)
420 {
421 png_error(png_ptr, "Read Error");
422 }
423 }
424
425 static void
png_write_data(png_structp png_ptr,png_bytep data,png_size_t length)426 png_write_data(png_structp png_ptr, png_bytep data, png_size_t length)
427 {
428 png_uint_32 check;
429
430 check = fwrite(data, 1, length, (FILE *)(png_ptr->io_ptr));
431 if (check != length)
432 {
433 png_error(png_ptr, "Write Error");
434 }
435 }
436
437 static void
png_flush(png_structp png_ptr)438 png_flush(png_structp png_ptr)
439 {
440 FILE *io_ptr;
441 io_ptr = (FILE *)CVT_PTR((png_ptr->io_ptr));
442 if (io_ptr != NULL)
443 fflush(io_ptr);
444 }
445
446 #endif
447
448 /*-----------------
449 * end of source
450 *-----------------
451 */
452