#include "instanceStore.h"
#include <boost/foreach.hpp>
#include <boost/assert.hpp>

template <class Type, class StoreType>
const StoreType &
InstanceStore<Type, StoreType>::GetAll()
{
	return *getInstances();
}

template <class Type, class StoreType>
void
InstanceStore<Type, StoreType>::Add(const typename StoreType::value_type & p)
{
	getInstances()->insert(p);
}

template <class Type, class StoreType>
void
InstanceStore<Type, StoreType>::prune()
{
	auto & ps = getInstances();
	if (ps->empty()) {
		delete ps;
		ps = NULL;
	}
}

template <class Type, class StoreType>
StoreType * &
InstanceStore<Type, StoreType>::getInstances()
{
	static StoreType * instances = NULL;
	if (!instances) {
		instances = new StoreType();
	}
	return instances;
}

template <class Type, class KeyType>
void
InstanceMap<Type, KeyType>::Add(const KeyType & k, Type * p)
{
	BOOST_ASSERT(IStore::getInstances()->find(k) == IStore::getInstances()->end());
	IStore::Add(Value(k, boost::shared_ptr<Type>(p)));
}

template <class Type, class KeyType>
void
InstanceMap<Type, KeyType>::Add(const KeyType & k, const boost::shared_ptr<Type> & p)
{
	BOOST_ASSERT(IStore::getInstances()->find(k) == IStore::getInstances()->end());
	IStore::Add(Value(k, p));
}

template <class Type, class KeyType>
void
InstanceMap<Type, KeyType>::Remove(const KeyType & k)
{
	IStore::getInstances()->erase(k);
	IStore::prune();
}

template <class Type, class KeyType>
void
InstanceMap<Type, KeyType>::OnEach(const boost::function<void(const Value &)> & func)
{
	BOOST_FOREACH(const auto & l, IStore::GetAll()) {
		func(l);
	}
	IStore::prune();
}

template <class Type>
void
InstanceSet<Type>::OnAll(const boost::function<void(Type *)> & func)
{
	BOOST_FOREACH(const auto & l, IStore::GetAll()) {
		func(l.get());
	}
	IStore::prune();
}

template <class Type>
void
InstanceSet<Type>::Remove(const boost::shared_ptr<Type> & p)
{
	IStore::getInstances()->erase(p);
	IStore::prune();
}

#define INSTANTIATESTORE(K, T) \
template class InstanceStore<T, std::map<K, boost::shared_ptr<T>>>; \
template class InstanceMap<T, K>; \
template class InstanceStore<T, std::set<boost::shared_ptr<T>>>; \
template class InstanceSet<T>