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得到函數指針,然後調用函數。