#include "toonz/txshnoteset.h"
#include "tstream.h"
#include "texception.h"

#include <QString>

//=============================================================================
// TXshNoteSet

TXshNoteSet::TXshNoteSet()
{
}

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

int TXshNoteSet::addNote(Note note)
{
	m_notes.push_back(note);
	return getCount() - 1;
}

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

void TXshNoteSet::removeNote(int index)
{
	if (m_notes.empty() || index >= (int)m_notes.size())
		return;
	m_notes.removeAt(index);
}

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

int TXshNoteSet::getCount() const
{
	if (m_notes.empty())
		return 0;
	return m_notes.size();
}

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

int TXshNoteSet::getNoteColorIndex(int noteIndex) const
{
	assert(noteIndex < m_notes.size());
	if (noteIndex >= m_notes.size())
		return 0;
	return m_notes[noteIndex].m_colorIndex;
}

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

void TXshNoteSet::setNoteColorIndex(int noteIndex, int colorIndex)
{
	assert(noteIndex < m_notes.size());
	if (noteIndex >= m_notes.size())
		return;
	m_notes[noteIndex].m_colorIndex = colorIndex;
}

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

QString TXshNoteSet::getNoteHtmlText(int noteIndex) const
{
	assert(noteIndex < m_notes.size());
	if (noteIndex >= m_notes.size())
		return QString();
	return m_notes[noteIndex].m_text;
}

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

void TXshNoteSet::setNoteHtmlText(int noteIndex, QString text)
{
	assert(noteIndex < m_notes.size());
	if (noteIndex >= m_notes.size())
		return;
	m_notes[noteIndex].m_text = text;
}

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

int TXshNoteSet::getNoteRow(int noteIndex) const
{
	assert(noteIndex < m_notes.size());
	if (noteIndex >= m_notes.size())
		return 0;
	return m_notes[noteIndex].m_row;
}

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

void TXshNoteSet::setNoteRow(int noteIndex, int row)
{
	assert(noteIndex < m_notes.size());
	if (noteIndex >= m_notes.size())
		return;
	m_notes[noteIndex].m_row = row;
}

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

int TXshNoteSet::getNoteCol(int noteIndex) const
{
	assert(noteIndex < m_notes.size());
	if (noteIndex >= m_notes.size())
		return 0;
	return m_notes[noteIndex].m_col;
}

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

void TXshNoteSet::setNoteCol(int noteIndex, int col)
{
	assert(noteIndex < m_notes.size());
	if (noteIndex >= m_notes.size())
		return;
	m_notes[noteIndex].m_col = col;
}

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

TPointD TXshNoteSet::getNotePos(int noteIndex) const
{
	assert(noteIndex < m_notes.size());
	if (noteIndex >= m_notes.size())
		return TPointD(5, 5);
	return m_notes[noteIndex].m_pos;
}

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

void TXshNoteSet::setNotePos(int noteIndex, TPointD pos)
{
	assert(noteIndex < m_notes.size());
	if (noteIndex >= m_notes.size())
		return;
	m_notes[noteIndex].m_pos = pos;
}

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

void TXshNoteSet::loadData(TIStream &is)
{
	while (!is.eos()) {
		std::string tagName;
		if (is.matchTag(tagName)) {
			if (tagName == "notes") {
				while (!is.eos()) {
					std::string tagName;
					if (is.matchTag(tagName)) {
						if (tagName == "note") {
							Note note;
							is >> note.m_colorIndex;
							std::wstring text;
							is >> text;
							note.m_text = QString::fromStdWString(text);
							is >> note.m_row;
							is >> note.m_col;
							is >> note.m_pos.x;
							is >> note.m_pos.y;
							m_notes.push_back(note);
						}
					} else
						throw TException("expected <note>");
					is.closeChild();
				}
			} else
				throw TException("expected <defaultColor> or <notes>");
			is.closeChild();
		} else
			throw TException("expected tag");
	}
}

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

void TXshNoteSet::saveData(TOStream &os)
{
	int i;
	os.openChild("notes");
	for (i = 0; i < getCount(); i++) {
		os.openChild("note");
		Note note = m_notes.at(i);
		os << note.m_colorIndex;
		os << note.m_text.toStdWString();
		os << note.m_row;
		os << note.m_col;
		os << note.m_pos.x;
		os << note.m_pos.y;
		os.closeChild();
	}
	os.closeChild();
}