infcubes crashes
[dosdemo] / libs / imago / src / filepng.c
1 /*
2 libimago - a multi-format image file input/output library.
3 Copyright (C) 2010 John Tsiombikas <nuclear@member.fsf.org>
4
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU Lesser General Public License as published
7 by the Free Software Foundation, either version 3 of the License, or
8 (at your option) any later version.
9
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 GNU Lesser General Public License for more details.
14
15 You should have received a copy of the GNU Lesser General Public License
16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
17 */
18
19 /* -- PNG module -- */
20
21 #include <stdlib.h>
22 #include <string.h>
23 #include <png.h>
24 #include "imago2.h"
25 #include "ftmodule.h"
26
27 static int check_file(struct img_io *io);
28 static int read_file(struct img_pixmap *img, struct img_io *io);
29 static int write_file(struct img_pixmap *img, struct img_io *io);
30
31 static void read_func(png_struct *png, unsigned char *data, size_t len);
32 static void write_func(png_struct *png, unsigned char *data, size_t len);
33 static void flush_func(png_struct *png);
34
35 static int png_type_to_fmt(int color_type, int channel_bits);
36 static int fmt_to_png_type(enum img_fmt fmt);
37
38
39 int img_register_png(void)
40 {
41         static struct ftype_module mod = {".png", check_file, read_file, write_file};
42         return img_register_module(&mod);
43 }
44
45 static int check_file(struct img_io *io)
46 {
47         unsigned char sig[8];
48         int res;
49         long pos = io->seek(0, SEEK_CUR, io->uptr);
50
51         if(io->read(sig, 8, io->uptr) < 8) {
52                 io->seek(pos, SEEK_SET, io->uptr);
53                 return -1;
54         }
55
56         res = png_sig_cmp(sig, 0, 8) == 0 ? 0 : -1;
57         io->seek(pos, SEEK_SET, io->uptr);
58         return res;
59 }
60
61 static int read_file(struct img_pixmap *img, struct img_io *io)
62 {
63         png_struct *png;
64         png_info *info;
65         int channel_bits, color_type, ilace_type, compression, filtering, fmt;
66         png_uint_32 xsz, ysz;
67
68         if(!(png = png_create_read_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0))) {
69                 return -1;
70         }
71
72         if(!(info = png_create_info_struct(png))) {
73                 png_destroy_read_struct(&png, 0, 0);
74                 return -1;
75         }
76
77         if(setjmp(png_jmpbuf(png))) {
78                 png_destroy_read_struct(&png, &info, 0);
79                 return -1;
80         }
81
82         png_set_read_fn(png, io, read_func);
83         png_set_sig_bytes(png, 0);
84         png_read_png(png, info, 0, 0);
85
86         png_get_IHDR(png, info, &xsz, &ysz, &channel_bits, &color_type, &ilace_type,
87                         &compression, &filtering);
88         if((fmt = png_type_to_fmt(color_type, channel_bits)) == -1) {
89                 png_destroy_read_struct(&png, &info, 0);
90                 return -1;
91         }
92
93         if(img_set_pixels(img, xsz, ysz, fmt, 0) == -1) {
94                 png_destroy_read_struct(&png, &info, 0);
95                 return -1;
96         }
97
98
99         if(channel_bits == 8) {
100                 unsigned int i;
101                 unsigned char **lineptr = (unsigned char**)png_get_rows(png, info);
102                 unsigned char *dest = img->pixels;
103
104                 for(i=0; i<ysz; i++) {
105                         memcpy(dest, lineptr[i], xsz * img->pixelsz);
106                         dest += xsz * img->pixelsz;
107                 }
108         } else {
109                 unsigned int i, j, num_elem;
110                 unsigned char **lineptr = (unsigned char**)png_get_rows(png, info);
111                 float *dest = img->pixels;
112
113                 num_elem = img->pixelsz / sizeof(float);
114                 for(i=0; i<ysz; i++) {
115                         for(j=0; j<xsz * num_elem; j++) {
116                                 unsigned short val = (lineptr[i][j * 2] << 8) | lineptr[i][j * 2 + 1];
117                                 *dest++ = (float)val / 65535.0;
118                         }
119                 }
120         }
121
122
123         png_destroy_read_struct(&png, &info, 0);
124         return 0;
125 }
126
127
128 static int write_file(struct img_pixmap *img, struct img_io *io)
129 {
130         png_struct *png;
131         png_info *info;
132         png_text txt;
133         struct img_pixmap tmpimg;
134         unsigned char **rows;
135         unsigned char *pixptr;
136         int i, coltype;
137
138         img_init(&tmpimg);
139
140         if(!(png = png_create_write_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0))) {
141                 return -1;
142         }
143         if(!(info = png_create_info_struct(png))) {
144                 png_destroy_write_struct(&png, 0);
145                 return -1;
146         }
147
148         /* if the input image is floating-point, we need to convert it to integer */
149         if(img_is_float(img)) {
150                 if(img_copy(&tmpimg, img) == -1) {
151                         return -1;
152                 }
153                 if(img_to_integer(&tmpimg) == -1) {
154                         img_destroy(&tmpimg);
155                         return -1;
156                 }
157                 img = &tmpimg;
158         }
159
160         txt.compression = PNG_TEXT_COMPRESSION_NONE;
161         txt.key = "Software";
162         txt.text = "libimago2";
163         txt.text_length = 0;
164
165         if(setjmp(png_jmpbuf(png))) {
166                 png_destroy_write_struct(&png, &info);
167                 img_destroy(&tmpimg);
168                 return -1;
169         }
170         png_set_write_fn(png, io, write_func, flush_func);
171
172         coltype = fmt_to_png_type(img->fmt);
173         png_set_IHDR(png, info, img->width, img->height, 8, coltype, PNG_INTERLACE_NONE,
174                         PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
175         png_set_text(png, info, &txt, 1);
176
177         if(!(rows = malloc(img->height * sizeof *rows))) {
178                 png_destroy_write_struct(&png, &info);
179                 img_destroy(&tmpimg);
180                 return -1;
181         }
182
183         pixptr = img->pixels;
184         for(i=0; i<img->height; i++) {
185                 rows[i] = pixptr;
186                 pixptr += img->width * img->pixelsz;
187         }
188         png_set_rows(png, info, rows);
189
190         png_write_png(png, info, 0, 0);
191         png_write_end(png, info);
192         png_destroy_write_struct(&png, &info);
193
194         free(rows);
195
196         img_destroy(&tmpimg);
197         return 0;
198 }
199
200 static void read_func(png_struct *png, unsigned char *data, size_t len)
201 {
202         struct img_io *io = (struct img_io*)png_get_io_ptr(png);
203
204         if(io->read(data, len, io->uptr) == -1) {
205                 longjmp(png_jmpbuf(png), 1);
206         }
207 }
208
209 static void write_func(png_struct *png, unsigned char *data, size_t len)
210 {
211         struct img_io *io = (struct img_io*)png_get_io_ptr(png);
212
213         if(io->write(data, len, io->uptr) == -1) {
214                 longjmp(png_jmpbuf(png), 1);
215         }
216 }
217
218 static void flush_func(png_struct *png)
219 {
220         /* XXX does it matter that we can't flush? */
221 }
222
223 static int png_type_to_fmt(int color_type, int channel_bits)
224 {
225         /* only 8 and 16 bits per channel ar supported at the moment */
226         if(channel_bits != 8 && channel_bits != 16) {
227                 return -1;
228         }
229
230         switch(color_type) {
231         case PNG_COLOR_TYPE_RGB:
232                 return channel_bits == 16 ? IMG_FMT_RGBF : IMG_FMT_RGB24;
233
234         case PNG_COLOR_TYPE_RGB_ALPHA:
235                 return channel_bits == 16 ? IMG_FMT_RGBAF : IMG_FMT_RGBA32;
236
237         case PNG_COLOR_TYPE_GRAY:
238                 return channel_bits == 16 ? IMG_FMT_GREYF : IMG_FMT_GREY8;
239
240         default:
241                 break;
242         }
243         return -1;
244 }
245
246 static int fmt_to_png_type(enum img_fmt fmt)
247 {
248         switch(fmt) {
249         case IMG_FMT_GREY8:
250                 return PNG_COLOR_TYPE_GRAY;
251
252         case IMG_FMT_RGB24:
253                 return PNG_COLOR_TYPE_RGB;
254
255         case IMG_FMT_RGBA32:
256                 return PNG_COLOR_TYPE_RGBA;
257
258         default:
259                 break;
260         }
261         return -1;
262 }