// TnzCore includes

#include "tfilepath.h"

// TnzLib includes

#include "toonz/txshsimplelevel.h"

#include "toonz/preferences.h"

#include "toonz/onionskinmask.h"

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

//    Macros

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

#define MINFADE 0.35

#define MAXFADE 0.95

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

//    Local namespace

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

namespace {

double inline getIncrement(int paperThickness) {

  struct locals {

    inline static void fillIncrements(double *values, int a, int b) {

      double slope = (values[b] - values[a]) / (b - a);

      for (int i = a + 1; i < b; ++i) values[i] = values[i - 1] + slope;

    }

  };  // locals

  static double Incr[101] = {-1.0};

  if (Incr[0] == -1.0) {

    Incr[0]   = 0.0;

    Incr[10]  = 0.05;

    Incr[50]  = 0.12;

    Incr[90]  = 0.3;

    Incr[100] = MAXFADE - MINFADE;

    locals::fillIncrements(Incr, 0, 10);

    locals::fillIncrements(Incr, 10, 50);

    locals::fillIncrements(Incr, 50, 90);

    locals::fillIncrements(Incr, 90, 100);

  }

  return Incr[paperThickness];

}

}  // namespace

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

//    OnionSkinMask  implementation

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

void OnionSkinMask::clear() {

  m_fos.clear();

  m_mos.clear();

  m_shiftTraceStatus = DISABLED;

  m_ghostAff[0]    = TAffine();

  m_ghostAff[1]    = TAffine();

  m_ghostCenter[0] = TPointD();

  m_ghostCenter[1] = TPointD();

}

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

void OnionSkinMask::getAll(int currentRow, std::vector<int> &output) const {</int>

  output.clear();

  output.reserve(m_fos.size() + m_mos.size());

  std::vector<int>::const_iterator fosIt, fosEnd(m_fos.end());</int>

  std::vector<int>::const_iterator mosIt, mosEnd(m_mos.end());</int>

  for (fosIt = m_fos.begin(), mosIt = m_mos.begin();

       fosIt != fosEnd && mosIt != mosEnd;) {

    int fos = *fosIt, mos = *mosIt + currentRow;

    if (fos < mos) {

      if (fos != currentRow) output.push_back(fos);

      ++fosIt;

    } else {

      if (mos != currentRow) output.push_back(mos);

      ++mosIt;

    }

  }

  for (; fosIt != fosEnd; ++fosIt)

    if (*fosIt != currentRow) output.push_back(*fosIt);

  for (; mosIt != mosEnd; ++mosIt) {

    int mos = *mosIt + currentRow;

    if (mos != currentRow) output.push_back(mos);

  }

}

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

void OnionSkinMask::setMos(int drow, bool on) {

  assert(drow != 0);

  typedef std::vector<int>::iterator Iter;</int>

  std::pair<iter, iter=""> r = std::equal_range(m_mos.begin(), m_mos.end(), drow);</iter,>

  if (on) {

    if (r.first == r.second) m_mos.insert(r.first, drow);

  } else {

    if (r.first != r.second) m_mos.erase(r.first, r.second);

  }

}

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

void OnionSkinMask::setFos(int row, bool on) {

  typedef std::vector<int>::iterator Iter;</int>

  std::pair<iter, iter=""> r = std::equal_range(m_fos.begin(), m_fos.end(), row);</iter,>

  if (on) {

    if (r.first == r.second) m_fos.insert(r.first, row);

  } else {

    if (r.first != r.second) m_fos.erase(r.first, r.second);

  }

}

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

bool OnionSkinMask::isFos(int row) const {

  return std::binary_search(m_fos.begin(), m_fos.end(), row);

}

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

bool OnionSkinMask::isMos(int drow) const {

  return std::binary_search(m_mos.begin(), m_mos.end(), drow);

}

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

bool OnionSkinMask::getMosRange(int &drow0, int &drow1) const {

  if (m_mos.empty()) {

    drow0 = 0, drow1 = -1;

    return false;

  } else {

    drow0 = m_mos.front(), drow1 = m_mos.back();

    return true;

  }

}

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

double OnionSkinMask::getOnionSkinFade(int rowsDistance) {

  if (rowsDistance == 0) return 0.9;

  double fade =

      MINFADE +

      abs(rowsDistance) *

          getIncrement(Preferences::instance()->getOnionPaperThickness());

  return tcrop(fade, MINFADE, MAXFADE);

}

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

void OnionSkinMask::setShiftTraceGhostAff(int index, const TAffine &aff) {

  assert(0 <= index && index < 2);

  m_ghostAff[index] = aff;

}

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

void OnionSkinMask::setShiftTraceGhostCenter(int index, const TPointD ¢er) {

  assert(0 <= index && index < 2);

  m_ghostCenter[index] = center;

}

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

//    OnionSkinMaskModifier  implementation

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

OnionSkinMaskModifier::OnionSkinMaskModifier(OnionSkinMask mask, int currentRow)

    : m_oldMask(mask)

    , m_curMask(mask)

    , m_firstRow(0)

    , m_lastRow(0)

    , m_curRow(currentRow)

    , m_status(0) {}

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

void OnionSkinMaskModifier::click(int row, bool isFos) {

  assert(m_status == 0);

  m_firstRow = m_lastRow = row;

  if (isFos) {

    assert(row != m_curRow);

    if (m_curMask.isEnabled() && m_curMask.isFos(row)) {

      m_status = 2;  // spegnere fos

      m_curMask.setFos(row, false);

    } else {

      if (!m_curMask.isEnabled()) {

        m_curMask.clear();

        m_curMask.enable(true);

      }

      m_curMask.setFos(row, true);

      m_status = 3;  // accendere fos

    }

  } else {

    int drow = row - m_curRow;

    if (drow != 0 && m_curMask.isEnabled() && m_curMask.isMos(drow)) {

      m_status = 4;  // spegnere mos

      m_curMask.setMos(drow, false);

    } else if (drow == 0) {

      m_status = 8 + 4 + 1;  // accendere mos; partito da 0

    } else {

      if (!m_curMask.isEnabled()) m_curMask.enable(true);

      m_curMask.setMos(drow, true);

      m_status = 4 + 1;  // accendere mos;

    }

  }

}

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

void OnionSkinMaskModifier::drag(int row) {

  if (m_status & 128) return;

  if (row == m_lastRow) return;

  m_status |= 64;  // moved

  int n = row - m_lastRow, d = 1;

  if (n < 0) n = -n, d = -d;

  int oldr = m_lastRow, r = oldr + d;

  for (int i = 0; i < n; ++i, r += d) {

    if (m_status & 4) {

      if (!m_curMask.isEnabled()) {

        m_curMask.clear();

        m_curMask.enable(true);

      }

      if (r != m_curRow) m_curMask.setMos(r - m_curRow, (m_status & 1) != 0);

    } else

      m_curMask.setFos(r, (m_status & 1) != 0);

  }

  m_lastRow = row;

}

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

void OnionSkinMaskModifier::release(int row) {

  if (m_status & 128) return;

  if ((m_status & 64) == 0    // non si e' mosso

      && (m_status & 8) == 8  // e' partito da zero

      && row == m_curRow) {

    if (!m_curMask.isEmpty() && m_curMask.isEnabled())

      m_curMask.enable(false);

    else {

      m_curMask.enable(true);

      if (m_curMask.isEmpty()) {

        m_curMask.setMos(-1, true);

        m_curMask.setMos(-2, true);

        m_curMask.setMos(-3, true);

      }

    }

  }

}