#include <tools assistant.h=""></tools>

#include <tools tool.h=""></tools>

#include <toonz tapplication.h=""></toonz>

#include <toonz txsheet.h=""></toonz>

#include <toonz txsheethandle.h=""></toonz>

#include <toonz txshlevelhandle.h=""></toonz>

#include <toonz tframehandle.h=""></toonz>

#include <toonz tobjecthandle.h=""></toonz>

#include <toonz dpiscale.h=""></toonz>

#include <tgl.h></tgl.h>

#include <tproperty.h></tproperty.h>

#include <limits></limits>

#include <cassert></cassert>

#ifdef MACOSX

const double TAssistantBase::lineWidthScale = 1.5;

#else

const double TAssistantBase::lineWidthScale = 1.0;

#endif

unsigned int TAssistantBase::drawFlags = 0;

//************************************************************************

//    TGuideline implementation

//************************************************************************

void

TGuideline::drawSegment(

  const TPointD &p0,

  const TPointD &p1,

  double pixelSize,

  bool active,

  bool enabled ) const

{

  double colorBlack[4] = { 0.0, 0.0, 0.0, 0.25 };

  double colorWhite[4] = { 1.0, 1.0, 1.0, 0.25 };

  if (!this->enabled || !enabled)

    colorBlack[3] = (colorWhite[3] = 0.125);

  else if (active)

    colorBlack[3] = (colorWhite[3] = 0.5);

  glPushAttrib(GL_ALL_ATTRIB_BITS);

  tglEnableBlending();

  tglEnableLineSmooth(true, 1.0 * TAssistant::lineWidthScale);

  TPointD d = p1 - p0;

  double k = norm2(d);

  if (k > TConsts::epsilon*TConsts::epsilon) {

    k = 0.5*pixelSize*TAssistant::lineWidthScale/sqrt(k);

    d = TPointD(-k*d.y, k*d.x);

    glColor4dv(colorWhite);

    tglDrawSegment(p0 - d, p1 - d);

    glColor4dv(colorBlack);

    tglDrawSegment(p0 + d, p1 + d);

  }

  glPopAttrib();

}

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

double

TGuideline::calcTrackWeight(const TTrack &track, const TAffine &toScreen, bool &outLongEnough) const {

  outLongEnough = false;

  if (!enabled || track.size() < 2)

    return std::numeric_limits<double>::infinity();</double>

  const double snapLenght = 20.0;

  const double snapScale = 1.0;

  const double maxLength = 2.0*snapLenght*snapScale;

  double sumWeight = 0.0;

  double sumLength = 0.0;

  double sumDeviation = 0.0;

  TPointD prev = toScreen*track[0].position;

  for(int i = 1; i < track.size(); ++i) {

    const TTrackPoint &tp = track[i];

    TTrackPoint mid = TTrack::interpolationLinear(track[i-1], track[i], 0.5);

    TPointD p = toScreen*tp.position;

    double length = tdistance(p, prev);

    sumLength += length;

    double midStepLength = sumLength - 0.5*length;

    if (midStepLength > TConsts::epsilon) {

      double weight = length*logNormalDistribuitionUnscaled(midStepLength, snapLenght, snapScale);

      sumWeight += weight;

      double deviation = tdistance(

        toScreen*mid.position,

        toScreen*nearestPoint(mid.position) );

      sumDeviation += weight*deviation;

    }

    prev = p;

    if (sumLength >= maxLength)

      { outLongEnough = i < track.fixedSize(); break; }

  }

  return sumWeight > TConsts::epsilon

       ? sumDeviation/sumWeight

       : std::numeric_limits<double>::infinity();</double>

}

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

TGuidelineP

TGuideline::findBest(const TGuidelineList &guidelines, const TTrack &track, const TAffine &toScreen, bool &outLongEnough) {

  outLongEnough = true;

  double bestWeight = 0.0;

  TGuidelineP best;

  for(TGuidelineList::const_iterator i = guidelines.begin(); i != guidelines.end(); ++i) {

    double weight = (*i)->calcTrackWeight(track, toScreen, outLongEnough);

    if (!best || weight < bestWeight)

      { bestWeight = weight; best = *i; }

  }

  return best;

}

//************************************************************************

//    TAssistantPoint implementation

//************************************************************************

TAssistantPoint::TAssistantPoint(const TStringId &name, const TPointD &defPosition):

  name(name),

  defPosition(defPosition),

  type(Circle),

  position(defPosition),

  radius(10.0),

  visible(true),

  selected() { }

