#include "toonz/tcolumnfxset.h"

// TnzLib includes

#include "toonz/tcolumnfx.h"

// TnzBase includes

#include "tfx.h"

// TnzCore includes

#include "tstream.h"

#include "texception.h"

TFxSet::TFxSet()

//: m_xsheetFx(0)

{

}

TFxSet::~TFxSet()

{

	// delete m_xsheetFx;

	clear();

}

void TFxSet::addFx(TFx *fx)

{

	std::set<tfx *="">::iterator it = m_fxs.find(fx);</tfx>

	if (it == m_fxs.end()) {

		fx->addRef();

		m_fxs.insert(fx);

		fx->setNewIdentifier();

	}

}

void TFxSet::getFxs(std::set<tfx *=""> &fxs)</tfx>

{

	fxs.insert(m_fxs.begin(), m_fxs.end());

}

void TFxSet::saveData(TOStream &os, int occupiedColumnCount)

{

	std::set<tfx *="">::iterator it;</tfx>

	for (it = m_fxs.begin(); it != m_fxs.end(); ++it) {

		TRasterFx *rasterFx = dynamic_cast<trasterfx *="">(*it);</trasterfx>

		if (rasterFx) {

			TColumnFx *columnFx = dynamic_cast<tcolumnfx *="">(rasterFx);</tcolumnfx>

			if (columnFx &&

				(columnFx->getColumnIndex() == -1 ||

				 columnFx->getColumnIndex() >= occupiedColumnCount))

				continue;

		}

		os.openChild("fxnode");

		os << (*it);

		os.closeChild();

	}

}

int TFxSet::getFxCount() const

{

	return m_fxs.size();

}

bool TFxSet::removeFx(TFx *fx)

{

	std::set<tfx *="">::iterator it = m_fxs.find(fx);</tfx>

	if (it != m_fxs.end()) {

		TFx *fx = *it;

		fx->release();

		m_fxs.erase(fx);

		return true;

	} else

		return false;

}

TFx *TFxSet::getFx(int index) const

{

	assert(0 <= index && index < getFxCount());

	std::set<tfx *="">::const_iterator it = m_fxs.begin();</tfx>

	std::advance(it, index);

	return *it;

}

TFx *TFxSet::getFx(const std::string &id) const

{

	std::set<tfx *="">::const_iterator it;</tfx>

	for (it = m_fxs.begin(); it != m_fxs.end(); ++it) {

		//     TFx *fx = *it;

		//if (fx->getId() == id)

		//  return fx;

	}

	return 0;

}

void TFxSet::clear()

{

	std::set<tfx *="">::iterator it;</tfx>

	for (it = m_fxs.begin(); it != m_fxs.end(); ++it) {

		TFx *fx = *it;

		fx->release();

	}

	m_fxs.clear();

}

void TFxSet::loadData(TIStream &is)

{

	clear();

	std::string tagName;

	while (is.openChild(tagName)) {

		if (tagName == "fxnode") {

			TPersist *p = 0;

			is >> p;

			// NOTE: In current implementation p is sharedly owned by is - it's automatically

			//       released upon stream destruction if the below assignment fails

			if (TFx *fx = dynamic_cast<tfx *="">(p)) {</tfx>

				addFx(fx);

			}

		} else {

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

		}

		is.closeChild();

	}

}

TFx *getActualFx(TFx *fx)

{

	// Zerary fxs and zerary COLUMN fxs are separate, and fx port connections

	// are stored in the actual zerary fx.

	// It's sad - couldn't we do something about it?

	TZeraryColumnFx *zeraryColumnFx = dynamic_cast<tzerarycolumnfx *="">(fx);</tzerarycolumnfx>

	return zeraryColumnFx ? zeraryColumnFx->getZeraryFx() : fx;

}

TFx *searchFx(const std::map<tfx *="" *,="" tfx=""> &table, TFx *fx)</tfx>

{

	std::map<tfx *="" *,="" tfx="">::const_iterator it = table.find(fx);</tfx>

	return it == table.end() ? 0 : it->second;

}

void updateFxLinks(const std::map<tfx *="" *,="" tfx=""> &table)</tfx>

{

	// The input table is made of (original, cloned) couples.

	// The purpose of this function is that of replicating the hierarchycal

	// structure from the first fxs to the seconds.

	std::map<tfx *="" *,="" tfx="">::const_iterator it;</tfx>

	for (it = table.begin(); it != table.end(); ++it) {

		TFx *fx = getActualFx(it->first), *fx2 = getActualFx(it->second);

		if (!fx || !fx2)

			continue;

		for (int i = 0; i < fx->getInputPortCount(); ++i) {

			TFx *inputFx = fx->getInputPort(i)->getFx();

			TFx *inputFx2 = 0;

			if (inputFx) {

				inputFx2 = searchFx(table, inputFx);

				inputFx = getActualFx(inputFx);

				// Normally, the above should be enough. However, it seems that

				// the possibility of the input table being 'incomplete' must be dealt with

				// (why is it not asserted >_< !)

				while (inputFx && !inputFx2 && (inputFx->getInputPortCount() > 0)) {

					// So, well... this block tries to delve deeper in the hierarchy

					// until a suitable mapped children is found... Yeah, sure...

					TFxPort *port = inputFx->getInputPort(0);

					inputFx = port->getFx();

					inputFx2 = searchFx(table, inputFx);

					inputFx = getActualFx(inputFx);

				}

			}

			if (inputFx2)

				fx2->getInputPort(i)->setFx(inputFx2);

		}

	}

}