mcinit.hpp

#ifndef __MC_INIT_HPP__
#define __MC_INIT_HPP__
 
#include <cassert>
#include <common/execinfo.hpp>
#include <cstdint>
#include <mc/domain.hpp>
#include <mc/particles_container.hpp>
#include <mc/traits.hpp>
#include <mc/unit.hpp>
#include <memory>
#include <stdexcept>
#include <utility>
#include <span> 
#include <biocma_cst_config.hpp>
 
namespace MC
{
 
  void impl_init(double& total_mass,
                 uint64_t n_particles,
                 MonteCarloUnit& unit,
                 AutoGenerated::ContainerVariant&& container);
  template <ModelType Model>
  std::unique_ptr<MonteCarloUnit> init(uint64_t n_particles,
                                       std::span<double> volumes,
                                       const NeighborsView<HostSpace>& neighbors,
                                       double& total_mass)
  {
    // Kokkos::printf("Using model: %s\r\n",Model::); //TODO Add model's name to trait (optional)
    if constexpr (ConstWeightModelType<Model>)
    {
      Kokkos::printf("Const Weights\r\n");
    }
 
    auto unit = std::make_unique<MonteCarloUnit>();
    unit->domain = ReactorDomain(volumes, neighbors);
    auto container = ParticlesContainer<Model>(n_particles);
    try
    {
      impl_init(total_mass, n_particles, *unit, std::move(container));
    }
    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__