#include "toonz/trasterimageutils.h"
#include "toonz/ttileset.h"
#include "tstroke.h"
#include "tofflinegl.h"
#include "tpalette.h"
#include "tvectorrenderdata.h"
#include "ttzpimagefx.h"
#include "tvectorgl.h"
#include "trop.h"
#include "tregion.h"
#include "tcolorstyles.h"

#include <QPainter>
#include <QColor>
#include <QImage>

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

namespace
{
QImage rasterToQImage(const TRasterP &ras, bool premultiplied, bool mirrored)
{
	if (TRaster32P ras32 = ras) {
		QImage image(ras->getRawData(), ras->getLx(), ras->getLy(),
					 premultiplied ? QImage::Format_ARGB32_Premultiplied : QImage::Format_ARGB32);
		if (mirrored)
			return image.mirrored();
		return image;
	} else if (TRasterGR8P ras8 = ras) {
		QImage image(ras->getRawData(), ras->getLx(), ras->getLy(), ras->getWrap(), QImage::Format_Indexed8);
		static QVector<QRgb> colorTable;
		if (colorTable.size() == 0) {
			int i;
			for (i = 0; i < 256; i++)
				colorTable.append(QColor(i, i, i).rgb());
		}
		image.setColorTable(colorTable);
		if (mirrored)
			return image.mirrored();
		return image;
	}
	return QImage();
}

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

void rasterizeWholeStroke(TOfflineGL *&gl, TStroke *stroke, TPalette *palette, bool doAnialias)
{
	TRectD bbox = stroke->getBBox();
	TRect rect = convert(bbox).enlarge(1);
	gl = new TOfflineGL(rect.getSize());
	gl->makeCurrent();
	gl->clear(TPixel32(0, 0, 0, 0));
	glEnable(GL_ALPHA_TEST);
	glAlphaFunc(GL_GREATER, 0);

	TPaletteP plt = palette->clone();
	int styleId = stroke->getStyle();
	TColorStyleP style = plt->getStyle(styleId);
	TTranslation affine(-convert(rect.getP00()));
	TVectorRenderData rd(affine, gl->getBounds(), plt.getPointer(), 0, true, true);
	if (doAnialias)
		tglDraw(rd, stroke);
	else {
		TStrokeProp *prop = stroke->getProp();
		if (prop)
			prop->getMutex()->lock();

		if (!style->isStrokeStyle() || style->isEnabled() == false) {
			if (prop)
				prop->getMutex()->unlock();

			prop = 0;
		} else {
			if (!prop || style.getPointer() != prop->getColorStyle()) {
				if (prop)
					prop->getMutex()->unlock();

				stroke->setProp(style->makeStrokeProp(stroke));
				prop = stroke->getProp();
				if (prop)
					prop->getMutex()->lock();
			}
		}

		if (!prop)
			return;
		prop->getMutex()->lock();
		glPushAttrib(GL_ALL_ATTRIB_BITS);
		prop->draw(rd);
		glPopAttrib();
	}
	glDisable(GL_ALPHA_TEST);
	glFinish();
}

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

TRect fastAddInkStroke(const TRasterImageP &ri, TStroke *stroke, TRectD clip, double opacity, bool doAntialiasing)
{
	TOfflineGL *gl = 0;
	TRectD bbox = stroke->getBBox();
	TRect sBBox = convert(bbox).enlarge(1);
	TRect rectRender = sBBox * ri->getRaster()->getBounds();

	if (!rectRender.isEmpty()) {
		if (opacity < 1.0) {
			int styleId = stroke->getStyle();
			TPalette *plt = ri->getPalette();
			TPixel32 color = plt->getStyle(styleId)->getMainColor();
			color.m = 255 * opacity;
			TPaletteP newPlt(plt);
			newPlt->getStyle(styleId)->setMainColor(color);
			rasterizeWholeStroke(gl, stroke, newPlt.getPointer(), doAntialiasing);
		} else
			rasterizeWholeStroke(gl, stroke, ri->getPalette(), doAntialiasing);
		TRect tmp = rectRender - sBBox.getP00();
		TRaster32P glRas = gl->getRaster()->extract(tmp);
		TRop::over(ri->getRaster(), glRas, rectRender.getP00());
		delete gl;
	}
	return rectRender;
}

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

TRect rasterizeRegion(TOfflineGL *&gl, TRect rasBounds, TRegion *region, TPalette *palette, TRectD clip)
{
	TRectD regionBBox = region->getBBox();
	if (!clip.isEmpty())
		regionBBox = regionBBox * clip;

	TRect rect = convert(regionBBox) * rasBounds;
	if (!rect.isEmpty()) {
		gl = new TOfflineGL(rect.getSize());
		gl->makeCurrent();
		gl->clear(TPixel32::Transparent);
		glPushAttrib(GL_ALL_ATTRIB_BITS);
		glEnable(GL_ALPHA_TEST);
		glAlphaFunc(GL_GREATER, 0);

		TTranslation affine(-convert(rect.getP00()));
		TVectorRenderData rd(affine, gl->getBounds(), palette, 0, true, true);

		tglDraw(rd, region);

		glDisable(GL_ALPHA_TEST);
		glPopAttrib();

		glFinish();
	}
	return rect;
}

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

void fastAddPaintRegion(const TRasterImageP &ri, TRegion *region,
						int newPaintId, int maxStyleId, TRectD clip = TRectD())
{
	TRaster32P ras = ri->getRaster();
	TOfflineGL *gl;
	TRect rect = rasterizeRegion(gl, ras->getBounds(), region, ri->getPalette(), clip);
	if (rect.isEmpty())
		return;

	TRaster32P glRas = gl->getRaster();
	assert(TPixelCM32::getMaxTone() == 255);
	TRop::over(ri->getRaster(), glRas, rect.getP00());
	delete gl;

	TRegion *subregion;
	UINT i = 0;
	for (; i < region->getSubregionCount(); ++i) {
		subregion = region->getSubregion(i);
		fastAddPaintRegion(ri, subregion, tmin(maxStyleId, subregion->getStyle()), maxStyleId);
	}
}
}

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