//************************************************************************

//    TAssistantType implementation

//************************************************************************

TMetaObjectHandler*

TAssistantType::createHandler(TMetaObject &obj) const

  { return createAssistant(obj); }

//************************************************************************

//    TAssistantBase implementation

//************************************************************************

TAssistantBase::TAssistantBase(TMetaObject &object):

  TMetaObjectHandler(object),

  m_idEnabled("enabled"),

  m_idPoints("points"),

  m_idX("x"),

  m_idY("y"),

  m_basePoint()

{

  addProperty( new TBoolProperty(m_idEnabled.str(), getEnabled()) );

}

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

TAssistantPoint&

TAssistantBase::addPoint(

  const TStringId &name,

  TAssistantPoint::Type type,

  const TPointD &defPosition,

  bool visible,

  double radius )

{

  assert(!m_points.count(name));

  TAssistantPoint &p = m_points.insert(

    TAssistantPointMap::value_type(name, TAssistantPoint(name, defPosition)) ).first->second;

  m_pointsOrder.push_back(&p);

  p.type     = type;

  p.radius   = radius;

  p.visible  = visible;

  if (!m_basePoint) m_basePoint = &p;

  return p;

}

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

TAssistantPoint&

TAssistantBase::addPoint(

  const TStringId &name,

  TAssistantPoint::Type type,

  const TPointD &defPosition,

  bool visible )

    { return addPoint(name, type, defPosition, visible, 10.0); }

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

const TAssistantPoint&

TAssistantBase::getBasePoint() const

  { assert(m_basePoint); return *m_basePoint; }

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

void

TAssistantBase::addProperty(TProperty *p)

  { m_properties.add(p); }

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

void

TAssistantBase::setTranslation(const TStringId &name, const QString &localName) const

  { m_properties.getProperty(name)->setQStringName( localName ); }

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

void

TAssistantBase::updateTranslation() const

  { setTranslation(m_idEnabled, tr("Enabled")); }

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

void

TAssistantBase::onSetDefaults() {

  setEnabled(true);

  for(TAssistantPointMap::iterator i = m_points.begin(); i != m_points.end(); ++i)

    i->second.position = i->second.defPosition;

  fixPoints();

  fixData();

}

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

void

TAssistantBase::fixPoints()

  { onFixPoints(); }

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

bool

TAssistantBase::move(const TPointD &position) {

  TPointD d = position - getBasePoint().position;

  if (d != TPointD()) {

    for(TAssistantPointMap::iterator i = m_points.begin(); i != m_points.end(); ++i)

      i->second.position += d;

    fixPoints();

    return true;

  }

  return false;

}

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

bool

TAssistantBase::movePoint(const TStringId &name, const TPointD &position) {

  TAssistantPointMap::iterator i = m_points.find(name);

  if (i != m_points.end() && i->second.position != position) {

    onMovePoint(i->second, position);

    return true;

  }

  return false;

}

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

void

TAssistantBase::setPointSelection(const TStringId &name, bool selected)  const {

  if (const TAssistantPoint *p = findPoint(name))

    p->selected = selected;

}

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

void

TAssistantBase::setAllPointsSelection(bool selected) const {

  for(TAssistantPointMap::const_iterator i = points().begin(); i != points().end(); ++i)

    i->second.selected = selected;

}

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

void

TAssistantBase::onDataChanged(const TVariant &value) {

  const TVariant& pointsData = data()[m_idPoints];

  TVariantPathEntry entry;

  if (&value == &data() || &value == &pointsData)

    onAllDataChanged();

  else

  if (pointsData.getChildPathEntry(value, entry) && entry.isField()) {

    const TVariant& pointData = pointsData[entry];

    TPointD position = TPointD(

      pointData[m_idX].getDouble(),

      pointData[m_idY].getDouble() );

    movePoint(entry.field(), position);

  } else

  if (data().getChildPathEntry(value, entry) && entry.isField()) {

    onDataFieldChanged(entry.field(), data()[entry.field()]);

  }

}

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

void

TAssistantBase::onDataFieldChanged(const TStringId &name, const TVariant &value)

  { updateProperty(name, value); }

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

void

TAssistantBase::onAllDataChanged() {

  const TVariant& pointsData = data()[m_idPoints];

  for(TAssistantPointMap::iterator i = m_points.begin(); i != m_points.end(); ++i) {

    const TVariant& pointData = pointsData[i->first];

    i->second.position = TPointD(

      pointData[m_idX].getDouble(),

      pointData[m_idY].getDouble() );

  }

  fixPoints();

  updateProperties();

}

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

