#include "toonzqt/rasterimagedata.h"
#include "toonzqt/strokesdata.h"
#include "tpaletteutil.h"
#include "trasterimage.h"
#include "toonz/stage.h"
#include "toonz/tcenterlinevectorizer.h"
#include "toonz/ttileset.h"
#include "toonz/toonzimageutils.h"
#include "toonz/toonzscene.h"
#include "toonz/sceneproperties.h"
#include "toonz/fullcolorpalette.h"

#include <memory>

namespace
{
TVectorImageP vectorize(const TImageP &source, const TRectD &rect, const VectorizerConfiguration &config, TAffine transform)
{
	VectorizerCore vc;
	TVectorImageP vi = vc.vectorize(source.getPointer(), config, source->getPalette());
	assert(vi);
	vi->setPalette(source->getPalette());

	double dpiX, dpiY;
	TToonzImageP ti(source);
	TRasterImageP ri(source);
	if (ti)
		ti->getDpi(dpiX, dpiY);
	else if (ri)
		ri->getDpi(dpiX, dpiY);
	else
		return vi;

	TScale sc(dpiX / Stage::inch, dpiY / Stage::inch);
	TTranslation tr(rect.getP00());

	int i;
	for (i = 0; i < (int)vi->getStrokeCount(); i++) {
		TStroke *stroke = vi->getStroke(i);
		stroke->transform((sc.inv() * tr) * transform, true);
	}
	return vi;
}

} //namspace

//===================================================================
// RasterImageData
//-------------------------------------------------------------------

RasterImageData::RasterImageData()
	: m_dpiX(0), m_dpiY(0), m_transformation(), m_dim()
{
}

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

RasterImageData::~RasterImageData()
{
}

//===================================================================
// ToonzImageData
//-------------------------------------------------------------------

ToonzImageData::ToonzImageData()
	: RasterImageData(), m_copiedRaster(0), m_palette(new TPalette())
{
}

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

ToonzImageData::ToonzImageData(const ToonzImageData &src)
	: m_copiedRaster(src.m_copiedRaster), m_palette(src.m_palette), m_usedStyles(src.m_usedStyles)
{
	m_dpiX = src.m_dpiX;
	m_dpiY = src.m_dpiY;
	m_rects = src.m_rects;
	m_strokes = src.m_strokes;
	m_transformation = src.m_transformation;
	m_originalStrokes = src.m_originalStrokes;
	m_dim = src.m_dim;
	assert(m_palette);
}

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

ToonzImageData::~ToonzImageData()
{
}

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

void ToonzImageData::setData(const TRasterP &copiedRaster, const TPaletteP &palette, double dpiX, double dpiY, const TDimension &dim,
							 const vector<TRectD> &rects, const vector<TStroke> &strokes,
							 const vector<TStroke> &originalStrokes,
							 const TAffine &transformation)
{
	m_copiedRaster = copiedRaster;
	m_palette = palette;
	m_dpiX = dpiX;
	m_dpiY = dpiY;
	m_rects = rects;
	m_strokes = strokes;
	m_transformation = transformation;
	m_originalStrokes = originalStrokes;
	m_dim = dim;

	/*-- 使用されているStyleの一覧を作る --*/
	TToonzImageP ti(m_copiedRaster, m_copiedRaster->getBounds());
	ToonzImageUtils::getUsedStyles(m_usedStyles, ti);
}

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

