在caffe.cpp里有一种根据命令行参数不同,调用不同函数的方法
#include <gflags/gflags.h>
#include <glog/logging>
#include <map>
#include <string>
typedef int (*Brewfunction)();
typedef std::map<std::string, Brewfunction> BrewMap;
BrewMap g_brew_map;
/*用宏定义了一个函数注册器,以后只需要用RegisterBrewFunction,就可以
完成某个函数的注册. 没有名字的namespace的作用是将该namespace中参
数的作用域限定在该namespace中.在该namespace中,对需要注册的函数定义
一个类,在该类的构造函数中,将需要注册的函数加入字典g_brew_map中.最后
需要实例化该类,来将函数真正添加到g_brew_map.*/
//但是想不明白的是, 为什么不直接在需要注册函数时用g_brew_map[funcname]就行了?这样不是更简单吗?
#define RegisterBrewFunction(func)\
namespace{\
class{\
public:\
__Registerer_##func(){\
g_brew_map[#func]=&func;\
}\
};\
__Registerer_##func g_registerer_##func;
}
int f1()
{
LOG(INFO)<<"This is f1";
return 0;
}
RegisterBrewFunction(f1); //注册f1
int f2()
{
LOG(INFO)<<"This is f2";
return 0;
}
RegisterBrewFunction(f2);//注册f2
BrewFunction GetBrewFunction(std::string name)
{
if (g_brew_map.count(name))
{
return g_brew_map[name];
}
else
{
LOG(ERROR)<<"Available action:";
for (BrewMap::iterator it=g_brew_map.begin();it!=g_brew_map.end();++it)
{
LOG(ERROR)<<"\t"<<it->first;
}
return NULL;
}
}
int main(int argc, char** argv)
{
gflags::ParseCommandLineFlags(&argc, &argv, true);
if (argc==2)
{
//这里本来是要用到异常处理的,以防止用户输入的第二个参数是没有注册过的函数.异常处理还没系统学过,以后再来补充
return GetBrewFunction(std::string(argv[1]))();
}
gflags::ShutDownCommandLineFlags();
}