mcinit.hpp

#ifndef __MC_INIT_HPP__
#define __MC_INIT_HPP__
 
#include <biocma_cst_config.hpp>
#include <cassert>
#include <common/execinfo.hpp>
#include <common/logger.hpp>
#include <cstdint>
#include <mc/domain.hpp>
#include <mc/particles_container.hpp>
#include <mc/traits.hpp>
#include <mc/unit.hpp>
#include <memory>
#include <span>
#include <stdexcept>
#include <utility>
 
namespace MC
{
 
  void impl_init(double& total_mass,
                 uint64_t n_particles,
                 MonteCarloUnit& unit,
                 AutoGenerated::ContainerVariant&& container,
                 bool uniform_init);
  template <ModelType Model>
  std::unique_ptr<MonteCarloUnit>
  init(const std::shared_ptr<IO::Logger>& _logger,
       uint64_t n_particles,
       std::span<double> volumes,
       const NeighborsView<HostSpace>& neighbors,
       bool uniform_mc_init,
       double& total_mass)
  {
    if constexpr (ConstWeightModelType<Model>)
    {
      if (_logger)
      {
        _logger->print(
            "Model",
            "Const Weights"); // TODO Add model's name to trait (optional)
      }
    }
 
    auto unit = std::make_unique<MonteCarloUnit>();
    unit->domain = ReactorDomain(volumes, neighbors);
    // bool flag_virtual = unit->domain.getNumberCompartments() == 1;
    auto container = ParticlesContainer<Model>(n_particles);
    try
    {
      impl_init(total_mass,
                n_particles,
                *unit,
                std::move(container),
                uniform_mc_init);
    }
    catch (const std::runtime_error& e)
    {
      return nullptr;
    }
 
    return unit;
  }
 
  void post_init_weight(std::unique_ptr<MonteCarloUnit>& unit,
                        double x0,
                        double total_mass);
 
} // namespace MC
 
#endif //__MC_INIT_HPP__