PointsDevice.hpp

Go to the documentation of this file.

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

  template <uint8_t Ndim, concepts::device TDev = clue::Device>
  class PointsDevice : public internal::points_interface<PointsDevice<Ndim, TDev>> {
  private:
    device_buffer<TDev, std::byte[]> m_buffer;
    PointsView m_view;
    int32_t m_size;
 
  public:
    template <concepts::queue TQueue>
    PointsDevice(TQueue& queue, int32_t n_points);

    template <concepts::queue TQueue>
    PointsDevice(TQueue& queue, int32_t n_points, std::span<std::byte> buffer);

    template <concepts::queue TQueue, concepts::contiguous_raw_data... TBuffers>
      requires(sizeof...(TBuffers) == 2 || sizeof...(TBuffers) == 4)
    PointsDevice(TQueue& queue, int32_t n_points, TBuffers... buffers);
 
    PointsDevice(const PointsDevice&) = delete;
    PointsDevice& operator=(const PointsDevice&) = delete;
    PointsDevice(PointsDevice&&) = default;
    PointsDevice& operator=(PointsDevice&&) = default;
    ~PointsDevice() = 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
 
    ALPAKA_FN_HOST auto rho() const;
    ALPAKA_FN_HOST auto rho();
 
    ALPAKA_FN_HOST auto delta() const;
    ALPAKA_FN_HOST auto delta();
 
    ALPAKA_FN_HOST auto nearestHigher() const;
    ALPAKA_FN_HOST auto nearestHigher();
 
  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<PointsDevice<Ndim, TDev>>;
  };
 
}  // namespace clue
 
#include "CLUEstering/data_structures/detail/PointsDevice.hpp"