#include "tiio_tzp.h"
//#include "tiio.h"
#include "trastercm.h"
#include "toonztags.h"
#include "texception.h"

#include "tiffio.h"
#include "tiffiop.h"
//#include "tspecialstyleid.h"
#include <set>

#ifdef _WIN32
#pragma warning(disable : 4996)
#include "windows.h"
#endif

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

class TzpReader : public Tiio::Reader
{

	TIFF *m_tiff;
	int m_row;
	bool m_tiled, m_stripped;
	int m_rowsPerStrip;
	int m_stripIndex;
	int m_rowLength;
	UCHAR *m_stripBuffer;
	int m_x, m_y, m_lx, m_ly;
	bool m_isCmap24;
	string m_history;
	int m_nColor;
	int m_nPencil;
	bool m_isBigEndian;
	bool m_isFirstLineRead;

public:
	TzpReader();
	~TzpReader();

	void open(FILE *file);

	Tiio::RowOrder getRowOrder() const
	{
		return Tiio::BOTTOM2TOP;
	}

	int skipLines(int lineCount);
	void readLine(char *buffer, int x0, int x1, int shrink);
};

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

TzpReader::TzpReader()
	: m_tiff(0), m_row(0), m_rowsPerStrip(0), m_stripIndex(-1), m_stripBuffer(0), m_rowLength(0), m_x(0), m_y(0), m_lx(0), m_ly(0), m_isCmap24(false), m_nColor(0), m_nPencil(0), m_isBigEndian(false), m_isFirstLineRead(true)
{
}

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

TzpReader::~TzpReader()
{
	delete m_stripBuffer;
}

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

void TzpReader::open(FILE *file)
{
	int fd = fileno(file);
	TIFFErrorHandler oldhandler;
	oldhandler = TIFFSetWarningHandler(NULL);
	m_tiff = TIFFFdOpen(fd, "", "rb");
	TIFFSetWarningHandler(oldhandler);
	if (!m_tiff)
		return;

	uint32 w = 0, h = 0, bps = 0, spp = 0, rps = 0;
	uint32 tileWidth = 0, tileLength = 0;

	TIFFGetField(m_tiff, TIFFTAG_IMAGEWIDTH, &w);
	TIFFGetField(m_tiff, TIFFTAG_IMAGELENGTH, &h);
	TIFFGetField(m_tiff, TIFFTAG_BITSPERSAMPLE, &bps);
	TIFFGetField(m_tiff, TIFFTAG_SAMPLESPERPIXEL, &spp);
	TIFFGetField(m_tiff, TIFFTAG_ROWSPERSTRIP, &rps);

	float xdpi, ydpi;
	TIFFGetField(m_tiff, TIFFTAG_XRESOLUTION, &xdpi);
	TIFFGetField(m_tiff, TIFFTAG_YRESOLUTION, &ydpi);
	m_info.m_dpix = xdpi;
	m_info.m_dpiy = ydpi;

	TIFFGetField(m_tiff, TIFFTAG_TILEWIDTH, &tileWidth);
	TIFFGetField(m_tiff, TIFFTAG_TILELENGTH, &tileLength);

	uint32 risCount = 0;
	USHORT *risArray = 0;

	m_info.m_lx = w;
	m_info.m_ly = h;
	m_x = 0;
	m_y = 0;
	m_lx = w;
	m_ly = h;

	if (TIFFGetField(m_tiff, TIFFTAG_TOONZWINDOW, &risCount, &risArray)) {
		m_info.m_x0 = m_x = risArray[0];
		m_info.m_y0 = m_y = risArray[1];
		m_info.m_lx = risArray[2];
		m_info.m_ly = risArray[3];
	} else {
		m_info.m_x0 = 0;
		m_info.m_y0 = 0;
	}

	m_info.m_x1 = m_info.m_x0 + w;
	m_info.m_y1 = m_info.m_y0 + h;

	if (TIFFIsTiled(m_tiff)) {
		m_rowsPerStrip = tileLength;
		int tilesPerRow = (w + tileWidth - 1) / tileWidth;
		m_rowLength = tileWidth * tilesPerRow;
		int stripSize = m_rowsPerStrip * m_rowLength * 4;
		m_stripBuffer = new UCHAR[stripSize];
	} else {
		m_rowsPerStrip = rps;
		int stripSize = rps * w * 4 + 4096; // TIFFStripSize(m_tiff);

		m_stripBuffer = new UCHAR[stripSize];
		m_rowLength = w;
	}

	uint32 paletteCount;
	USHORT *palette;

	TIFFGetField(m_tiff, TIFFTAG_TOONZPALETTE, &paletteCount, &palette);

	m_nColor = palette[10];
	m_nPencil = palette[11];

	if (m_nColor == 128 && m_nPencil == 32) {
		throw TException("Old 4.1 Palette");
	}

	if (bps == 32)
		m_isCmap24 = true;
	else
		m_isCmap24 = false;

	char *data;
	uint32 count;
	TIFFGetField(m_tiff, TIFFTAG_TOONZHISTORY, &count, &data);
	string history(data);

	uint16 planarconfig;
	TIFFGetField(m_tiff, TIFFTAG_PLANARCONFIG, &planarconfig);
	if (planarconfig == PLANARCONFIG_SEPARATE) {
		//tmsg_error("separate buffer image not supported yet in .tz(up) files");
	}
}

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

