Play-/Source/OsStructManager.h

#ifndef _OSSTRUCTMANAGER_H_
#define _OSSTRUCTMANAGER_H_

template<typename StructType>
class COsStructManager
{
public:
	typedef uint32 iterator;

	COsStructManager(StructType* structBase, uint32 idBase, uint32 structMax) :
	  m_structBase(structBase),
	  m_idBase(idBase),
	  m_structMax(structMax)
	{

	}

	StructType* GetBase() const
	{
		return m_structBase;
	}

	StructType* operator [](uint32 index) const
	{
		index -= m_idBase;
		if(index >= m_structMax)
		{
			return NULL;
		}
		StructType* structPtr = m_structBase + index;
		if(!structPtr->isValid)
		{
			return NULL;
		}
		return structPtr;
	}

	uint32 Allocate() const
	{
		for(unsigned int i = 0; i < m_structMax; i++)
		{
			if(!m_structBase[i].isValid) 
			{
				m_structBase[i].isValid = true;
				return (i + m_idBase);
			}
		}
		return -1;
	}

	void Free(uint32 id)
	{
		StructType* structPtr = (*this)[id];
		if(!structPtr->isValid)
		{
			throw std::exception();
		}
		structPtr->isValid = false;
	}

	void FreeAll()
	{
		for(unsigned int i = 0; i < m_structMax; i++)
		{
			m_structBase[i].isValid = false;
		}
	}

	iterator Begin() const
	{
		return m_idBase;
	}

	iterator End() const
	{
		return m_idBase + m_structMax;
	}

private:
	StructType* m_structBase;
	uint32		m_structMax;
	uint32		m_idBase;
};

#endif