Program Listing for File KinGroupIDGetter.hpp

↰ Return to documentation for file (include/dish2/viz/getters/KinGroupIDGetter.hpp)

#pragma once
#ifndef DISH2_VIZ_GETTERS_KINGROUPIDGETTER_HPP_INCLUDE
#define DISH2_VIZ_GETTERS_KINGROUPIDGETTER_HPP_INCLUDE

#include <functional>

#include "../../cell/Cell.hpp"
#include "../../config/cfg.hpp"
#include "../../genome/Genome.hpp"
#include "../../genome/KinGroupID.hpp"
#include "../../world/ThreadWorld.hpp"

namespace dish2 {

template<typename Spec>
class KinGroupIDGetter {

  std::reference_wrapper<const dish2::ThreadWorld<Spec>> thread_world;

public:

  using value_type = dish2::KinGroupID<Spec>;

  template< typename... Args >
  KinGroupIDGetter(
    const dish2::ThreadWorld<Spec>& thread_world_,
    Args&&...
  )
  : thread_world(thread_world_)
  {}

  const value_type Get(
    const size_t cell_idx, const size_t cardinal_idx=0
  ) const {
    if ( thread_world.get().GetCell( cell_idx ).genome.has_value() ) {
      return thread_world.get().GetCell( cell_idx ).GetGenome().kin_group_id;
    } else {
      const static Genome< Spec > genome{};
      return genome.kin_group_id;
    }
  }

  size_t GetNumCells() const { return thread_world.get().GetSize(); }

  size_t GetNumCardinals( const size_t cell_idx=0 ) const {
    return thread_world.get().GetCell( cell_idx ).GetNumCardinals();
  }

};

} // namespace dish2

#endif // #ifndef DISH2_VIZ_GETTERS_KINGROUPIDGETTER_HPP_INCLUDE