Factory.h

Go to the documentation of this file.

//
// Copyright Contributors to the MaterialX Project
// SPDX-License-Identifier: Apache-2.0
//
 
#ifndef MATERIALX_FACTORY_H
#define MATERIALX_FACTORY_H
 
 
#include <MaterialXCore/Library.h>
 
MATERIALX_NAMESPACE_BEGIN
 
template <class T> class Factory
{
  public:
    using Ptr = shared_ptr<T>;
    using CreatorFunction = Ptr (*)();
    using CreatorMap = std::unordered_map<string, CreatorFunction>;
 
    void registerClass(const string& typeName, CreatorFunction f)
    {
        _creatorMap[typeName] = f;
    }
 
    bool classRegistered(const string& typeName) const
    {
        return _creatorMap.find(typeName) != _creatorMap.end();
    }
 
    void unregisterClass(const string& typeName)
    {
        auto it = _creatorMap.find(typeName);
        if (it != _creatorMap.end())
        {
            _creatorMap.erase(it);
        }
    }
 
    Ptr create(const string& typeName) const
    {
        auto it = _creatorMap.find(typeName);
        return (it != _creatorMap.end() ? it->second() : nullptr);
    }
 
  private:
    CreatorMap _creatorMap;
};
 
MATERIALX_NAMESPACE_END
 
#endif // MATERIALX_FACTORY_H