// TnzCore includes

#include "tundo.h"

#include "tconvert.h"

#include "tstream.h"

// TnzBase includes

#include "tdoubleparam.h"

// STD includes

#include <set></set>

#include "tspectrumparam.h"

typedef pair<tdoubleparamp, tpixelparamp=""> ColorKeyParam;</tdoubleparamp,>

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

class TSpectrumParamImp

{

	TSpectrumParam *m_sp;

	std::vector<colorkeyparam> m_keys;</colorkeyparam>

public:

	bool m_draggingEnabled;

	bool m_notificationEnabled;

	bool m_isMatteEnabled;

	std::set<tparamobserver *=""> m_observers;</tparamobserver>

public:

	TSpectrumParamImp(TSpectrumParam *sp) : m_sp(sp), m_keys(), m_draggingEnabled(false), m_notificationEnabled(true), m_isMatteEnabled(true)

	{

	}

	TSpectrumParamImp(const TSpectrumParamImp &s)

	{

		copy(s);

	}

	void copy(const TSpectrumParamImp &src)

	{

		m_keys.clear();

		std::vector<colorkeyparam>::const_iterator it = src.m_keys.begin();</colorkeyparam>

		for (; it != src.m_keys.end(); ++it) {

			TDoubleParamP s(it->first->clone());

			TPixelParamP c(it->second->clone());

			m_keys.push_back(std::make_pair(s, c));

		}

	}

	void addKey(const ColorKeyParam &colorKey)

	{

		/*

    m_sp->addParam(colorKey.first);

    m_sp->addParam(colorKey.second);

*/

		m_keys.push_back(colorKey);

	}

	void insertKey(int index, ColorKeyParam &colorKey)

	{

		/*

    m_sp->addParam(colorKey.first);

    m_sp->addParam(colorKey.second);

*/

		std::vector<colorkeyparam>::iterator it = m_keys.begin() + index;</colorkeyparam>

		m_keys.insert(it, colorKey);

	}

	void eraseKey(int index)

	{

		std::vector<colorkeyparam>::iterator colorKeyIt = m_keys.begin() + index;</colorkeyparam>

		/*

    m_sp->removeParam((*colorKeyIt).first);

    m_sp->removeParam((*colorKeyIt).second);

*/

		m_keys.erase(colorKeyIt);

	}

	int getKeyCount() const

	{

		return m_keys.size();

	}

	ColorKeyParam getKey(int index) const

	{

		return m_keys[index];

	}

	void clearKeys()

	{

		m_keys.clear();

	}

	void notify(const TParamChange &change)

	{

		for (std::set<tparamobserver *="">::iterator it = m_observers.begin();</tparamobserver>

			 it != m_observers.end(); ++it)

			(*it)->onChange(change);

	}

private:

	TSpectrumParamImp &operator=(const TSpectrumParamImp &); //not implemented

};

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

PERSIST_IDENTIFIER(TSpectrumParam, "spectrumParam")

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

TSpectrumParam::TSpectrumParam()

{

	m_imp = new TSpectrumParamImp(this); //brutto...

	ColorKeyParam ck1(TDoubleParamP(0.0), TPixelParamP(TPixel32::Black));

	ColorKeyParam ck2(TDoubleParamP(1.0), TPixelParamP(TPixel32::White));

	m_imp->addKey(ck1);

	m_imp->addKey(ck2);

}

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

TSpectrumParam::TSpectrumParam(const TSpectrumParam &src)

	: TParam(src.getName())

{

	m_imp = new TSpectrumParamImp(*src.m_imp);

}

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

void TSpectrumParam::addObserver(TParamObserver *obs)

{

	m_imp->m_observers.insert(obs);

}

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

void TSpectrumParam::removeObserver(TParamObserver *obs)

{

	m_imp->m_observers.erase(obs);

}

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

TSpectrumParam::TSpectrumParam(int keyCount, TSpectrum::ColorKey keys[])

