cafFactory.h

//##################################################################################################
//
//   Custom Visualization Core library
//   Copyright (C) 2011-2013 Ceetron AS
//
//   This library may be used under the terms of either the GNU General Public License or
//   the GNU Lesser General Public License as follows:
//
//   GNU General Public License Usage
//   This library is free software: you can redistribute it and/or modify
//   it under the terms of the GNU General Public License as published by
//   the Free Software Foundation, either version 3 of the License, or
//   (at your option) any later version.
//
//   This library is distributed in the hope that it will be useful, but WITHOUT ANY
//   WARRANTY; without even the implied warranty of MERCHANTABILITY or
//   FITNESS FOR A PARTICULAR PURPOSE.
//
//   See the GNU General Public License at <<http://www.gnu.org/licenses/gpl.html>>
//   for more details.
//
//   GNU Lesser General Public License Usage
//   This library is free software; you can redistribute it and/or modify
//   it under the terms of the GNU Lesser General Public License as published by
//   the Free Software Foundation; either version 2.1 of the License, or
//   (at your option) any later version.
//
//   This library is distributed in the hope that it will be useful, but WITHOUT ANY
//   WARRANTY; without even the implied warranty of MERCHANTABILITY or
//   FITNESS FOR A PARTICULAR PURPOSE.
//
//   See the GNU Lesser General Public License at <<http://www.gnu.org/licenses/lgpl-2.1.html>>
//   for more details.
//
//##################################################################################################
 
#pragma once
 
#include "cafAssert.h"
 
#include <cstddef>
#include <map>
#include <vector>
 
// Taken from gtest.h
//
// Due to C++ preprocessor weirdness, we need double indirection to
// concatenate two tokens when one of them is __LINE__.  Writing
//
//   foo ## __LINE__
//
// will result in the token foo__LINE__, instead of foo followed by
// the current line number.  For more details, see
// http://www.parashift.com/c++-faq-lite/misc-technical-issues.html#faq-39.6
#define CAFFA_FACTORY_CONCATENATE_STRINGS( foo, bar ) CAFFA_FACTORY_CONCATENATE_STRINGS_IMPL_( foo, bar )
#define CAFFA_FACTORY_CONCATENATE_STRINGS_IMPL_( foo, bar ) foo##bar
 
#define CAFFA_UNIQUE_COMPILE_UNIT_VAR_NAME( foo ) CAFFA_FACTORY_CONCATENATE_STRINGS( foo, __LINE__ )
 
 
#define CAFFA_FACTORY_REGISTER( BaseType, TypeToCreate, KeyType, key )   \
    static bool CAFFA_UNIQUE_COMPILE_UNIT_VAR_NAME( my##TypeToCreate ) = \
        caffa::Factory<BaseType, KeyType>::instance()->registerCreator<TypeToCreate>( key )
#define CAFFA_FACTORY_REGISTER2( BaseType, TypeToCreate, KeyType, key )   \
    static bool CAFFA_UNIQUE_COMPILE_UNIT_VAR_NAME( my2##TypeToCreate ) = \
        caffa::Factory<BaseType, KeyType>::instance()->registerCreator<TypeToCreate>( key )
 
namespace caffa
{
//==================================================================================================
//==================================================================================================
 
template <typename BaseType, typename KeyType>
class Factory
{
    class ObjectCreatorBase;
 
public:
    typedef typename std::map<KeyType, ObjectCreatorBase*>::iterator iterator_type;
 
    static Factory<BaseType, KeyType>* instance()
    {
        static Factory<BaseType, KeyType>* fact = new Factory<BaseType, KeyType>;
        return fact;
    }
 
    template <typename TypeToCreate>
    bool registerCreator( const KeyType& key )
    {
        iterator_type entryIt;
 
        entryIt = m_factoryMap.find( key );
        if ( entryIt == m_factoryMap.end() )
        {
            m_factoryMap[key] = new ObjectCreator<TypeToCreate>();
            return true;
        }
        return false;
    }
 
    BaseType* create( const KeyType& key )
    {
        iterator_type entryIt;
 
        entryIt = m_factoryMap.find( key );
        if ( entryIt != m_factoryMap.end() )
        {
            return entryIt->second->create();
        }
        else
        {
            return nullptr;
        }
    }
 
    std::vector<KeyType> allKeys()
    {
        std::vector<KeyType> keys;
 
        iterator_type entryIt;
        for ( entryIt = m_factoryMap.begin(); entryIt != m_factoryMap.end(); ++entryIt )
        {
            keys.push_back( entryIt->first );
        }
 
        return keys;
    }
 
private:
    Factory() {}
    ~Factory()
    {
        iterator_type entryIt;
 
        for ( entryIt = m_factoryMap.begin(); entryIt != m_factoryMap.end(); ++entryIt )
        {
            delete ( entryIt->second );
        }
    }
 
    // Internal helper classes
 
    class ObjectCreatorBase
    {
    public:
        ObjectCreatorBase() {}
        virtual ~ObjectCreatorBase() {}
        virtual BaseType* create() = 0;
    };
 
    template <typename TypeToCreate>
    class ObjectCreator : public ObjectCreatorBase
    {
    public:
        BaseType* create() override { return new TypeToCreate(); }
    };
 
    // Map to store factory
    std::map<KeyType, ObjectCreatorBase*> m_factoryMap;
};
 
} // End of namespace caffa