#include "toonz/tpinnedrangeset.h"
#include "tstream.h"

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

TPinnedRangeSet::TPinnedRangeSet()
	: m_stageObject()
{
}

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

int TPinnedRangeSet::getRangeIndex(int frame) const
{
	if (m_ranges.empty())
		return -1;
	if (frame < m_ranges.front().first || m_ranges.back().second < frame)
		return -1;
	int start = 0, end = (int)m_ranges.size() - 1;
	while (start < end) {
		if (start + 1 == end) {
			if (m_ranges[end].first <= frame)
				start = end;
			else
				end = start;
			break;
		}
		int m = (start + end) / 2;
		assert(start < m && m < end);
		if (m_ranges[m].first <= frame)
			start = m;
		else
			end = m;
	}
	int index = start;
	if (m_ranges[index].first <= frame && frame <= m_ranges[index].second)
		return index;
	else
		return -1;
}

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

void TPinnedRangeSet::setRange(int first, int second)
{
	std::vector<Range>::iterator it;
	// finds the range which could possibly be merged before : range.second+1 >= first
	for (it = m_ranges.begin(); it != m_ranges.end(); ++it)
		if (it->second + 1 >= first)
			break;

	if (it == m_ranges.end()) {
		// not found => for every range : range.second+1 < first. Just add the new one
		m_ranges.push_back(Range(first, second));
	} else if (it->first > second + 1) {
		// it comes after and can not be merged. Just insert the new one
		m_ranges.insert(it, Range(first, second));
	} else {
		// merge
		if (first < it->first)
			it->first = first;
		// update it->second, and find all the ranges which should be deleted
		std::vector<Range>::iterator it2 = it;
		for (++it2; it2 != m_ranges.end(); ++it2)
			if (it2->first > second + 1)
				break;
			else
				it->second = it2->second;
		if (second > it->second)
			it->second = second;
		++it;
		if (it != it2)
			m_ranges.erase(it, it2);
	}
}

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

void TPinnedRangeSet::removeRange(int first, int second)
{
	std::vector<Range>::iterator it;
	// finds the first range which could possibly overlap [first,second] : range.second>=first
	for (it = m_ranges.begin(); it != m_ranges.end(); ++it)
		if (it->second >= first)
			break;

	if (it == m_ranges.end() || it->first > second) {
		// nothing to do: for each range range.second < first or range.first > second
	} else if (it->first < first && it->second > second) {
		// [first,second] is a "hole"
		Range range(it->first, first - 1);
		it->first = second + 1;
		m_ranges.insert(it, range);
	} else {
		std::vector<Range>::iterator it2;
		if (it->first < first) {
			it->second = first - 1;
			++it;
		}
		it2 = it;
		while (it != m_ranges.end() && it->second <= second)
			++it;
		if (it != m_ranges.end() && it->first <= second)
			it->first = second + 1;
		if (it2 != it)
			m_ranges.erase(it2, it);
	}
}

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

void TPinnedRangeSet::removeAllRanges()
{
	m_ranges.clear();
}

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

void TPinnedRangeSet::setPlacement(const TAffine &placement)
{
	m_placement = placement;
}

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

TPinnedRangeSet *TPinnedRangeSet::clone()
{
	TPinnedRangeSet *clonedPinnedRangeSet = new TPinnedRangeSet();
	clonedPinnedRangeSet->m_stageObject = m_stageObject;
	clonedPinnedRangeSet->m_placement = m_placement;
	clonedPinnedRangeSet->m_ranges = m_ranges;
	return clonedPinnedRangeSet;
}

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

void TPinnedRangeSet::loadData(TIStream &is)
{
	m_ranges.clear();
	int prevFrame = 0;
	int i = 0;
	std::string tagName;
	int count = 0;
	while (is.matchTag(tagName) && count < 3) {
		if (tagName == "permanent") {
			while (!is.matchEndTag()) {
				i++;
				int frame = 0;
				is >> frame;
				if (i % 2 == 0)
					m_ranges.push_back(Range(prevFrame, frame));
				prevFrame = frame;
			}
			count++;
		} else if (tagName == "temp") {
			assert(0);
			// OBSOLETO
			while (!is.matchEndTag()) {
				int frame = 0;
				is >> frame;
				//m_tempPinned.push_back(frame);
			}
			count++;
		} else if (tagName == "lockedAngle") {
			assert(0);
			// OBSOLETO
			while (!is.matchEndTag()) {
				int rangeIndex = -1;
				is >> rangeIndex;
			}

			count++;
		} else if (tagName == "placement") {
			is >> m_placement.a11 >> m_placement.a12 >> m_placement.a13;
			is >> m_placement.a21 >> m_placement.a22 >> m_placement.a23;
			is.matchEndTag();
		}
	}
}

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

void TPinnedRangeSet::saveData(TOStream &os)
{
	if ((int)m_ranges.size() == 0)
		return;
	os.openChild("pinnedStatus");
	if (m_ranges.size() > 0) {
		os.openChild("permanent");
		for (int p = 0; p < (int)m_ranges.size(); p++) {
			os << m_ranges[p].first << m_ranges[p].second;
		}
		os.closeChild();
	}
	if (m_placement != TAffine()) {
		os.openChild("placement");
		os << m_placement.a11 << m_placement.a12 << m_placement.a13;
		os << m_placement.a21 << m_placement.a22 << m_placement.a23;
		os.closeChild();
	}
	os.closeChild();
}