Blob Blame Raw


#if _MSC_VER >= 1400
#define _CRT_SECURE_NO_DEPRECATE 1
#pragma warning(disable : 4996)
#endif

#include "texception.h"
#include "tpixel.h"
#include "tiio_sgi.h"
#include "tsystem.h"
#include "tpixelgr.h"
#include "../compatibility/tfile_io.h"
#ifdef LINUX
//#define _XOPEN_SOURCE_EXTENDED
#include <string.h>
#endif

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#if defined(MACOSX)
#include <sys/malloc.h>
#else
#include <malloc.h>
#endif

#include <assert.h>

#ifdef _WIN32
#include <io.h>
#include <windows.h>
#else
#ifndef LINUX
#include <strings.h>

#endif
#include <unistd.h>
#endif
#include <stdarg.h>

using namespace std;
//
// IMAGERGB:
// header file per le immagini sgi
//
struct IMAGERGB {
	unsigned short imagic; /* stuff saved on disk . . */
	unsigned short type;
	unsigned short dim;
	unsigned short xsize;
	unsigned short ysize;
	unsigned short zsize;
	TUINT32 min;
	TUINT32 max;
	TUINT32 wastebytes;
	char name[80];
	TUINT32 colormap;

	TINT32 file; /* stuff used in core only */
	unsigned short flags;
	short dorev;
	short x;
	short y;
	short z;
	short cnt;
	unsigned short *ptr;
	unsigned short *base;
	unsigned short *tmpbuf;
	TUINT32 offset;
	TUINT32 rleend;	/* for rle images */
	TUINT32 *rowstart; /* for rle images */
	TINT32 *rowsize;   /* for rle images */
};

const int IMAGERGB_HEADER_SIZE = sizeof(IMAGERGB);

//
// macro definite nel vecchio ImageP/filergb.h
//
#define IMAGIC 0732
#define TYPEMASK 0xff00
#define ITYPE_RLE 0x0100
#define ISRLE(type) (((type)&0xff00) == ITYPE_RLE)
#define ITYPE_VERBATIM 0x0000
#define ISVERBATIM(type) (((type)&0xff00) == ITYPE_VERBATIM)
#define BPPMASK 0x00ff
#define BPP(type) ((type)&BPPMASK)
#define RLE(bpp) (ITYPE_RLE | (bpp))
#define VERBATIM(bpp) (ITYPE_VERBATIM | (bpp))
#define IBUFSIZE(pixels) ((pixels + (pixels >> 6)) << 2)
#define RLE_NOP 0x00

#ifndef _IORW
#define _IOREAD 0x1
#define _IOWRT 0x2
#define _IORW 0x80
#define _IOERR 0x20
#endif

static USHORT *ibufalloc(IMAGERGB *image, int bpp);
static void cvtshorts(USHORT buffer[], TINT32 n);
static void cvtTINT32s(TUINT32 *buffer, TINT32 n);
static void cvtimage(IMAGERGB *image);
static void img_rle_expand(USHORT *rlebuf, int ibpp, USHORT *expbuf, int obpp);
static int img_getrowsize(IMAGERGB *image);
static TUINT32 img_optseek(IMAGERGB *image, TUINT32 offset);
static TINT32 rgb_img_read(IMAGERGB *image, char *buffer, TINT32 count);
static int img_badrow(IMAGERGB *image, int y, int z);
static TUINT32 img_seek(IMAGERGB *image, UINT y, UINT z, UINT offs);
static TINT32 RGB_img_write(IMAGERGB *image, char *buffer, TINT32 count);
static void img_setrowsize(IMAGERGB *image, UINT cnt, UINT y, UINT z);
static TINT32 img_rle_compact(USHORT *expbuf, int ibpp, USHORT *rlebuf,
							  int obpp, int cnt);
static int iflush(IMAGERGB *image);

/*-------------------------------------------------------------------------*/

static int do_rgb_write_header(IMAGERGB *img, int fd)
{
	// DOBBIAMO SWAPPPPARE ?????????????????
	int count = 0;
	count += write(fd, &img->imagic, (int)sizeof(unsigned short));
	count += write(fd, &img->type, (int)sizeof(unsigned short));
	count += write(fd, &img->dim, (int)sizeof(unsigned short));
	count += write(fd, &img->xsize, (int)sizeof(unsigned short));
	count += write(fd, &img->ysize, (int)sizeof(unsigned short));
	count += write(fd, &img->zsize, (int)sizeof(unsigned short));
	count += write(fd, &img->min, (int)sizeof(TUINT32));
	count += write(fd, &img->max, (int)sizeof(TUINT32));
	count += write(fd, &img->wastebytes, (int)sizeof(TUINT32));
	count += write(fd, img->name, (int)sizeof(img->name));
	count += write(fd, &img->colormap, (int)sizeof(TUINT32));

	count += write(fd, &img->file, (int)sizeof(TINT32));
	count += write(fd, &img->flags, (int)sizeof(unsigned short));
	count += write(fd, &img->dorev, (int)sizeof(short));
	count += write(fd, &img->x, (int)sizeof(short));
	count += write(fd, &img->y, (int)sizeof(short));
	count += write(fd, &img->z, (int)sizeof(short));
	count += write(fd, &img->cnt, (int)sizeof(short));
	count += write(fd, &img->ptr, (int)sizeof(unsigned short *));
	count += write(fd, &img->base, (int)sizeof(unsigned short *));
	count += write(fd, &img->tmpbuf, (int)sizeof(unsigned short *));
	count += write(fd, &img->offset, (int)sizeof(TUINT32));
	count += write(fd, &img->rleend, (int)sizeof(TUINT32));
	count += write(fd, &img->rowstart, (int)sizeof(TUINT32 *));
	count += write(fd, &img->rowsize, (int)sizeof(TINT32 *));
	if (sizeof(void *) == 8) //siamo a 64bit: l'header side ha dei padding bytes.
		count = (count + 0x7) & (~0x7);
	return count;
}

