#include "tparamset.h"
#include "tundo.h"
//#include "tparam.h"
#include "tdoubleparam.h"
#include "tstream.h"

#include <set>

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

namespace
{
void doRelease(const std::pair<TParam *, std::string> &param)
{
	param.first->release();
}
};

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

class TParamSetImp : public TParamObserver
{
	friend class TParamSet;
	TParamSet *m_param;
	std::vector<std::pair<TParam *, std::string>> m_params;

	// ChangeBlock *m_changeBlock;
	bool m_draggingEnabled, m_notificationEnabled;

public:
	TParamSetImp(TParamSet *param) : m_param(param)
									 //, m_changeBlock(0)
									 ,
									 m_draggingEnabled(false), m_notificationEnabled(true)
	{
	}

	~TParamSetImp()
	{
		std::for_each(m_params.begin(), m_params.end(), doRelease);
	}
	// std::set<TParamSetObserver*> m_observers;
	std::set<TParamObserver *> m_paramObservers;

	template <typename T>
	void notify(const T &change);

	void onChange(const TParamChange &change)
	{
	}
};

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

TParamSet::TParamSet(std::string name)
	: TParam(name)
	, m_imp(new TParamSetImp(this))
{
}

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

TParamSet::TParamSet(const TParamSet &src)
	: TParam(src.getName())
	, m_imp(new TParamSetImp(this))
{
}

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

TParamSet::~TParamSet()
{
	delete m_imp;
}

//---------------------------------------------------------
/*
template <class Container >
class MyBackInsertIterator : public std::iterator<std::output_iterator_tag, void, void, void, void>
{
protected:
  Container &container;
public:
  MyBackInsertIterator(Container &c) : container(c) {}
  MyBackInsertIterator<Container > &operator=(const typename Container::value_type &value)
  {
    container.push_back(value);
    return *this;
  }
  MyBackInsertIterator<Container>&operator*()
  {
  return *this;
  }
  MyBackInsertIterator<Container>&operator++()
  {
  return *this;
  }
  MyBackInsertIterator<Container>&operator++(int)
  {
  return *this;
  }
};
*/

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

void TParamSet::beginParameterChange()
{
	//  assert(0);

	//std::set<TParamSetObserver*>::iterator it = m_imp->m_observers.begin();
	//for (;it != m_imp->m_observers.end(); ++it)
	//  (*it)->onBeginChangeBlock(this);

	//assert(!m_imp->m_changeBlock);

	std::vector<TParam *> params;

	/*
MyBackInsertIterator<vector<pair<TParam*, string> > > myBackInsertIterator(params);
copy(m_imp->m_params.begin(), m_imp->m_params.end(), myIterator);
*/

	std::vector<std::pair<TParam *, std::string>>::iterator it2 = m_imp->m_params.begin();
	for (; it2 != m_imp->m_params.end(); ++it2)
		params.push_back(it2->first);

	// m_imp->m_changeBlock = new ChangeBlock;
};
//---------------------------------------------------------

void TParamSet::endParameterChange(){
	//  assert(0);
	//assert(m_imp->m_changeBlock);
	/*
TParamSetChange change(this, m_imp->m_changeBlock->m_firstAffectedFrame, 
                             m_imp->m_changeBlock->m_lastAffectedFrame, 
                             m_imp->m_changeBlock->m_changes, false);

change.m_dragging = m_imp->m_draggingEnabled;

//delete m_imp->m_changeBlock;
//m_imp->m_changeBlock = 0;
//m_imp->notify(change);

//std::set<TParamSetObserver*>::iterator it = m_imp->m_observers.begin();
//for (;it != m_imp->m_observers.end(); ++it)
//  (*it)->onEndChangeBlock(this);

m_imp->notify(change);
*/
};

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

void TParamSet::addParam(const TParamP &param, const std::string &name)
{
	std::pair<TParam *, std::string> paramToInsert = std::make_pair(param.getPointer(), name);
	std::vector<std::pair<TParam *, std::string>>::iterator it =
		std::find(m_imp->m_params.begin(), m_imp->m_params.end(), paramToInsert);

	if (it == m_imp->m_params.end()) {
		param->addRef();
		param->addObserver(m_imp);
		m_imp->m_params.push_back(paramToInsert);
		//TParamSetParamAdded psParamAdded(this, param.getPointer(), name, false);
		if (param->getName().empty())
			param->setName(name);
		//m_imp->notify(psParamAdded);
	}
}

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

