// 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;
	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;
	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();
}