#include "toonz/txshnoteset.h"

#include "tstream.h"

#include "texception.h"

#include <qstring></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>");</note>

          is.closeChild();

        }

      } else

        throw TException("expected <defaultcolor> or <notes>");</notes></defaultcolor>

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

}