/*-------------------------------------------------------------------------*/

static int do_rgb_read_header(IMAGERGB *img, int fd)
{
	// DOBBIAMO SWAPPPPARE ?????????????????
	int count = 0;
	count += read(fd, &img->imagic, sizeof(unsigned short));
	count += read(fd, &img->type, sizeof(unsigned short));
	count += read(fd, &img->dim, sizeof(unsigned short));
	count += read(fd, &img->xsize, sizeof(unsigned short));
	count += read(fd, &img->ysize, sizeof(unsigned short));
	count += read(fd, &img->zsize, sizeof(unsigned short));
	count += read(fd, &img->min, sizeof(TUINT32));
	count += read(fd, &img->max, sizeof(TUINT32));
	count += read(fd, &img->wastebytes, sizeof(TUINT32));
	count += read(fd, img->name, sizeof(img->name));
	count += read(fd, &img->colormap, sizeof(TUINT32));

	count += read(fd, &img->file, sizeof(TINT32));
	count += read(fd, &img->flags, sizeof(unsigned short));
	count += read(fd, &img->dorev, sizeof(short));
	count += read(fd, &img->x, sizeof(short));
	count += read(fd, &img->y, sizeof(short));
	count += read(fd, &img->z, sizeof(short));
	count += read(fd, &img->cnt, sizeof(short));
	count += read(fd, &img->ptr, sizeof(unsigned short *));
	count += read(fd, &img->base, sizeof(unsigned short *));
	count += read(fd, &img->tmpbuf, sizeof(unsigned short *));
	count += read(fd, &img->offset, sizeof(TUINT32));
	count += read(fd, &img->rleend, sizeof(TUINT32));
	count += read(fd, &img->rowstart, sizeof(TUINT32 *));
	count += read(fd, &img->rowsize, sizeof(TINT32 *));
	if (sizeof(void *) == 8) //siamo a 64bit: l'header side ha dei padding bytes.
		count = (count + 0x7) & (~0x7);
	return count;
}

/*-------------------------------------------------------------------------*/
enum OpenMode {
	OpenRead,
	OpenWrite
};

/*-------------------------------------------------------------------------*/

static IMAGERGB *iopen(int fd, OpenMode openMode,
					   unsigned int type, unsigned int dim, unsigned int xsize,
					   unsigned int ysize, unsigned int zsize, short dorev)
{
	IMAGERGB *image;
	extern int errno;
	int tablesize, f = fd;

	image = (IMAGERGB *)malloc((int)sizeof(IMAGERGB));

	memset(image, 0, sizeof(IMAGERGB));

	if (openMode == OpenWrite) {
		//WRITE

		image->imagic = IMAGIC;
		image->type = type;
		image->xsize = xsize;
		image->ysize = 1;
		image->zsize = 1;

		if (dim > 1)
			image->ysize = ysize;

		if (dim > 2)
			image->zsize = zsize;

		if (image->zsize == 1) {
			image->dim = 2;
			if (image->ysize == 1)
				image->dim = 1;
		} else {
			image->dim = 3;
		}

		image->min = 10000000;
		image->max = 0;
		strncpy(image->name, "no name", 80);
		image->wastebytes = 0;
		image->dorev = dorev;
		if (do_rgb_write_header(image, f) != IMAGERGB_HEADER_SIZE) {
			cout << "iopen: error on write of image header\n" << endl;
			return NULL;
		}
		image->flags = _IOWRT;
	} else {
		//READ
		if (do_rgb_read_header(image, f) != IMAGERGB_HEADER_SIZE) {
			cout << "iopen: error on read of image header" << endl;
			return NULL;
		}

		if (((image->imagic >> 8) | ((image->imagic & 0xff) << 8)) == IMAGIC) {
			image->dorev = 1;
			cvtimage(image);
		} else
			image->dorev = 0;

		if (image->imagic != IMAGIC) {
			cout << "iopen: bad magic in image file " << image->imagic << endl;
			return NULL;
		}
		image->flags = _IOREAD;
	}

	if (ISRLE(image->type)) {
		tablesize = image->ysize * image->zsize * (int)sizeof(TINT32);
		image->rowstart = (TUINT32 *)malloc(tablesize);
		image->rowsize = (TINT32 *)malloc(tablesize);

		if (image->rowstart == 0 || image->rowsize == 0) {
			cout << "iopen: error on table alloc" << endl;
			return NULL;
		}

		image->rleend = 512L + 2 * tablesize;

		if (openMode == OpenWrite) {
			//WRITE
			int max = image->ysize * image->zsize;
			for (int i = 0; i < max; i++) {
				image->rowstart[i] = 0;
				image->rowsize[i] = -1;
			}
		} else {
			//READ
			tablesize = image->ysize * image->zsize * (int)sizeof(TINT32);
			lseek(f, 512L, 0);
			if (read(f, image->rowstart, tablesize) != tablesize) {
#ifdef _WIN32
				DWORD error;
				error = GetLastError();
#endif
				TSystem::outputDebug("iopen: error on read of rowstart\n");
				return NULL;
			}

			if (image->dorev)
				cvtTINT32s(image->rowstart, tablesize);
			if (read(f, image->rowsize, tablesize) != tablesize) {
#ifdef _WIN32
				DWORD error;
				error = GetLastError();
#endif
				TSystem::outputDebug("iopen: error on read of rowsize\n");
				return NULL;
			}

			if (image->dorev)
				cvtTINT32s((TUINT32 *)image->rowsize, tablesize);
		}
	} //END ISRLE

	image->cnt = 0;
	image->ptr = 0;
	image->base = 0;

	if ((image->tmpbuf = ibufalloc(image, BPP(image->type))) == 0) {
		char xs[1024];
		sprintf(xs, "%d", image->xsize);

		TSystem::outputDebug(string("iopen: error on tmpbuf alloc %d\n") + xs);
		return NULL;
	}

	image->x = image->y = image->z = 0;
	image->file = fd;
	image->offset = 512L; // set up for img_optseek
	lseek((int)image->file, 512L, 0);

	return (image);
}

