FairMQ/fairmq/options/FairProgOptions.h

/* 
 * File:   FairProgOptions.h
 * Author: winckler
 *
 * Created on March 11, 2015, 5:38 PM
 */

#ifndef FAIRPROGOPTIONS_H
#define	FAIRPROGOPTIONS_H

#include "FairMQLogger.h"
#include <boost/program_options.hpp>
#include <boost/filesystem.hpp>

#include <string>
#include <vector>
#include <iostream>
#include <fstream>
#include <iterator>
#include <tuple>

/*
 * FairProgOptions abstract base class 
 * parse command line, configuration file options as well as environment variables.
 * 
 * The user defines in the derived class the option descriptions and
 * the pure virtual ParseAll() method
 * 
 * class MyOptions : public FairProgOptions
 * {
 *      public : 
 *      MyOptions() : FairProgOptions()
 *      {
 *          fCmdline_options.add(fGenericDesc);
 *          fVisible_options.add(fCmdline_options);
 *      }
 *      virtual ~MyOptions(){}
 *      virtual int ParseAll(const int argc, char** argv, bool AllowUnregistered = false)
 *      {
 *          if(ParseCmdLine(argc,argv,fCmdline_options,fvarmap,AllowUnregistered))
 *              return 1;
 *          
 *          PrintOptions();
 *          return 0;
 *      }
 * }
 */

template<class T>
std::ostream& operator<<(std::ostream& os, const std::vector<T>& v)
{
    std::copy(v.begin(), v.end(), std::ostream_iterator<T>(os, "  "));
    return os;
}

namespace po = boost::program_options;

class FairProgOptions
{
public:
    FairProgOptions();
    virtual ~FairProgOptions();

    //  add options_description
    int AddToCmdLineOptions(const po::options_description& optdesc, bool visible = true);
    int AddToCfgFileOptions(const po::options_description& optdesc, bool visible = true);
    int AddToEnvironmentOptions(const po::options_description& optdesc);

    void EnableCfgFile(const std::string& filename = "")
    {
        fUseConfigFile = true;
        if (filename.empty())
        {
            fCmdline_options.add_options()
                ("config,c", po::value<std::string>(&fConfigFile)->required(), "Path to configuration file");
        }
        else
        {
            fConfigFile = filename;
        }
    }

    void UseConfigFile(const std::string& filename = "")
    {
        fUseConfigFile = true;
        if (filename.empty())
        {
            fCmdline_options.add_options()
                ("config,c", po::value<std::string>(&fConfigFile)->required(), "Path to configuration file");
        }
        else
        {
            fConfigFile = filename;
        }
    }

    // set value corresponding to the key
    template<typename T>
    T GetValue(const std::string& key) const
    {
        T val;
        try
        {
            if (fvarmap.count(key))
            {
                val = fvarmap[key].as<T>();
            }
            else
            {
                LOG(ERROR) << "Key '"<< key <<"' not found in boost variable map";
                LOG(INFO) << "Command line / txt config file options are the following : ";
                this->PrintHelp();
            }
        }
        catch(std::exception& e)
        {
            LOG(ERROR) << "Exception thorwn for the key '" << key << "'";
            LOG(ERROR) << e.what();
            this->PrintHelp();
        }

        return val;
    }

    // convert value to string that corresponds to the key
    std::string GetStringValue(const std::string& key);
    const po::variables_map& GetVarMap() const {return fvarmap;}
    
    // boost prog options parsers
    int ParseCmdLine(const int argc, char** argv, const po::options_description& desc, po::variables_map& varmap, bool AllowUnregistered = false);
    int ParseCmdLine(const int argc, char** argv, const po::options_description& desc, bool AllowUnregistered = false);
    
    int ParseCfgFile(const std::string& filename, const po::options_description& desc, po::variables_map& varmap, bool AllowUnregistered = false);
    int ParseCfgFile(const std::string& filename, const po::options_description& desc, bool AllowUnregistered = false);
    int ParseCfgFile(std::ifstream& ifs, const po::options_description& desc, po::variables_map& varmap, bool AllowUnregistered = false);
    int ParseCfgFile(std::ifstream& ifs, const po::options_description& desc, bool AllowUnregistered = false);

    int ParseEnvironment(const std::function<std::string(std::string)>&);

    virtual int ParseAll(const int argc, char** argv, bool AllowUnregistered = false) = 0;

    virtual int PrintOptions();
    int PrintHelp() const;

protected:
    // options container
    po::variables_map fvarmap;

    // basic description categories
    po::options_description fGenericDesc;
    po::options_description fConfigDesc;
    po::options_description fEnvironmentDesc;
    po::options_description fHiddenDesc;

    // Description of cmd line and simple configuration file (configuration file like txt, but not like xml json ini)
    po::options_description fCmdline_options;
    po::options_description fConfig_file_options;

    // Description which is printed in help command line
    po::options_description fVisible_options;

    int fVerboseLvl;
    bool fUseConfigFile;
    std::string fConfigFile;
    virtual int NotifySwitchOption();

    // UpadateVarMap() and replace() --> to modify the value of variable map after calling po::store
    template<typename T>
    void UpadateVarMap(const std::string& key, const T& val)
    {
        replace(fvarmap, key, val);
    }

    template<typename T>
    void replace(std::map<std::string, po::variable_value>& vm, const std::string& opt, const T& val)
    {
        vm[opt].value() = boost::any(val);
    }

private:
    // /////////////////////////////////////////////
    // Methods below are helper functions used in the PrintOptions method
    typedef std::tuple<std::string, std::string,std::string, std::string> VarValInfo_t;
    typedef std::map<std::string, VarValInfo_t > MapVarValInfo_t;

    VarValInfo_t Get_variable_value_info(const po::variable_value& var_val);

    template<typename T>
    std::string variable_value_to_string(const po::variable_value& var_val)
    {
        auto& value = var_val.value();
        std::ostringstream ostr;
        if (auto q = boost::any_cast<T>(&value))
        {
            ostr << *q;
        }
        std::string val_str = ostr.str();
        return val_str;
    }

    static void Max(int &val, const int &comp)
    {
        if (comp > val)
        {
            val = comp;
        }
    }
};


#endif	/* FAIRPROGOPTIONS_H */