events.hpp

#ifndef __MC_EVENTS_HPP__
#define __MC_EVENTS_HPP__
 
#include "biocma_cst_config.hpp"
#include "impl/Kokkos_Profiling.hpp"
#include <Kokkos_Core.hpp>
#include <Kokkos_Core_fwd.hpp>
#include <common/execinfo.hpp>
#include <cstddef>
#include <span>
 
namespace MC
{
  enum class EventType : char
  {
    NewParticle = 0, 
    Exit,            
    Move,            
    Death,           
    Overflow,
    ChangeWeight, 
    __COUNT__
  };
 
  constexpr size_t number_event_type = static_cast<std::size_t>(
      EventType::__COUNT__); //< Number of different events
  template <EventType event>
  KOKKOS_INLINE_FUNCTION consteval size_t
  event_index()
  {
    return static_cast<size_t>(event);
  }
 
  // class Events
  // {
  // public:
  //   using value_type = Kokkos::Array<std::size_t, number_event_type>;
  //   using view_type = Kokkos::View<value_type,
  //   Kokkos::DefaultExecutionSpace>;
 
  //   value_type counts;
  //   // std::size_t counts[number_event_type];
 
  //   KOKKOS_INLINE_FUNCTION void
  //   add_into(Events& dst) const
  //   {
  //     for (std::size_t i = 0; i < number_event_type; ++i)
  //     {
  //       dst.counts[i] += counts[i];
  //     }
  //   }
 
  //   KOKKOS_INLINE_FUNCTION
  //   Events() : counts()
  //   {
  //     init();
  //   }
 
  //   KOKKOS_INLINE_FUNCTION
  //   Events(const Events& rhs) : counts()
  //   {
  //     for (std::size_t i = 0; i < number_event_type; i++)
  //     {
  //       counts[i] = rhs.counts[i];
  //     }
  //   }
 
  //   template <EventType Event>
  //   KOKKOS_INLINE_FUNCTION void
  //   inc()
  //   {
  //     counts[event_index<Event>()]++;
  //   }
  //   template <EventType Event>
  //   KOKKOS_INLINE_FUNCTION void
  //   add(std::size_t val)
  //   {
  //     counts[event_index<Event>()] += val;
  //   }
 
  //   KOKKOS_INLINE_FUNCTION
  //   void
  //   init()
  //   {
  //     for (std::size_t i = 0; i < number_event_type; i++)
  //     {
  //       counts[i] = 0;
  //     }
  //   }
  // };
 
  // struct TallyReducer
  // {
  //   using reducer = TallyReducer;
  //   using value_type = Events;
  //   using result_view_type = Kokkos::View<value_type, Kokkos::HostSpace>;
 
  // private:
  //   result_view_type result_;
 
  // public:
  //   explicit TallyReducer() : result_("reducer")
  //   {
  //   }
 
  //   KOKKOS_INLINE_FUNCTION
  //   void
  //   join(value_type& dst, const value_type& src) const
  //   {
  //     src.add_into(dst);
  //   }
 
  //   KOKKOS_INLINE_FUNCTION
  //   void
  //   init(value_type& val) const
  //   {
  //     val.init();
  //   }
 
  //   [[nodiscard]] KOKKOS_INLINE_FUNCTION value_type&
  //   reference() const
  //   {
  //     return *result_.data();
  //   }
 
  //   [[nodiscard]] KOKKOS_INLINE_FUNCTION result_view_type
  //   view() const
  //   {
  //     return result_;
  //   }
 
  //   [[nodiscard]] KOKKOS_INLINE_FUNCTION bool
  //   references_scalar() const
  //   {
  //     return true;
  //   }
  // };

  struct EventContainer
  {
 
    // Use SharedHostPinnedSpace as this struct is meant to be small enough to
    // be shared between Host and Device According to SharedHostPinnedSpace
    // documentation, the size of this data can fit into one cache line so
    // transfer is not a botteneck
    Kokkos::View<std::size_t[number_event_type], Kokkos::SharedSpace> // NOLINT
        _events; // NOLINT(cppcoreguidelines-avoid-c-arrays,modernize-avoid-c-arrays)
 