TRect TRasterImageUtils::addStroke(const TRasterImageP &ri, TStroke *stroke, TRectD clip,
								   double opacity, bool doAntialiasing)
{
	TStroke *s = new TStroke(*stroke);
	TPoint riCenter = ri->getRaster()->getCenter();
	s->transform(TTranslation(riCenter.x, riCenter.y));
	TRect rect = fastAddInkStroke(ri, s, clip, opacity, doAntialiasing);
	rect -= riCenter;
	delete s;
	return rect;
}

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

TRect TRasterImageUtils::convertWorldToRaster(const TRectD &area, const TRasterImageP ri)
{
	if (area.isEmpty())
		return TRect();
	if (!ri || !ri->getRaster())
		return TRect(tfloor(area.x0), tfloor(area.y0), tfloor(area.x1) - 1, tfloor(area.y1) - 1);
	TRasterP ras = ri->getRaster();
	TRectD rect(area + ras->getCenterD());
	return TRect(tfloor(rect.x0), tfloor(rect.y0), tceil(rect.x1) - 1, tceil(rect.y1) - 1);
}

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

TRectD TRasterImageUtils::convertRasterToWorld(const TRect &area, const TRasterImageP ri)
{
	if (area.isEmpty())
		return TRectD();

	TRectD rect(area.x0, area.y0, area.x1 + 1, area.y1 + 1);
	if (ri && ri->getRaster())
		rect = rect - ri->getRaster()->getCenterD();
	return rect;
}

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

// DA RIFARE
// e' lenta da far schifo

