vmec_test_commandline_parser.hpp

Go to the documentation of this file.

//******************************************************************************
//******************************************************************************
 
#ifndef vmec_test_commandline_parser_hpp
#define vmec_test_commandline_parser_hpp
 
#include <map>
#include <string>
#include <sstream>
#include <functional>
 
namespace vmec_test {
 
    typedef std::map<std::string, std::string> arg_map;
    typedef std::pair<std::string, std::string> arg_element;
 
//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
    class commandline_parser {
    public:
        const arg_map commands;
        const std::function<void(void)> help;
 
//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
        static arg_map parse_commands(const size_t argc,
                                      const char * argv[],
                                      const std::function<void(void)> help) {
            if (argc == 0) {
                help();
            }
 
            arg_map commands;
 
            for (size_t i = 1; i < argc; i++) {
                std::string arg(argv[i]);
 
                if (arg == "-h") {
                    help();
                } else {
                    size_t eqpos = arg.find('=');
                    if (eqpos != std::string::npos) {
                        std::string key = arg.substr(0, eqpos);
                        std::string value = arg.substr(eqpos + 1, std::string::npos);
 
                        commands.insert(arg_element(key, value));
                    } else {
                        commands.insert(arg_element(arg, ""));
                    }
                }
            }
 
            return commands;
        }
 
//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
        commandline_parser(const size_t argc,
                           const char * argv[],
                           const std::function<void(void)> help) :
        commands(parse_commands(argc, argv, help)) {}
 
//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
        bool is_set(const std::string &key) const {
            return commands.find(key) != commands.end();
        }
 
//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
        template<typename TYPE>
        TYPE get(const std::string &key) const {
            if (!is_set(key)) {
                help();
            }
 
            std::stringstream value_stream(commands.at(key));
            TYPE temp;
            value_stream >> temp;
 
            return temp;
        }
    };
}
 
#endif