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

	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();</vectorizerconfiguration>

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

	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();</vectorizerconfiguration>

		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 ¶ms)

{

	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 ¶ms)

{

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

	}

}