/*-------------------------------------------------------------------------*/

static USHORT *ibufalloc(IMAGERGB *image, int bpp)
{
	return (USHORT *)malloc(IBUFSIZE(image->xsize) * bpp);
}

/*-------------------------------------------------------------------------*/
/*
   Inverte gli short del buffer
*/

static void cvtshorts(unsigned short buffer[], TINT32 n)
{
	TINT32 nshorts = n >> 1;
	for (int i = 0; i < nshorts; i++) {
		unsigned short swrd = *buffer;
		*buffer++ = (swrd >> 8) | (swrd << 8);
	}
	return;
}

/*-----------------------------------------------------------------------------*/
/*
 *  INVERTE I LONG DEL BUFFER 
   */
/*-----------------------------------------------------------------------------*/

static void cvtTINT32s(TUINT32 buffer[], TINT32 n)
{
	TINT32 nTINT32s = n >> 2;
	for (int i = 0; i < nTINT32s; i++) {
		TUINT32 lwrd = buffer[i];
		buffer[i] = ((lwrd >> 24) |
					 (lwrd >> 8 & 0xff00) |
					 (lwrd << 8 & 0xff0000) |
					 (lwrd << 24));
	}
	return;
}

/*-----------------------------------------------------------------------------*/
/*
 *  INVERTE I LONG E GLI SHORT DEL BUFFER 
*/
/*-----------------------------------------------------------------------------*/

static void cvtimage(IMAGERGB *image)
{
	TUINT32 *buffer = (TUINT32 *)image;
	cvtshorts((unsigned short *)buffer, 12);
	cvtTINT32s(buffer + 3, 12);
	cvtTINT32s(buffer + 26, 4);
	return;
}

/*-----------------------------------------------------------------------------*/

#define EXPAND_CODE(TYPE)                \
	while (1) {                          \
		pixel = *iptr++;                 \
		if (!(count = (pixel & 0x7f)))   \
			return;                      \
		if (pixel & 0x80) {              \
			while (count--)              \
				*optr++ = (TYPE)*iptr++; \
		} else {                         \
			pixel = *iptr++;             \
			while (count--)              \
				*optr++ = (TYPE)pixel;   \
		}                                \
	}

/*-----------------------------------------------------------------------------*/
/*
 *  ESPANDE UNA IMMAGINE FORMATO RGB-RLE  
*/
/*-----------------------------------------------------------------------------*/

static void img_rle_expand(unsigned short *rlebuf, int ibpp,
						   unsigned short *expbuf, int obpp)
{
	if (ibpp == 1 && obpp == 1) {
		unsigned char *iptr = (unsigned char *)rlebuf;
		unsigned char *optr = (unsigned char *)expbuf;
		unsigned short pixel, count;

		EXPAND_CODE(unsigned char);
	} else if (ibpp == 1 && obpp == 2) {
		unsigned char *iptr = (unsigned char *)rlebuf;
		unsigned short *optr = expbuf;
		unsigned short pixel, count;

		EXPAND_CODE(unsigned short);
	} else if (ibpp == 2 && obpp == 1) {
		unsigned short *iptr = rlebuf;
		unsigned char *optr = (unsigned char *)expbuf;
		unsigned short pixel, count;

		EXPAND_CODE(unsigned char);
	} else if (ibpp == 2 && obpp == 2) {
		unsigned short *iptr = rlebuf;
		unsigned short *optr = expbuf;
		unsigned short pixel, count;

		EXPAND_CODE(unsigned short);
	} else
		cout << "rle_expand: bad bpp: " << ibpp << obpp << endl;
}

/*-----------------------------------------------------------------------------*/
/*
 *  RITORNA L'AMPIEZZA DELLA RIGA DI UN IMMAGINE RGB
*/
/*-----------------------------------------------------------------------------*/

static int img_getrowsize(IMAGERGB *image)
{
	switch (image->dim) {
	case 1:
		return (int)image->rowsize[0];
	case 2:
		return (int)image->rowsize[image->y];
	case 3:
		return (int)image->rowsize[image->y + image->z * image->ysize];
	}
	return 0;
}