int TzpReader::skipLines(int lineCount)
{
	//m_row += lineCount;
	return 0;
}

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

extern int ComboInkIndex[]; //a bad patch....

void TzpReader::readLine(char *buffer, int x0, int x1, int shrink)
{
	TPixelCM32 *pix = (TPixelCM32 *)buffer;
	for (int i = 0; i < m_info.m_lx; i++)
		pix[i] = TPixelCM32();
	int y = m_row++;

	const int paintOffset = 0;				// paint#1 --> 1
	const int inkOffset = 1 + m_nColor - 1; // ink#0 --> nColor-1
	// (n.b. c'e' un colore in meno visto che il primo colore e' il BG)

	if (m_isCmap24) {
		if (m_y <= y && y < m_y + m_ly) {
			vector<TUINT32> line(m_lx);
			TIFFReadScanline(m_tiff, (char *)&line[0], y - m_y, 0);
			if (m_isFirstLineRead) {
				m_isFirstLineRead = false;
				for (int i = 0; i < m_lx; i++)
					if ((line[i] >> 24) != 0) {
						m_isBigEndian = true;
						break;
					}
			}

			pix += m_x;
			for (int i = 0; i < m_lx; i++) {
				TUINT32 inPix = m_isBigEndian ? swapTINT32(line[i]) : line[i];
				int tone = inPix & 0xFF;
				int paint = ((inPix >> 8) & 0xFF);
				int ink = ((inPix >> 16) & 0xFF);
				if (paint > 0)
					paint += paintOffset;
				if (ComboInkIndex[ink] != -1)
					ink = ComboInkIndex[ink];
				else
					ink += inkOffset;
				pix[i] = TPixelCM32(ink, paint, tone);
			}
		}
	} else {
		if (m_y <= y && y < m_y + m_ly) {
			vector<unsigned short> line(m_lx);
			TIFFReadScanline(m_tiff, (char *)&line[0], y - m_y, 0);
			pix += m_x;
			static std::set<int> table;

			///per le tzp che vengono da Irix
			bool bigEndian = (m_tiff->tif_header.classic.tiff_magic == TIFF_BIGENDIAN);

			for (int i = 0; i < m_lx; i++) {
				unsigned short inPix = line[i];
				if (bigEndian)
					inPix = swapUshort(inPix);

				int tone = inPix & 0xF;
				int paint = ((inPix >> 4) & 0x7F);
				int ink = ((inPix >> 11) & 0x1F);
				tone |= tone << 4;
				if (paint > 0)
					paint += paintOffset;
				ink += inkOffset;
				pix[i] = TPixelCM32(ink, paint, tone);
				if (tone < 255) {
					//           int w = ink;
				}
				if (table.find(ink) == table.end()) {
					table.insert(ink);
				}
			}
		}
	}
}

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

Tiio::Reader *Tiio::makeTzpReader()
{
	return new TzpReader();
}

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

Tiio::Writer *Tiio::makeTzpWriter()
{
	assert(0);
	return 0;
}

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