{

	m_imp = new TSpectrumParamImp(this);

	for (int i = 0; i < keyCount; i++) {

		double v = keys[i].first;

		TPixel32 pix = keys[i].second;

		TDoubleParamP dp(v);

		TPixelParamP pp(pix);

		pp->enableMatte(m_imp->m_isMatteEnabled);

		ColorKeyParam ck(dp, pp);

		m_imp->addKey(ck);

	}

}

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

void TSpectrumParam::copy(TParam *src)

{

	TSpectrumParam *p = dynamic_cast<tspectrumparam *="">(src);</tspectrumparam>

	if (!p)

		throw TException("invalid source for copy");

	setName(src->getName());

	m_imp->copy(*(p->m_imp));

}

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

TSpectrumParam::~TSpectrumParam()

{

	delete m_imp;

}

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

TSpectrum TSpectrumParam::getValue(double frame) const

{

	assert(m_imp);

	std::vector<tspectrum::colorkey> keys;</tspectrum::colorkey>

	int keyCount = m_imp->getKeyCount();

	for (int i = 0; i < keyCount; i++) {

		ColorKeyParam paramKey = m_imp->getKey(i);

		TSpectrum::ColorKey key(

			paramKey.first->getValue(frame),

			paramKey.second->getValue(frame));

		keys.push_back(key);

	}

	return TSpectrum(keys.size(), &keys[0]);

}

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

TSpectrum64 TSpectrumParam::getValue64(double frame) const

{

	assert(m_imp);

	std::vector<tspectrum64::colorkey> keys;</tspectrum64::colorkey>

	int keyCount = m_imp->getKeyCount();

	for (int i = 0; i < keyCount; i++) {

		ColorKeyParam paramKey = m_imp->getKey(i);

		TSpectrum64::ColorKey key(

			paramKey.first->getValue(frame),

			toPixel64(paramKey.second->getValue(frame)));

		keys.push_back(key);

	}

	return TSpectrum64(keys.size(), &keys[0]);

}

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

void TSpectrumParam::setValue(double frame, const TSpectrum &spectrum, bool undoing)

{

	assert(getKeyCount() == spectrum.getKeyCount());

	int keyCount = getKeyCount();

	for (int i = 0; i < keyCount; i++) {

		TSpectrum::Key key = spectrum.getKey(i);

		setValue(frame, i, key.first, key.second, undoing);

	}

}

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

TDoubleParamP TSpectrumParam::getPosition(int index) const

{

	assert(index <= m_imp->getKeyCount());

	return m_imp->getKey(index).first;

}

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

TPixelParamP TSpectrumParam::getColor(int index) const

{

	assert(index <= m_imp->getKeyCount());

	return m_imp->getKey(index).second;

}

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

int TSpectrumParam::getKeyCount() const

{

	assert(m_imp);

	return m_imp->getKeyCount();

}

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

void TSpectrumParam::setValue(double frame, int index, double s, const TPixel32 &color, bool undoing)

{

	assert(m_imp);

	int keyCount = m_imp->getKeyCount();

	if (index < 0 || index >= keyCount)

		throw TException("TSpectrumParam::setValue. Index out of range");

	ColorKeyParam key = m_imp->getKey(index);

	//  beginParameterChange();

	key.first->setValue(frame, s);

	key.second->setValue(frame, color);

	//  endParameterChange();

	m_imp->notify(TParamChange(this, TParamChange::m_minFrame, TParamChange::m_maxFrame, true, m_imp->m_draggingEnabled, false));

}

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

void TSpectrumParam::setDefaultValue(const TSpectrum &value)

{

	assert(value.getKeyCount() == getKeyCount());

	for (int i = 0; i < getKeyCount(); i++) {

		ColorKeyParam dstKeyParam = m_imp->getKey(i);

		TSpectrum::Key srcKey = value.getKey(i);

		dstKeyParam.first->setDefaultValue(srcKey.first);

		dstKeyParam.second->setDefaultValue(srcKey.second);

	}

}

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

void TSpectrumParam::insertKey(int index, double s, const TPixel32 &color)

{

	assert(m_imp);

	int keyCount = m_imp->getKeyCount();

	if (index < 0)

		index = 0;

	else if (index >= keyCount)

		index = keyCount;

	TDoubleParamP dp(s);

	TPixelParamP pp(color);

	pp->enableMatte(m_imp->m_isMatteEnabled);

	ColorKeyParam ck(dp, pp);

	m_imp->insertKey(index, ck);

}

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