/*-----------------------------------------------------------------------------*/
/*
 * SPOSTA IL PUNTATORE AL FILE RGB
*/
/*-----------------------------------------------------------------------------*/

static TUINT32 img_optseek(IMAGERGB *image, TUINT32 offset)
{
	if (image->offset != offset) {
		image->offset = offset;
		return (TUINT32)lseek(image->file, (TINT32)offset, 0);
	}
	return offset;
}

/*-----------------------------------------------------------------------------*/
/*
 *  LEGGE DAL FILE RGB E RIEMPE IL BUFFER 
*/
/*-----------------------------------------------------------------------------*/

static TINT32 rgb_img_read(IMAGERGB *image, char *buffer, TINT32 count)
{
	TINT32 retval;

	retval = read(image->file, buffer, count);
	if (retval == count)
		image->offset += count;
	else
		// BRUTTO: ma qui non ci si deve mai passare, serve per fare un crash
		image->offset = (TUINT32)-1;
	return retval;
}

/*-----------------------------------------------------------------------------*/
/*
 * CONTROLLA SE LA RIGA CORRENTE DELL'IMMAGINE RGB E' VALIDA
*/
/*-----------------------------------------------------------------------------*/

static int img_badrow(IMAGERGB *image, int y, int z)
{
	if (y >= image->ysize || z >= image->zsize)
		return 1;
	else
		return 0;
}

/*-----------------------------------------------------------------------------*/
/*
 *  POSIZIONA IL PUNTATORE AL FILE RGB ALL'INIZIO DELL'AREA DATI IMMAGINE
 */
/*-----------------------------------------------------------------------------*/

static TUINT32 img_seek(IMAGERGB *image,
						unsigned int y, unsigned int z, unsigned int offs)
{
	if (img_badrow(image, y, z)) {
		cout << "imglib: row number out of range" << endl;
		return (TUINT32)EOF;
	}
	image->x = 0;
	image->y = y;
	image->z = z;
	if (ISVERBATIM(image->type)) {
		switch (image->dim) {
		case 1:
			return img_optseek(image, 512L + offs);
		case 2:
			return img_optseek(image, 512L + offs + (y * image->xsize) * BPP(image->type));
		case 3:
			return img_optseek(image,
							   512L + offs + (y * image->xsize + z * image->xsize * image->ysize) * BPP(image->type));
		default:
			cout << "img_seek: wierd dim" << endl;
			break;
		}
	} else if (ISRLE(image->type)) {
		switch (image->dim) {
		case 1:
			return img_optseek(image, offs + image->rowstart[0]);
		case 2:
			return img_optseek(image, offs + image->rowstart[y]);
		case 3:
			return img_optseek(image, offs + image->rowstart[y + z * image->ysize]);
		default:
			cout << "img_seek: wierd dim" << endl;
			break;
		}
	} else
		cout << "img_seek: wierd image type" << endl;
	return 0;
}

/*-----------------------------------------------------------------------------*/
/*
   Legge una riga (compressa/non compressa) di un file RGB.
*/

static int new_getrow(IMAGERGB *image, void *buffer, UINT y, UINT z)
{
	short cnt;

	if (!(image->flags & (_IORW | _IOREAD)))
		return -1;

	if (image->dim < 3)
		z = 0;

	if (image->dim < 2)
		y = 0;

	img_seek(image, y, z, 0);

	if (ISVERBATIM(image->type)) {
		switch (BPP(image->type)) {
		case 1:
			if (rgb_img_read(image, (char *)buffer, image->xsize) != image->xsize)
				return -1;

			return image->xsize;
		case 2:
			cnt = image->xsize << 1;
			if (rgb_img_read(image, (char *)buffer, cnt) != cnt)
				return -1;
			else {
				if (image->dorev)
					cvtshorts((unsigned short *)buffer, cnt);
				return image->xsize;
			}
		default:
			cout << "getrow: wierd bpp" << endl;
			break;
		}
	} else if (ISRLE(image->type)) {
		switch (BPP(image->type)) {
		case 1:
			if ((cnt = img_getrowsize(image)) == -1)
				return -1;
			if (rgb_img_read(image, (char *)image->tmpbuf, cnt) != cnt)
				return -1;
			else {
				img_rle_expand(image->tmpbuf, 1, (unsigned short *)buffer, 1);
				return image->xsize;
			}
		case 2:
			if ((cnt = img_getrowsize(image)) == -1)
				return -1;
			if (cnt != rgb_img_read(image, (char *)image->tmpbuf, cnt))
				return -1;
			else {
				if (image->dorev)
					cvtshorts(image->tmpbuf, cnt);
				img_rle_expand(image->tmpbuf, 2, (unsigned short *)buffer, 2);
				return image->xsize;
			}
		default:
			cout << "getrow: weird bpp" << endl;
			break;
		}
	} else
		cout << "getrow: weird image type" << endl;
	return -1;
}

/*-----------------------------------------------------------------------------*/
/*
 *  ROBA PRESA DA ASMWRITERGB.C
 */
/*-----------------------------------------------------------------------------*/

static TINT32 RGB_img_write(IMAGERGB *image, char *buffer, TINT32 count)
{
	TINT32 retval;

	retval = write(image->file, buffer, count);
	if (retval == count)
		image->offset += count;
	else
		image->offset = (TUINT32)-1;
	return retval;
}

