// TnzLib includes

#include "toonz/txsheet.h"

#include "toonz/txshcell.h"

#include "toonz/hook.h"

#include "toonz/dpiscale.h"

#include "toonz/txshsimplelevel.h"

#include "toonz/stage.h"

#include "xshhandlemanager.h"

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

//    Local namespace

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

namespace {

// Returns the sum of each pass hook between fid and precFid.

TPointD forwardPass(const TFrameId &fid, const TFrameId &precFid,

                    TXshSimpleLevel *sl, Hook *hook) {

  TPointD delta;

  std::vector<tframeid> levelFids;</tframeid>

  sl->getFids(levelFids);

  int f, fCount = levelFids.size();

  for (f = 0; f != fCount; ++f) {

    const TFrameId &levelFid = levelFids[f];

    if (levelFid < precFid) continue;

    if (levelFid >= fid) break;

    if (hook->isKeyframe(levelFid))

      delta += hook->getAPos(levelFid) - hook->getBPos(levelFid);

  }

  return delta;

}

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

// Returns subtraction of each pass hook between fid and precFid.

TPointD backwardPass(const TFrameId &fid, const TFrameId &precFid,

                     TXshSimpleLevel *sl, Hook *hook) {

  TPointD delta;

  std::vector<tframeid> levelFids;</tframeid>

  sl->getFids(levelFids);

  int f, fCount = levelFids.size();

  for (f = 0; f != fCount; ++f) {

    const TFrameId &levelFid = levelFids[f];

    if (levelFid < fid) continue;

    if (levelFid >= precFid) break;

    if (hook->isKeyframe(levelFid))

      delta -= hook->getAPos(levelFid) - hook->getBPos(levelFid);

  }

  return delta;

}

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

// Compute pass hook between fid and precFid

TPointD computePassHook(const TFrameId &fid, const TFrameId &precFid,

                        TXshSimpleLevel *sl, Hook *hook) {

  if (fid < precFid && (precFid.getNumber() - fid.getNumber()) <= 2)

    return backwardPass(fid, precFid, sl, hook);

  return forwardPass(fid, precFid, sl, hook);

}

}  // namespace

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

//    XshHandleManager  implementation

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

TPointD XshHandleManager::getHandlePos(const TStageObjectId &id,

                                       const std::string &handle,

                                       int row) const {

  static const double unit = 8.0;

  assert(m_xsh->getScene());

  if (handle == "")

    return TPointD();

  else if (handle[0] == 'H' && handle.length() > 1) {

    // Hook port case

    if (!id.isColumn()) return TPointD();

    int hookIndex = 0;

    {

      int h, hLength = handle.length();

      for (h = 1; h != hLength; ++h)

        hookIndex = hookIndex * 10 + (handle[h] - '0');

    }

    TStageObject *obj = m_xsh->getStageObject(id);

    if (const PlasticSkeletonDeformationP &def =

            obj->getPlasticSkeletonDeformation()) {

      int skelId = def->skeletonId(row);

      PlasticSkeleton skel;

      def->storeDeformedSkeleton(skelId, row, skel);

      int v = def->vertexIndex(hookIndex, skelId);

      return (v >= 0) ? TScale(1.0 / Stage::inch) * skel.vertex(v).P()

                      : TPointD();

    }

    --hookIndex;

    int col = id.getIndex();

    TXshCell cell(m_xsh->getCell(row, col));

    TXshLevel *xl = cell.m_level.getPointer();

    if (!xl) return TPointD();

    TXshSimpleLevel *sl = xl->getSimpleLevel();

    if (!sl) return TPointD();

    Hook *hook = sl->getHookSet()->getHook(hookIndex);

    if (!hook) return TPointD();

    TFrameId fid(cell.m_frameId);

    TPointD pos = hook->getAPos(fid) * (1.0 / Stage::inch);

    TPointD delta;

    for (int r = row - 1; r >= 0; --r) {

      cell = m_xsh->getCell(r, col);

      if (cell.m_level.getPointer() != xl) break;

      const TFrameId &precFid = cell.m_frameId;

      delta += computePassHook(fid, precFid, sl, hook);

      fid = precFid;

    }

    pos += delta * (1.0 / Stage::inch);

    pos = getDpiAffine(sl, cell.m_frameId, true) * pos;

    return pos;

  } else if ('A' <= handle[0] && handle[0] <= 'Z')

    return TPointD(unit * (handle[0] - 'B'), 0);

  else if ('a' <= handle[0] && handle[0] <= 'z')

    return TPointD(0.5 * unit * (handle[0] - 'b'), 0);

  else

    return TPointD();

}