// TnzCore includes
#include "tgeometry.h"
#include "tstream.h"
#include "texception.h"

#include "toonz/vectorizerparameters.h"

//**************************************************************************
//    Local namespace  strings
//**************************************************************************

namespace
{

const char *s_version = "version",

		   *s_generalConfiguration = "generalConfiguration",
		   *s_outline = "outline",
		   *s_threshold = "threshold",
		   *s_leaveUnpainted = "leaveUnpainted",
		   *s_visibilityBits = "visibilityBits",

		   *s_Centerline = "Centerline",
		   *s_despeckling = "despeckling",
		   *s_maxThickness = "maxThickness",
		   *s_penalty = "penalty",
		   *s_thicknessRatio = "thicknessRatio",
		   *s_makeFrame = "makeFrame",
		   *s_naaSource = "naaSource",

		   *s_accuracy = "accuracy",
		   *s_thicknessRatioFirst = "thicknessRatioFirst",
		   *s_thicknessRatioLast = "thicknessRatioLast",
		   *s_paintFill = "paintFill",

		   *s_Outline = "Outline",
		   *s_adherenceTol = "adherenceTol",
		   *s_angleTol = "angleTol",
		   *s_relativeTol = "relativeTol",
		   *s_mergeTol = "mergeTol",
		   *s_maxColors = "maxColors",
		   *s_transparentColor = "transparentColor",
		   *s_toneTol = "toneTol",

		   *s_adherence = "adherence",
		   *s_angle = "angle",
		   *s_relative = "relative",
		   *s_toneThreshold = "toneThreshold";

} // namespace

//**************************************************************************
//    Local namespace  stuff
//**************************************************************************

namespace
{

const VersionNumber l_versionNumber(71, 0);

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

void saveData(const VectorizerConfiguration &conf, TOStream &os)
{
	int leaveUnpainted = conf.m_leaveUnpainted ? 1 : 0;

	os.child(s_threshold) << conf.m_threshold;
	os.child(s_leaveUnpainted) << leaveUnpainted;
}

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

void loadData(VectorizerConfiguration &conf, TIStream &is)
{
	string tagName;
	while (is.matchTag(tagName)) {
		if (tagName == s_threshold)
			is >> conf.m_threshold, is.matchEndTag();
		else if (tagName == s_leaveUnpainted) {
			int leaveUnpainted;
			is >> leaveUnpainted, conf.m_leaveUnpainted = (leaveUnpainted == 0) ? false : true;

			is.matchEndTag();
		} else
			is.skipCurrentTag();
	}
}

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

void saveData(const CenterlineConfiguration &conf, TOStream &os)
{
	os.openChild(s_generalConfiguration);
	saveData(static_cast<const VectorizerConfiguration &>(conf), os);
	os.closeChild();

	int makeFrame = conf.m_makeFrame ? 1 : 0;

	os.child(s_despeckling) << conf.m_despeckling;
	os.child(s_maxThickness) << conf.m_maxThickness;
	os.child(s_penalty) << conf.m_penalty;
	os.child(s_thicknessRatio) << conf.m_thicknessRatio;
	os.child(s_makeFrame) << makeFrame;
	os.child(s_naaSource) << (conf.m_naaSource ? 1 : 0);
}

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

void loadData(CenterlineConfiguration &conf, TIStream &is)
{
	string tagName;
	while (is.matchTag(tagName)) {
		if (tagName == s_generalConfiguration)
			loadData(static_cast<VectorizerConfiguration &>(conf), is), is.matchEndTag();
		else if (tagName == s_despeckling)
			is >> conf.m_despeckling, is.matchEndTag();
		else if (tagName == s_maxThickness)
			is >> conf.m_maxThickness, is.matchEndTag();
		else if (tagName == s_penalty)
			is >> conf.m_penalty, is.matchEndTag();
		else if (tagName == s_thicknessRatio)
			is >> conf.m_thicknessRatio, is.matchEndTag();
		else if (tagName == s_makeFrame) {
			int makeFrame;
			is >> makeFrame, conf.m_makeFrame = (makeFrame == 0) ? false : true;
			is.matchEndTag();
		} else if (tagName == s_naaSource) {
			int naaSource;
			is >> naaSource, conf.m_naaSource = (naaSource == 0) ? false : true;
			is.matchEndTag();
		} else
			is.skipCurrentTag();
	}
}

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

void saveData(const NewOutlineConfiguration &conf, TOStream &os)
{
	os.openChild(s_generalConfiguration);
	saveData(static_cast<const VectorizerConfiguration &>(conf), os);
	os.closeChild();

	os.child(s_despeckling) << conf.m_despeckling;
	os.child(s_adherenceTol) << conf.m_adherenceTol;
	os.child(s_angleTol) << conf.m_angleTol;
	os.child(s_relativeTol) << conf.m_relativeTol;
	os.child(s_mergeTol) << conf.m_mergeTol;
	os.child(s_maxColors) << conf.m_maxColors;
	os.child(s_transparentColor) << conf.m_transparentColor;
	os.child(s_toneTol) << conf.m_toneTol;
}

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

void loadData(NewOutlineConfiguration &conf, TIStream &is)
{
	string tagName;
	while (is.matchTag(tagName)) {
		if (tagName == s_generalConfiguration)
			loadData(static_cast<VectorizerConfiguration &>(conf), is), is.matchEndTag();
		else if (tagName == s_despeckling)
			is >> conf.m_despeckling, is.matchEndTag();
		else if (tagName == s_adherenceTol)
			is >> conf.m_adherenceTol, is.matchEndTag();
		else if (tagName == s_angleTol)
			is >> conf.m_angleTol, is.matchEndTag();
		else if (tagName == s_relativeTol)
			is >> conf.m_relativeTol, is.matchEndTag();
		else if (tagName == s_mergeTol)
			is >> conf.m_mergeTol, is.matchEndTag();
		else if (tagName == s_maxColors)
			is >> conf.m_maxColors, is.matchEndTag();
		else if (tagName == s_transparentColor)
			is >> conf.m_transparentColor, is.matchEndTag();
		else if (tagName == s_toneTol)
			is >> conf.m_toneTol, is.matchEndTag();
		else
			is.skipCurrentTag();
	}
}

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

void convert(const CenterlineConfiguration &conf, VectorizerParameters &params)
{
	params.m_cThreshold = conf.m_threshold / 25.0;
	params.m_cAccuracy = 10 - conf.m_penalty;
	params.m_cDespeckling = conf.m_despeckling / 2.0;
	params.m_cMaxThickness = conf.m_maxThickness * 2;
	//params.m_cThicknessRatio        = centConf.m_thicknessRatio;
	params.m_cThicknessRatioFirst = conf.m_thicknessRatio;
	params.m_cThicknessRatioLast = conf.m_thicknessRatio;
	params.m_cMakeFrame = conf.m_makeFrame;
	params.m_cPaintFill = !conf.m_leaveUnpainted;
	params.m_cNaaSource = conf.m_naaSource;
}

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

void convert(const NewOutlineConfiguration &conf, VectorizerParameters &params)
{
	params.m_oDespeckling = conf.m_despeckling;
	params.m_oAccuracy = tround((5.0 - conf.m_mergeTol) * 2.0);
	params.m_oAdherence = tround(conf.m_adherenceTol * 100.0);
	params.m_oAngle = tround(conf.m_angleTol * 180.0);
	params.m_oRelative = tround(conf.m_relativeTol * 100.0);
	params.m_oMaxColors = conf.m_maxColors;
	params.m_oToneThreshold = conf.m_toneTol;
	params.m_oTransparentColor = conf.m_transparentColor;
	params.m_oPaintFill = !conf.m_leaveUnpainted;
}

} // namespace

