abstractNeighborGraph.h

/*
This Source Code Form is subject to the terms of the Mozilla Public
 License, v. 2.0. If a copy of the MPL was not distributed with this
 file, You can obtain one at http://mozilla.org/MPL/2.0/.
*/
 
#pragma once
 
#include "neighborGraphTraits.h"
 
#include <cstddef> // size_t
 
#define WRITE_NEIGHBOR_GRAPH_ALIASES                                                                                    \
    using Traits         = _Traits;                            \
    using DataPoint      = typename Traits::DataPoint;         \
    using Scalar         = typename DataPoint::Scalar;         \
    using VectorType     = typename DataPoint::VectorType;     \
    using IndexType      = typename Traits::IndexType;         \
    using PointContainer = typename Traits::PointContainer;     \
    using PointContainerConstRef =                                                                                      \
        typename Traits::PointContainerConstRef;                \
    using IndexContainer    = typename Traits::IndexContainer;  \
    using IndexContainerRef = typename Traits::IndexContainerRef; 
namespace Ponca
{
    template <typename _Traits>
    struct NeighborGraphBufferBase
    {
        WRITE_NEIGHBOR_GRAPH_ALIASES
 
        PointContainer points;
        IndexContainer indices;
 
        size_t points_size{0};
        size_t indices_size{0};
 
        PONCA_MULTIARCH inline NeighborGraphBufferBase() = default;
 
        PONCA_MULTIARCH inline NeighborGraphBufferBase(PointContainerConstRef _points) : points(_points) {}
 
        PONCA_MULTIARCH inline NeighborGraphBufferBase(PointContainerConstRef _points, IndexContainerRef _indices,
                                                       const size_t _points_size, const size_t _indices_size)
            : points(_points), indices(_indices), points_size(_points_size), indices_size(_indices_size)
        {
        }
    };
 
    template <typename _Traits, template <typename> typename BufferType, typename _OneConnectedIndexQuery,
              typename _RangeIndexQuery>
    class AbstractNeighborGraph
    {
    public:
        WRITE_NEIGHBOR_GRAPH_ALIASES
        using OneConnectedIndexQuery = _OneConnectedIndexQuery;
        using RangeIndexQuery        = _RangeIndexQuery;
 
        using Buffers = BufferType<Traits>;
        static_assert(std::is_base_of_v<NeighborGraphBufferBase<Traits>, Buffers>,
                      "BufferType must inherit NeighborGraphBufferBase");
 
    protected:
        PONCA_MULTIARCH inline const IndexType* getIndexPtr() const { return Traits::getIndexRawPtr(m_bufs.indices); }
        PONCA_MULTIARCH inline IndexType* getIndexPtr() { return Traits::getIndexRawPtr(m_bufs.indices); }
 
    public:
        // Data --------------------------------------------------------------------
 
        PONCA_MULTIARCH inline AbstractNeighborGraph(const Buffers& _bufs) : m_bufs(_bufs) {}
 
        PONCA_MULTIARCH [[nodiscard]] inline IndexType edgeCount() const { return (IndexType)m_bufs.indices_size; }
        PONCA_MULTIARCH [[nodiscard]] inline IndexType pointCount() const { return (IndexType)m_bufs.points_size; }
        PONCA_MULTIARCH [[nodiscard]] inline PointContainer points() const { return m_bufs.points; };
        PONCA_MULTIARCH [[nodiscard]] inline IndexContainer edges() const { return m_bufs.indices; };
        PONCA_MULTIARCH [[nodiscard]] inline const Buffers& buffers() const { return m_bufs; }
 
        // Query -------------------------------------------------------------------
 
        PONCA_MULTIARCH [[nodiscard]] inline OneConnectedIndexQuery oneConnectedNeighbors(int index = 0) const
        {
            return OneConnectedIndexQuery(static_cast<const typename OneConnectedIndexQuery::NeighborGraph*>(this),
                                          index);
        }
 
        PONCA_MULTIARCH [[nodiscard]] inline RangeIndexQuery rangeNeighbors(int index, Scalar r) const
        {
            return RangeIndexQuery(static_cast<const typename RangeIndexQuery::NeighborGraph*>(this), r, index);
        }
 
        PONCA_MULTIARCH [[nodiscard]] inline RangeIndexQuery rangeNeighborsIndexQuery() const
        {
            return RangeIndexQuery(static_cast<const typename RangeIndexQuery::NeighborGraph*>(this), 0, 0);
        }
 
    protected:          // for friends relations
        Buffers m_bufs; 
    };
} // namespace Ponca