一種根據輸入動態執行函數

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

發佈了74 篇原創文章 · 獲贊 56 · 訪問量 24萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章