//**************************************************************************
//    VectorizerConfiguration  implementation
//**************************************************************************

PERSIST_IDENTIFIER(VectorizerParameters, "vectorizerParameters")

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

VectorizerParameters::VectorizerParameters()
	: m_visibilityBits(-1) // All visible by default
	  ,
	  m_isOutline(false)
{
	CenterlineConfiguration cConf;
	convert(cConf, *this);

	NewOutlineConfiguration oConf;
	convert(oConf, *this);
}

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

CenterlineConfiguration VectorizerParameters::getCenterlineConfiguration(double frame) const
{
	CenterlineConfiguration conf;

	conf.m_outline = false;
	conf.m_threshold = m_cThreshold * 25;
	conf.m_penalty = 10 - m_cAccuracy; // m_cAccuracy in [1,10]
	conf.m_despeckling = m_cDespeckling * 2;
	conf.m_maxThickness = m_cMaxThickness / 2.0;
	//conf.m_thicknessRatio   = m_cThicknessRatio;
	conf.m_thicknessRatio = (1 - frame) * m_cThicknessRatioFirst + frame * m_cThicknessRatioLast;
	conf.m_leaveUnpainted = !m_cPaintFill;
	conf.m_makeFrame = m_cMakeFrame;
	conf.m_naaSource = m_cNaaSource;

	return conf;
}

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

NewOutlineConfiguration VectorizerParameters::getOutlineConfiguration(double frame) const
{
	NewOutlineConfiguration conf;

	conf.m_outline = true;
	conf.m_despeckling = m_oDespeckling;
	conf.m_adherenceTol = m_oAdherence * 0.01;
	conf.m_angleTol = m_oAngle / 180.0;
	conf.m_relativeTol = m_oRelative * 0.01;
	conf.m_mergeTol = 5.0 - m_oAccuracy * 0.5;
	conf.m_leaveUnpainted = !m_oPaintFill;
	conf.m_maxColors = m_oMaxColors;
	conf.m_transparentColor = m_oTransparentColor;
	conf.m_toneTol = m_oToneThreshold;

	return conf;
}

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