void TParamSet::insertParam(const TParamP &param, const std::string &name, int index)
{
	std::pair<TParam *, std::string> paramToInsert = std::make_pair(param.getPointer(), name);
	std::vector<std::pair<TParam *, std::string>>::iterator it =
		std::find(m_imp->m_params.begin(), m_imp->m_params.end(), paramToInsert);

	if (it == m_imp->m_params.end()) {
		param->addRef();
		param->addObserver(m_imp);
		it = m_imp->m_params.begin();
		int f;
		for (f = 0; f < index; f++)
			it++;
		m_imp->m_params.insert(it, paramToInsert);
		if (param->getName().empty())
			param->setName(name);
	}
}

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

namespace
{
class matchesParam
{
	TParamP m_param;

public:
	matchesParam(const TParamP &param) : m_param(param) {}
	bool operator()(const std::pair<TParam *, std::string> &param)
	{
		return m_param.getPointer() == param.first;
	}
};
}

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

void TParamSet::removeParam(const TParamP &param)
{
	std::vector<std::pair<TParam *, std::string>>::iterator it =
		std::find_if(m_imp->m_params.begin(), m_imp->m_params.end(), matchesParam(param));
	if (it != m_imp->m_params.end()) {
		param->removeObserver(m_imp);

		//TParamSetParamRemoved psParamRemoved(this, param.getPointer(), it->second, false);
		//m_imp->notify(psParamRemoved);

		param->release();
		m_imp->m_params.erase(it);
	}
}

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

void TParamSet::removeAllParam()
{
	while (!m_imp->m_params.empty()) {
		std::vector<std::pair<TParam *, std::string>>::iterator it = m_imp->m_params.begin();
		TParam *param = it->first;
		param->removeObserver(m_imp);
		param->release();
		m_imp->m_params.erase(it);
	}
}

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

int TParamSet::getParamCount() const
{
	return m_imp->m_params.size();
}

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

TParamP TParamSet::getParam(int i) const
{
	assert(i >= 0 && i < (int)m_imp->m_params.size());
	return m_imp->m_params[i].first;
}

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

std::string TParamSet::getParamName(int i) const
{
	assert(i >= 0 && i < (int)m_imp->m_params.size());
	return m_imp->m_params[i].second;
}

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

int TParamSet::getParamIdx(const std::string &name) const
{
	int i, paramsCount = m_imp->m_params.size();
	for (i = 0; i < paramsCount; ++i)
		if (m_imp->m_params[i].second == name)
			break;

	return i;
}

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

void TParamSet::getAnimatableParams(std::vector<TParamP> &params, bool recursive)
{
	std::vector<std::pair<TParam *, std::string>>::iterator it = m_imp->m_params.begin();
	for (; it != m_imp->m_params.end(); ++it) {
		TParam *param = it->first;

		TDoubleParamP dparam = TParamP(param);
		if (dparam)
			params.push_back(dparam);
		else {
			TParamSetP paramset = TParamP(param);
			if (paramset && recursive)
				paramset->getAnimatableParams(params, recursive);
		}
	}
}

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

void TParamSet::addObserver(TParamObserver *observer)
{
	//TParamSetObserver *obs = dynamic_cast<TParamSetObserver *>(observer);
	//if (obs)
	//  m_imp->m_observers.insert(obs);
	//else
	m_imp->m_paramObservers.insert(observer);
}

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

template <typename T>
void TParamSetImp::notify(const T &change)
{
	if (m_notificationEnabled) {
		//  for (std::set<TParamSetObserver*>::iterator it = m_observers.begin();
		//                                              it!= m_observers.end();
		//                                              ++it)
		//    (*it)->onChange(change);
		for (std::set<TParamObserver *>::iterator paramIt = m_paramObservers.begin();
			 paramIt != m_paramObservers.end();
			 ++paramIt)
			(*paramIt)->onChange(change);
	}
}

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

void TParamSet::removeObserver(TParamObserver *observer)
{
	//TParamSetObserver *obs = dynamic_cast<TParamSetObserver *>(observer);
	//if (obs)
	//  m_imp->m_observers.erase(obs);
	//else
	m_imp->m_paramObservers.erase(observer);
}

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

void TParamSet::enableDragging(bool on)
{
	std::vector<std::pair<TParam *, std::string>>::iterator it = m_imp->m_params.begin();
	for (; it != m_imp->m_params.end(); ++it) {
		TDoubleParamP dparam(it->first);
		//if (dparam)
		//  dparam->enableDragging(on);
	}

	m_imp->m_draggingEnabled = on;
}

//---------------------------------------------------------
/*
namespace {

class DoEnableNotification : public std::binary_function {
public:
  DoEnableNotification() {}

  void operator() (const pair<TParam*, string> &param, bool on) 
  {
    return param->first->enableNotification(on);
  }
};
}
*/
//---------------------------------------------------------

