Options.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
#ifndef INCLUDE_GUARD_BELLE2_MVA_OPTIONS_HEADER
#define INCLUDE_GUARD_BELLE2_MVA_OPTIONS_HEADER
 
#include <boost/program_options.hpp>
#include <boost/program_options/errors.hpp>
#include <boost/property_tree/ptree_fwd.hpp>
 
#include <string>
#include <vector>
#include <functional>
 
namespace po = boost::program_options;
 
namespace Belle2 {
  namespace MVA {
 
 
    class Options {
    public:
      virtual void load(const boost::property_tree::ptree& pt) = 0;
 
      virtual void save(boost::property_tree::ptree& pt) const = 0;
 
      virtual po::options_description getDescription() = 0;
 
      virtual ~Options() = default;
    };
 
    class GeneralOptions : public Options {
 
    public:
      virtual void load(const boost::property_tree::ptree& pt) override;
 
      virtual void save(boost::property_tree::ptree& pt) const override;
 
      virtual po::options_description getDescription() override;
 
      std::string m_method; 
      std::string m_identifier; 
      std::vector<std::string> m_datafiles; 
      std::string m_treename = "ntuple"; 
      std::vector<std::string> m_variables; 
      std::vector<std::string> m_spectators; 
      int m_signal_class = 1; 
      unsigned int m_nClasses = 2; 
      std::string m_target_variable = "isSignal"; 
      std::string m_weight_variable = "__weight__"; 
      unsigned int m_max_events = 0; 
    };
 
    class SpecificOptions : public Options {
 
    public:
      virtual std::string getMethod() const = 0;
 
    };
 
    class MetaOptions : public Options {
 
    public:
      virtual void load(const boost::property_tree::ptree& pt) override;
 
      virtual void save(boost::property_tree::ptree& pt) const override;
 
      virtual po::options_description getDescription() override;
 
      bool m_use_splot = false; 
      std::string m_splot_variable = "M"; 
      std::vector<std::string> m_splot_mc_files; 
      bool m_splot_combined = false; 
      bool m_splot_boosted = false; 
      bool m_use_sideband_subtraction = false; 
      std::vector<std::string> m_sideband_mc_files; 
      std::string m_sideband_variable =
        ""; 
      bool m_use_reweighting = false; 
      std::string m_reweighting_identifier = ""; 
      std::string m_reweighting_variable =
        ""; 
      std::vector<std::string> m_reweighting_data_files; 
      std::vector<std::string> m_reweighting_mc_files; 
    };
 
    template<typename T>
    std::function<void(T)> check_bounds(T min, T max, const std::string& name)
    {
      return [name, min, max](T v) -> void {
        if (v <= min || v >= max)
        {
          throw po::validation_error(po::validation_error::invalid_option_value, name,
                                     std::to_string(min) + " <= " + name + " <= " + std::to_string(max) + ": provided value " + std::to_string(v));
        }
      };
    }
 
    template<typename T>
    std::function<void(std::vector<T>)> check_bounds_vector(T min, T max, const std::string& name)
    {
      return [name, min, max](const std::vector<T>& vec) -> void {
        for (auto v : vec)
        {
          if (v <= min || v >= max) {
            throw po::validation_error(po::validation_error::invalid_option_value, name,
                                       std::to_string(min) + " <= " + name + " <= " + std::to_string(max) + ": provided value " + std::to_string(v));
          }
        }
      };
    }
 
  }
}
#endif