void TSpectrumParam::addKey(double s, const TPixel32 &color)

{

	/*

  assert(m_imp);

  insertKey(m_imp->getKeyCount(), s,color);

*/

	int index = m_imp->getKeyCount();

	assert(m_imp);

	int keyCount = m_imp->getKeyCount();

	if (index < 0)

		index = 0;

	else if (index >= keyCount)

		index = keyCount;

	TDoubleParamP dp(s);

	TPixelParamP pp(color);

	pp->enableMatte(m_imp->m_isMatteEnabled);

	ColorKeyParam ck(dp, pp);

	m_imp->insertKey(index, ck);

}

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

void TSpectrumParam::removeKey(int index)

{

	assert(m_imp);

	int keyCount = m_imp->getKeyCount();

	if (index < 0 || index >= keyCount)

		throw TException("TSpectrumParam::removeKey. Index out of range");

	m_imp->eraseKey(index);

}

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

bool TSpectrumParam::isKeyframe(double frame) const

{

	int keyCount = m_imp->getKeyCount();

	for (int i = 0; i < keyCount; i++) {

		ColorKeyParam currentKey = m_imp->getKey(i);

		if (currentKey.first->isKeyframe(frame))

			return true;

		if (currentKey.second->isKeyframe(frame))

			return true;

	}

	return false;

}

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

void TSpectrumParam::deleteKeyframe(double frame)

{

	int keyCount = m_imp->getKeyCount();

	for (int i = 0; i < keyCount; i++) {

		ColorKeyParam currentKey = m_imp->getKey(i);

		currentKey.first->deleteKeyframe(frame);

		currentKey.second->deleteKeyframe(frame);

	}

}

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

void TSpectrumParam::clearKeyframes()

{

	assert(m_imp);

	int k, keyCount = m_imp->getKeyCount();

	for (k = 0; k < keyCount; ++k) {

		const ColorKeyParam &key = m_imp->getKey(k);

		key.first->clearKeyframes();

		key.second->clearKeyframes();

	}

	m_imp->notify(TParamChange(this, TParamChange::m_minFrame, TParamChange::m_maxFrame, true, m_imp->m_draggingEnabled, false));

}

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

void TSpectrumParam::assignKeyframe(

	double frame,

	const TParamP &src, double srcFrame,

	bool changedOnly)

{

	TSpectrumParamP spectrum = src;

	if (!spectrum)

		return;

	int keyCount = m_imp->getKeyCount();

	if (keyCount != spectrum->m_imp->getKeyCount())

		return;

	for (int i = 0; i < keyCount; i++) {

		ColorKeyParam dstKey = m_imp->getKey(i);

		ColorKeyParam srcKey = spectrum->m_imp->getKey(i);

		dstKey.first->setValue(frame, srcKey.first->getValue(srcFrame));

		dstKey.second->setValue(frame, srcKey.second->getValue(srcFrame));

	}

}

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

void TSpectrumParam::loadData(TIStream &is)

{

	assert(m_imp);

	m_imp->clearKeys();

	string tagName;

	is.openChild(tagName);

	assert(tagName == "spectrum");

	while (!is.eos()) {

		TDoubleParamP pos(0.0);

		TPixelParamP color(TPixel32::Black);

		is.openChild(tagName);

		pos->loadData(is);

		is.closeChild();

		is.openChild(tagName);

		color->loadData(is);

		is.closeChild();

		ColorKeyParam ck(pos, color);

		m_imp->addKey(ck);

	}

	is.closeChild();

}

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

void TSpectrumParam::saveData(TOStream &os)

{

	assert(m_imp);

	int keyCount = m_imp->getKeyCount();

	os.openChild("spectrum");

	for (int i = 0; i < keyCount; i++) {

		ColorKeyParam key = m_imp->getKey(i);

		os.openChild("s_value");

		key.first->saveData(os);

		os.closeChild();

		os.openChild("col_value");

		key.second->saveData(os);

		os.closeChild();

	}

	os.closeChild();

}

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