void TParamSet::enableNotification(bool on)
{
	//  std::for_each(m_imp->m_params.begin(), m_imp->m_params.end(), std::bind2nd(DoEnableNotification, on));

	std::vector<std::pair<TParam *, std::string>>::iterator it = m_imp->m_params.begin();
	for (; it != m_imp->m_params.end(); ++it) {
		it->first->enableNotification(on);
	}

	m_imp->m_notificationEnabled = on;
}

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

bool TParamSet::isNotificationEnabled() const
{
	return m_imp->m_notificationEnabled;
}

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

bool TParamSet::isKeyframe(double frame) const
{
	for (int i = 0; i < getParamCount(); i++)
		if (getParam(i)->isKeyframe(frame))
			return true;
	return false;
}

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

void TParamSet::getKeyframes(std::set<double> &frames) const
{
	for (int i = 0; i < getParamCount(); i++)
		getParam(i)->getKeyframes(frames);
}

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

double TParamSet::keyframeIndexToFrame(int index) const
{
	std::set<double> frames;
	getKeyframes(frames);
	assert(0 <= index && index < (int)frames.size());
	std::set<double>::const_iterator it = frames.begin();
	std::advance(it, index);
	return *it;
}

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

int TParamSet::getNextKeyframe(double frame) const
{
	std::set<double> frames;
	getKeyframes(frames);
	std::set<double>::iterator it = frames.upper_bound(frame);
	if (it == frames.end())
		return -1;
	else
		return std::distance(frames.begin(), it);
}

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

int TParamSet::getPrevKeyframe(double frame) const
{
	std::set<double> frames;
	getKeyframes(frames);
	std::set<double>::iterator it = frames.lower_bound(frame);
	if (it == frames.begin())
		return -1;
	else {
		--it;
		return std::distance(frames.begin(), it);
	}
}

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

bool TParamSet::hasKeyframes() const
{
	for (int i = 0; i < getParamCount(); i++)
		if (getParam(i)->hasKeyframes())
			return true;
	return false;
}

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

int TParamSet::getKeyframeCount() const
{
	std::set<double> frames;
	getKeyframes(frames);
	return frames.size();
}

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

void TParamSet::deleteKeyframe(double frame)
{
	for (int i = 0; i < getParamCount(); i++)
		getParam(i)->deleteKeyframe(frame);
}

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

void TParamSet::clearKeyframes()
{
	for (int i = 0; i < getParamCount(); i++)
		getParam(i)->clearKeyframes();
}

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

void TParamSet::assignKeyframe(
	double frame,
	const TParamP &src, double srcFrame,
	bool changedOnly)
{
	TParamSetP paramSetSrc = src;
	if (!paramSetSrc)
		return;
	if (getParamCount() != paramSetSrc->getParamCount())
		return;
	for (int i = 0; i < getParamCount(); i++)
		getParam(i)->assignKeyframe(
			frame, paramSetSrc->getParam(i), srcFrame, changedOnly);
}

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

TParam *TParamSet::clone() const
{
	return new TParamSet(*this);
}

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

void TParamSet::copy(TParam *src)
{
	TParamSet *p = dynamic_cast<TParamSet *>(src);
	if (!p)
		throw TException("invalid source for copy");
	int srcParamCount = p->getParamCount();
	removeAllParam();
	int i;
	for (i = 0; i < srcParamCount; i++) {
		TParamP param = p->getParam(i);
		addParam(param->clone(), param->getName());
	}
}

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

void TParamSet::loadData(TIStream &is)
{
	std::string tagName;
	is.openChild(tagName);
	while (!is.eos()) {
		std::string paramName;
		is.openChild(paramName);
		TPersist *p = 0;
		is >> p;
		TParam *param = dynamic_cast<TParam *>(p);
		assert(param);
		addParam(param, paramName);
		is.closeChild();
	}
	is.closeChild();
}

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

void TParamSet::saveData(TOStream &os)
{
	os.openChild(getName());
	std::vector<std::pair<TParam *, std::string>>::iterator it = m_imp->m_params.begin();
	std::vector<std::pair<TParam *, std::string>>::iterator end = m_imp->m_params.end();
	while (it != end) {
		os.openChild(it->second);
		//it->first->saveData(os);
		os << it->first;
		os.closeChild();
		++it;
	}
	os.closeChild();
}

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

std::string TParamSet::getValueAlias(double frame, int precision)
{
	std::string alias = "(";

	std::vector<std::pair<TParam *, std::string>>::iterator end = m_imp->m_params.begin();
	std::advance(end, m_imp->m_params.size() - 1);

	std::vector<std::pair<TParam *, std::string>>::iterator it = m_imp->m_params.begin();
	for (; it != end; ++it)
		alias += it->first->getValueAlias(frame, precision) + ",";

	alias += it->first->getValueAlias(frame, precision);

	alias += ")";
	return alias;
}

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

TPersistDeclarationT<TParamSet> TParamSet::m_declaration("TParamSet");