void

TAssistantBase::onFixPoints()

  { }

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

void

TAssistantBase::onMovePoint(TAssistantPoint &point, const TPointD &position)

  { point.position = position; }

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

void

TAssistantBase::onFixData() {

  TVariant& pointsData = data()[m_idPoints];

  for(TAssistantPointMap::const_iterator i = points().begin(); i != points().end(); ++i) {

    TVariant& pointData = pointsData[i->first];

    pointData[m_idX].setDouble( i->second.position.x );

    pointData[m_idY].setDouble( i->second.position.y );

  }

}

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

void

TAssistantBase::updateProperties() {

  const TVariantMap &map = data().getMap();

  for(TVariantMap::const_iterator i = map.begin(); i != map.end(); ++i)

    if (i->first != m_idPoints)

      updateProperty(i->first, i->second);

}

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

void

TAssistantBase::updateProperty(const TStringId &name, const TVariant &value) {

  TProperty *property = m_properties.getProperty(name);

  if (!property)

    return;

  if (TBoolProperty *boolProperty = dynamic_cast<tboolproperty*>(property)) {</tboolproperty*>

    boolProperty->setValue( value.getBool() );

  } else

  if (TDoubleProperty *doubleProperty = dynamic_cast<tdoubleproperty*>(property)) {</tdoubleproperty*>

    doubleProperty->setValue( value.getDouble() );

  } else

  if (TIntProperty *intProperty = dynamic_cast<tintproperty*>(property)) {</tintproperty*>

    intProperty->setValue( (int)value.getDouble() );

  } else

  if (TStringProperty *stringProperty = dynamic_cast<tstringproperty*>(property)) {</tstringproperty*>

    stringProperty->setValue( to_wstring(value.getString()) );

  } else

  if (TEnumProperty *enumProperty = dynamic_cast<tenumproperty*>(property)) {</tenumproperty*>

    enumProperty->setValue( to_wstring(value.getString()) );

  }

}

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

void

TAssistantBase::onPropertyChanged(const TStringId &name) {

  TProperty *property = m_properties.getProperty(name);

  if (!property)

    return;

  if (name == m_idPoints)

    return;

  if (TBoolProperty *boolProperty = dynamic_cast<tboolproperty*>(property)) {</tboolproperty*>

    data()[name].setBool( boolProperty->getValue() );

  } else

  if (TDoubleProperty *doubleProperty = dynamic_cast<tdoubleproperty*>(property)) {</tdoubleproperty*>

    data()[name].setDouble( doubleProperty->getValue() );

  } else

  if (TIntProperty *intProperty = dynamic_cast<tintproperty*>(property)) {</tintproperty*>

    data()[name].setDouble( (double)intProperty->getValue() );

  } else

  if (TStringProperty *stringProperty = dynamic_cast<tstringproperty*>(property)) {</tstringproperty*>

    data()[name].setString( to_string(stringProperty->getValue()) );

  } else

  if (TEnumProperty *enumProperty = dynamic_cast<tenumproperty*>(property)) {</tenumproperty*>

    data()[name].setString( to_string(enumProperty->getValue()) );

  }

}

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

double

TAssistantBase::getDrawingAlpha(bool enabled) const

  { return enabled && this->getEnabled() ? 0.5 : 0.25; }

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

double

TAssistantBase::getDrawingGridAlpha() const

  { return 0.2; }

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

void

TAssistantBase::drawSegment(const TPointD &p0, const TPointD &p1, double pixelSize, double alpha) const {

  double colorBlack[4] = { 0.0, 0.0, 0.0, alpha };

  double colorWhite[4] = { 1.0, 1.0, 1.0, alpha };

  if (drawFlags & DRAW_ERROR) colorBlack[0] = 1;

  glPushAttrib(GL_ALL_ATTRIB_BITS);

  tglEnableBlending();

  tglEnableLineSmooth(true, 1.0 * lineWidthScale);

  TPointD d = p1 - p0;

  double k = norm2(d);

  if (k > TConsts::epsilon*TConsts::epsilon) {

    k = 0.5*pixelSize*lineWidthScale/sqrt(k);

    d = TPointD(-k*d.y, k*d.x);

    glColor4dv(colorWhite);

    tglDrawSegment(p0 - d, p1 - d);

    glColor4dv(colorBlack);

    tglDrawSegment(p0 + d, p1 + d);

  }

  glPopAttrib();

}

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

