mirror of https://github.com/odrling/Aegisub
118 lines
3.9 KiB
C++
118 lines
3.9 KiB
C++
// Copyright (c) 2012, Thomas Goyne <plorkyeran@aegisub.org>
|
|
//
|
|
// Permission to use, copy, modify, and distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice appear in all copies.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
//
|
|
// Aegisub Project http://www.aegisub.org/
|
|
|
|
/// @file factory_manager.h
|
|
/// @brief Template/base-class for factory classes
|
|
/// @ingroup utility
|
|
///
|
|
|
|
#pragma once
|
|
|
|
#include <algorithm>
|
|
#include <map>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
template <typename func>
|
|
class FactoryBase {
|
|
protected:
|
|
typedef std::map<std::string, std::pair<bool, func>> map;
|
|
|
|
static map& classes() {
|
|
static map classes;
|
|
return classes;
|
|
}
|
|
|
|
static void DoRegister(func function, std::string name, bool hide, std::vector<std::string> &subtypes) {
|
|
if (subtypes.empty())
|
|
classes().insert(std::make_pair(name, std::make_pair(hide, function)));
|
|
else {
|
|
for (auto const& subtype : subtypes)
|
|
classes().insert(std::make_pair(name + '/' + subtype, std::make_pair(hide, function)));
|
|
}
|
|
}
|
|
|
|
static func Find(std::string const& name) {
|
|
auto factory = classes().find(name);
|
|
return factory != classes().end() ? factory->second.second : nullptr;
|
|
}
|
|
|
|
public:
|
|
static std::vector<std::string> GetClasses(std::string favourite="") {
|
|
std::vector<std::string> list;
|
|
std::string cmp;
|
|
std::transform(favourite.begin(), favourite.end(), favourite.begin(), ::tolower);
|
|
for (auto const& cls : classes()) {
|
|
cmp.clear();
|
|
std::transform(cls.first.begin(), cls.first.end(), std::back_inserter(cmp), ::tolower);
|
|
if (cmp == favourite)
|
|
list.insert(list.begin(), cls.first);
|
|
else if (!cls.second.first)
|
|
list.push_back(cls.first);
|
|
}
|
|
return list;
|
|
}
|
|
};
|
|
|
|
template<typename Base, typename Arg1=void, typename Arg2=void>
|
|
class Factory : public FactoryBase<Base *(*)(Arg1, Arg2)> {
|
|
typedef Base *(*func)(Arg1, Arg2);
|
|
|
|
public:
|
|
static std::unique_ptr<Base> Create(std::string const& name, Arg1 a1, Arg2 a2) {
|
|
auto factory = FactoryBase<func>::Find(name);
|
|
return factory ? std::unique_ptr<Base>(factory(a1, a2)) : nullptr;
|
|
}
|
|
|
|
template<class T>
|
|
static void Register(std::string name, bool hide = false, std::vector<std::string> subTypes = {}) {
|
|
FactoryBase<func>::DoRegister([](Arg1 a1, Arg2 a2) -> Base * { return new T(a1, a2); }, name, hide, subTypes);
|
|
}
|
|
};
|
|
|
|
template<typename Base, typename Arg1>
|
|
class Factory<Base, Arg1, void> : public FactoryBase<Base *(*)(Arg1)> {
|
|
typedef Base *(*func)(Arg1);
|
|
|
|
public:
|
|
static std::unique_ptr<Base> Create(std::string const& name, Arg1 a1) {
|
|
auto factory = FactoryBase<func>::Find(name);
|
|
return factory ? std::unique_ptr<Base>(factory(a1)) : nullptr;
|
|
}
|
|
|
|
template<class T>
|
|
static void Register(std::string name, bool hide = false, std::vector<std::string> subTypes = {}) {
|
|
FactoryBase<func>::DoRegister([](Arg1 a1) -> Base * { return new T(a1); }, name, hide, subTypes);
|
|
}
|
|
};
|
|
|
|
template<typename Base>
|
|
class Factory<Base, void, void> : public FactoryBase<Base *(*)()> {
|
|
typedef Base *(*func)();
|
|
|
|
public:
|
|
static std::unique_ptr<Base> Create(std::string const& name) {
|
|
auto factory = FactoryBase<func>::Find(name);
|
|
return factory ? std::unique_ptr<Base>(factory()) : nullptr;
|
|
}
|
|
|
|
template<class T>
|
|
static void Register(std::string name, bool hide = false, std::vector<std::string> subTypes = {}) {
|
|
FactoryBase<func>::DoRegister([]() -> Base * { return new T; }, name, hide, subTypes);
|
|
}
|
|
};
|