#include "toonz/txshpalettelevel.h"

#include "toonz/txshleveltypes.h"

#include "toonz/toonzscene.h"

#include "tconvert.h"

#include "tstream.h"

#include "tfilepath_io.h"

#include "tsystem.h"

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

DEFINE_CLASS_CODE(TXshPaletteLevel, 52)

PERSIST_IDENTIFIER(TXshPaletteLevel, "paletteLevel")

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

// TXshPaletteLevel

TXshPaletteLevel::TXshPaletteLevel(std::wstring name)

    : TXshLevel(m_classCode, name), m_palette(0) {

  m_type = PLT_XSHLEVEL;

}

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

TXshPaletteLevel::~TXshPaletteLevel() {}

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

TPalette *TXshPaletteLevel::getPalette() const { return m_palette; }

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

void TXshPaletteLevel::setPalette(TPalette *palette) {

  if (m_palette != palette) {

    if (m_palette) m_palette->release();

    m_palette = palette;

    if (m_palette) m_palette->addRef();

  }

}

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

void TXshPaletteLevel::setPath(const TFilePath &path) { m_path = path; }

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

void TXshPaletteLevel::loadData(TIStream &is) {

  std::string tagName;

  while (is.matchTag(tagName)) {

    if (tagName == "name") {

      std::wstring name;

      is >> name;

      setName(name);

    } else if (tagName == "path") {

      is >> m_path;

    } else {

      throw TException("TXshPaletteLevel, unknown tag: " + tagName);

    }

    is.closeChild();

  }

}

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

void TXshPaletteLevel::saveData(TOStream &os) {

  os.child("path") << m_path;

  os.child("name") << getName();

}

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

void TXshPaletteLevel::load() {

  TFilePath path = getScene()->decodeFilePath(m_path);

  if (TSystem::doesExistFileOrLevel(path)) {

    TFileStatus fs(path);

    TPersist *p = 0;

    TIStream is(path);

    TPalette *palette = nullptr;

    if (is && fs.doesExist()) {

      std::string tagName;

      if (is.matchTag(tagName) && tagName == "palette") {

        std::string gname;

        is.getTagParam("name", gname);

        palette = new TPalette();

        palette->loadData(is);

        palette->setGlobalName(::to_wstring(gname));

        is.matchEndTag();

        palette->setPaletteName(path.getWideName());

        setPalette(palette);

      }

    }

    assert(m_palette);

  }

}

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

void TXshPaletteLevel::save() {

  TFilePath path = getScene()->decodeFilePath(m_path);

  if (TSystem::doesExistFileOrLevel(path) && m_palette) {

    TOStream os(path);

    os << m_palette;

  }

}

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

int TXshPaletteLevel::getFrameCount() const { return 0; }