void TSpectrumParam::enableDragging(bool on)

{

	m_imp->m_draggingEnabled = on;

}

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

void TSpectrumParam::enableNotification(bool on)

{

	m_imp->m_notificationEnabled = on;

}

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

bool TSpectrumParam::isNotificationEnabled() const

{

	return m_imp->m_notificationEnabled;

}

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

namespace

{

inline string toString(const TPixel32 &color)

{

	string alias = "(";

	alias += ::toString(color.r) + ",";

	alias += ::toString(color.g) + ",";

	alias += ::toString(color.b) + ",";

	alias += ::toString(color.m);

	alias += ")";

	return alias;

}

inline string toString(const TSpectrum::ColorKey &key, int precision)

{

	string alias = "(";

	alias += ::toString(key.first, precision) + ",";

	alias += toString(key.second);

	alias += ")";

	return alias;

}

} // namespace

string TSpectrumParam::getValueAlias(double frame, int precision)

{

	std::vector<tspectrum::colorkey> keys;</tspectrum::colorkey>

	int keyCount = m_imp->getKeyCount();

	for (int i = 0; i < keyCount; i++) {

		ColorKeyParam paramKey = m_imp->getKey(i);

		TSpectrum::ColorKey key(

			paramKey.first->getValue(frame),

			paramKey.second->getValue(frame));

		keys.push_back(key);

	}

	string alias = "(";

	if (!keys.empty()) {

		std::vector<tspectrum::colorkey>::iterator it = keys.begin();</tspectrum::colorkey>

		std::vector<tspectrum::colorkey>::iterator end = keys.begin();</tspectrum::colorkey>

		std::advance(end, keys.size() - 1);

		for (; it != end; ++it) {

			alias += toString(*it, precision);

			alias += ",";

		}

		alias += toString(*it, precision);

	}

	alias += ")";

	return alias;

}

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

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

void TSpectrumParam::enableMatte(bool on)

{

	m_imp->m_isMatteEnabled = on;

}

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

bool TSpectrumParam::isMatteEnabled() const

{

	return m_imp->m_isMatteEnabled;

}

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

bool TSpectrumParam::hasKeyframes() const

{

	int keyCount = m_imp->getKeyCount();

	for (int i = 0; i < keyCount; i++) {

		ColorKeyParam currentKey = m_imp->getKey(i);

		if (currentKey.first->hasKeyframes() || currentKey.second->hasKeyframes())

			return true;

	}

	return false;

}

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

void TSpectrumParam::getKeyframes(std::set<double> &frames) const</double>

{

	int keyCount = m_imp->getKeyCount();

	for (int i = 0; i < keyCount; i++) {

		ColorKeyParam currentKey = m_imp->getKey(i);

		currentKey.first->getKeyframes(frames);

		currentKey.second->getKeyframes(frames);

	}

}

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

int TSpectrumParam::getNextKeyframe(double frame) const

{

	std::set<double> frames;</double>

	getKeyframes(frames);

	std::set<double>::iterator it = frames.upper_bound(frame);</double>

	if (it == frames.end())

		return -1;

	else

		return std::distance(frames.begin(), it);

}

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

int TSpectrumParam::getPrevKeyframe(double frame) const

{

	std::set<double> frames;</double>

	getKeyframes(frames);

	std::set<double>::iterator it = frames.lower_bound(frame);</double>

	if (it == frames.begin())

		return -1;

	else {

		--it;

		return std::distance(frames.begin(), it);

	}

}

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

double TSpectrumParam::keyframeIndexToFrame(int index) const

{

	std::set<double> frames;</double>

	getKeyframes(frames);

	assert(0 <= index && index < (int)frames.size());

	std::set<double>::const_iterator it = frames.begin();</double>

	std::advance(it, index);

	return *it;

}

TIStream &operator>>(TIStream &in, TSpectrumParamP &p)

{

	TPersist *tmp;

	in >> tmp;

	p = TSpectrumParamP(dynamic_cast<tspectrumparam *="">(tmp));</tspectrumparam>

	return in;

}