/*-----------------------------------------------------------------------------*/

static void img_setrowsize(IMAGERGB *image, UINT cnt, UINT y, UINT z)
{
	TINT32 *sizeptr = 0;

	if (img_badrow(image, y, z))
		return;
	switch (image->dim) {
	case 1:
		sizeptr = &image->rowsize[0];
		image->rowstart[0] = image->rleend;
		break;
	case 2:
		sizeptr = &image->rowsize[y];
		image->rowstart[y] = image->rleend;
		break;
	case 3:
		sizeptr = &image->rowsize[y + z * image->ysize];
		image->rowstart[y + z * image->ysize] = image->rleend;
	}
	if (*sizeptr != -1)
		image->wastebytes += *sizeptr;
	*sizeptr = (TINT32)cnt;
	image->rleend += cnt;
}

/*-----------------------------------------------------------------------------*/

#define COMPACT_CODE(TYPE)                                                            \
	while (iptr < ibufend) {                                                          \
		sptr = iptr;                                                                  \
		iptr += 2;                                                                    \
		while ((iptr < ibufend) && ((iptr[-2] != iptr[-1]) || (iptr[-1] != iptr[0]))) \
			iptr++;                                                                   \
		iptr -= 2;                                                                    \
		count = iptr - sptr;                                                          \
		while (count) {                                                               \
			todo = (TYPE)(count > 126 ? 126 : count);                                 \
			count -= todo;                                                            \
			*optr++ = (TYPE)(0x80 | todo);                                            \
			while (todo--)                                                            \
				*optr++ = (TYPE)*sptr++;                                              \
		}                                                                             \
		sptr = iptr;                                                                  \
		cc = *iptr++;                                                                 \
		while ((iptr < ibufend) && (*iptr == cc))                                     \
			iptr++;                                                                   \
		count = iptr - sptr;                                                          \
		while (count) {                                                               \
			todo = (TYPE)(count > 126 ? 126 : count);                                 \
			count -= todo;                                                            \
			*optr++ = (TYPE)todo;                                                     \
			*optr++ = (TYPE)cc;                                                       \
		}                                                                             \
	}                                                                                 \
	*optr++ = 0;

/*-----------------------------------------------------------------------------*/

static TINT32 img_rle_compact(unsigned short *expbuf, int ibpp, unsigned short *rlebuf,
							  int obpp, int cnt)
{
	if (ibpp == 1 && obpp == 1) {
		unsigned char *iptr = (unsigned char *)expbuf;
		unsigned char *ibufend = iptr + cnt;
		unsigned char *sptr;
		unsigned char *optr = (unsigned char *)rlebuf;
		TUINT32 todo, cc;
		TINT32 count;

		COMPACT_CODE(unsigned char);
		return optr - (unsigned char *)rlebuf;
	} else if (ibpp == 1 && obpp == 2) {
		unsigned char *iptr = (unsigned char *)expbuf;
		unsigned char *ibufend = iptr + cnt;
		unsigned char *sptr;
		unsigned short *optr = rlebuf;
		TUINT32 todo, cc;
		TINT32 count;

		COMPACT_CODE(unsigned short);
		return optr - rlebuf;
	} else if (ibpp == 2 && obpp == 1) {
		unsigned short *iptr = expbuf;
		unsigned short *ibufend = iptr + cnt;
		unsigned short *sptr;
		unsigned char *optr = (unsigned char *)rlebuf;
		TUINT32 todo, cc;
		TINT32 count;

		COMPACT_CODE(unsigned char);
		return optr - (unsigned char *)rlebuf;
	} else if (ibpp == 2 && obpp == 2) {
		unsigned short *iptr = expbuf;
		unsigned short *ibufend = iptr + cnt;
		unsigned short *sptr;
		unsigned short *optr = rlebuf;
		unsigned short todo, cc;
		TINT32 count;

		COMPACT_CODE(unsigned short);
		return optr - rlebuf;
	} else {
		cout << "rle_compact: bad bpp: %d %d" << ibpp << obpp;
		return 0;
	}
}

/*-----------------------------------------------------------------------------*/

void iclose(IMAGERGB *image)
{
	TINT32 tablesize;

	iflush(image);
	img_optseek(image, 0);

	/* CONTROLLARE!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!*/

	if (image->flags & _IOWRT) {
		if (image->dorev)
			cvtimage(image);
		if (do_rgb_write_header(image, image->file) != IMAGERGB_HEADER_SIZE) {
			fprintf(stderr, "iflush: error on write of image header\n");
			return;
		}

		if (image->dorev)
			cvtimage(image);
		if (ISRLE(image->type)) {
			img_optseek(image, 512L);
			tablesize = image->ysize * image->zsize * (int)sizeof(TINT32);
			if (image->dorev)
				cvtTINT32s(image->rowstart, tablesize);
			if (RGB_img_write(image, (char *)(image->rowstart), tablesize) != tablesize) {
				fprintf(stderr, "iflush: error on write of rowstart\n");
				return;
			}
			if (image->dorev)
				cvtTINT32s((TUINT32 *)image->rowsize, tablesize);
			if (RGB_img_write(image, (char *)(image->rowsize), tablesize) != tablesize) {
				fprintf(stderr, "iflush: error on write of rowsize\n");
				return;
			}
		}
	}
	if (image->base) {
		free(image->base);
		image->base = 0;
	}
	if (image->tmpbuf) {
		free(image->tmpbuf);
		image->tmpbuf = 0;
	}
	if (ISRLE(image->type)) {
		free(image->rowstart);
		image->rowstart = 0;
		free(image->rowsize);
		image->rowsize = 0;
	}
	free(image);
	return;
}

