serdes_class.hpp

#ifndef __UTILS_SERDES_HPP__
#define __UTILS_SERDES_HPP__
 
#include "mc/alias.hpp"
#include <cassert>
#include <mc/traits.hpp>
#include <optional>
 
// NOLINTBEGIN
struct SerdeModel
{
  using uniform_weight = std::true_type; // Using type alias
  using Self = SerdeModel;
  using FloatType = float;
  using Config = std::nullopt_t;
 
  enum class particle_var : int
  {
    a = 0,
    b,
    c,
    COUNT
  };
 
  static constexpr std::size_t n_var = INDEX_FROM_ENUM(particle_var::COUNT);
  static constexpr std::string_view name = "simple";
  using SelfParticle = MC::ParticlesModel<Self::n_var, Self::FloatType>;
 
  MODEL_CONSTANT FloatType a_i = 5e-6; // m
 
  KOKKOS_INLINE_FUNCTION static void
  init([[maybe_unused]] const MC::KPRNG::pool_type& random_pool,
       std::size_t idx,
       const SelfParticle& arr)
  {
    GET_PROPERTY(SerdeModel::particle_var::a) = a_i;
  }
 
  KOKKOS_INLINE_FUNCTION static double
  mass([[maybe_unused]] std::size_t idx,
       [[maybe_unused]] const SelfParticle& arr)
  {
    return 1.;
  }
 
  KOKKOS_INLINE_FUNCTION static MC::Status
  update([[maybe_unused]] const MC::KPRNG::pool_type& random_pool,
         [[maybe_unused]] FloatType d_t,
         [[maybe_unused]] std::size_t idx,
         [[maybe_unused]] const SelfParticle& arr,
         [[maybe_unused]] const MC::LocalConcentration& c)
  {
 
    return MC::Status::Idle;
  }
 
  KOKKOS_INLINE_FUNCTION static void
  division([[maybe_unused]] const MC::KPRNG::pool_type& random_pool,
           [[maybe_unused]] std::size_t idx,
           [[maybe_unused]] std::size_t idx2,
           [[maybe_unused]] const SelfParticle& arr,
           [[maybe_unused]] const SelfParticle& buffer_arr)
  {
  }
 
  static MC::ContribIndexBounds get_bounds()
  {
    return {0, 0};
  }
};
static_assert(ModelType<SerdeModel>, "Check non serde model");
 
#endif