一种根据输入动态执行函数

一种根据输入动态执行函数

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得到函数指针，然后调用函数。