// 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_alignBoundaryStrokesDirection = "alignBoundaryStrokesDirection",

           *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) {

  std::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) {

  std::string tagName;

  while (is.matchTag(tagName)) {

    if (tagName == s_generalConfiguration)

      loadData(static_cast<vectorizerconfiguration &="">(conf), is),</vectorizerconfiguration>

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

  std::string tagName;

  while (is.matchTag(tagName)) {

    if (tagName == s_generalConfiguration)

      loadData(static_cast<vectorizerconfiguration &="">(conf), is),</vectorizerconfiguration>

          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 ¶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_cAlignBoundaryStrokesDirection =

      conf.m_alignBoundaryStrokesDirection;

  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;

  params.m_oAlignBoundaryStrokesDirection =

      conf.m_alignBoundaryStrokesDirection;

}

}  // 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_alignBoundaryStrokesDirection = m_cAlignBoundaryStrokesDirection;

  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_alignBoundaryStrokesDirection = m_oAlignBoundaryStrokesDirection;

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

        << (m_cAlignBoundaryStrokesDirection ? 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.child(s_alignBoundaryStrokesDirection)

        << (m_oAlignBoundaryStrokesDirection ? 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_alignBoundaryStrokesDirection)

          is >> val, m_cAlignBoundaryStrokesDirection = (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 if (tagName == s_alignBoundaryStrokesDirection)

          is >> val, m_oAlignBoundaryStrokesDirection = (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();

  }

}