#include "toonz/hook.h"
#include "tstream.h"
#include "tfilepath.h"

#include "tconvert.h"
#include "toonz/txsheet.h"
#include "toonz/txshcell.h"

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

Hook::Hook()
	: m_id(0), m_trackerObjectId(-1)
{
}

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

bool Hook::isEmpty() const
{
	return m_frames.empty();
}

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

bool Hook::isKeyframe(const TFrameId &fid) const
{
	return m_frames.count(fid) > 0;
}

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

Hook::Frames::const_iterator Hook::find(TFrameId fid) const
{
	if (m_frames.empty())
		return m_frames.end();
	Frames::const_iterator it;
	it = m_frames.lower_bound(fid);
	if (it == m_frames.end()) {
		// dopo l'ultimo
		assert(it != m_frames.begin());
		--it;
	} else if (it->first == fid || it == m_frames.begin()) {
		// giusto o prima del primo
	} else
		--it;
	return it;
}

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

TPointD Hook::getPos(const TFrameId &fid) const
{
	//return TPointD(0,0);
	Frames::const_iterator it = find(fid);
	if (it == m_frames.end())
		return TPointD();
	else
		return it->second.m_pos;
}

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

TPointD Hook::getAPos(const TFrameId &fid) const
{
	Frames::const_iterator it = find(fid);
	if (it == m_frames.end())
		return TPointD();
	else if (it->first == fid)
		return it->second.m_aPos;
	else
		return it->second.m_bPos;
}

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

TPointD Hook::getBPos(const TFrameId &fid) const
{
	Frames::const_iterator it = find(fid);
	if (it == m_frames.end())
		return TPointD();
	else
		return it->second.m_bPos;
}

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

void Hook::setAPos(const TFrameId &fid, TPointD pos)
{
	Frames::iterator it = m_frames.lower_bound(fid);
	Frame f;
	if (it != m_frames.end() && it->first == fid) {
		f = it->second;
		if (f.m_aPos == f.m_bPos)
			f.m_aPos = f.m_bPos = pos;
		else if (tdistance2(pos, f.m_bPos) <= 1)
			f.m_aPos = f.m_bPos;
		else
			f.m_aPos = pos;
	} else {
		f.m_aPos = f.m_bPos = pos;
	}
	m_frames[fid] = f;
	update();
}

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

void Hook::setBPos(const TFrameId &fid, TPointD pos)
{
	Frames::iterator it = m_frames.lower_bound(fid);
	Frame f;
	if (it != m_frames.end() && it->first == fid) {
		f = it->second;
		if (tdistance2(pos, f.m_aPos) <= 1)
			f.m_bPos = f.m_aPos;
		else
			f.m_bPos = pos;
	} else {
		f.m_aPos = getAPos(fid);
		f.m_bPos = pos;
	}
	m_frames[fid] = f;
	update();
}

//---------------------------------------------------------
TRectD Hook::getTrackerRegion(const TFrameId &fid)
{
	TRectD rect;
	rect.x0 = getPos(fid).x - (getTrackerRegionWidth() * 0.5);
	rect.y0 = getPos(fid).y - (getTrackerRegionHeight() * 0.5);
	rect.x1 = getPos(fid).x + (getTrackerRegionWidth() * 0.5);
	rect.y1 = getPos(fid).y + (getTrackerRegionHeight() * 0.5);
	return rect;
}

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

void Hook::update()
{
	TPointD delta;
	for (Frames::iterator it = m_frames.begin();
		 it != m_frames.end(); ++it) {
		it->second.m_pos = it->second.m_aPos + delta;
		delta -= it->second.m_bPos - it->second.m_aPos;
	}
	m_delta = delta;
}

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

void Hook::renumber(const std::map<TFrameId, TFrameId> &renumberTable)
{
	Frames oldFrames = m_frames;
	m_frames.clear();
	for (Frames::iterator it = oldFrames.begin(); it != oldFrames.end(); ++it) {
		std::map<TFrameId, TFrameId>::const_iterator j = renumberTable.find(it->first);
		assert(j != renumberTable.end());
		if (j == renumberTable.end())
			continue;
		m_frames[j->second] = it->second;
	}
}

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

void Hook::eraseFrame(const TFrameId &fid)
{
	m_frames.erase(fid);
}

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

