kdTreeNearestQueries.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 "kdTreeQuery.h"
#include "../../query.h"
#include "../Iterator/kdTreeNearestIterator.h"
 
namespace Ponca {
 
template <typename Traits,
          template <typename> typename IteratorType,
          typename QueryType>
class KdTreeNearestQueryBase : public KdTreeQuery<Traits>, public QueryType
{
public:
    using DataPoint      = typename Traits::DataPoint;
    using IndexType      = typename Traits::IndexType;
    using Scalar         = typename DataPoint::Scalar;
    using VectorType     = typename DataPoint::VectorType;
    using QueryAccelType = KdTreeQuery<Traits>;
    using Iterator       = IteratorType<typename Traits::IndexType>;
 
    PONCA_MULTIARCH KdTreeNearestQueryBase(const StaticKdTreeBase<Traits>* kdtree, typename QueryType::InputType input) :
            KdTreeQuery<Traits>(kdtree), QueryType(input){}
 
    PONCA_MULTIARCH inline Iterator begin(){
        QueryAccelType::reset();
        QueryType::reset();
        this->search();
        return Iterator(QueryType::m_nearest);
    }
 
    PONCA_MULTIARCH inline Iterator end(){
        return Iterator(QueryType::m_nearest + 1);
    }
 
protected:
    PONCA_MULTIARCH inline void search(){
        KdTreeQuery<Traits>::searchInternal(QueryType::template getInputPosition<VectorType>(QueryAccelType::m_kdtree->points()),
                                             [](IndexType, IndexType){},
                                             [this](){return QueryType::descentDistanceThreshold();},
                                             [this](IndexType idx){return QueryType::skipIndexFunctor(idx);},
                                             [this](IndexType idx, IndexType, Scalar d)
                                             {
                                                 QueryType::m_nearest = idx;
                                                 QueryType::m_squared_distance = d;
                                                 return false;
                                             }
        );
    }
};
template <typename Traits>
using KdTreeNearestIndexQuery = KdTreeNearestQueryBase< Traits, KdTreeNearestIterator,
                                NearestIndexQuery<typename Traits::IndexType, typename Traits::DataPoint::Scalar>>;
 
template <typename Traits>
using KdTreeNearestPointQuery = KdTreeNearestQueryBase< Traits, KdTreeNearestIterator,
                                NearestPointQuery<typename Traits::IndexType, typename Traits::DataPoint>>;
} // namespace ponca