// 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 "toonz/hook.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();

}