nuclear@26: /*
nuclear@26: libimago - a multi-format image file input/output library.
nuclear@26: Copyright (C) 2010 John Tsiombikas <nuclear@member.fsf.org>
nuclear@26: 
nuclear@26: This program is free software: you can redistribute it and/or modify
nuclear@26: it under the terms of the GNU Lesser General Public License as published
nuclear@26: by the Free Software Foundation, either version 3 of the License, or
nuclear@26: (at your option) any later version.
nuclear@26: 
nuclear@26: This program is distributed in the hope that it will be useful,
nuclear@26: but WITHOUT ANY WARRANTY; without even the implied warranty of
nuclear@26: MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
nuclear@26: GNU Lesser General Public License for more details.
nuclear@26: 
nuclear@26: You should have received a copy of the GNU Lesser General Public License
nuclear@26: along with this program.  If not, see <http://www.gnu.org/licenses/>.
nuclear@26: */
nuclear@26: 
nuclear@26: /* -- PNG module -- */
nuclear@26: 
nuclear@26: #include <stdlib.h>
nuclear@26: #include <png.h>
nuclear@26: #include "imago2.h"
nuclear@26: #include "ftype_module.h"
nuclear@26: 
nuclear@26: static int check_file(struct img_io *io);
nuclear@26: static int read_file(struct img_pixmap *img, struct img_io *io);
nuclear@26: static int write_file(struct img_pixmap *img, struct img_io *io);
nuclear@26: 
nuclear@26: static void read_func(png_struct *png, unsigned char *data, size_t len);
nuclear@26: static void write_func(png_struct *png, unsigned char *data, size_t len);
nuclear@26: static void flush_func(png_struct *png);
nuclear@26: 
nuclear@26: static int png_type_to_fmt(int color_type, int channel_bits);
nuclear@26: static int fmt_to_png_type(enum img_fmt fmt);
nuclear@26: 
nuclear@26: 
nuclear@26: int img_register_png(void)
nuclear@26: {
nuclear@26: 	static struct ftype_module mod = {".png", check_file, read_file, write_file};
nuclear@26: 	return img_register_module(&mod);
nuclear@26: }
nuclear@26: 
nuclear@26: static int check_file(struct img_io *io)
nuclear@26: {
nuclear@26: 	unsigned char sig[8];
nuclear@26: 	int res;
nuclear@26: 	long pos = io->seek(0, SEEK_CUR, io->uptr);
nuclear@26: 
nuclear@26: 	if(io->read(sig, 8, io->uptr) < 8) {
nuclear@26: 		io->seek(pos, SEEK_SET, io->uptr);
nuclear@26: 		return -1;
nuclear@26: 	}
nuclear@26: 
nuclear@26: 	res = png_sig_cmp(sig, 0, 8) == 0 ? 0 : -1;
nuclear@26: 	io->seek(pos, SEEK_SET, io->uptr);
nuclear@26: 	return res;
nuclear@26: }
nuclear@26: 
nuclear@26: static int read_file(struct img_pixmap *img, struct img_io *io)
nuclear@26: {
nuclear@26: 	png_struct *png;
nuclear@26: 	png_info *info;
nuclear@26: 	unsigned char **lineptr, *dest;
nuclear@26: 	int i, channel_bits, color_type, ilace_type, compression, filtering, fmt;
nuclear@26: 	unsigned long xsz, ysz;
nuclear@26: 
nuclear@26: 	if(!(png = png_create_read_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0))) {
nuclear@26: 		return -1;
nuclear@26: 	}
nuclear@26: 
nuclear@26: 	if(!(info = png_create_info_struct(png))) {
nuclear@26: 		png_destroy_read_struct(&png, 0, 0);
nuclear@26: 		return -1;
nuclear@26: 	}
nuclear@26: 
nuclear@26: 	if(setjmp(png_jmpbuf(png))) {
nuclear@26: 		png_destroy_read_struct(&png, &info, 0);
nuclear@26: 		return -1;
nuclear@26: 	}
nuclear@26: 
nuclear@26: 	png_set_read_fn(png, io, read_func);
nuclear@26: 	png_set_sig_bytes(png, 0);
nuclear@26: 	png_read_png(png, info, 0, 0);
nuclear@26: 
nuclear@26: 	png_get_IHDR(png, info, &xsz, &ysz, &channel_bits, &color_type, &ilace_type,
nuclear@26: 			&compression, &filtering);
nuclear@26: 	if((fmt = png_type_to_fmt(color_type, channel_bits)) == -1) {
nuclear@26: 		png_destroy_read_struct(&png, &info, 0);
nuclear@26: 		return -1;
nuclear@26: 	}
nuclear@26: 
nuclear@26: 	if(img_set_pixels(img, xsz, ysz, fmt, 0) == -1) {
nuclear@26: 		png_destroy_read_struct(&png, &info, 0);
nuclear@26: 		return -1;
nuclear@26: 	}
nuclear@26: 
nuclear@26: 	lineptr = (unsigned char**)png_get_rows(png, info);
nuclear@26: 
nuclear@26: 	dest = img->pixels;
nuclear@26: 	for(i=0; i<ysz; i++) {
nuclear@26: 		memcpy(dest, lineptr[i], xsz * img->pixelsz);
nuclear@26: 		dest += xsz * img->pixelsz;
nuclear@26: 	}
nuclear@26: 	png_destroy_read_struct(&png, &info, 0);
nuclear@26: 	return 0;
nuclear@26: }
nuclear@26: 
nuclear@26: 
nuclear@26: static int write_file(struct img_pixmap *img, struct img_io *io)
nuclear@26: {
nuclear@26: 	png_struct *png;
nuclear@26: 	png_info *info;
nuclear@26: 	png_text txt;
nuclear@26: 	struct img_pixmap tmpimg;
nuclear@26: 	unsigned char **rows;
nuclear@26: 	unsigned char *pixptr;
nuclear@26: 	int i, coltype;
nuclear@26: 
nuclear@26: 	img_init(&tmpimg);
nuclear@26: 
nuclear@26: 	if(!(png = png_create_write_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0))) {
nuclear@26: 		return -1;
nuclear@26: 	}
nuclear@26: 	if(!(info = png_create_info_struct(png))) {
nuclear@26: 		png_destroy_write_struct(&png, 0);
nuclear@26: 		return -1;
nuclear@26: 	}
nuclear@26: 
nuclear@26: 	/* if the input image is floating-point, we need to convert it to integer */
nuclear@26: 	if(img_is_float(img)) {
nuclear@26: 		if(img_copy(&tmpimg, img) == -1) {
nuclear@26: 			return -1;
nuclear@26: 		}
nuclear@26: 		if(img_to_integer(&tmpimg) == -1) {
nuclear@26: 			img_destroy(&tmpimg);
nuclear@26: 			return -1;
nuclear@26: 		}
nuclear@26: 		img = &tmpimg;
nuclear@26: 	}
nuclear@26: 
nuclear@26: 	txt.compression = PNG_TEXT_COMPRESSION_NONE;
nuclear@26: 	txt.key = "Software";
nuclear@26: 	txt.text = "libimago2";
nuclear@26: 	txt.text_length = 0;
nuclear@26: 
nuclear@26: 	if(setjmp(png_jmpbuf(png))) {
nuclear@26: 		png_destroy_write_struct(&png, &info);
nuclear@26: 		img_destroy(&tmpimg);
nuclear@26: 		return -1;
nuclear@26: 	}
nuclear@26: 	png_set_write_fn(png, io, write_func, flush_func);
nuclear@26: 
nuclear@26: 	coltype = fmt_to_png_type(img->fmt);
nuclear@26: 	png_set_IHDR(png, info, img->width, img->height, 8, coltype, PNG_INTERLACE_NONE,
nuclear@26: 			PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
nuclear@26: 	png_set_text(png, info, &txt, 1);
nuclear@26: 
nuclear@26: 	if(!(rows = malloc(img->height * sizeof *rows))) {
nuclear@26: 		png_destroy_write_struct(&png, &info);
nuclear@26: 		img_destroy(&tmpimg);
nuclear@26: 		return -1;
nuclear@26: 	}
nuclear@26: 
nuclear@26: 	pixptr = img->pixels;
nuclear@26: 	for(i=0; i<img->height; i++) {
nuclear@26: 		rows[i] = pixptr;
nuclear@26: 		pixptr += img->width * img->pixelsz;
nuclear@26: 	}
nuclear@26: 	png_set_rows(png, info, rows);
nuclear@26: 
nuclear@26: 	png_write_png(png, info, 0, 0);
nuclear@26: 	png_write_end(png, info);
nuclear@26: 	png_destroy_write_struct(&png, &info);
nuclear@26: 
nuclear@26: 	free(rows);
nuclear@26: 
nuclear@26: 	img_destroy(&tmpimg);
nuclear@26: 	return 0;
nuclear@26: }
nuclear@26: 
nuclear@26: static void read_func(png_struct *png, unsigned char *data, size_t len)
nuclear@26: {
nuclear@26: 	struct img_io *io = (struct img_io*)png_get_io_ptr(png);
nuclear@26: 
nuclear@26: 	if(io->read(data, len, io->uptr) == -1) {
nuclear@26: 		longjmp(png_jmpbuf(png), 1);
nuclear@26: 	}
nuclear@26: }
nuclear@26: 
nuclear@26: static void write_func(png_struct *png, unsigned char *data, size_t len)
nuclear@26: {
nuclear@26: 	struct img_io *io = (struct img_io*)png_get_io_ptr(png);
nuclear@26: 
nuclear@26: 	if(io->write(data, len, io->uptr) == -1) {
nuclear@26: 		longjmp(png_jmpbuf(png), 1);
nuclear@26: 	}
nuclear@26: }
nuclear@26: 
nuclear@26: static void flush_func(png_struct *png)
nuclear@26: {
nuclear@26: 	/* XXX does it matter that we can't flush? */
nuclear@26: }
nuclear@26: 
nuclear@26: static int png_type_to_fmt(int color_type, int channel_bits)
nuclear@26: {
nuclear@26: 	/* we don't support non-8bit-per-channel images yet */
nuclear@26: 	if(channel_bits > 8) {
nuclear@26: 		return -1;
nuclear@26: 	}
nuclear@26: 
nuclear@26: 	switch(color_type) {
nuclear@26: 	case PNG_COLOR_TYPE_RGB:
nuclear@26: 		return IMG_FMT_RGB24;
nuclear@26: 
nuclear@26: 	case PNG_COLOR_TYPE_RGB_ALPHA:
nuclear@26: 		return IMG_FMT_RGBA32;
nuclear@26: 
nuclear@26: 	case PNG_COLOR_TYPE_GRAY:
nuclear@26: 		return IMG_FMT_GREY8;
nuclear@26: 
nuclear@26: 	default:
nuclear@26: 		break;
nuclear@26: 	}
nuclear@26: 	return -1;
nuclear@26: }
nuclear@26: 
nuclear@26: static int fmt_to_png_type(enum img_fmt fmt)
nuclear@26: {
nuclear@26: 	switch(fmt) {
nuclear@26: 	case IMG_FMT_GREY8:
nuclear@26: 		return PNG_COLOR_TYPE_GRAY;
nuclear@26: 
nuclear@26: 	case IMG_FMT_RGB24:
nuclear@26: 		return PNG_COLOR_TYPE_RGB;
nuclear@26: 
nuclear@26: 	case IMG_FMT_RGBA32:
nuclear@26: 		return PNG_COLOR_TYPE_RGBA;
nuclear@26: 
nuclear@26: 	default:
nuclear@26: 		break;
nuclear@26: 	}
nuclear@26: 	return -1;
nuclear@26: }