PointsHost.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "CLUEstering/data_structures/internal/PointsCommon.hpp"
#include "CLUEstering/internal/alpaka/memory.hpp"
 
#include <optional>
#include <ranges>
#include <span>
#include <alpaka/alpaka.hpp>
 
namespace clue {

  template <uint8_t Ndim>
  class PointsHost : public internal::points_interface<PointsHost<Ndim>> {
  private:
    std::optional<host_buffer<std::byte[]>> m_buffer;
    PointsView m_view;
    int32_t m_size;
 
  public:
    class Point {
      std::array<float, Ndim> m_coordinates;
      float m_weight;
      int m_clusterIndex;
 
    public:
      Point(const std::array<float, Ndim>& coordinates, float weight, int cluster_index);
      float operator[](size_t dim) const;
 
      float weight() const;
      float cluster_index() const;
    };
 
    template <concepts::queue TQueue>
    PointsHost(TQueue& queue, int32_t n_points);
 
    template <concepts::queue TQueue>
    PointsHost(TQueue& queue, int32_t n_points, std::span<std::byte> buffer);
 
    template <concepts::queue TQueue, std::ranges::contiguous_range... TBuffers>
      requires(sizeof...(TBuffers) == 2 || sizeof...(TBuffers) == 4)
    PointsHost(TQueue& queue, int32_t n_points, TBuffers&&... buffers);
 
    template <concepts::queue TQueue, concepts::contiguous_raw_data... TBuffers>
      requires(sizeof...(TBuffers) == 2 || sizeof...(TBuffers) == 4)
    PointsHost(TQueue& queue, int32_t n_points, TBuffers... buffers);
 
    PointsHost(const PointsHost&) = delete;
    PointsHost& operator=(const PointsHost&) = delete;
    PointsHost(PointsHost&&) = default;
    PointsHost& operator=(PointsHost&&) = default;
    ~PointsHost() = default;
 
#ifdef CLUE_BUILD_DOXYGEN
    ALPAKA_FN_HOST int32_t size() const;
    ALPAKA_FN_HOST auto coords() const;
    ALPAKA_FN_HOST auto coords();
    ALPAKA_FN_HOST auto coords(size_t dim) const;
    ALPAKA_FN_HOST auto coords(size_t dim);
    ALPAKA_FN_HOST auto weights() const;
    ALPAKA_FN_HOST auto weights();
    ALPAKA_FN_HOST auto clusterIndexes() const;
    ALPAKA_FN_HOST auto clusterIndexes();
    ALPAKA_FN_HOST auto isSeed() const;
    ALPAKA_FN_HOST auto isSeed();
    ALPAKA_FN_HOST const auto& view() const;
    ALPAKA_FN_HOST auto& view();
#endif
 
    Point operator[](std::size_t idx) const;
 
  private:
    inline static constexpr uint8_t Ndim_ = Ndim;
 
    template <concepts::queue _TQueue, uint8_t _Ndim, concepts::device _TDev>
    friend void copyToHost(_TQueue& queue,
                           PointsHost<_Ndim>& h_points,
                           const PointsDevice<_Ndim, _TDev>& d_points);
    template <concepts::queue _TQueue, uint8_t _Ndim, concepts::device _TDev>
    friend void copyToDevice(_TQueue& queue,
                             PointsDevice<_Ndim, _TDev>& d_points,
                             const PointsHost<_Ndim>& h_points);
    friend struct internal::points_interface<PointsHost<Ndim>>;
  };
 
}  // namespace clue
 
#include "CLUEstering/data_structures/detail/PointsHost.hpp"
#include "CLUEstering/data_structures/detail/Point.hpp"