    // Kokkos::View<std::size_t[number_event_type], Kokkos::SharedSpace> //
    // NOLINT
    //     m_cumulative; //
    //     NOLINT(cppcoreguidelines-avoid-c-arrays,modernize-avoid-c-arrays)

    EventContainer() : _events("events")
    // , m_cumulative("cumulativr_events")
    {
 
      Kokkos::deep_copy(_events, 0); // Ensure all event to 0 occurence
 
      // Kokkos::deep_copy(m_cumulative, 0); // Ensure all event to 0 occurence
    }

    [[nodiscard]] std::span<std::size_t>
    get_span() const
    {
      // As we use SharedHostPinnedSpace, we can deal with classic std
      // containers to use host manipulation
      return { _events.data(), number_event_type };
    }

    KOKKOS_INLINE_FUNCTION void
    clear() const
    {
 
      // const auto& cumview = m_cumulative;
      // const auto& evview = _events;
      // Kokkos::parallel_for(
      //     "_sync_cumulative_event",
      //     number_event_type,
      //     KOKKOS_LAMBDA(const int i) { cumview[i] += evview[i]; });
      Kokkos::deep_copy(_events, 0);
    }

    template <EventType event>
    [[nodiscard]] constexpr std::size_t
    get() const
    {
      static_assert(event != EventType::__COUNT__,
                    "Count is not a valid event");
      return _events(event_index<event>());
    }

    template <EventType event>
    [[nodiscard]] constexpr std::size_t
    get_cumulative() const
    {
      static_assert(event != EventType::__COUNT__,
                    "Count is not a valid event");
      return m_cumulative(event_index<event>());
    }

    template <EventType event>
    KOKKOS_FORCEINLINE_FUNCTION constexpr void
    incr() const
    {
      static_assert(event != EventType::__COUNT__,
                    "Count is not a valid event");
      Kokkos::atomic_add(&_events[event_index<event>()], 1);
    }
 
    template <EventType event>
    KOKKOS_FORCEINLINE_FUNCTION constexpr void
    add(size_t val) const
    {
      static_assert(event != EventType::__COUNT__,
                    "Count is not a valid event");
      Kokkos::atomic_add(&_events[event_index<event>()], val);
    }
 
    template <EventType event>
    KOKKOS_FORCEINLINE_FUNCTION constexpr void
    wrap_incr() const
    {
      if constexpr (AutoGenerated::FlagCompileTime::enable_event_counter)
      {
        incr<event>();
      }
    }
 
    template <class Archive>
    void
    save(Archive& ar) const
    {
      std::array<std::size_t, number_event_type> array{};
 
      auto rd = std::span<std::size_t>(_events.data(), number_event_type);
      std::copy(rd.begin(), rd.end(), array.begin());
      assert(rd[0] == _events[0] && rd[0] == array[0]);
 
      // std::array<std::size_t, number_event_type> array_cumulative{};
      // rd = std::span<std::size_t>(m_cumulative.data(), number_event_type);
      // std::copy(rd.begin(), rd.end(), array_cumulative.begin());
      // assert(rd[0] == m_cumulative[0] && rd[0] == array_cumulative[0]);
      // ar(array, array_cumulative);
      //
      ar(array);
    }
 
    template <class Archive>
    void
    load(Archive& ar)
    {
 
      std::array<std::size_t, number_event_type> array{};
      // std::array<std::size_t, number_event_type> array_cumulative{};
      // ar(array, array_cumulative);
      ar(array);
 
      auto rd = std::span<std::size_t>(_events.data(), number_event_type);
      std::copy(array.begin(), array.end(), rd.begin());
      assert(rd[0] == _events[0]);
 
      // rd = std::span<std::size_t>(m_cumulative.data(), number_event_type);
      // std::copy(array_cumulative.begin(), array_cumulative.end(),
      // rd.begin()); assert(rd[0] == m_cumulative[0]);
    }
  };
 
}; // namespace MC
 
#endif //__MC_EVENTS_HPP__