void Hook::saveData(TOStream &os)
{
	Frames::iterator j;
	for (j = m_frames.begin(); j != m_frames.end(); ++j) {
		os.openChild("frame");
		os << j->first.getNumber();
		os << j->second.m_aPos.x << j->second.m_aPos.y;
		os << j->second.m_bPos.x << j->second.m_bPos.y;
		os.closeChild();
	}
	if (m_trackerObjectId >= 0) {
		os.openChild("tracker");
		os << m_trackerObjectId << m_width << m_height;
		os.closeChild();
	}
}

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

void Hook::loadData(TIStream &is)
{
	m_frames.clear();
	std::string tagName;
	while (is.matchTag(tagName)) {
		if (tagName == "frame") {
			Frame f;
			int frameNumber = 0;
			is >> frameNumber;
			is >> f.m_aPos.x >> f.m_aPos.y;
			is >> f.m_bPos.x >> f.m_bPos.y;
			m_frames[TFrameId(frameNumber)] = f;
			m_trackerObjectId = -1;
		} else if (tagName == "tracker") {
			is >> m_trackerObjectId;
			is >> m_width;
			is >> m_height;
		} else
			throw TException("expected <frame>");
		is.matchEndTag();
	}
	update();
}

//=========================================================

HookSet::HookSet()
	: m_trackerObjectsSet(new TrackerObjectsSet)
{
}

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

HookSet::~HookSet()
{
	clearPointerContainer(m_hooks);
	delete m_trackerObjectsSet;
}

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

HookSet::HookSet(const HookSet &other)
	: m_hooks(other.m_hooks), m_trackerObjectsSet(new TrackerObjectsSet)
{
	int h, hCount = m_hooks.size();
	for (h = 0; h != hCount; ++h)
		if (m_hooks[h])
			m_hooks[h] = new Hook(*m_hooks[h]);
}

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

HookSet &HookSet::operator=(const HookSet &other)
{
	clearPointerContainer(m_hooks);
	m_hooks = other.m_hooks;

	int h, hCount = m_hooks.size();
	for (h = 0; h != hCount; ++h)
		if (m_hooks[h])
			m_hooks[h] = new Hook(*m_hooks[h]);

	return *this;
}

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

int HookSet::getHookCount() const
{
	return m_hooks.size();
}

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

Hook *HookSet::getHook(int index) const
{
	return 0 <= index && index < getHookCount() ? m_hooks[index] : 0;
}

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

Hook *HookSet::touchHook(int index)
{
	assert(0 <= index && index < maxHooksCount);
	if (index < 0 || index >= maxHooksCount)
		return 0;

	while (index >= (int)m_hooks.size())
		m_hooks.push_back(0);

	if (m_hooks[index] == 0) {
		Hook *hook = new Hook();
		m_hooks[index] = hook;
		hook->m_id = index;
	}

	return m_hooks[index];
}

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

Hook *HookSet::addHook()
{
	int h, hCount = m_hooks.size();
	for (h = 0; h != hCount; ++h) {
		if (m_hooks[h] == 0) {
			Hook *hook = new Hook();
			m_hooks[h] = hook;
			hook->m_id = h;

			return hook;
		} else if (m_hooks[h]->isEmpty())
			return m_hooks[h];
	}

	if (m_hooks.size() >= maxHooksCount)
		return 0;

	Hook *hook = new Hook();
	hook->m_id = m_hooks.size();
	m_hooks.push_back(hook);

	return hook;
}

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

void HookSet::clearHook(Hook *hook)
{
	for (int i = 0; i < (int)m_hooks.size(); i++)
		if (m_hooks[i] == hook)
			m_hooks[i] = 0;
	delete hook;
}

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

void HookSet::clearHooks()
{
	for (int i = 0; i < (int)m_hooks.size(); i++)
		delete m_hooks[i];
	m_hooks.clear();
}

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

TrackerObjectsSet *HookSet::getTrackerObjectsSet() const
{
	// Ripulisco l'insieme dei tracker objects
	m_trackerObjectsSet->clearAll();

	// costruisco il tracker object in base agli Hook
	for (int i = 0; i < getHookCount(); ++i) {
		Hook *hook = getHook(i);
		if (!hook || hook->isEmpty())
			continue;

		int trackerObjectId = hook->getTrackerObjectId();
		if (trackerObjectId >= 0) // se l'Hook ha anche una regione
		{
			TrackerObject *trackerObject = m_trackerObjectsSet->getObject(trackerObjectId);
			if (trackerObject == NULL) {
				trackerObject = new TrackerObject(trackerObjectId);
				m_trackerObjectsSet->addObject(trackerObject);
			}
			trackerObject = m_trackerObjectsSet->getObject(trackerObjectId);
			assert(trackerObject != NULL);
			trackerObject->addHook(hook);
		}
	}
	return m_trackerObjectsSet;
}

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

