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