#include "tiio_plt.h"
//#include "tiio.h"
#include "trastercm.h"
#include "toonztags.h"

#include "tiffio.h"

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

extern "C" {
#include "avl.h"
#include "tcm.h"
}

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

class PltReader : 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;
	std::string m_history;
	int m_pltType; //[1..4]
	int m_nColor;
	int m_nPencil;
	std::vector<TPixel32> m_infoRow;
	std::map<int, std::pair<std::string, std::string>> m_pltNames; // colorIndex(<256: paint) , pageName, colorName
public:
	PltReader();
	~PltReader();

	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);

	void getTzpPaletteColorNames(std::map<int, std::pair<std::string, std::string>> &pltColorNames) const;
};

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

PltReader::PltReader()
	: 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_pltType(-1), m_nColor(0), m_nPencil(0)
{
}

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

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

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

int decode_group_name(
	char group_name[],
	char **name,
	int *key,
	int *sister_index)
{
	char *s, *t;
	*key = 0;
	*sister_index = -1;
	s = group_name;
	while (1) {
		t = strchr(s, '!');
		if (!t)
			break;
		*t = '\0';
		if (s[0] == 'C' && s[1] >= '0' && s[1] <= '9')
			*sister_index = atoi(s + 1);
		else if (s[0] >= '0' && s[0] <= '9')
			*key = atoi(s);
		*t = '!';
		s = t + 1;
	}
	*name = s;
	return 1;
}

void PltReader::getTzpPaletteColorNames(std::map<int, std::pair<std::string, std::string>> &pltColorNames) const
{
	pltColorNames = m_pltNames;
}

//--------------------------------------------------------------------------------------
namespace
{
	bool isDefaultName(const std::string &_name)
{
	int i;
	QString name = QString::fromStdString(_name);
	if (name.startsWith("Ink_")) {
		for (i = 4; i < name.size(); i++)
			if (!name.at(i).isDigit())
				return false;
		return true;
	} else if (name.startsWith("Paint_")) {
		for (i = 6; i < name.size(); i++)
			if (!name.at(i).isDigit())
				return false;
		return true;
	} else
		return false;
}

} //namespace
//------------------------------------------------------------

bool PaletteRead = false;
int ComboInkIndex[256]; //a bad patch....

void PltReader::open(FILE *file)
{
	char *data;
	uint32 count;
	int i;
	for (i = 0; i < 256; i++)
		ComboInkIndex[i] = -1;

	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, rps = 0;
	uint32 tileWidth = 0, tileLength = 0;
	uint16 bps = 0, spp = 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;

	assert(h == 1);
	m_ly = m_info.m_ly = 2; // per l'infoRow

	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);
	assert(paletteCount);

	m_pltType = palette[0];

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

	std::string colorNames;
	if (TIFFGetField(m_tiff, TIFFTAG_TOONZCOLORNAMES, &count, &data))
		colorNames = data;

	TREE *names = cdb_decode_all(data, Tcm_24_default_info);

	CDB_TREE_ITEM *item;

	char *pageName;
	int key, sisterIndex;

	int maxCount = m_nColor + m_nPencil;
	m_infoRow.resize(maxCount, TPixel32(0, 0, 0, 0));

	for (i = 0; i < maxCount; i++) {
		item = (CDB_TREE_ITEM *)avl_locate(names, i);
		if (!item)
			continue;
		decode_group_name(item->group, &pageName, &key, &sisterIndex);

		if (strcmp(pageName, "_UNUSED_PAGE") == 0)
			continue;
		if (sisterIndex == -1 || i < m_nColor)
			m_infoRow[i].r = 255;
		if (i < m_nColor)
			m_infoRow[i].g = 255;

		while (isdigit(item->name[0])) //in toonz colors cannot begin with digits
			item->name++;

		std::map<int, std::pair<std::string, std::string>>::iterator it;
		if ((it = m_pltNames.find(i)) == m_pltNames.end() || isDefaultName(it->second.second))
			m_pltNames[i] = std::pair<std::string, std::string>(pageName, item->name);

		if (sisterIndex != -1) {
			int comboindex = (i < 256) ? (sisterIndex >> 16) + 256 : sisterIndex >> 8;
			if (i >= 256)
				ComboInkIndex[i - 256] = comboindex;
			std::map<int, std::pair<std::string, std::string>>::iterator it;
			if ((it = m_pltNames.find(comboindex)) == m_pltNames.end() || isDefaultName(it->second.second)) {
				m_pltNames[comboindex] = std::pair<std::string, std::string>(pageName, item->name);
				if (comboindex < m_nColor)
					m_infoRow[comboindex].r = m_infoRow[comboindex].g = 255;
			}
		}
	}

	if (TIFFGetField(m_tiff, TIFFTAG_TOONZHISTORY, &count, &data))
		m_history = data;

	uint16 photometric, planarconfig;

	TIFFGetField(m_tiff, TIFFTAG_PHOTOMETRIC, &photometric);
	TIFFGetField(m_tiff, TIFFTAG_PLANARCONFIG, &planarconfig);

	if (photometric != PHOTOMETRIC_RGB || planarconfig != PLANARCONFIG_CONTIG) {
		//tmsg_error("bad !");
	}
}

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

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

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

void PltReader::readLine(char *buffer, int x0, int x1, int shrink)
{
	TPixelRGBM32 *pix = (TPixelRGBM32 *)buffer;
	int i;

	for (i = 0; i < m_info.m_lx; i++)
		pix[i] = TPixelRGBM32();
	int y = m_row++;

	if (y == 1) {
		for (i = 0; i < m_info.m_lx; i++)
			pix[i] = m_infoRow[i];
		return;
	} else if (y > 1)
		return;

	unsigned char line[4096 * 4];

	TIFFReadScanline(m_tiff, (char *)line, y - m_y, 0);

	switch (m_pltType) {
	case 1:
		throw "Unsupported palette type";
		break;

	case 2:
	case 4:
		for (i = 0; i < (m_nColor + m_nPencil) * 4; i += 4) //prima i color, poi i pencil
		{
			pix->r = line[i + 0];
			pix->g = line[i + 1];
			pix->b = line[i + 2];
			pix->m = line[i + 3];
			++pix;
		}
		break;

	case 3:
		for (i = 0; i < (m_nColor + m_nPencil) * 4; i += 4) //prima i color, poi i pencil
		{
			pix->m = line[i + 0]; /* different from type 2 */
			pix->b = line[i + 1];
			pix->g = line[i + 2];
			pix->r = line[i + 3];
			++pix;
		}
		break;
	default:
		throw "Unknown palette type";
		break;
	}
}

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

Tiio::Reader *Tiio::makePltReader()
{
	return new PltReader();
}

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

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

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