FairMQ/fairmq/options/FairProgOptions.h

/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
/*
 * 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()
 *      {
 *          fCmdlineOptions.add(fGenericDesc);
 *          fVisibleOptions.add(fCmdlineOptions);
 *      }
 *      virtual ~MyOptions() {}
 *      virtual int ParseAll(const int argc, char** argv, bool allowUnregistered = false)
 *      {
 *          if(ParseCmdLine(argc, argv, fCmdlineOptions, 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;
namespace fs = boost::filesystem;

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 UseConfigFile(const std::string& filename = "");

    // get value corresponding to the key
    template<typename T>
    T GetValue(const std::string& key) const
    {
        T val = T();
        try
        {
            if (fVarMap.count(key))
            {
                val = fVarMap[key].as<T>();
            }
        }
        catch(std::exception& e)
        {
            LOG(ERROR) << "Exception thrown 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 fCmdLineOptions;
    po::options_description fConfigFileOptions;

    // Description which is printed in help command line
    // to handle logger severity
    std::map<std::string,fairmq::severity_level> fSeverityMap;
    po::options_description fVisibleOptions;

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

    // UpadateVarMap() and replace() --> helper functions 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 GetVariableValueInfo(const po::variable_value& varValue);

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

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

#endif /* FAIRPROGOPTIONS_H */