SimulationOptions.cpp

Go to the documentation of this file.

//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      Resample/Options/SimulationOptions.cpp
//! @brief     Implements class SimulationOptions
//!
//! @homepage  http://www.bornagainproject.org
//! @license   GNU General Public License v3 or higher (see COPYING)
//! @copyright Forschungszentrum Jülich GmbH 2018
//! @authors   Scientific Computing Group at MLZ (see CITATION, AUTHORS)
//
//  ************************************************************************************************
 
#include "Resample/Options/SimulationOptions.h"
#include "Base/Util/Assert.h"
#include <cstdlib>
#include <iostream>
#include <string>
#include <thread>
 
SimulationOptions::SimulationOptions()
    : m_mc_integration(false)
    , m_include_specular(false)
    , m_use_avg_materials(false)
    , m_mc_points(1)
{
    if (const char* env_p = std::getenv("BA_NTHREADS")) {
        m_thread_info.n_threads = std::stoi(env_p);
        if (m_thread_info.n_threads > 0)
            return;
    }
    m_thread_info.n_threads = getHardwareConcurrency();
}
 
bool SimulationOptions::isIntegrate() const
{
    return m_mc_integration && m_mc_points > 1;
}
 
void SimulationOptions::setMonteCarloIntegration(bool flag, size_t mc_points)
{
    m_mc_integration = flag;
    m_mc_points = mc_points;
}
 
void SimulationOptions::setNumberOfThreads(int nthreads)
{
    ASSERT(nthreads >= 0);
    if (nthreads == 0)
        m_thread_info.n_threads = getHardwareConcurrency();
    else
        m_thread_info.n_threads = nthreads;
    ASSERT(m_thread_info.n_threads > 0);
}
 
unsigned SimulationOptions::getNumberOfThreads() const
{
    ASSERT(m_thread_info.n_threads > 0);
    return m_thread_info.n_threads;
}
 
void SimulationOptions::setNumberOfBatches(int nbatches)
{
    ASSERT(nbatches > 0);
    m_thread_info.n_batches = nbatches;
}
 
unsigned SimulationOptions::getNumberOfBatches() const
{
    ASSERT(m_thread_info.n_batches > 0);
    return m_thread_info.n_batches;
}
 
unsigned SimulationOptions::getCurrentBatch() const
{
    unsigned result = m_thread_info.current_batch;
    ASSERT(result < m_thread_info.n_batches);
    return result;
}
 
unsigned SimulationOptions::getHardwareConcurrency() const
{
    return std::thread::hardware_concurrency();
}