[libpicofe.git] / readpng.c

/*
 * (C) Gražvydas "notaz" Ignotas, 2008-2011
 *
 * This work is licensed under the terms of any of these licenses
 * (at your option):
 *  - GNU GPL, version 2 or later.
 *  - GNU LGPL, version 2.1 or later.
 *  - MAME license.
 * See the COPYING file in the top-level directory.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <png.h>
#include "readpng.h"
#include "lprintf.h"

int readpng(void *dest, const char *fname, readpng_what what, int req_w, int req_h)
{
	FILE *fp;
	png_structp png_ptr = NULL;
	png_infop info_ptr = NULL;
	png_bytepp row_ptr = NULL;
	int ret = -1;

	if (dest == NULL || fname == NULL)
	{
		return -1;
	}

	fp = fopen(fname, "rb");
	if (fp == NULL)
	{
		lprintf(__FILE__ ": failed to open: %s\n", fname);
		return -1;
	}

	png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
	if (!png_ptr)
	{
		lprintf(__FILE__ ": png_create_read_struct() failed\n");
		fclose(fp);
		return -1;
	}

	info_ptr = png_create_info_struct(png_ptr);
	if (!info_ptr)
	{
		lprintf(__FILE__ ": png_create_info_struct() failed\n");
		goto done;
	}

	// Start reading
	png_init_io(png_ptr, fp);
	png_read_png(png_ptr, info_ptr, PNG_TRANSFORM_STRIP_16 | PNG_TRANSFORM_STRIP_ALPHA | PNG_TRANSFORM_PACKING, NULL);
	row_ptr = png_get_rows(png_ptr, info_ptr);
	if (row_ptr == NULL)
	{
		lprintf(__FILE__ ": png_get_rows() failed\n");
		goto done;
	}

	// lprintf("%s: %ix%i @ %ibpp\n", fname, (int)png_get_image_width(png_ptr, info_ptr),
	// 	(int)png_get_image_height(png_ptr, info_ptr), png_get_bit_depth(png_ptr, info_ptr));

	switch (what)
	{
		case READPNG_BG:
		{
			int height, width, h, x_ofs = 0, y_ofs = 0;
			unsigned short *dst = dest;

			if (png_get_bit_depth(png_ptr, info_ptr) != 8)
			{
				lprintf(__FILE__ ": bg image uses %ibpc, needed 8bpc\n", png_get_bit_depth(png_ptr, info_ptr));
				break;
			}
			width = png_get_image_width(png_ptr, info_ptr);
			if (width > req_w) {
				x_ofs = (width - req_w) / 2;
				width = req_w;
			} else
				dst += (req_w - width) / 2;
			height = png_get_image_height(png_ptr, info_ptr);
			if (height > req_h) {
				y_ofs = (height - req_h) / 2;
				height = req_h;
			} else
				dst += (req_h - height) / 2 * req_w;

			for (h = 0; h < height; h++)
			{
				unsigned char *src = row_ptr[h + y_ofs] + x_ofs * 3;
				int len = width;
				while (len--)
				{
#ifdef PSP
					*dst++ = ((src[2]&0xf8)<<8) | ((src[1]&0xf8)<<3) | (src[0] >> 3); // BGR
#else
					*dst++ = ((src[0]&0xf8)<<8) | ((src[1]&0xf8)<<3) | (src[2] >> 3); // RGB
#endif
					src += 3;
				}
				dst += req_w - width;
			}
			break;
		}

		case READPNG_FONT:
		{
			int x, y, x1, y1;
			unsigned char *dst = dest;
			if (png_get_image_width(png_ptr, info_ptr) != req_w || png_get_image_height(png_ptr, info_ptr) != req_h)
			{
				lprintf(__FILE__ ": unexpected font image size %dx%d, needed %dx%d\n",
					(int)png_get_image_width(png_ptr, info_ptr), (int)png_get_image_height(png_ptr, info_ptr), req_w, req_h);
				break;
			}
			if (png_get_bit_depth(png_ptr, info_ptr) != 8)
			{
				lprintf(__FILE__ ": font image uses %ibpp, needed 8bpp\n", png_get_bit_depth(png_ptr, info_ptr));
				break;
			}
			for (y = 0; y < 16; y++)
			{
				for (x = 0; x < 16; x++)
				{
					/* 16x16 grid of syms */
					int sym_w = req_w / 16;
					int sym_h = req_h / 16;
					for (y1 = 0; y1 < sym_h; y1++)
					{
						unsigned char *src = row_ptr[y*sym_h + y1] + x*sym_w;
						for (x1 = sym_w/2; x1 > 0; x1--, src+=2)
							*dst++ = ((src[0]^0xff) & 0xf0) | ((src[1]^0xff) >> 4);
					}
				}
			}
			break;
		}

		case READPNG_SELECTOR:
		{
			int x1, y1;
			unsigned char *dst = dest;
			if (png_get_image_width(png_ptr, info_ptr) != req_w || png_get_image_height(png_ptr, info_ptr) != req_h)
			{
				lprintf(__FILE__ ": unexpected selector image size %ix%i, needed %dx%d\n",
					(int)png_get_image_width(png_ptr, info_ptr), (int)png_get_image_height(png_ptr, info_ptr), req_w, req_h);
				break;
			}
			if (png_get_bit_depth(png_ptr, info_ptr) != 8)
			{
				lprintf(__FILE__ ": selector image uses %ibpp, needed 8bpp\n", png_get_bit_depth(png_ptr, info_ptr));
				break;
			}
			for (y1 = 0; y1 < req_h; y1++)
			{
				unsigned char *src = row_ptr[y1];
				for (x1 = req_w/2; x1 > 0; x1--, src+=2)
					*dst++ = ((src[0]^0xff) & 0xf0) | ((src[1]^0xff) >> 4);
			}
			break;
		}

		case READPNG_24:
		{
			int height, width, h;
			unsigned char *dst = dest;
			if (png_get_bit_depth(png_ptr, info_ptr) != 8)
			{
				lprintf(__FILE__ ": image uses %ibpc, needed 8bpc\n", png_get_bit_depth(png_ptr, info_ptr));
				break;
			}
			width = png_get_image_width(png_ptr, info_ptr);
			if (width > req_w)
				width = req_w;
			height = png_get_image_height(png_ptr, info_ptr);
			if (height > req_h)
				height = req_h;

			for (h = 0; h < height; h++)
			{
				int len = width;
				unsigned char *src = row_ptr[h];
				dst += (req_w - width) * 3;
				for (len = width; len > 0; len--, dst+=3, src+=3)
					dst[0] = src[2], dst[1] = src[1], dst[2] = src[0];
			}
			break;
		}
	}


	ret = 0;