//!Converts a TVectorImage into a TRasterImage. The input vector image
//!is transformed through the passed affine \b aff, and put into a
//!TRasterImage strictly covering the bounding box of the transformed
//!vector image. The output image has its lower-left position in the
//!world reference specified by the \b pos parameter, which is granted to
//!be an integer displacement of the passed value. Additional parameters
//!include an integer \b enlarge by which the output image is enlarged with
//!respect to the transformed image's bbox, and the bool \b transformThickness
//!to specify whether the transformation should involve strokes' thickensses
//!or not.
TRasterImageP TRasterImageUtils::vectorToFullColorImage(
	const TVectorImageP &vimage, const TAffine &aff, TPalette *palette,
	const TPointD &outputPos, const TDimension &outputSize,
	const std::vector<TRasterFxRenderDataP> *fxs, bool transformThickness)
{
	if (!vimage || !palette)
		return 0;

	//Transform the vector image through aff
	TVectorImageP vi = vimage->clone();
	vi->transform(aff, transformThickness);

	//Allocate the output ToonzImage
	TRaster32P raster(outputSize.lx, outputSize.ly);
	raster->clear();
	TRasterImageP ri(raster);
	ri->setPalette(palette->clone());

	//Shift outputPos to the origin
	vi->transform(TTranslation(-outputPos));

	int strokeCount = vi->getStrokeCount();
	std::vector<int> strokeIndex(strokeCount);
	std::vector<TStroke *> strokes(strokeCount);
	int i;
	for (i = 0; i < strokeCount; ++i) {
		strokeIndex[i] = i;
		strokes[i] = vi->getStroke(i);
	}
	vi->notifyChangedStrokes(strokeIndex, strokes);

	int maxStyleId = palette->getStyleCount() - 1;
	for (i = 0; i < (int)vi->getRegionCount(); ++i) {
		TRegion *region = vi->getRegion(i);
		fastAddPaintRegion(ri, region, tmin(maxStyleId, region->getStyle()), maxStyleId);
	}

	set<int> colors;
	if (fxs) {
		for (i = 0; i < (int)fxs->size(); i++) {
			SandorFxRenderData *sandorData = dynamic_cast<SandorFxRenderData *>((*fxs)[i].getPointer());
			if (sandorData && sandorData->m_type == BlendTz) {
				std::string indexes = toString(sandorData->m_blendParams.m_colorIndex);
				std::vector<std::string> items;
				parseIndexes(indexes, items);
				PaletteFilterFxRenderData paletteFilterData;
				insertIndexes(items, &paletteFilterData);
				colors = paletteFilterData.m_colors;
				break;
			}
		}
	}

	for (i = 0; i < strokeCount; ++i) {
		TStroke *stroke = vi->getStroke(i);

		bool visible = false;
		int styleId = stroke->getStyle();
		TColorStyleP style = palette->getStyle(styleId);
		assert(style);
		int colorCount = style->getColorParamCount();
		if (colorCount == 0)
			visible = true;
		else {
			visible = false;
			for (int j = 0; j < style->getColorParamCount() && !visible; j++) {
				TPixel32 color = style->getColorParamValue(j);
				if (color.m != 0)
					visible = true;
			}
		}
		if (visible)
			fastAddInkStroke(ri, stroke, TRectD(), 1, true);
	}
	return ri;
}

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

TRect TRasterImageUtils::eraseRect(const TRasterImageP &ri, const TRectD &area)
{
	TRasterP ras = ri->getRaster();
	TRect rect = convertWorldToRaster(area, ri) * ras->getBounds();
	if (rect.isEmpty())
		return rect;
	ras->lock();
	TRasterP workRas = ras->extract(rect);
	if (workRas->getPixelSize() == 4)
		workRas->clear();
	else {
		TRasterGR8P rasGR8(workRas);
		if (rasGR8)
			rasGR8->fill(TPixelGR8::White);
	}
	ras->unlock();
	return rect;
}

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

std::vector<TRect> TRasterImageUtils::paste(const TRasterImageP &ri, const TTileSetFullColor *tileSet)
{
	std::vector<TRect> rects;
	TRasterP raster = ri->getRaster();
	for (int i = 0; i < tileSet->getTileCount(); i++) {
		const TTileSetFullColor::Tile *tile = tileSet->getTile(i);
		TRasterP ras;
		tile->getRaster(ras);
		assert(ras);
		raster->copy(ras, tile->m_rasterBounds.getP00());
		rects.push_back(tile->m_rasterBounds);
	}
	return rects;
}
//-------------------------------------------------------------------