/*-------------------------------------------------------------------------*/

static int new_putrow(IMAGERGB *image, void *buffer, UINT y, UINT z)
{
	TINT32 cnt;
	int dorev, bpp;

	if (!(image->flags & (_IORW | _IOWRT)))
		return -1;

	if (image->dim < 3)
		z = 0;

	if (image->dim < 2)
		y = 0;

	image->min = 0;
	bpp = BPP(image->type);
	dorev = image->dorev && bpp == 2;

	if (bpp == 1)
		image->max = 255; /*commento tutti i calcoli su max e min per velocizzare */
	else
		image->max = 65535;

	if (ISVERBATIM(image->type)) {
		img_seek(image, y, z, 0);
		cnt = image->xsize << (bpp - 1);
		if (dorev)
			cvtshorts((unsigned short *)buffer, cnt);
		if (RGB_img_write(image, (char *)buffer, cnt) != cnt) {
			if (dorev)
				cvtshorts((unsigned short *)buffer, cnt);
			return -1;
		} else {
			if (dorev)
				cvtshorts((unsigned short *)buffer, cnt);
			return image->xsize;
		}
	} else if (ISRLE(image->type)) {
		cnt = img_rle_compact((unsigned short *)buffer, bpp, image->tmpbuf, bpp, image->xsize);
		cnt <<= (bpp - 1);
		img_setrowsize(image, cnt, y, z);
		img_seek(image, y, z, 0);
		if (dorev)
			cvtshorts(image->tmpbuf, cnt);
		if (RGB_img_write(image, (char *)(image->tmpbuf), cnt) != cnt) {
			if (dorev)
				cvtshorts(image->tmpbuf, cnt);
			return -1;
		} else {
			if (dorev)
				cvtshorts(image->tmpbuf, cnt);
			return image->xsize;
		}
	} else
		fprintf(stderr, "putrow: wierd image type\n");

	return -1;
}

/*----------------------------------------------------------------------------*/

static int iflush(IMAGERGB *image)
{
	unsigned short *base;

	if ((image->flags & _IOWRT) && (base = image->base) != NULL && (image->ptr - base) > 0) {
		if (new_putrow(image, base, image->y, image->z) != image->xsize) {
			image->flags |= _IOERR;
			return (EOF);
		}
	}
	return 0;
}

/*----------------------------------------------------------------------------*/

class SgiReader : public Tiio::Reader
{

	IMAGERGB *m_header;
	int m_currentY;

public:
	SgiReader() : m_header(0), m_currentY(0) {}

	~SgiReader();

	void open(FILE *file);

	TPropertyGroup *getProperties() const { return 0; }

	void readLine(char *buffer, int x0, int x1, int shrink);
	void readLine(short *buffer, int x0, int x1, int shrink);

	int skipLines(int lineCount);
};

//-------------------------------------------------------------------

void SgiReader::open(FILE *file)
{
	m_header = iopen(fileno(file), OpenRead, 0, 0, 0, 0, 0, 0);

	if (!m_header) {
		string str("Can't open file");
		throw(str);
	}

	m_info.m_lx = m_header->xsize;
	m_info.m_ly = m_header->ysize;
	m_info.m_bitsPerSample = BPP(m_header->type) * 8;
	m_info.m_samplePerPixel = m_header->zsize;
	Tiio::SgiWriterProperties *prop = new Tiio::SgiWriterProperties();
	m_info.m_properties = prop;
	prop->m_endianess.setValue(m_header->dorev == 1 ? L"Big Endian" : L"Little Endian");
	prop->m_compressed.setValue(ISRLE(m_header->type) ? true : false);
	wstring pixelSize;
	int ps = m_info.m_bitsPerSample * m_info.m_samplePerPixel;
	if (ps == 8)
		pixelSize = L"8 bits (Greyscale)";
	else if (ps == 24)
		pixelSize = L"24 bits";
	else if (ps == 32)
		pixelSize = L"32 bits";
	else if (ps == 48)
		pixelSize = L"48 bits";
	else if (ps == 64)
		pixelSize = L"64 bits";
	prop->m_pixelSize.setValue(pixelSize);
}

//-------------------------------------------------------------------

SgiReader::~SgiReader()
{
	if (!m_header)
		return;
	if (m_header->base)
		free(m_header->base);
	if (m_header->tmpbuf)
		free(m_header->tmpbuf);
	if ((ISRLE(m_header->type))) {
		free(m_header->rowstart);
		free(m_header->rowsize);
	}

	free(m_header);
	m_header = 0;
}

//-------------------------------------------------------------------

