nuclear@2: /*
nuclear@2: libimago - a multi-format image file input/output library.
nuclear@2: Copyright (C) 2010 John Tsiombikas <nuclear@member.fsf.org>
nuclear@2: 
nuclear@2: This program is free software: you can redistribute it and/or modify
nuclear@2: it under the terms of the GNU Lesser General Public License as published
nuclear@2: by the Free Software Foundation, either version 3 of the License, or
nuclear@2: (at your option) any later version.
nuclear@2: 
nuclear@2: This program is distributed in the hope that it will be useful,
nuclear@2: but WITHOUT ANY WARRANTY; without even the implied warranty of
nuclear@2: MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
nuclear@2: GNU Lesser General Public License for more details.
nuclear@2: 
nuclear@2: You should have received a copy of the GNU Lesser General Public License
nuclear@2: along with this program.  If not, see <http://www.gnu.org/licenses/>.
nuclear@2: */
nuclear@2: 
nuclear@2: #include <stdio.h>
nuclear@2: #include <stdlib.h>
nuclear@2: #include <string.h>
nuclear@2: #include "imago2.h"
nuclear@2: #include "ftype_module.h"
nuclear@2: 
nuclear@2: static int pixel_size(enum img_fmt fmt);
nuclear@2: static size_t def_read(void *buf, size_t bytes, void *uptr);
nuclear@2: static size_t def_write(void *buf, size_t bytes, void *uptr);
nuclear@2: static long def_seek(long offset, int whence, void *uptr);
nuclear@2: 
nuclear@2: 
nuclear@2: void img_init(struct img_pixmap *img)
nuclear@2: {
nuclear@2: 	img->pixels = 0;
nuclear@2: 	img->width = img->height = 0;
nuclear@2: 	img->fmt = IMG_FMT_RGBA32;
nuclear@2: 	img->pixelsz = pixel_size(img->fmt);
nuclear@2: 	img->name = 0;
nuclear@2: }
nuclear@2: 
nuclear@2: 
nuclear@2: void img_destroy(struct img_pixmap *img)
nuclear@2: {
nuclear@2: 	free(img->pixels);
nuclear@2: 	img->pixels = 0;	/* just in case... */
nuclear@2: 	img->width = img->height = 0xbadbeef;
nuclear@2: 	free(img->name);
nuclear@2: }
nuclear@2: 
nuclear@2: struct img_pixmap *img_create(void)
nuclear@2: {
nuclear@2: 	struct img_pixmap *p;
nuclear@2: 
nuclear@2: 	if(!(p = malloc(sizeof *p))) {
nuclear@2: 		return 0;
nuclear@2: 	}
nuclear@2: 	img_init(p);
nuclear@2: 	return p;
nuclear@2: }
nuclear@2: 
nuclear@2: void img_free(struct img_pixmap *img)
nuclear@2: {
nuclear@2: 	img_destroy(img);
nuclear@2: 	free(img);
nuclear@2: }
nuclear@2: 
nuclear@2: int img_set_name(struct img_pixmap *img, const char *name)
nuclear@2: {
nuclear@2: 	char *tmp;
nuclear@2: 
nuclear@2: 	if(!(tmp = malloc(strlen(name) + 1))) {
nuclear@2: 		return -1;
nuclear@2: 	}
nuclear@2: 	strcpy(tmp, name);
nuclear@2: 	img->name = tmp;
nuclear@2: 	return 0;
nuclear@2: }
nuclear@2: 
nuclear@2: int img_set_format(struct img_pixmap *img, enum img_fmt fmt)
nuclear@2: {
nuclear@2: 	if(img->pixels) {
nuclear@2: 		return img_convert(img, fmt);
nuclear@2: 	}
nuclear@2: 	img->fmt = fmt;
nuclear@2: 	return 0;
nuclear@2: }
nuclear@2: 
nuclear@2: int img_copy(struct img_pixmap *dest, struct img_pixmap *src)
nuclear@2: {
nuclear@2: 	return img_set_pixels(dest, src->width, src->height, src->fmt, src->pixels);
nuclear@2: }
nuclear@2: 
nuclear@2: int img_set_pixels(struct img_pixmap *img, int w, int h, enum img_fmt fmt, void *pix)
nuclear@2: {
nuclear@2: 	void *newpix;
nuclear@2: 	int pixsz = pixel_size(fmt);
nuclear@2: 
nuclear@2: 	if(!(newpix = malloc(w * h * pixsz))) {
nuclear@2: 		return -1;
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	if(pix) {
nuclear@2: 		memcpy(newpix, pix, w * h * pixsz);
nuclear@2: 	} else {
nuclear@2: 		memset(newpix, 0, w * h * pixsz);
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	free(img->pixels);
nuclear@2: 	img->pixels = newpix;
nuclear@2: 	img->width = w;
nuclear@2: 	img->height = h;
nuclear@2: 	img->pixelsz = pixsz;
nuclear@2: 	img->fmt = fmt;
nuclear@2: 	return 0;
nuclear@2: }
nuclear@2: 
nuclear@2: void *img_load_pixels(const char *fname, int *xsz, int *ysz, enum img_fmt fmt)
nuclear@2: {
nuclear@2: 	struct img_pixmap img;
nuclear@2: 
nuclear@2: 	img_init(&img);
nuclear@2: 
nuclear@2: 	if(img_load(&img, fname) == -1) {
nuclear@2: 		return 0;
nuclear@2: 	}
nuclear@2: 	if(img.fmt != fmt) {
nuclear@2: 		if(img_convert(&img, fmt) == -1) {
nuclear@2: 			img_destroy(&img);
nuclear@2: 			return 0;
nuclear@2: 		}
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	*xsz = img.width;
nuclear@2: 	*ysz = img.height;
nuclear@2: 	return img.pixels;
nuclear@2: }
nuclear@2: 
nuclear@2: int img_save_pixels(const char *fname, void *pix, int xsz, int ysz, enum img_fmt fmt)
nuclear@2: {
nuclear@2: 	struct img_pixmap img;
nuclear@2: 
nuclear@2: 	img_init(&img);
nuclear@2: 	img.fmt = fmt;
nuclear@2: 	img.name = (char*)fname;
nuclear@2: 	img.width = xsz;
nuclear@2: 	img.height = ysz;
nuclear@2: 	img.pixels = pix;
nuclear@2: 
nuclear@2: 	return img_save(&img, fname);
nuclear@2: }
nuclear@2: 
nuclear@2: void img_free_pixels(void *pix)
nuclear@2: {
nuclear@2: 	free(pix);
nuclear@2: }
nuclear@2: 
nuclear@2: int img_load(struct img_pixmap *img, const char *fname)
nuclear@2: {
nuclear@2: 	int res;
nuclear@2: 	FILE *fp;
nuclear@2: 
nuclear@2: 	if(!(fp = fopen(fname, "rb"))) {
nuclear@2: 		return -1;
nuclear@2: 	}
nuclear@2: 	res = img_read_file(img, fp);
nuclear@2: 	fclose(fp);
nuclear@2: 	return res;
nuclear@2: }
nuclear@2: 
nuclear@2: /* TODO implement filetype selection */
nuclear@2: int img_save(struct img_pixmap *img, const char *fname)
nuclear@2: {
nuclear@2: 	int res;
nuclear@2: 	FILE *fp;
nuclear@2: 
nuclear@2: 	img_set_name(img, fname);
nuclear@2: 
nuclear@2: 	if(!(fp = fopen(fname, "wb"))) {
nuclear@2: 		return -1;
nuclear@2: 	}
nuclear@2: 	res = img_write_file(img, fp);
nuclear@2: 	fclose(fp);
nuclear@2: 	return res;
nuclear@2: }
nuclear@2: 
nuclear@2: int img_read_file(struct img_pixmap *img, FILE *fp)
nuclear@2: {
nuclear@2: 	struct img_io io = {0, def_read, def_write, def_seek};
nuclear@2: 
nuclear@2: 	io.uptr = fp;
nuclear@2: 	return img_read(img, &io);
nuclear@2: }
nuclear@2: 
nuclear@2: int img_write_file(struct img_pixmap *img, FILE *fp)
nuclear@2: {
nuclear@2: 	struct img_io io = {0, def_read, def_write, def_seek};
nuclear@2: 
nuclear@2: 	io.uptr = fp;
nuclear@2: 	return img_write(img, &io);
nuclear@2: }
nuclear@2: 
nuclear@2: int img_read(struct img_pixmap *img, struct img_io *io)
nuclear@2: {
nuclear@2: 	struct ftype_module *mod;
nuclear@2: 
nuclear@2: 	if((mod = img_find_format_module(io))) {
nuclear@2: 		return mod->read(img, io);
nuclear@2: 	}
nuclear@2: 	return -1;
nuclear@2: }
nuclear@2: 
nuclear@2: int img_write(struct img_pixmap *img, struct img_io *io)
nuclear@2: {
nuclear@2: 	struct ftype_module *mod;
nuclear@2: 
nuclear@2: 	if(!img->name || !(mod = img_guess_format(img->name))) {
nuclear@2: 		/* TODO throw some sort of warning? */
nuclear@2: 		/* TODO implement some sort of module priority or let the user specify? */
nuclear@2: 		if(!(mod = img_get_module(0))) {
nuclear@2: 			return -1;
nuclear@2: 		}
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	return mod->write(img, io);
nuclear@2: }
nuclear@2: 
nuclear@2: int img_to_float(struct img_pixmap *img)
nuclear@2: {
nuclear@2: 	enum img_fmt targ_fmt;
nuclear@2: 
nuclear@2: 	switch(img->fmt) {
nuclear@2: 	case IMG_FMT_GREY8:
nuclear@2: 		targ_fmt = IMG_FMT_GREYF;
nuclear@2: 		break;
nuclear@2: 
nuclear@2: 	case IMG_FMT_RGB24:
nuclear@2: 		targ_fmt = IMG_FMT_RGBF;
nuclear@2: 		break;
nuclear@2: 
nuclear@2: 	case IMG_FMT_RGBA32:
nuclear@2: 		targ_fmt = IMG_FMT_RGBAF;
nuclear@2: 		break;
nuclear@2: 
nuclear@2: 	default:
nuclear@2: 		return 0;	/* already float */
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	return img_convert(img, targ_fmt);
nuclear@2: }
nuclear@2: 
nuclear@2: int img_to_integer(struct img_pixmap *img)
nuclear@2: {
nuclear@2: 	enum img_fmt targ_fmt;
nuclear@2: 
nuclear@2: 	switch(img->fmt) {
nuclear@2: 	case IMG_FMT_GREYF:
nuclear@2: 		targ_fmt = IMG_FMT_GREY8;
nuclear@2: 		break;
nuclear@2: 
nuclear@2: 	case IMG_FMT_RGBF:
nuclear@2: 		targ_fmt = IMG_FMT_RGB24;
nuclear@2: 		break;
nuclear@2: 
nuclear@2: 	case IMG_FMT_RGBAF:
nuclear@2: 		targ_fmt = IMG_FMT_RGBA32;
nuclear@2: 		break;
nuclear@2: 
nuclear@2: 	default:
nuclear@2: 		return 0;	/* already integer */
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	return img_convert(img, targ_fmt);
nuclear@2: }
nuclear@2: 
nuclear@2: int img_is_float(struct img_pixmap *img)
nuclear@2: {
nuclear@2: 	return img->fmt >= IMG_FMT_GREYF && img->fmt <= IMG_FMT_RGBAF;
nuclear@2: }
nuclear@2: 
nuclear@2: int img_has_alpha(struct img_pixmap *img)
nuclear@2: {
nuclear@2: 	return img->fmt >= IMG_FMT_GREY8 && img->fmt <= IMG_FMT_RGBA32;
nuclear@2: }
nuclear@2: 
nuclear@2: void img_io_set_user_data(struct img_io *io, void *uptr)
nuclear@2: {
nuclear@2: 	io->uptr = uptr;
nuclear@2: }
nuclear@2: 
nuclear@2: void img_io_set_read_func(struct img_io *io, size_t (*read)(void*, size_t, void*))
nuclear@2: {
nuclear@2: 	io->read = read;
nuclear@2: }
nuclear@2: 
nuclear@2: void img_io_set_write_func(struct img_io *io, size_t (*write)(void*, size_t, void*))
nuclear@2: {
nuclear@2: 	io->write = write;
nuclear@2: }
nuclear@2: 
nuclear@2: void img_io_set_seek_func(struct img_io *io, long (*seek)(long, int, void*))
nuclear@2: {
nuclear@2: 	io->seek = seek;
nuclear@2: }
nuclear@2: 
nuclear@2: 
nuclear@2: static int pixel_size(enum img_fmt fmt)
nuclear@2: {
nuclear@2: 	switch(fmt) {
nuclear@2: 	case IMG_FMT_GREY8:
nuclear@2: 		return 1;
nuclear@2: 	case IMG_FMT_RGB24:
nuclear@2: 		return 3;
nuclear@2: 	case IMG_FMT_RGBA32:
nuclear@2: 		return 4;
nuclear@2: 	case IMG_FMT_GREYF:
nuclear@2: 		return sizeof(float);
nuclear@2: 	case IMG_FMT_RGBF:
nuclear@2: 		return 3 * sizeof(float);
nuclear@2: 	case IMG_FMT_RGBAF:
nuclear@2: 		return 4 * sizeof(float);
nuclear@2: 	default:
nuclear@2: 		break;
nuclear@2: 	}
nuclear@2: 	return 0;
nuclear@2: }
nuclear@2: 
nuclear@2: static size_t def_read(void *buf, size_t bytes, void *uptr)
nuclear@2: {
nuclear@2: 	return uptr ? fread(buf, 1, bytes, uptr) : 0;
nuclear@2: }
nuclear@2: 
nuclear@2: static size_t def_write(void *buf, size_t bytes, void *uptr)
nuclear@2: {
nuclear@2: 	return uptr ? fwrite(buf, 1, bytes, uptr) : 0;
nuclear@2: }
nuclear@2: 
nuclear@2: static long def_seek(long offset, int whence, void *uptr)
nuclear@2: {
nuclear@2: 	if(!uptr || fseek(uptr, offset, whence) == -1) {
nuclear@2: 		return -1;
nuclear@2: 	}
nuclear@2: 	return ftell(uptr);
nuclear@2: }
nuclear@2: