kdTreeKNearestIterator.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 "../../indexSquaredDistance.h"
#include "../../../Common/Containers/limitedPriorityQueue.h"
#include <cstddef>
 
namespace Ponca
{
 
    template <typename Index, typename DataPoint, int MAX_KNN_SIZE>
    class KdTreeKNearestIterator
    {
    public:
        using iterator_category = std::input_iterator_tag;
        using difference_type   = std::ptrdiff_t;
        using value_type        = Index;
        using pointer           = Index*;
        using reference         = const Index&;
 
        using Scalar   = typename DataPoint::Scalar;
        using Iterator = typename LimitedPriorityQueue<IndexSquaredDistance<Index, Scalar>, MAX_KNN_SIZE>::iterator;
 
        PONCA_MULTIARCH inline KdTreeKNearestIterator() = default;
        PONCA_MULTIARCH inline KdTreeKNearestIterator(const Iterator& iterator) : m_iterator(iterator) {}
        PONCA_MULTIARCH virtual inline ~KdTreeKNearestIterator() = default;
 
    public:
        PONCA_MULTIARCH inline bool operator!=(const KdTreeKNearestIterator& other) const
        {
            return m_iterator != other.m_iterator;
        }
 
        PONCA_MULTIARCH inline bool operator==(const KdTreeKNearestIterator& other) const
        {
            return m_iterator == other.m_iterator;
        }
 
        PONCA_MULTIARCH inline KdTreeKNearestIterator& operator++()
        {
            ++m_iterator;
            return *this;
        }
 
        PONCA_MULTIARCH inline KdTreeKNearestIterator operator++(int)
        {
            KdTreeKNearestIterator tmp = *this;
            ++m_iterator;
            return tmp;
        }
 
        PONCA_MULTIARCH inline void operator+=(int i) { m_iterator += i; }
 
        PONCA_MULTIARCH inline reference operator*() const { return const_cast<reference>(m_iterator->index); }
 
    protected:
        Iterator m_iterator;
    };
} // namespace Ponca