#include "toonz/txshchildlevel.h"
#include "toonz/txshleveltypes.h"
#include "toonz/txsheet.h"
#include "toonz/imagemanager.h"
#include "toonz/toonzscene.h"
#include "toonz/txshcolumn.h"
#include "tconvert.h"
#include "trasterimage.h"
#include "timagecache.h"
#include "tstream.h"

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

DEFINE_CLASS_CODE(TXshChildLevel, 50)

PERSIST_IDENTIFIER(TXshChildLevel, "childLevel")

//=============================================================================
// TXshChildLevel

TXshChildLevel::TXshChildLevel(std::wstring name)
	: TXshLevel(m_classCode, name), m_xsheet(new TXsheet()), m_iconId()
{
	m_xsheet->addRef();
	m_type = CHILD_XSHLEVEL;
}

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

TXshChildLevel::~TXshChildLevel()
{
	m_xsheet->release();
	if (m_iconId != "") {
		ImageManager::instance()->bind(m_iconId, 0);
		TImageCache::instance()->remove(m_iconId);
	}
}

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

void TXshChildLevel::loadData(TIStream &is)
{
	if (m_xsheet)
		m_xsheet->release();
	m_xsheet = 0;
	is >> m_xsheet;
	m_xsheet->addRef();
	std::string tagName;
	if (is.matchTag(tagName)) {
		if (tagName == "name") {
			std::wstring name;
			is >> name;
			setName(name);
		}
		is.closeChild();
	}
}

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

void TXshChildLevel::saveData(TOStream &os)
{
	os << m_xsheet;
	os.child("name") << getName();
}

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

void TXshChildLevel::setXsheet(TXsheet *xsheet)
{
	xsheet->addRef();
	m_xsheet->release();
	m_xsheet = xsheet;
}

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

void TXshChildLevel::setScene(ToonzScene *scene)
{
	TXshLevel::setScene(scene);
	assert(m_xsheet);
	if (!m_xsheet)
		return;
	m_xsheet->setScene(scene);
	int i, columnCount = m_xsheet->getColumnCount();
	for (i = 0; i < columnCount; i++)
		if (m_xsheet->getColumn(i))
			m_xsheet->getColumn(i)->setXsheet(m_xsheet);
}

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

int TXshChildLevel::getFrameCount() const
{
	if (m_xsheet)
		return m_xsheet->getFrameCount();
	else
		return 0;
}

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

void TXshChildLevel::getFids(std::vector<TFrameId> &fids) const
{
	int i;
	for (i = 1; i <= getFrameCount(); i++)
		fids.push_back(TFrameId(i));
}