void ToonzImageData::getData(TRasterP &copiedRaster, double &dpiX, double &dpiY,
							 vector<TRectD> &rects, vector<TStroke> &strokes, vector<TStroke> &originalStrokes,
							 TAffine &transformation, TPalette *targetPalette) const
{
	if (!m_copiedRaster || (m_rects.empty() && m_strokes.empty()))
		return;
	copiedRaster = m_copiedRaster->clone();
	dpiX = m_dpiX;
	dpiY = m_dpiY;
	assert(m_palette);
	int i;
	for (i = 0; i < (int)m_rects.size(); i++)
		rects.push_back(m_rects[i]);
	for (i = 0; i < (int)m_strokes.size(); i++)
		strokes.push_back(m_strokes[i]);
	for (i = 0; i < (int)m_originalStrokes.size(); i++)
		originalStrokes.push_back(m_originalStrokes[i]);

	transformation = m_transformation;
	TRasterCM32P cmRas = copiedRaster;

	if (!targetPalette)
		targetPalette = new TPalette();

	if (!cmRas)
		return;
	std::set<int> usedStyles(m_usedStyles);
	TToonzImageP ti(cmRas, cmRas->getBounds());
	if (usedStyles.size() == 0)
		ToonzImageUtils::getUsedStyles(usedStyles, ti);
	std::map<int, int> indexTable;
	mergePalette(targetPalette, indexTable, m_palette, usedStyles);
	ToonzImageUtils::scrambleStyles(ti, indexTable);
	ti->setPalette(m_palette.getPointer());
}

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

StrokesData *ToonzImageData::toStrokesData(ToonzScene *scene) const
{
	assert(scene);
	TRectD rect;
	if (!m_rects.empty())
		rect = m_rects[0];
	else if (!m_strokes.empty())
		rect = m_strokes[0].getBBox();
	unsigned int i;
	for (i = 0; i < m_rects.size(); i++)
		rect += m_rects[i];
	for (i = 0; i < m_strokes.size(); i++)
		rect += m_strokes[i].getBBox();
	TToonzImageP image(m_copiedRaster, m_copiedRaster->getBounds());
	image->setPalette(m_palette.getPointer());
	image->setDpi(m_dpiX, m_dpiY);

	const VectorizerParameters &vParams = *scene->getProperties()->getVectorizerParameters();

	CenterlineConfiguration cConf = vParams.getCenterlineConfiguration(0.0);
	NewOutlineConfiguration oConf = vParams.getOutlineConfiguration(0.0);

	const VectorizerConfiguration &config = vParams.m_isOutline ? static_cast<const VectorizerConfiguration &>(oConf) : static_cast<const VectorizerConfiguration &>(cConf);

	TVectorImageP vi = vectorize(image, rect, config, m_transformation);

	StrokesData *sd = new StrokesData();

	std::set<int> indexes;
	for (i = 0; i < vi->getStrokeCount(); i++)
		indexes.insert(i);

	sd->setImage(vi, indexes);
	return sd;
}

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

int ToonzImageData::getMemorySize() const
{
	int i, size = 0;
	for (i = 0; i < (int)m_strokes.size(); i++)
		size += m_strokes[i].getControlPointCount() * sizeof(TThickPoint) + 100;
	for (i = 0; i < (int)m_originalStrokes.size(); i++)
		size += m_originalStrokes[i].getControlPointCount() * sizeof(TThickPoint) + 100;
	return size + sizeof(*(m_copiedRaster.getPointer())) +
		   sizeof(*(m_palette.getPointer())) +
		   sizeof(*this);
}

//===================================================================
// FullColorImageData
//-------------------------------------------------------------------

FullColorImageData::FullColorImageData()
	: RasterImageData(), m_copiedRaster(0), m_palette(new TPalette())
{
}

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

FullColorImageData::FullColorImageData(const FullColorImageData &src)
	: m_copiedRaster(src.m_copiedRaster), m_palette(src.m_palette)
{
	m_dpiX = src.m_dpiX;
	m_dpiY = src.m_dpiY;
	m_rects = src.m_rects;
	m_strokes = src.m_strokes;
	m_transformation = src.m_transformation;
	m_originalStrokes = src.m_originalStrokes;
	m_dim = src.m_dim;
}

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

FullColorImageData::~FullColorImageData()
{
}

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

