MultiThreadingConfig.cpp

Go to the documentation of this file.

/*
 * MultiThreadingConfig.cpp
 *
 *  Created on: May 22, 2018
 *      Author: mschefer
 */

#include <boost/thread.hpp>

#include "SEImplementation/Configuration/MultiThreadingConfig.h"

using namespace Euclid::Configuration;
namespace po = boost::program_options;

namespace SourceXtractor {

static const std::string THREADS_NB {"thread-count"};
static const std::string MAX_QUEUE_SIZE {"thread-max-queue-size"};

MultiThreadingConfig::MultiThreadingConfig(long manager_id) : Configuration(manager_id), m_threads_nb(-1), m_max_queue_size(1000) {}

auto MultiThreadingConfig::getProgramOptions() -> std::map<std::string, OptionDescriptionList> {
  return { {"Multi-threading", {
      {THREADS_NB.c_str(), po::value<int>()->default_value(-1), "Number of worker threads (-1=automatic, 0=disable all multithreading)"},
      {MAX_QUEUE_SIZE.c_str(), po::value<int>()->default_value(1000), "Limit the size of the internal queues"}
  }}};
}

void MultiThreadingConfig::initialize(const UserValues& args) {
  m_threads_nb = args.at(THREADS_NB).as<int>();
  if (m_threads_nb == -1) {
    m_threads_nb = boost::thread::hardware_concurrency();
  }
  else if (m_threads_nb < -1) {
    throw Elements::Exception("Invalid number of threads.");
  }
  if (m_threads_nb > 0) {
    m_thread_pool = std::make_shared<Euclid::ThreadPool>(m_threads_nb);
  }

  m_max_queue_size = args.at(MAX_QUEUE_SIZE).as<int>();
  if (m_max_queue_size <= 0) {
    throw Elements::Exception(MAX_QUEUE_SIZE + " must be strictly positive");
  }
}

} // SourceXtractor namespace