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 {
 
  namespace concepts = detail::concepts;

  template <uint8_t Ndim>
  class PointsHost {
  public:
    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;
 
    ALPAKA_FN_HOST int32_t size() const;
 
    ALPAKA_FN_HOST std::span<const float> coords() const;
    ALPAKA_FN_HOST std::span<float> coords();
 
    ALPAKA_FN_HOST std::span<const float> coords(size_t dim) const;
    ALPAKA_FN_HOST std::span<float> coords(size_t dim);
 
    ALPAKA_FN_HOST std::span<const float> weights() const;
    ALPAKA_FN_HOST std::span<float> weights();
 
    ALPAKA_FN_HOST std::span<const int> clusterIndexes() const;
    ALPAKA_FN_HOST std::span<int> clusterIndexes();
 
    ALPAKA_FN_HOST std::span<const int> isSeed() const;
    ALPAKA_FN_HOST std::span<int> isSeed();
 
    ALPAKA_FN_HOST const PointsView* view() const;
    ALPAKA_FN_HOST PointsView* view();
 
    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);
 
  private:
    std::optional<host_buffer<std::byte[]>> m_buffer;
    host_buffer<PointsView> m_view;
    int32_t m_size;
  };
 
}  // namespace clue
 
#include "CLUEstering/data_structures/detail/PointsHost.hpp"