void FullColorImageData::setData(const TRasterP &copiedRaster, const TPaletteP &palette, double dpiX, double dpiY, const TDimension &dim,
								 const vector<TRectD> &rects, const vector<TStroke> &strokes,
								 const vector<TStroke> &originalStrokes,
								 const TAffine &transformation)
{
	m_copiedRaster = copiedRaster;
	m_palette = palette;
	m_dpiX = dpiX;
	m_dpiY = dpiY;
	m_rects = rects;
	m_strokes = strokes;
	m_originalStrokes = originalStrokes;
	m_transformation = transformation;
	m_dim = dim;
}

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

void FullColorImageData::getData(TRasterP &copiedRaster, double &dpiX, double &dpiY,
								 vector<TRectD> &rects, vector<TStroke> &strokes, vector<TStroke> &originalStrokes,
								 TAffine &transformation, TPalette *targetPalette) const
{
	if (!m_copiedRaster || (m_rects.empty() && m_strokes.empty()))
		return;
	copiedRaster = m_copiedRaster->clone();
	dpiX = m_dpiX;
	dpiY = m_dpiY;
	int i;
	for (i = 0; i < (int)m_rects.size(); i++)
		rects.push_back(m_rects[i]);
	for (i = 0; i < (int)m_strokes.size(); i++)
		strokes.push_back(m_strokes[i]);
	for (i = 0; i < (int)m_originalStrokes.size(); i++)
		originalStrokes.push_back(m_originalStrokes[i]);
	transformation = m_transformation;

	TRasterP ras = copiedRaster;
	if (!ras)
		return;
	if (!m_palette)
		return;
	if (!targetPalette)
		targetPalette = new TPalette();
	std::set<int> usedStyles;
	TRasterImageP ri(ras);

	for (i = 0; i < m_palette->getPageCount(); i++) {
		TPalette::Page *page = m_palette->getPage(i);
		int j;
		for (j = 0; j < page->getStyleCount(); j++)
			usedStyles.insert(page->getStyleId(j));
	}
	std::map<int, int> indexTable;
	mergePalette(targetPalette, indexTable, m_palette, usedStyles);
	ri->setPalette(m_palette.getPointer());
}

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

StrokesData *FullColorImageData::toStrokesData(ToonzScene *scene) const
{
	assert(scene);
	TRectD rect;
	if (!m_rects.empty())
		rect = m_rects[0];
	else if (!m_strokes.empty())
		rect = m_strokes[0].getBBox();
	unsigned int i;
	for (i = 0; i < m_rects.size(); i++)
		rect += m_rects[i];
	for (i = 0; i < m_strokes.size(); i++)
		rect += m_strokes[i].getBBox();
	TRasterImageP image(m_copiedRaster);
	image->setPalette(FullColorPalette::instance()->getPalette(scene));
	image->setDpi(m_dpiX, m_dpiY);

	const VectorizerParameters *vParams = scene->getProperties()->getVectorizerParameters();
	assert(vParams);

	std::auto_ptr<VectorizerConfiguration> config(vParams->getCurrentConfiguration(0.0));
	TVectorImageP vi = vectorize(image, rect, *config, m_transformation);

	StrokesData *sd = new StrokesData();

	std::set<int> indexes;
	for (i = 0; i < vi->getStrokeCount(); i++)
		indexes.insert(i);

	sd->setImage(vi, indexes);
	return sd;
}

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

int FullColorImageData::getMemorySize() const
{
	int i, size = 0;
	for (i = 0; i < (int)m_strokes.size(); i++)
		size += m_strokes[i].getControlPointCount() * sizeof(TThickPoint) + 100;
	for (i = 0; i < (int)m_originalStrokes.size(); i++)
		size += m_originalStrokes[i].getControlPointCount() * sizeof(TThickPoint) + 100;
	return size + sizeof(*(m_copiedRaster.getPointer())) +
		   sizeof(*(m_palette.getPointer())) +
		   sizeof(*this);
}