698517be |
1 | /* |
29a8c4f3 |
2 | * (C) GraÅžvydas "notaz" Ignotas, 2008-2011 |
698517be |
3 | * |
4 | * This work is licensed under the terms of any of these licenses |
5 | * (at your option): |
6 | * - GNU GPL, version 2 or later. |
7 | * - GNU LGPL, version 2.1 or later. |
8 | * See the COPYING file in the top-level directory. |
9 | */ |
10 | |
11 | #include <stdio.h> |
29a8c4f3 |
12 | #include <stdlib.h> |
698517be |
13 | #include <string.h> |
14 | #include <png.h> |
15 | #include "readpng.h" |
16 | #include "lprintf.h" |
17 | |
18 | int readpng(void *dest, const char *fname, readpng_what what, int req_w, int req_h) |
19 | { |
20 | FILE *fp; |
21 | png_structp png_ptr = NULL; |
22 | png_infop info_ptr = NULL; |
23 | png_bytepp row_ptr = NULL; |
24 | int ret = -1; |
25 | |
26 | if (dest == NULL || fname == NULL) |
27 | { |
28 | return -1; |
29 | } |
30 | |
31 | fp = fopen(fname, "rb"); |
32 | if (fp == NULL) |
33 | { |
34 | lprintf(__FILE__ ": failed to open: %s\n", fname); |
35 | return -1; |
36 | } |
37 | |
38 | png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL); |
39 | if (!png_ptr) |
40 | { |
41 | lprintf(__FILE__ ": png_create_read_struct() failed\n"); |
42 | fclose(fp); |
43 | return -1; |
44 | } |
45 | |
46 | info_ptr = png_create_info_struct(png_ptr); |
47 | if (!info_ptr) |
48 | { |
49 | lprintf(__FILE__ ": png_create_info_struct() failed\n"); |
50 | goto done; |
51 | } |
52 | |
53 | // Start reading |
54 | png_init_io(png_ptr, fp); |
55 | png_read_png(png_ptr, info_ptr, PNG_TRANSFORM_STRIP_16 | PNG_TRANSFORM_STRIP_ALPHA | PNG_TRANSFORM_PACKING, NULL); |
56 | row_ptr = png_get_rows(png_ptr, info_ptr); |
57 | if (row_ptr == NULL) |
58 | { |
59 | lprintf(__FILE__ ": png_get_rows() failed\n"); |
60 | goto done; |
61 | } |
62 | |
63 | // lprintf("%s: %ix%i @ %ibpp\n", fname, (int)info_ptr->width, (int)info_ptr->height, info_ptr->pixel_depth); |
64 | |
65 | switch (what) |
66 | { |
67 | case READPNG_BG: |
68 | { |
69 | int height, width, h; |
70 | unsigned short *dst = dest; |
71 | if (info_ptr->pixel_depth != 24) |
72 | { |
73 | lprintf(__FILE__ ": bg image uses %ibpp, needed 24bpp\n", info_ptr->pixel_depth); |
74 | break; |
75 | } |
76 | height = info_ptr->height; |
77 | if (height > req_h) |
78 | height = req_h; |
79 | width = info_ptr->width; |
80 | if (width > req_w) |
81 | width = req_w; |
82 | |
83 | for (h = 0; h < height; h++) |
84 | { |
85 | unsigned char *src = row_ptr[h]; |
86 | int len = width; |
87 | while (len--) |
88 | { |
89 | #ifdef PSP |
90 | *dst++ = ((src[2]&0xf8)<<8) | ((src[1]&0xf8)<<3) | (src[0] >> 3); // BGR |
91 | #else |
92 | *dst++ = ((src[0]&0xf8)<<8) | ((src[1]&0xf8)<<3) | (src[2] >> 3); // RGB |
93 | #endif |
94 | src += 3; |
95 | } |
96 | dst += req_w - width; |
97 | } |
98 | break; |
99 | } |
100 | |
101 | case READPNG_FONT: |
102 | { |
103 | int x, y, x1, y1; |
104 | unsigned char *dst = dest; |
105 | if (info_ptr->width != req_w || info_ptr->height != req_h) |
106 | { |
107 | lprintf(__FILE__ ": unexpected font image size %dx%d, needed %dx%d\n", |
108 | (int)info_ptr->width, (int)info_ptr->height, req_w, req_h); |
109 | break; |
110 | } |
111 | if (info_ptr->pixel_depth != 8) |
112 | { |
113 | lprintf(__FILE__ ": font image uses %ibpp, needed 8bpp\n", info_ptr->pixel_depth); |
114 | break; |
115 | } |
116 | for (y = 0; y < 16; y++) |
117 | { |
118 | for (x = 0; x < 16; x++) |
119 | { |
120 | /* 16x16 grid of syms */ |
121 | int sym_w = req_w / 16; |
122 | int sym_h = req_h / 16; |
123 | for (y1 = 0; y1 < sym_h; y1++) |
124 | { |
125 | unsigned char *src = row_ptr[y*sym_h + y1] + x*sym_w; |
126 | for (x1 = sym_w/2; x1 > 0; x1--, src+=2) |
127 | *dst++ = ((src[0]^0xff) & 0xf0) | ((src[1]^0xff) >> 4); |
128 | } |
129 | } |
130 | } |
131 | break; |
132 | } |
133 | |
134 | case READPNG_SELECTOR: |
135 | { |
136 | int x1, y1; |
137 | unsigned char *dst = dest; |
138 | if (info_ptr->width != req_w || info_ptr->height != req_h) |
139 | { |
140 | lprintf(__FILE__ ": unexpected selector image size %ix%i, needed %dx%d\n", |
141 | (int)info_ptr->width, (int)info_ptr->height, req_w, req_h); |
142 | break; |
143 | } |
144 | if (info_ptr->pixel_depth != 8) |
145 | { |
146 | lprintf(__FILE__ ": selector image uses %ibpp, needed 8bpp\n", info_ptr->pixel_depth); |
147 | break; |
148 | } |
149 | for (y1 = 0; y1 < req_h; y1++) |
150 | { |
151 | unsigned char *src = row_ptr[y1]; |
152 | for (x1 = req_w/2; x1 > 0; x1--, src+=2) |
153 | *dst++ = ((src[0]^0xff) & 0xf0) | ((src[1]^0xff) >> 4); |
154 | } |
155 | break; |
156 | } |
157 | |
158 | case READPNG_24: |
159 | { |
160 | int height, width, h; |
161 | unsigned char *dst = dest; |
162 | if (info_ptr->pixel_depth != 24) |
163 | { |
164 | lprintf(__FILE__ ": image uses %ibpp, needed 24bpp\n", info_ptr->pixel_depth); |
165 | break; |
166 | } |
167 | height = info_ptr->height; |
168 | if (height > req_h) |
169 | height = req_h; |
170 | width = info_ptr->width; |
171 | if (width > req_w) |
172 | width = req_w; |
173 | |
174 | for (h = 0; h < height; h++) |
175 | { |
176 | int len = width; |
177 | unsigned char *src = row_ptr[h]; |
178 | dst += (req_w - width) * 3; |
179 | for (len = width; len > 0; len--, dst+=3, src+=3) |
180 | dst[0] = src[2], dst[1] = src[1], dst[2] = src[0]; |
181 | } |
182 | break; |
183 | } |
184 | } |
185 | |
186 | |
187 | ret = 0; |
188 | done: |
189 | png_destroy_read_struct(&png_ptr, info_ptr ? &info_ptr : NULL, (png_infopp)NULL); |
190 | fclose(fp); |
191 | return ret; |
192 | } |
193 | |
29a8c4f3 |
194 | int writepng(const char *fname, unsigned short *src, int w, int h) |
195 | { |
196 | png_structp png_ptr = NULL; |
197 | png_infop info_ptr = NULL; |
198 | png_bytepp row_pointers; |
199 | int i, j, ret = -1; |
200 | FILE *f; |
201 | |
202 | f = fopen(fname, "wb"); |
203 | if (f == NULL) { |
204 | lprintf(__FILE__ ": failed to open \"%s\"\n", fname); |
205 | return -1; |
206 | } |
207 | |
208 | row_pointers = calloc(h, sizeof(row_pointers[0])); |
209 | if (row_pointers == NULL) |
210 | goto end1; |
211 | |
212 | for (i = 0; i < h; i++) { |
213 | unsigned char *dst = malloc(w * 3); |
214 | if (dst == NULL) |
215 | goto end2; |
216 | row_pointers[i] = dst; |
217 | for (j = 0; j < w; j++, src++, dst += 3) { |
218 | dst[0] = (*src & 0xf800) >> 8; |
219 | dst[1] = (*src & 0x07e0) >> 3; |
220 | dst[2] = (*src & 0x001f) << 3; |
221 | } |
222 | } |
223 | |
224 | /* initialize stuff */ |
225 | png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL); |
226 | if (png_ptr == NULL) { |
227 | fprintf(stderr, "png_create_write_struct() failed"); |
228 | goto end2; |
229 | } |
230 | |
231 | info_ptr = png_create_info_struct(png_ptr); |
232 | if (info_ptr == NULL) { |
233 | fprintf(stderr, "png_create_info_struct() failed"); |
234 | goto end3; |
235 | } |
236 | |
237 | if (setjmp(png_jmpbuf(png_ptr)) != 0) { |
238 | fprintf(stderr, "error in png code\n"); |
239 | goto end4; |
240 | } |
241 | |
242 | png_init_io(png_ptr, f); |
243 | |
244 | png_set_IHDR(png_ptr, info_ptr, w, h, |
245 | 8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE, |
246 | PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT); |
247 | |
248 | png_write_info(png_ptr, info_ptr); |
249 | png_write_image(png_ptr, row_pointers); |
250 | png_write_end(png_ptr, NULL); |
251 | |
252 | ret = 0; |
253 | |
254 | end4: |
255 | // png_destroy_info_struct(png_ptr, &info_ptr); // freed below |
256 | end3: |
257 | png_destroy_write_struct(&png_ptr, &info_ptr); |
258 | end2: |
259 | for (i = 0; i < h; i++) |
260 | free(row_pointers[i]); |
261 | free(row_pointers); |
262 | end1: |
263 | fclose(f); |
264 | return ret; |
265 | } |
698517be |
266 | |