typedef int (*BrewFunction)();//函数指针类型变量
typedef std::map<string, BrewFunction> BrewMap;
BrewMap g_brew_map;//全局变量
#define RegisterBrewFunction(func) \
namespace { \
class __Registerer_##func { \
public: /* NOLINT */ \
__Registerer_##func() { \
g_brew_map[#func] = &func; \
} \
}; \
__Registerer_##func g_registerer_##func; \
}
匿名命名空间,使得文件外无法访问这些变量,定义一个类,以数组方式往map中添加一个函数指针。‘#’表示将一个变量变成字符串。‘##’表示拼接一个字符串和一个变量。
//返回函数指针
static BrewFunction GetBrewFunction(const string& name) {
if (g_brew_map.count(name)) {//查找元素
return g_brew_map[name];
} else {
std:cerr << "Available caffe actions:";
for (BrewMap::iterator it = g_brew_map.begin();
it != g_brew_map.end(); ++it) {
std:cerr << "\t" << it->first;
}
std:cerr << "Unknown action: " << name;
return NULL; // not reachable, just to suppress old compiler warnings.
}
}
int name1(){
//add code here
return 0;
}
RegisterBrewFunction(name1);
int name2(){
//add code here
return 0;
}
RegisterBrewFunction(name2);
使用的方法:
int main(int argc, char** argv) {
//argv[1]是输入的函数名字,根据这个名词,执行不同的函数。
int ret = GetBrewFunction(caffe::string(argv[1]))();
//()表示函数调用,函数形参为空。
return 0;
}
示例:
./main name1
./main name2
程序的整个思路是:
开始程序,执行宏展开,宏展开的过程会定义两个类别和两个全局对象,其名称含有name1和name2字段。类定义对象的时候会调用类的构造函数,在构造函数里将函数指针加入map中。
根据终端输入的函数名字,由名字查询map得到函数指针,然后调用函数。