basf2_mva_extract.cc

/**************************************************************************
 * 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.                  *
 **************************************************************************/
 
#include <iostream>
#include <vector>
 
#include <mva/utility/Utility.h>
 
namespace po = boost::program_options;
using namespace Belle2::MVA;
 
int main(int argc, char* argv[])
{
 
  std::vector<std::string> filenames;
  std::string directory;
 
  po::options_description description("Options");
  description.add_options()
  ("help", "print this message")
  ("identifiers", po::value<std::vector<std::string>>(&filenames)->multitoken(), "Identifiers of the trained methods")
  ("directory", po::value<std::string>(&directory)->required(),
   "Directory which will contain the temporary directories generated by the the mva package");
 
  po::variables_map vm;
 
  try {
    po::parsed_options parsed = po::command_line_parser(argc, argv).options(description).run();
    po::store(parsed, vm);
 
    if (vm.count("help")) {
      std::cout << description << std::endl;
      return 1;
    }
    po::notify(vm);
  } catch (po::error& err) {
    std::cerr << "Error: " << err.what() << "\n";
    return 1;
  }
 
  for (const auto& filename : filenames)
    Belle2::MVA::Utility::extract(filename, directory);
  return 0;
 
}