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