]> de.git.xonotic.org Git - xonotic/darkplaces.git/blob - image_png.c
changed DP_MODPLUG_STATIC_PREFIX to DP_MODPLUG_STATIC_LIBDIR (so it works on Fedora...
[xonotic/darkplaces.git] / image_png.c
1 /*
2         Copyright (C) 2006  Serge "(515)" Ziryukin, Forest "LordHavoc" Hale
3
4         This program is free software; you can redistribute it and/or
5         modify it under the terms of the GNU General Public License
6         as published by the Free Software Foundation; either version 2
7         of the License, or (at your option) any later version.
8
9         This program is distributed in the hope that it will be useful,
10         but WITHOUT ANY WARRANTY; without even the implied warranty of
11         MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
12
13         See the GNU General Public License for more details.
14
15         You should have received a copy of the GNU General Public License
16         along with this program; if not, write to:
17
18                 Free Software Foundation, Inc.
19                 59 Temple Place - Suite 330
20                 Boston, MA  02111-1307, USA
21
22 */
23
24 //[515]: png implemented into DP ONLY FOR TESTING 2d stuff with csqc
25 // so delete this bullshit :D
26 //
27 //LordHavoc: rewrote most of this.
28
29 #include "quakedef.h"
30 #include "image.h"
31 #include "image_png.h"
32
33 static void                             (*qpng_set_sig_bytes)           (void*, int);
34 static int                              (*qpng_sig_cmp)                         (const unsigned char*, size_t, size_t);
35 static void*                    (*qpng_create_read_struct)      (const char*, void*, void*, void*);
36 static void*                    (*qpng_create_info_struct)      (void*);
37 static void                             (*qpng_read_info)                       (void*, void*);
38 static void                             (*qpng_set_expand)                      (void*);
39 static void                             (*qpng_set_gray_1_2_4_to_8)     (void*);
40 static void                             (*qpng_set_palette_to_rgb)      (void*);
41 static void                             (*qpng_set_tRNS_to_alpha)       (void*);
42 static void                             (*qpng_set_gray_to_rgb)         (void*);
43 static void                             (*qpng_set_filler)                      (void*, unsigned int, int);
44 static void                             (*qpng_read_update_info)        (void*, void*);
45 static void                             (*qpng_read_image)                      (void*, unsigned char**);
46 static void                             (*qpng_read_end)                        (void*, void*);
47 static void                             (*qpng_destroy_read_struct)     (void**, void**, void**);
48 static void                             (*qpng_set_read_fn)                     (void*, void*, void*);
49 static unsigned int             (*qpng_get_valid)                       (void*, void*, unsigned int);
50 static unsigned int             (*qpng_get_rowbytes)            (void*, void*);
51 static unsigned char    (*qpng_get_channels)            (void*, void*);
52 static unsigned char    (*qpng_get_bit_depth)           (void*, void*);
53 static unsigned int             (*qpng_get_IHDR)                        (void*, void*, unsigned long*, unsigned long*, int *, int *, int *, int *, int *);
54 static char*                    (*qpng_get_libpng_ver)          (void*);
55
56 static dllfunction_t pngfuncs[] =
57 {
58         {"png_set_sig_bytes",           (void **) &qpng_set_sig_bytes},
59         {"png_sig_cmp",                         (void **) &qpng_sig_cmp},
60         {"png_create_read_struct",      (void **) &qpng_create_read_struct},
61         {"png_create_info_struct",      (void **) &qpng_create_info_struct},
62         {"png_read_info",                       (void **) &qpng_read_info},
63         {"png_set_expand",                      (void **) &qpng_set_expand},
64         {"png_set_gray_1_2_4_to_8",     (void **) &qpng_set_gray_1_2_4_to_8},
65         {"png_set_palette_to_rgb",      (void **) &qpng_set_palette_to_rgb},
66         {"png_set_tRNS_to_alpha",       (void **) &qpng_set_tRNS_to_alpha},
67         {"png_set_gray_to_rgb",         (void **) &qpng_set_gray_to_rgb},
68         {"png_set_filler",                      (void **) &qpng_set_filler},
69         {"png_read_update_info",        (void **) &qpng_read_update_info},
70         {"png_read_image",                      (void **) &qpng_read_image},
71         {"png_read_end",                        (void **) &qpng_read_end},
72         {"png_destroy_read_struct",     (void **) &qpng_destroy_read_struct},
73         {"png_set_read_fn",                     (void **) &qpng_set_read_fn},
74         {"png_get_valid",                       (void **) &qpng_get_valid},
75         {"png_get_rowbytes",            (void **) &qpng_get_rowbytes},
76         {"png_get_channels",            (void **) &qpng_get_channels},
77         {"png_get_bit_depth",           (void **) &qpng_get_bit_depth},
78         {"png_get_IHDR",                        (void **) &qpng_get_IHDR},
79         {"png_get_libpng_ver",          (void **) &qpng_get_libpng_ver},
80         {NULL, NULL}
81 };
82
83 // Handle for PNG DLL
84 dllhandle_t png_dll = NULL;
85
86
87 /*
88 =================================================================
89
90   DLL load & unload
91
92 =================================================================
93 */
94
95 /*
96 ====================
97 PNG_OpenLibrary
98
99 Try to load the PNG DLL
100 ====================
101 */
102 qboolean PNG_OpenLibrary (void)
103 {
104         const char* dllnames [] =
105         {
106 #ifdef WIN64
107                 "libpng12_64.dll",
108 #elif WIN32
109                 "libpng12.dll",
110 #elif defined(MACOSX)
111                 "libpng12.0.dylib",
112 #else
113                 "libpng12.so.0",
114                 "libpng.so", // FreeBSD
115 #endif
116                 NULL
117         };
118
119         // Already loaded?
120         if (png_dll)
121                 return true;
122
123         // Load the DLL
124         if (! Sys_LoadLibrary (dllnames, &png_dll, pngfuncs))
125         {
126                 Con_Printf ("PNG support disabled\n");
127                 return false;
128         }
129
130         Con_Printf ("PNG support enabled\n");
131         return true;
132 }
133
134
135 /*
136 ====================
137 PNG_CloseLibrary
138
139 Unload the PNG DLL
140 ====================
141 */
142 void PNG_CloseLibrary (void)
143 {
144         Sys_UnloadLibrary (&png_dll);
145 }
146
147
148 /*
149 =================================================================
150
151         PNG decompression
152
153 =================================================================
154 */
155
156 #define PNG_LIBPNG_VER_STRING "1.2.4"
157
158 #define PNG_COLOR_MASK_PALETTE    1
159 #define PNG_COLOR_MASK_COLOR      2
160 #define PNG_COLOR_MASK_ALPHA      4
161
162 #define PNG_COLOR_TYPE_GRAY 0
163 #define PNG_COLOR_TYPE_PALETTE  (PNG_COLOR_MASK_COLOR | PNG_COLOR_MASK_PALETTE)
164 #define PNG_COLOR_TYPE_RGB        (PNG_COLOR_MASK_COLOR)
165 #define PNG_COLOR_TYPE_RGB_ALPHA  (PNG_COLOR_MASK_COLOR | PNG_COLOR_MASK_ALPHA)
166 #define PNG_COLOR_TYPE_GRAY_ALPHA (PNG_COLOR_MASK_ALPHA)
167
168 #define PNG_COLOR_TYPE_RGBA  PNG_COLOR_TYPE_RGB_ALPHA
169 #define PNG_COLOR_TYPE_GA  PNG_COLOR_TYPE_GRAY_ALPHA
170
171 #define PNG_INFO_tRNS 0x0010
172
173 // this struct is only used for status information during loading
174 static struct
175 {
176         const unsigned char     *tmpBuf;
177         int             tmpBuflength;
178         int             tmpi;
179         //int           FBgColor;
180         //int           FTransparent;
181         unsigned int    FRowBytes;
182         //double        FGamma;
183         //double        FScreenGamma;
184         unsigned char   **FRowPtrs;
185         unsigned char   *Data;
186         //char  *Title;
187         //char  *Author;
188         //char  *Description;
189         int             BitDepth;
190         int             BytesPerPixel;
191         int             ColorType;
192         unsigned long   Height; // retarded libpng 1.2 pngconf.h uses long (64bit/32bit depending on arch)
193         unsigned long   Width; // retarded libpng 1.2 pngconf.h uses long (64bit/32bit depending on arch)
194         int             Interlace;
195         int             Compression;
196         int             Filter;
197         //double        LastModified;
198         //int           Transparent;
199 } my_png;
200
201 //LordHavoc: removed __cdecl prefix, added overrun protection, and rewrote this to be more efficient
202 void PNG_fReadData(void *png, unsigned char *data, size_t length)
203 {
204         size_t l;
205         l = my_png.tmpBuflength - my_png.tmpi;
206         if (l < length)
207         {
208                 Con_Printf("PNG_fReadData: overrun by %i bytes\n", (int)(length - l));
209                 // a read going past the end of the file, fill in the remaining bytes
210                 // with 0 just to be consistent
211                 memset(data + l, 0, length - l);
212                 length = l;
213         }
214         memcpy(data, my_png.tmpBuf + my_png.tmpi, length);
215         my_png.tmpi += (int)length;
216         //Com_HexDumpToConsole(data, (int)length);
217 }
218
219 void PNG_error_fn(void *png, const char *message)
220 {
221         Con_Printf("PNG_LoadImage: error: %s\n", message);
222 }
223
224 void PNG_warning_fn(void *png, const char *message)
225 {
226         Con_Printf("PNG_LoadImage: warning: %s\n", message);
227 }
228
229 extern int      image_width;
230 extern int      image_height;
231
232 unsigned char *PNG_LoadImage_BGRA (const unsigned char *raw, int filesize)
233 {
234         unsigned int c;
235         unsigned int    y;
236         void *png, *pnginfo;
237         unsigned char *imagedata = NULL;
238         unsigned char ioBuffer[8192];
239
240         // FIXME: register an error handler so that abort() won't be called on error
241
242         // No DLL = no PNGs
243         if (!png_dll)
244                 return NULL;
245
246         if(qpng_sig_cmp(raw, 0, filesize))
247                 return NULL;
248         png = (void *)qpng_create_read_struct(PNG_LIBPNG_VER_STRING, 0, (void *)PNG_error_fn, (void *)PNG_warning_fn);
249         if(!png)
250                 return NULL;
251
252         // this must be memset before the setjmp error handler, because it relies
253         // on the fields in this struct for cleanup
254         memset(&my_png, 0, sizeof(my_png));
255
256         // NOTE: this relies on jmp_buf being the first thing in the png structure
257         // created by libpng! (this is correct for libpng 1.2.x)
258 #ifdef __cplusplus
259 #ifdef MACOSX
260         if (setjmp((int *)png))
261 #else
262         if (setjmp((__jmp_buf_tag *)png))
263 #endif
264 #else
265         if (setjmp(png))
266 #endif
267         {
268                 if (my_png.Data)
269                         Mem_Free(my_png.Data);
270                 my_png.Data = NULL;
271                 if (my_png.FRowPtrs)
272                         Mem_Free(my_png.FRowPtrs);
273                 my_png.FRowPtrs = NULL;
274                 qpng_destroy_read_struct(&png, &pnginfo, 0);
275                 return NULL;
276         }
277         //
278
279         pnginfo = qpng_create_info_struct(png);
280         if(!pnginfo)
281         {
282                 qpng_destroy_read_struct(&png, &pnginfo, 0);
283                 return NULL;
284         }
285         qpng_set_sig_bytes(png, 0);
286
287         my_png.tmpBuf = raw;
288         my_png.tmpBuflength = filesize;
289         my_png.tmpi = 0;
290         //my_png.Data           = NULL;
291         //my_png.FRowPtrs       = NULL;
292         //my_png.Height         = 0;
293         //my_png.Width          = 0;
294         my_png.ColorType        = PNG_COLOR_TYPE_RGB;
295         //my_png.Interlace      = 0;
296         //my_png.Compression    = 0;
297         //my_png.Filter         = 0;
298         qpng_set_read_fn(png, ioBuffer, (void *)PNG_fReadData);
299         qpng_read_info(png, pnginfo);
300         qpng_get_IHDR(png, pnginfo, &my_png.Width, &my_png.Height,&my_png.BitDepth, &my_png.ColorType, &my_png.Interlace, &my_png.Compression, &my_png.Filter);
301
302         // this check guards against pngconf.h with unsigned int *width/height parameters on big endian systems by detecting the strange values and shifting them down 32bits
303         // (if it's little endian the unwritten bytes are the most significant
304         //  ones and we don't worry about that)
305         //
306         // this is only necessary because of retarded 64bit png_uint_32 types in libpng 1.2, which can (conceivably) vary by platform
307 #if LONG_MAX > 4000000000
308         if (my_png.Width > LONG_MAX || my_png.Height > LONG_MAX)
309         {
310                 my_png.Width >>= 32;
311                 my_png.Height >>= 32;
312         }
313 #endif
314
315         if (my_png.ColorType == PNG_COLOR_TYPE_PALETTE)
316                 qpng_set_palette_to_rgb(png);
317         if (my_png.ColorType == PNG_COLOR_TYPE_GRAY || my_png.ColorType == PNG_COLOR_TYPE_GRAY_ALPHA)
318         {
319                 qpng_set_gray_to_rgb(png);
320                 if (my_png.BitDepth < 8)
321                         qpng_set_gray_1_2_4_to_8(png);
322         }
323
324         if (qpng_get_valid(png, pnginfo, PNG_INFO_tRNS))
325                 qpng_set_tRNS_to_alpha(png);
326         if (my_png.BitDepth == 8 && !(my_png.ColorType  & PNG_COLOR_MASK_ALPHA))
327                 qpng_set_filler(png, 255, 1);
328         if (( my_png.ColorType == PNG_COLOR_TYPE_GRAY) || (my_png.ColorType == PNG_COLOR_TYPE_GRAY_ALPHA ))
329                 qpng_set_gray_to_rgb(png);
330         if (my_png.BitDepth < 8)
331                 qpng_set_expand(png);
332
333         qpng_read_update_info(png, pnginfo);
334
335         my_png.FRowBytes = qpng_get_rowbytes(png, pnginfo);
336         my_png.BytesPerPixel = qpng_get_channels(png, pnginfo);
337
338         my_png.FRowPtrs = (unsigned char **)Mem_Alloc(tempmempool, my_png.Height * sizeof(*my_png.FRowPtrs));
339         if (my_png.FRowPtrs)
340         {
341                 imagedata = (unsigned char *)Mem_Alloc(tempmempool, my_png.Height * my_png.FRowBytes);
342                 if(imagedata)
343                 {
344                         my_png.Data = imagedata;
345                         for(y = 0;y < my_png.Height;y++)
346                                 my_png.FRowPtrs[y] = my_png.Data + y * my_png.FRowBytes;
347                         qpng_read_image(png, my_png.FRowPtrs);
348                 }
349                 else
350                         Con_DPrintf("PNG_LoadImage : not enough memory\n");
351                 Mem_Free(my_png.FRowPtrs);
352                 my_png.FRowPtrs = NULL;
353         }
354         else
355                 Con_DPrintf("PNG_LoadImage : not enough memory\n");
356
357         qpng_read_end(png, pnginfo);
358         qpng_destroy_read_struct(&png, &pnginfo, 0);
359
360         image_width = (int)my_png.Width;
361         image_height = (int)my_png.Height;
362
363         if (my_png.BitDepth != 8)
364         {
365                 Con_Printf ("PNG_LoadImage : bad color depth\n");
366                 Mem_Free(imagedata);
367                 imagedata = NULL;
368         }
369
370         // swizzle RGBA to BGRA
371         for (y = 0;y < (unsigned int)(image_width*image_height*4);y += 4)
372         {
373                 c = imagedata[y+0];
374                 imagedata[y+0] = imagedata[y+2];
375                 imagedata[y+2] = c;
376         }
377
378         return imagedata;
379 }
380