void VectorizerParameters::saveData(TOStream &os)
{
	os.child(s_version) << l_versionNumber.first << l_versionNumber.second;
	os.child(s_outline) << (m_isOutline ? 1 : 0);
	os.child(s_visibilityBits) << int(m_visibilityBits);

	os.openChild(s_Centerline);
	{
		os.child(s_threshold) << m_cThreshold;
		os.child(s_accuracy) << m_cAccuracy;
		os.child(s_despeckling) << m_cDespeckling;
		os.child(s_maxThickness) << m_cMaxThickness;
		os.child(s_thicknessRatioFirst) << m_cThicknessRatioFirst;
		os.child(s_thicknessRatioLast) << m_cThicknessRatioLast;
		os.child(s_makeFrame) << (m_cMakeFrame ? 1 : 0);
		os.child(s_paintFill) << (m_cPaintFill ? 1 : 0);
		os.child(s_naaSource) << (m_cNaaSource ? 1 : 0);
	}
	os.closeChild();

	os.openChild(s_Outline);
	{
		os.child(s_despeckling) << m_oDespeckling;
		os.child(s_accuracy) << m_oAccuracy;
		os.child(s_adherence) << m_oAdherence;
		os.child(s_angle) << m_oAngle;
		os.child(s_relative) << m_oRelative;
		os.child(s_maxColors) << m_oMaxColors;
		os.child(s_toneThreshold) << m_oToneThreshold;
		os.child(s_transparentColor) << m_oTransparentColor;
		os.child(s_paintFill) << (m_oPaintFill ? 1 : 0);
	}
	os.closeChild();
}

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

void VectorizerParameters::loadData(TIStream &is)
{
	VersionNumber version;

	std::string tagName;
	int val;

	while (is.matchTag(tagName)) {
		if (tagName == s_version)
			is >> version.first >> version.second, is.matchEndTag();
		else if (tagName == s_outline)
			is >> val, m_isOutline = (val != 0), is.matchEndTag();
		else if (tagName == s_visibilityBits) {
			is >> val, is.matchEndTag();

			if (version == l_versionNumber) // Restore visibility bits only if the parameters
				m_visibilityBits = val;		// version is current - defaulting them is no big deal.
		} else if (tagName == s_Centerline) {
			while (is.matchTag(tagName)) {
				if (tagName == s_threshold)
					is >> m_cThreshold, is.matchEndTag();
				else if (tagName == s_accuracy)
					is >> m_cAccuracy, is.matchEndTag();
				else if (tagName == s_despeckling)
					is >> m_cDespeckling, is.matchEndTag();
				else if (tagName == s_maxThickness)
					is >> m_cMaxThickness, is.matchEndTag();
				else if (tagName == s_thicknessRatioFirst)
					is >> m_cThicknessRatioFirst, is.matchEndTag();
				else if (tagName == s_thicknessRatioLast)
					is >> m_cThicknessRatioLast, is.matchEndTag();
				else if (tagName == s_makeFrame)
					is >> val, m_cMakeFrame = (val != 0), is.matchEndTag();
				else if (tagName == s_paintFill)
					is >> val, m_cPaintFill = (val != 0), is.matchEndTag();
				else if (tagName == s_naaSource)
					is >> val, m_cNaaSource = (val != 0), is.matchEndTag();
				else
					is.skipCurrentTag();
			}

			is.matchEndTag();
		} else if (tagName == s_Outline) {
			while (is.matchTag(tagName)) {
				if (tagName == s_despeckling)
					is >> m_oDespeckling, is.matchEndTag();
				else if (tagName == s_accuracy)
					is >> m_oAccuracy, is.matchEndTag();
				else if (tagName == s_adherence)
					is >> m_oAdherence, is.matchEndTag();
				else if (tagName == s_angle)
					is >> m_oAngle, is.matchEndTag();
				else if (tagName == s_relative)
					is >> m_oRelative, is.matchEndTag();
				else if (tagName == s_maxColors)
					is >> m_oMaxColors, is.matchEndTag();
				else if (tagName == s_toneThreshold)
					is >> m_oToneThreshold, is.matchEndTag();
				else if (tagName == s_transparentColor)
					is >> m_oTransparentColor, is.matchEndTag();
				else if (tagName == s_paintFill)
					is >> val, m_oPaintFill = (val != 0), is.matchEndTag();
				else
					is.skipCurrentTag();
			}

			is.matchEndTag();
		} else if (tagName == "CenterlineConfiguration") // Old tags, not saved anymore
		{
			CenterlineConfiguration conf;
			::loadData(conf, is), is.matchEndTag();

			convert(conf, *this);
		} else if (tagName == "NewOutlineConfiguration") //
		{
			NewOutlineConfiguration conf;
			::loadData(conf, is), is.matchEndTag();

			convert(conf, *this);
		} else
			is.skipCurrentTag();
	}
}