void HookSet::renumber(const std::map<TFrameId, TFrameId> &renumberTable)
{
	for (int i = 0; i < getHookCount(); i++)
		if (getHook(i))
			getHook(i)->renumber(renumberTable);
}

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

void HookSet::eraseFrame(const TFrameId &fid)
{
	for (int i = 0; i < getHookCount(); i++)
		if (getHook(i))
			getHook(i)->eraseFrame(fid);
}

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

void HookSet::saveData(TOStream &os)
{
	for (int i = 0; i < getHookCount(); ++i) {
		os.openChild("hook");
		Hook *hook = getHook(i);
		// Pay attention: it is necessary that empty hook will be however saved
		// in the xml file.
		if (hook)
			hook->saveData(os);
		os.closeChild();
	}
}

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

void HookSet::loadData(TIStream &is)
{
	std::string tagName;
	while (is.matchTag(tagName)) {
		if (tagName == "hook") {
			Hook *hook = new Hook();
			hook->m_id = m_hooks.size();
			hook->loadData(is);
			is.matchEndTag();
			m_hooks.push_back(hook);
		} else
			return;

		is.matchEndTag();
	}
}

//=========================================================
//
//	TRACKER
//

Hook *TrackerObject::getHook(int index)
{
	assert(index >= 0 && index < getHooksCount());
	return m_hooks.at(index);
}
void TrackerObject::addHook(Hook *hook)
{
	m_hooks.push_back(hook);
}

void TrackerObject::removeHook(int index)
{
	assert(index >= 0 && index < (int)m_hooks.size());
	m_hooks.erase(m_hooks.begin() + index);
}

TrackerObject *TrackerObjectsSet::getObject(int objectId)
{
	assert(objectId >= 0);
	std::map<int, TrackerObject *>::iterator it = m_trackerObjects.find(objectId);
	if (it != m_trackerObjects.end())
		return it->second;
	else
		return NULL;
}

TrackerObject *TrackerObjectsSet::getObjectFromIndex(int index)
{
	assert(index >= 0 && index < (int)m_trackerObjects.size());
	return m_trackerObjects[index];
}

// return objectIndex, return -1 if objectId doesn't exist
int TrackerObjectsSet::getIndexFromId(int objectId)
{
	int index = -1;
	int i = 0;
	for (i = 0; i < (int)m_trackerObjects.size(); i++) {
		int id = m_trackerObjects[i]->getId();
		if (id == objectId) {
			index = i;
			break;
		}
	}
	return index;
}

int TrackerObjectsSet::getIdFromIndex(int index)
{
	assert(index >= 0 && index < (int)m_trackerObjects.size());
	return m_trackerObjects[index]->getId();
}

// add new object, return new object Id
int TrackerObjectsSet::addObject()
{
	// assegno l'id
	int id;
	if (m_trackerObjects.size() > 0) {
		std::map<int, TrackerObject *>::iterator it = m_trackerObjects.end();
		--it;
		id = it->first + 1;
	} else
		id = 0;

	TrackerObject *trackerObject = new TrackerObject(id);
	m_trackerObjects[id] = trackerObject;
	return id;
}

void TrackerObjectsSet::addObject(TrackerObject *trackerObject)
{
	assert(trackerObject);
	int id = trackerObject->getId();
	assert(id >= 0);

	m_trackerObjects[id] = trackerObject;
}

void TrackerObjectsSet::removeObject(int objectId)
{
	assert(objectId >= 0);
	std::map<int, TrackerObject *>::iterator tt = m_trackerObjects.find(objectId);
	if (tt != m_trackerObjects.end()) {
		delete tt->second;
		m_trackerObjects.erase(tt);
	}
}

void TrackerObjectsSet::clearAll()
{
	std::map<int, TrackerObject *>::iterator tt, tEnd(m_trackerObjects.end());
	for (tt = m_trackerObjects.begin(); tt != tEnd; ++tt)
		delete tt->second;

	m_trackerObjects.clear();
}

//=========================================================

std::string getHookName(int code)
{
	assert(0 <= code && code < 10);
	if (code == 0)
		return "B";
	else
		return "H" + toString(code);
}