EnumWrapper.hpp

Go to the documentation of this file.

 
#ifndef ENUMWRAPPER_HPP
#define ENUMWRAPPER_HPP
 
#include <boost/program_options.hpp>
#include <map>
#include <string>
 
// A helper class to make configuring enums easier with boost::program_options.
 
namespace EnumWrap {
template <typename E> class EnumWrapper;
template <typename E> std::istream& operator>>(std::istream&, EnumWrapper<E>&);
 
template <typename E> class EnumWrapper {
public:
    typedef std::map<std::string, E> MapType;
 
    E operator()(const std::string& key)
    {
        value = map.at(key);
        return value;
    }
    operator E() const { return value; }
 
    static void setMap(const MapType& inMap)
    {
        map.clear();
        map = inMap;
    }
 
    friend std::istream& operator>><E>(std::istream& is, EnumWrapper<E>&);
 
private:
    static MapType map;
 
    E value;
};
 
template <typename E> typename EnumWrapper<E>::MapType EnumWrapper<E>::map;
 
template <typename E> std::istream& operator>>(std::istream& is, EnumWrapper<E>& e)
{
    std::string tok;
    is >> tok;
    try {
        e(tok);
    } catch (const std::out_of_range& oor) {
        throw boost::program_options::validation_error(
            boost::program_options::validation_error::invalid_option);
    }
    return is;
}
}
 
#endif // ndef ENUMWRAPPER_HPP