FeederOptions.cxx

Go to the documentation of this file.

/* Copyright (C) 2025 Jagiellonian University, Krakow
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Bartosz Sobol [committer] */
 
#include "FeederOptions.h"
 
#include "log.hpp"
 
#include <boost/program_options.hpp>
 
#include <iostream>
 
#include <fmt/format.h>
 
namespace po = boost::program_options;
 
FeederOptions::FeederOptions(int argc, char** argv)
{
  uint32_t logLevel{severity_level::status};
  uint32_t logSyslog{severity_level::status};
  std::string logFile{};
 
  po::options_description options("Options");
 
  auto addOption = options.add_options();
  addOption("help,h", "produce help message");
  addOption("log-level,l", po::value<unsigned>(&logLevel)->default_value(logLevel)->value_name("N"),
            "set the file log level (all:0)");
  addOption("log-file,L", po::value<std::string>(&logFile)->value_name("FILENAME"), "name of target log file");
  addOption("log-syslog", po::value<unsigned>(&logSyslog)->implicit_value(logSyslog)->value_name("N"),
            "enable logging to syslog at given log level");
  addOption("input-uri,i", po::value<std::string>(&fInput)->value_name("URI"), "uri of a RecoResult source");
  addOption("output-uri,o",
            po::value<std::vector<std::string>>()->multitoken()->value_name("scheme://host/path?param=value..."), "");
  addOption("offset", po::value<uint64_t>(&fOffset)->value_name("N")->default_value(0),
            "set the offset of RecoResults to select for processing "
            "(default: 0)");
  addOption("stride", po::value<uint64_t>(&fStride)->value_name("N")->default_value(1),
            "set the stride of RecoResults to select for processing "
            "(default: 1)");
  addOption("release-mode,R", po::value<bool>(&fReleaseMode)->implicit_value(true)->default_value(false),
            "copy and release each RecoResult immediately after receiving it");
  addOption("maximum-number,n", po::value<uint64_t>(&fMaximumNumber)->value_name("N")->default_value(UINT64_MAX),
            "set the maximum number of RecoResults to process (default: "
            "unlimited)");
 
  po::variables_map vm;
  po::store(po::parse_command_line(argc, argv, options), vm);
  po::notify(vm);
 
  if (vm.count("help") != 0) {
    std::cout << options << std::endl;
    exit(EXIT_SUCCESS);
  }
 
  logging::add_console(static_cast<severity_level>(logLevel));
  if (vm.count("log-file") != 0u) {
    L_(info) << fmt::format("[FeederOptions] Logging output to file: {}", logFile);
    logging::add_file(logFile, static_cast<severity_level>(logLevel));
  }
  if (vm.count("log-syslog") != 0u) {
    logging::add_syslog(logging::syslog::local0, static_cast<severity_level>(logSyslog));
  }
 
  if (vm.count("output-uri") != 0) {
    fOutputs = vm["output-uri"].as<std::vector<std::string>>();
  }
 
  if (vm.count("input-uri") == 0) {
    throw std::runtime_error("[FeederOptions] No input source specified");
  }
  if (vm.count("input-uri") > 1) {
    throw std::runtime_error("[FeederOptions] More than one input source specified");
  }
 
  if (fStride == 0) {
    throw std::runtime_error("[FeederOptions] Stride must be greater than zero");
  }
}