工厂注册类
利用模版形式注册类
cpp
#include <iostream>
#include <memory>
#include <functional>
namespace cyn {
//自定义断言
//#ifndef _DEBUG // _RELEASE 或者 _DEBUG ,根据你的编译器/构建系统
#ifdef _DEBUG // _RELEASE 或者 _DEBUG ,根据你的编译器/构建系统
#define CHECK(cond) \
do { \
if (!(cond)) { \
std::cerr << "Assertion failed: (" << #cond << "), function " << __FUNCTION__ \
<< ", file " << __FILE__ << ", line " << __LINE__ << "." << std::endl; \
} \
} while (0)
#define CHECK_EX(cond, msg) \
do { \
if (!(cond)) { \
std::cerr << "Assertion failed: (" << #cond << "), function " << __FUNCTION__ \
<< ", file " << __FILE__ << ", line " << __LINE__ << "." << std::endl; \
std::cerr << "Message: " << msg << std::endl; \
throw std::runtime_error(msg); \
} \
} while (0)
#else
#define CHECK(cond) \
do { \
\
} while (0)
#define CHECK_EX(cond, msg) \
do { \
\
} while (0)
#endif
// 参数基类
class LayerParams
{
public:
virtual ~LayerParams() = default;
};
// 具体参数类型
class DenseLayerParams : public LayerParams {
public:
int units;
float learning_rate;
DenseLayerParams(int units, float learning_rate) : units(units), learning_rate(learning_rate) {}
};
// 基类
class Layer
{
public:
Layer(const std::string& name,const std::shared_ptr<LayerParams>& params) : name(name),layer_params(params){}
virtual int forward(int){return 0;};
public:
int input_;
std::string name;
std::shared_ptr<LayerParams> layer_params;
};
//获取类型名字
template <typename T>
constexpr auto class_type_name() noexcept
{
std::string_view name, prefix, suffix;
#ifdef _MSC_VER
name = __FUNCSIG__;
prefix = "auto __cdecl cyn::class_type_name<class cyn::";
suffix = ">(void) noexcept";
#elif defined(__GNUC__)
name = __PRETTY_FUNCTION__;
prefix = "constexpr auto cyn::class_type_name() [with T = cyn::";
suffix = "]";
#elif defined(__clang__)
name = __PRETTY_FUNCTION__;
prefix = "constexpr auto cyn::class_type_name() [with T = cyn::";
suffix = "]";
#endif
name.remove_prefix(prefix.size());
name.remove_suffix(suffix.size());
return name;
}
// 定义一个注册表基类模版
template <typename T, typename... Args>
class Factory
{
using CreateFunction = std::function<std::shared_ptr<T>(Args...)>;
Factory(){}
public:
virtual ~Factory() = default;
static Factory& Instance()
{
static Factory instance;
return instance;
}
void Register(const std::string & name,const CreateFunction& create)
{
std::cout <<"Factory Register name:"<< std::string(name) << std::endl;
registry[std::string(name)] = create;
}
std::shared_ptr<T> Create(const std::string& name,Args... args)
{
if (registry.find(name) != registry.end())
{
std::cout <<"Create succeed:"<< std::string(name) << std::endl;
return registry[name](args...);
}
std::cout <<"Create fail:"<< std::string(name) << std::endl;
return nullptr;
}
private:
std::unordered_map<std::string, CreateFunction> registry;
};
//注册类 统一接口
template <typename Base, typename Impl, typename... Args>
class Register
{
public:
explicit Register()
{
std::cout<<"Register name: "<<std::string(class_type_name<Impl>())<<std::endl;
//流程6
Factory<Base, Args...>::Instance().Register(std::string(class_type_name<Impl>()), [](Args... args)
{
return std::shared_ptr<Base>(new Impl(args...));
});
}
};
// LayerSub 在这很关键,实现了继承LayerSub (也就是继承基类)自动注册
//注册Layer子类,实现继承自动注册 T 为Layer子类类型
/*
* 继承LayerSub-构造函数->(void)registered->trigger()->_register(str)
*/
template <typename T>
class LayerSub : public Layer
{
//流程4
static bool trigger()
{
//流程5
Register<Layer,T,std::shared_ptr<LayerParams>> _register;
return true;
}
protected:
//流程2
LayerSub(const std::shared_ptr<LayerParams>& params) : Layer(std::string(class_type_name<T>()),params)
{
std::cout<<"LayerSub:"<<std::string(class_type_name<T>())<<std::endl;
//流程3
(void)registered;
}
public:
static bool registered;
};
template <typename T>
bool LayerSub<T>::registered = LayerSub<T>::trigger();
//定义Layer类 创建接口
typedef Factory<Layer, std::shared_ptr<LayerParams>> LayerFactory;
// 具体层类型
class DenseLayer : public LayerSub<DenseLayer>
{
public:
DenseLayer(const std::shared_ptr<LayerParams>& params) : LayerSub(params)
{
std::cout<<"DenseLayer constructor!"<<std::endl;
std::shared_ptr<DenseLayerParams> p = std::dynamic_pointer_cast<DenseLayerParams>(params);
//检查参数类型 - 调试用
CHECK_EX(p,"error");
//打印参数值
std::cout<<p->units<<std::endl;
std::cout<<p->learning_rate<<std::endl;
}
int forward(int s) override {
std::cout<<"wo shi zi lei!"<<s<<std::endl;
return s;
}
};
}
int main(int argc, char *argv[])
{
QCoreApplication a(argc, argv);
std::shared_ptr<cyn::DenseLayerParams> param = std::make_shared<cyn::DenseLayerParams>(100,1);
auto ss = cyn::LayerFactory::Instance().Create(std::string("DenseLayer"), param);
ss->forward(1);
return a.exec();
}
输出结果:
Register name: DenseLayer
Factory Register name:DenseLayer
Create succeed:DenseLayer
LayerSub:DenseLayer
DenseLayer constructor!
100
1
wo shi zi lei!1