void SgiReader::readLine(short *buffer, int x0, int x1, int shrink)
{
	//assert(shrink == 1);
	//assert(x0 == 0);
	//assert(x1 == m_info.m_lx-1);

	assert(BPP(m_header->type) == 2);

	{ // 64 -> 32
		TPixel64 *pix = (TPixel64 *)buffer;
		std::vector<USHORT> rbuf(m_info.m_lx), gbuf(m_info.m_lx), bbuf(m_info.m_lx), mbuf(m_info.m_lx);
		if (m_header->zsize == 4) {
			new_getrow(m_header, &rbuf[0], m_currentY, 0);
			new_getrow(m_header, &gbuf[0], m_currentY, 1);
			new_getrow(m_header, &bbuf[0], m_currentY, 2);
			new_getrow(m_header, &mbuf[0], m_currentY, 3);
			for (int i = 0; i < (x1 - x0 + 1); ++i) {
				pix->r = rbuf[i];
				pix->g = gbuf[i];
				pix->b = bbuf[i];
				pix->m = mbuf[i];
				++pix;
			}
		} else {
			new_getrow(m_header, &rbuf[0], m_currentY, 0);
			new_getrow(m_header, &gbuf[0], m_currentY, 1);
			new_getrow(m_header, &bbuf[0], m_currentY, 2);
			for (int i = 0; i < (x1 - x0 + 1); ++i) {
				pix->r = rbuf[i];
				pix->g = gbuf[i];
				pix->b = bbuf[i];
				pix->m = 0xffff;
				++pix;
			}
		}
	}
	m_currentY++;
}

//-------------------------------------------------------------------

void SgiReader::readLine(char *buffer, int x0, int x1, int shrink)
{
	//assert(shrink == 1);
	//assert(x0 == 0);
	//assert(x1 == m_info.m_lx-1);

	//Non ancora implementata la lettura parziale
	x0 = 0;
	x1 = m_info.m_lx - 1;
	shrink = 1;

	if (BPP(m_header->type) == 1) //32 -> 32
	{
		TPixel32 *pix = (TPixel32 *)buffer;
		std::vector<UCHAR> rbuf(m_info.m_lx), gbuf(m_info.m_lx), bbuf(m_info.m_lx), mbuf(m_info.m_lx);
		if (m_header->zsize == 4) {
			new_getrow(m_header, &rbuf[0], m_currentY, 0);
			new_getrow(m_header, &gbuf[0], m_currentY, 1);
			new_getrow(m_header, &bbuf[0], m_currentY, 2);
			new_getrow(m_header, &mbuf[0], m_currentY, 3);
			for (int i = 0; i < (x1 - x0 + 1); ++i) {
				pix->r = rbuf[i];
				pix->g = gbuf[i];
				pix->b = bbuf[i];
				pix->m = mbuf[i];
				++pix;
			}
		} else {
			new_getrow(m_header, &rbuf[0], m_currentY, 0);
			if (m_header->zsize == 1)
				for (int i = 0; i < (x1 - x0 + 1); ++i) {
					pix->r = rbuf[i];
					pix->g = rbuf[i];
					pix->b = rbuf[i];
					pix->m = 0xff;
					++pix;
				}
			else {
				new_getrow(m_header, &gbuf[0], m_currentY, 1);
				new_getrow(m_header, &bbuf[0], m_currentY, 2);
				for (int i = 0; i < (x1 - x0 + 1); ++i) {
					pix->r = rbuf[i];
					pix->g = gbuf[i];
					pix->b = bbuf[i];
					pix->m = 0xff;
					++pix;
				}
			}
		}
	} else { // 64 -> 32
		TPixel32 *pix = (TPixel32 *)buffer;
		std::vector<USHORT> rbuf(m_info.m_lx), gbuf(m_info.m_lx), bbuf(m_info.m_lx), mbuf(m_info.m_lx);
		if (m_header->zsize == 4) {
			new_getrow(m_header, &rbuf[0], m_currentY, 0);
			new_getrow(m_header, &gbuf[0], m_currentY, 1);
			new_getrow(m_header, &bbuf[0], m_currentY, 2);
			new_getrow(m_header, &mbuf[0], m_currentY, 3);
			for (int i = 0; i < (x1 - x0 + 1); ++i) {
				pix->r = rbuf[i] >> 8;
				pix->g = gbuf[i] >> 8;
				pix->b = bbuf[i] >> 8;
				pix->m = mbuf[i] >> 8;
				++pix;
			}
		} else {
			new_getrow(m_header, &rbuf[0], m_currentY, 0);
			new_getrow(m_header, &gbuf[0], m_currentY, 1);
			new_getrow(m_header, &bbuf[0], m_currentY, 2);
			for (int i = 0; i < (x1 - x0 + 1); ++i) {
				pix->r = rbuf[i] >> 8;
				pix->g = gbuf[i] >> 8;
				pix->b = bbuf[i] >> 8;
				pix->m = 0xff;
				++pix;
			}
		}
	}
	m_currentY++;
}

//-------------------------------------------------------------------

int SgiReader::skipLines(int lineCount)
{

	m_currentY += lineCount;
	return lineCount;
}

//-------------------------------------------------------------------

/*
						WRITER
*/

class SgiWriter : public Tiio::Writer
{
	int m_currentY;
	IMAGERGB *m_header;

protected:
	TImageInfo m_info;

public:
	SgiWriter() : m_currentY(0), m_header(0){};
	~SgiWriter()
	{
		if (m_header)
			iclose(m_header);
		delete m_properties;
	}

	void open(FILE *file, const TImageInfo &info);

	TPropertyGroup *getProperties() { return m_properties; }

	void writeLine(char *buffer);
	void writeLine(short *buffer);

	void flush() {}