void TRasterImageUtils::addSceneNumbering(const TRasterImageP &ri, int globalIndex,
										  const std::wstring &sceneName, int sceneIndex)
{
	if (!ri)
		return;
	TRasterP raster = ri->getRaster();
	int lx = raster->getLx(), ly = raster->getLy();
	QColor greyOverlay(100, 100, 100, 140);
	QImage image = rasterToQImage(raster, true, false);
	QPainter p(&image);
	QFont numberingFont = QFont();
	numberingFont.setPixelSize(ly * 0.04);
	numberingFont.setBold(true);
	p.setFont(numberingFont);
	QMatrix matrix;
	p.setMatrix(matrix.translate(0, ly).scale(1, -1), true);
	QFontMetrics fm = p.fontMetrics();
	int fontHeight = fm.height();
	int offset = fontHeight * 0.2;

	//write the scenename and the scene frame
	QString sceneFrame = QString::number(sceneIndex);
	while (sceneFrame.size() < 4)
		sceneFrame.push_front("0");
	QString sceneNumberingString = QString::fromStdWString(sceneName) + ": " + sceneFrame;

	int sceneNumberingWidth = fm.width(sceneNumberingString);
	p.setPen(Qt::NoPen);
	p.setBrush(QColor(255, 255, 255, 255));
	p.drawRect(offset, ly - offset - fontHeight, sceneNumberingWidth + offset * 2, fontHeight);
	p.setBrush(greyOverlay);
	p.drawRect(offset, ly - offset - fontHeight, sceneNumberingWidth + offset * 2, fontHeight);
	p.setPen(Qt::white);
	p.drawText(2 * offset, ly - 2 * offset, sceneNumberingString);

	//write the global frame
	QString globalFrame = QString::number(globalIndex);
	while (globalFrame.size() < 4)
		globalFrame.push_front("0");

	int gloablNumberingWidth = fm.width(globalFrame);
	p.setPen(Qt::NoPen);
	p.setBrush(QColor(255, 255, 255, 255));
	p.drawRect(lx - 3 * offset - gloablNumberingWidth, ly - offset - fontHeight, gloablNumberingWidth + offset * 2, fontHeight);
	p.setBrush(greyOverlay);
	p.drawRect(lx - 3 * offset - gloablNumberingWidth, ly - offset - fontHeight, gloablNumberingWidth + offset * 2, fontHeight);
	p.setPen(Qt::white);
	p.drawText(lx - 2 * offset - gloablNumberingWidth, ly - 2 * offset, globalFrame);
	p.end();
}

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

void TRasterImageUtils::addGlobalNumbering(const TRasterImageP &ri, const std::wstring &sceneName, int globalIndex)
{
	if (!ri)
		return;
	TRasterP raster = ri->getRaster();
	int lx = raster->getLx(), ly = raster->getLy();
	QColor greyOverlay(100, 100, 100, 140);
	QImage image = rasterToQImage(raster, true, false);
	QPainter p(&image);
	QFont numberingFont = QFont();
	numberingFont.setPixelSize(ly * 0.04);
	numberingFont.setBold(true);
	p.setFont(numberingFont);
	QMatrix matrix;
	p.setMatrix(matrix.translate(0, ly).scale(1, -1), true);
	QFontMetrics fm = p.fontMetrics();
	int fontHeight = fm.height();
	int offset = fontHeight * 0.2;
	QString globalFrame = QString::number(globalIndex);
	while (globalFrame.size() < 4)
		globalFrame.push_front("0");
	QString globalNumberingString = QString::fromStdWString(sceneName) + ": " + globalFrame;

	int globalNumberingWidth = fm.width(globalNumberingString);
	p.setPen(Qt::NoPen);
	p.setBrush(QColor(255, 255, 255, 255));
	p.drawRect(offset, ly - offset - fontHeight, globalNumberingWidth + offset * 2, fontHeight);
	p.setBrush(greyOverlay);
	p.drawRect(offset, ly - offset - fontHeight, globalNumberingWidth + offset * 2, fontHeight);
	p.setPen(Qt::white);
	p.drawText(2 * offset, ly - 2 * offset, globalNumberingString);
	p.end();
}