从caffe.cpp里学到的一种函数调用的模式

在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();
}