void

TAssistantBase::drawMark(const TPointD &p, const TPointD &normal, double pixelSize, double alpha) const {

  TPointD d = normal*5*pixelSize;

  drawSegment(p - d,p + d, pixelSize, alpha);

}

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

void

TAssistantBase::drawDot(const TPointD &p, double alpha) const {

  double colorBlack[4] = { 0.0, 0.0, 0.0, alpha };

  double colorWhite[4] = { 1.0, 1.0, 1.0, alpha };

  glPushAttrib(GL_ALL_ATTRIB_BITS);

  tglEnableBlending();

  glColor4dv(colorWhite);

  tglEnablePointSmooth(6.0);

  glBegin(GL_POINTS);

  glVertex2d(p.x, p.y);

  glEnd();

  glColor4dv(colorBlack);

  tglEnablePointSmooth(3.0);

  glBegin(GL_POINTS);

  glVertex2d(p.x, p.y);

  glEnd();

  glPopAttrib();

}

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

void

TAssistantBase::drawPoint(const TAssistantPoint &point, double pixelSize) const {

  if (!point.visible) return;

  double radius = point.radius;

  double crossSize = 1.2*radius;

  double alpha = 0.5;

  double colorBlack[4] = { 0.0, 0.0, 0.0, alpha };

  double colorGray[4]  = { 0.5, 0.5, 0.5, alpha };

  double colorWhite[4] = { 1.0, 1.0, 1.0, alpha };

  double width = 1.5;

  if (point.selected) {

    colorBlack[2] = 1.0;

    colorGray[2] = 1.0;

    width = 2.0;

  }

  glPushAttrib(GL_ALL_ATTRIB_BITS);

  // fill

  tglEnableBlending();

  if (point.type == TAssistantPoint::CircleFill) {

    glColor4dv(colorGray);

    tglDrawDisk(point.position, radius*pixelSize);

  }

  TPointD crossDx(pixelSize*crossSize, 0.0);

  TPointD crossDy(0.0, pixelSize*crossSize);

  TPointD gridDx(pixelSize*radius, 0.0);

  TPointD gridDy(0.0, pixelSize*radius);

  // back line

  tglEnableLineSmooth(true, 2.0*width*lineWidthScale);

  glColor4dv(colorWhite);

  if (point.type == TAssistantPoint::CircleCross) {

    tglDrawSegment(point.position - crossDx, point.position + crossDx);

    tglDrawSegment(point.position - crossDy, point.position + crossDy);

  }

  tglDrawCircle(point.position, radius*pixelSize);

  // front line

  glLineWidth(width * lineWidthScale);

  glColor4dv(colorBlack);

  if (point.type == TAssistantPoint::CircleCross) {

    tglDrawSegment(point.position - crossDx, point.position + crossDx);

    tglDrawSegment(point.position - crossDy, point.position + crossDy);

  }

  tglDrawCircle(point.position, radius*pixelSize);

  // dots

  switch(point.type) {

  case TAssistantPoint::CircleDoubleDots:

    drawDot(point.position - gridDx*0.5, alpha);

    drawDot(point.position + gridDx*0.5, alpha);

    drawDot(point.position - gridDy*0.5, alpha);

    drawDot(point.position + gridDy*0.5, alpha);

    //no break

  case TAssistantPoint::CircleDots:

    drawDot(point.position - gridDx, alpha);

    drawDot(point.position + gridDx, alpha);

    drawDot(point.position - gridDy, alpha);

    drawDot(point.position + gridDy, alpha);

    //no break

  case TAssistantPoint::Circle:

    drawDot(point.position, alpha);

    break;

  default:

    break;

  }

  glPopAttrib();

}

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

void