done:
	png_destroy_read_struct(&png_ptr, info_ptr ? &info_ptr : NULL, (png_infopp)NULL);
	fclose(fp);
	return ret;
}

int writepng(const char *fname, unsigned short *src, int w, int h)
{
	png_structp png_ptr = NULL;
	png_infop info_ptr = NULL;
	png_bytepp row_pointers;
	int i, j, ret = -1;
	FILE *f;

	f = fopen(fname, "wb");
	if (f == NULL) {
		lprintf(__FILE__ ": failed to open \"%s\"\n", fname);
		return -1;
	}

	row_pointers = calloc(h, sizeof(row_pointers[0]));
	if (row_pointers == NULL)
		goto end1;

	for (i = 0; i < h; i++) {
		unsigned char *dst = malloc(w * 3);
		if (dst == NULL)
			goto end2;
		row_pointers[i] = dst;
		for (j = 0; j < w; j++, src++, dst += 3) {
			dst[0] = (*src & 0xf800) >> 8;
			dst[1] = (*src & 0x07e0) >> 3;
			dst[2] = (*src & 0x001f) << 3;
		}
	}

	/* initialize stuff */
	png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
	if (png_ptr == NULL) {
		fprintf(stderr, "png_create_write_struct() failed");
		goto end2;
	}

	info_ptr = png_create_info_struct(png_ptr);
	if (info_ptr == NULL) {
		fprintf(stderr, "png_create_info_struct() failed");
		goto end3;
	}

	if (setjmp(png_jmpbuf(png_ptr)) != 0) {
		fprintf(stderr, "error in png code\n");
		goto end4;
	}

	png_init_io(png_ptr, f);

	png_set_IHDR(png_ptr, info_ptr, w, h,
		8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,
		PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

	png_write_info(png_ptr, info_ptr);
	png_write_image(png_ptr, row_pointers);
	png_write_end(png_ptr, NULL);

	ret = 0;

end4:
//	png_destroy_info_struct(png_ptr, &info_ptr); // freed below
end3:
	png_destroy_write_struct(&png_ptr, &info_ptr);
end2:
	for (i = 0; i < h; i++)
		free(row_pointers[i]);
	free(row_pointers);
end1:
	fclose(f);
	return ret;
}
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),
a085ae5e	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;
13059a60	78	}
a085ae5e	79	width = png_get_image_width(png_ptr, info_ptr);
8ef1ad32 GI	80	if (width > req_w) {
8ef1ad32 GI	81	x_ofs = (width - req_w) / 2;
f6eaae4f	82	width = req_w;
3ccc039b	83	} else
3ccc039b	84	dst += (req_w - width) / 2;
a085ae5e	85	height = png_get_image_height(png_ptr, info_ptr);
8ef1ad32 GI	86	if (height > req_h) {
8ef1ad32 GI	87	y_ofs = (height - req_h) / 2;
a085ae5e	88	height = req_h;
3ccc039b	89	} else
3ccc039b	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;
13059a60	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;
13059a60	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[ysym_h + y1] + x*sym_w;
f6eaae4f	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;
13059a60	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;
13059a60	157	}
f6eaae4f	158	for (y1 = 0; y1 < req_h; y1++)
13059a60	159	{
13059a60	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	{
ff63afa1	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;
ff63afa1	175	}
a085ae5e	176	width = png_get_image_width(png_ptr, info_ptr);
f6eaae4f	177	if (width > req_w)
f6eaae4f	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	}
13059a60	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