gflags是google开源的一个处理命令行参数的库, 由c++开发, 有python接口. caffe中有用到, 拿来学习下
//test.cpp
#include <gflags/gflags.h>
#include "other.h"
#include "test.h"
//自定义参数合法性检查
static bool Validatemyarg(const char* filename, gflags::int32 value)
{
if (value>0 && value<100)
return true;
std::cout<<"Invalid value for "<<filename<<' '<<value<<std::endl;
return false;
}
DEFINE_int32(myarg,10,'myarg');
//用gflags::RegisterFlagValidator来检查参数是否合法,用static是防止外部文件访问吗?
static const bool validmyarg=gflags::RegisterFlagValidator(myarg,&Validatemyarg);
int main(int argc, char** argv)
{
//设置版本号,当在命令行输入test --version的时候会输出该版本号
gflags::SetVersionString("1.0.0.0")
//设置参数的用法,在命令行中输入test --help,会输出
gflags::SetUsageMessage("usage: test <args>\n\n"
"args:\n"
" -myarg: int32 ,1~99\n");
//解析命令行参数, 当第三个参数为true时,解析结束后会将argc设为1,argv只保留原argv[0].当为false时,argc仍为原来的值,但是argv中的元素可能改变顺序
gflags::ParseCommandLineFlags(&argc, &argv, true);
//otherfunc()定义在others.cpp中,这个是用来检验DECLARE的
otherfunc();
gflags::ShutDownCommandLineFlags();
test.h
//test.h
#ifndef _TEST_H_
#define _TEST_H_
//DECLARE_int32相当于对变量FLAGS_myarg作extern外部声明
DECLARE_int32(myarg);
#endif
others.h
//others.h
#ifndef _OTHERS_H_
#define _OTHERS_H_
#include "test.h"
void otherfunc();
#endif
others.cpp
//others.cpp
#include <others.h>
#include <iostream>
void otherfunc()
{
//外部变量FLAGS_myarg
std::cout<<FLAGS_myarg<<std::endl;
}
除了FLAGS_int32外, 还有参数类型:
* DECLARE_bool: boolean
* DECLARE_int32: 32-bit integer
* DECLARE_int64: 64-bit integer
* DECLARE_uint64: unsigned 64-bit integer
* DECLARE_double: double
* DECLARE_string: C++ string