TAssistantBase::drawIndex(const TPointD &p, int index, bool selected, double pixelSize) const {

  static const int segments[7][4] = {

    { 0, 2, 1, 2 },   // A

    { 1, 1, 1, 2 },   // B    + A +

    { 1, 0, 1, 1 },   // C    F   B

    { 0, 0, 1, 0 },   // D    + G +

    { 0, 0, 0, 1 },   // E    E   C

    { 0, 1, 0, 2 },   // F    + D +

    { 0, 1, 1, 1 } }; // G

  static const int glyphs[][7] = {

  // A B C D E F G

    {1,1,1,1,1,1,0},   // 0

    {0,1,1,0,0,0,0},   // 1

    {1,1,0,1,1,0,1},   // 2

    {1,1,1,1,0,0,1},   // 3

    {0,1,1,0,0,1,1},   // 4

    {1,0,1,1,0,1,1},   // 5

    {1,0,1,1,1,1,1},   // 6

    {1,1,1,0,0,1,0},   // 7

    {1,1,1,1,1,1,1},   // 8

    {1,1,1,1,0,1,1} }; // 9

  if (index < 0) index = 0;

  int len = 0;

  int digits[16] = {};

  for(int i = index; i; i /= 10)

    digits[len++] = i%10;

  if (!len) len = 1;

  double w = 5, h = 5, d = 0.5, dx = w+2;

  double alpha = 0.5;

  double colorBlack[4] = { 0.0, 0.0, 0.0, alpha };

  double colorWhite[4] = { 1.0, 1.0, 1.0, alpha };

  if (selected) colorBlack[2] = 1.0;

  glPushAttrib(GL_ALL_ATTRIB_BITS);

  tglEnableBlending();

  tglEnableLineSmooth(true, 1.0 * lineWidthScale);

  double k = 0.5*pixelSize*lineWidthScale;

  double y = p.y;

  for(int i = 0; i < len; ++i) {

    double x = p.x + dx*(len-i-1);

    const int *g = glyphs[digits[i]];

    for(int i = 0; i < 7; ++i) {

      if (!g[i]) continue;

      const int *s = segments[i];

      if (s[0] == s[2]) {

        // vertical

        glColor4dv(colorWhite);

        tglDrawSegment(

          TPointD(x + s[0]*w + k, y + s[1]*h + d),

          TPointD(x + s[2]*w + k, y + s[3]*h - d) );

        glColor4dv(colorBlack);

        tglDrawSegment(

          TPointD(x + s[0]*w - k, y + s[1]*h + d),

          TPointD(x + s[2]*w - k, y + s[3]*h - d) );

      } else {

        // horisontal

        glColor4dv(colorWhite);

        tglDrawSegment(

          TPointD(x + s[0]*w + d, y + s[1]*h + k),

          TPointD(x + s[2]*w - d, y + s[3]*h + k) );

        glColor4dv(colorBlack);

        tglDrawSegment(

          TPointD(x + s[0]*w + d, y + s[1]*h - k),

          TPointD(x + s[2]*w - d, y + s[3]*h - k) );

      }

    }

  }

  glPopAttrib();

}

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

void

TAssistantBase::draw(TToolViewer*, bool) const

  { }

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

void

TAssistantBase::drawEdit(TToolViewer *viewer) const {

  // paint all points

  draw(viewer);

  double pixelSize = sqrt(tglGetPixelSize2());

  for(TAssistantPointMap::const_iterator i = points().begin(); i != points().end(); ++i)

    drawPoint(i->second, pixelSize);

}

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

void

TAssistantBase::drawEdit(TToolViewer *viewer, int index) const {

  drawEdit(viewer);

  drawIndex(

    getBasePoint().position + TPointD(8, 8),

    index, getBasePoint().selected, sqrt(tglGetPixelSize2()) );

}

//************************************************************************

//    TAssistant implementation

//************************************************************************

TAssistant::TAssistant(TMetaObject &object):

  TAssistantBase(object),

  m_idMagnetism("magnetism")

  { addProperty( new TDoubleProperty(m_idMagnetism.str(), 0.0, 1.0, getMagnetism()) ); }

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

void

TAssistant::updateTranslation() const {

  TAssistantBase::updateTranslation();

  setTranslation(m_idMagnetism, tr("Magnetism"));

}

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

void

TAssistant::onSetDefaults() {

  setMagnetism(1.0);

  TAssistantBase::onSetDefaults();

}

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

void

TAssistant::onFixData() {

  TAssistantBase::onFixData();

  setMagnetism( std::max(0.0, std::min(1.0, getMagnetism())) );

}

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

void

TAssistant::getGuidelines(const TPointD&, const TAffine&, TGuidelineList&) const

  { }

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

bool

TAssistant::calcPerspectiveStep(

  double minStep,

  double minX,

  double maxX,

  double x0,

  double x1,

  double x2,

  double &outK,

  double &outMin,

  double &outMax )

{

  outK = outMin = outMax = 0.0;

  double dx1 = x1 - x0;