	//  RowOrder getRowOrder() const { return BOTTOM2TOP; }
	bool write64bitSupported() const { return true; }

	void setProperties(TPropertyGroup *properties);

private:
	// not implemented
	SgiWriter(const SgiWriter &);
	SgiWriter &operator=(const SgiWriter &);
};

Tiio::Reader *Tiio::makeSgiReader()
{
	return new SgiReader();
}

//=========================================================

Tiio::Writer *Tiio::makeSgiWriter()
{
	return new SgiWriter();
}

//---------------------------------------------------------

void SgiWriter::open(FILE *file, const TImageInfo &info)
{
	if (!m_properties)
		m_properties = new Tiio::SgiWriterProperties();

	TEnumProperty *p = (TEnumProperty *)(m_properties->getProperty("Bits Per Pixel"));
	assert(p);
	string str = toString(p->getValue());
	int bitPerPixel = atoi(str.c_str());
	int channelBytesNum = 1;
	int dim = 3;
	int zsize = 1;

	m_info = info;

	switch (bitPerPixel) {
	case 8:
		dim = 2;
		zsize = 1;
		break;
	case 24:
		dim = 3;
		zsize = 3;
		break;
	case 32:
		dim = 3;
		zsize = 4;
		break;
	case 48:
		dim = 3;
		zsize = 3;
		channelBytesNum = 2;
		break;
	case 64:
		zsize = 4;
		dim = 3;
		channelBytesNum = 2;
		break;
	}

	TBoolProperty *bp = (TBoolProperty *)(m_properties->getProperty("RLE-Compressed"));
	assert(bp);
	bool compressed = bp->getValue();
	p = (TEnumProperty *)(m_properties->getProperty("Endianess"));
	assert(p);
	str = toString(p->getValue());
	bool bigEndian = (str == "Big Endian");

	m_header = iopen(fileno(file), OpenWrite, compressed ? RLE(BPP(channelBytesNum)) : VERBATIM(BPP(channelBytesNum)),
					 dim, m_info.m_lx, m_info.m_ly, zsize, bigEndian ? 1 : 0);
}
//---------------------------------------------------------

void SgiWriter::writeLine(char *buffer)
{
	if (BPP(m_header->type) == 1) {
		if (m_header->dim < 3) // 32->8
		{
			new_putrow(m_header, buffer, m_currentY, 0);
		} else {
			std::vector<UCHAR> rbuf(m_info.m_lx), gbuf(m_info.m_lx), bbuf(m_info.m_lx), mbuf(m_info.m_lx);
			TPixelRGBM32 *pix = (TPixelRGBM32 *)buffer;
			for (int i = 0; i < m_info.m_lx; ++i) {
				rbuf[i] = pix->r;
				gbuf[i] = pix->g;
				bbuf[i] = pix->b;
				mbuf[i] = pix->m;
				++pix;
			}
			new_putrow(m_header, &rbuf[0], m_currentY, 0);
			new_putrow(m_header, &gbuf[0], m_currentY, 1);
			new_putrow(m_header, &bbuf[0], m_currentY, 2);
			if (m_header->zsize == 4)
				new_putrow(m_header, &mbuf[0], m_currentY, 3);
		}
	}
	++m_currentY;
}

//---------------------------------------------------------

void SgiWriter::writeLine(short *buffer)
{
	assert(BPP(m_header->type) == 2);

	{
		if (m_header->dim < 3) {
			std::vector<USHORT> tmp(m_info.m_lx);
			TPixelRGBM64 *pix = (TPixelRGBM64 *)buffer;
			for (int i = 0; i < m_info.m_lx; ++i) {
				tmp[i] = TPixelGR16::from(*pix).value;
				++pix;
			}
			new_putrow(m_header, &tmp[0], m_currentY, 0);
		} else {
			std::vector<USHORT> rbuf(m_info.m_lx), gbuf(m_info.m_lx), bbuf(m_info.m_lx), mbuf(m_info.m_lx);
			TPixelRGBM64 *pix = (TPixelRGBM64 *)buffer;
			for (int i = 0; i < m_info.m_lx; ++i) {
				rbuf[i] = pix->r;
				gbuf[i] = pix->g;
				bbuf[i] = pix->b;
				mbuf[i] = pix->m;
				++pix;
			}
			new_putrow(m_header, &rbuf[0], m_currentY, 0);
			new_putrow(m_header, &gbuf[0], m_currentY, 1);
			new_putrow(m_header, &bbuf[0], m_currentY, 2);
			if (m_header->zsize == 4)
				new_putrow(m_header, &mbuf[0], m_currentY, 3);
		}
	}
	++m_currentY;
}
//=========================================================

Tiio::SgiWriterProperties::SgiWriterProperties()
	: m_pixelSize("Bits Per Pixel"), m_endianess("Endianess"), m_compressed("RLE-Compressed", false)
{
	m_pixelSize.addValue(L"24 bits");
	m_pixelSize.addValue(L"32 bits");
	m_pixelSize.addValue(L"48 bits");
	m_pixelSize.addValue(L"64 bits");
	m_pixelSize.addValue(L"8 bits (Greyscale)");
	m_pixelSize.setValue(L"32 bits");
	bind(m_pixelSize);
	bind(m_compressed);
	m_endianess.addValue(L"Big Endian");
	m_endianess.addValue(L"Little Endian");
	bind(m_endianess);
}