Factory.h

Go to the documentation of this file.

#ifndef __FACTORY_H__
#define __FACTORY_H__

#include <assert.h>

#include <map>
#include <set>

namespace Base
{
template <typename BaseClassType, typename ClassType>
BaseClassType* CreateObject()
{
    return new ClassType();
}


template <typename BaseClassType, typename UniqueIdType>
class Factory
{
protected:
    typedef BaseClassType* (*CreateObjectFunc)();

public:
    typedef typename std::map<UniqueIdType, CreateObjectFunc>::const_iterator ConstIterator;
    typedef typename std::map<UniqueIdType, CreateObjectFunc>::iterator Iterator;


    template <typename ClassType>
    bool Register(UniqueIdType unique_id)
    {
        if (m_object_creator.find(unique_id) != m_object_creator.end())
            return false;

        m_object_creator[unique_id] = &CreateObject<BaseClassType, ClassType>;

        return true;
    }

    bool Unregister(UniqueIdType unique_id) { return (m_object_creator.erase(unique_id) == 1); }

    BaseClassType* Create(UniqueIdType unique_id)
    {
        Iterator iter = m_object_creator.find(unique_id);

        if (iter == m_object_creator.end())
            return NULL;

        return ((*iter).second)();
    }

    ConstIterator GetBegin() const { return m_object_creator.begin(); }

    Iterator GetBegin() { return m_object_creator.begin(); }

    ConstIterator GetEnd() const { return m_object_creator.end(); }

    Iterator GetEnd() { return m_object_creator.end(); }

protected:
    std::map<UniqueIdType, CreateObjectFunc> m_object_creator;
};



}  // namespace Base


#endif  // __FACTORY_H__