knnGraphTraits.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 <Eigen/Geometry>
#include <stack>
#include <set>
 
#include "../../Common/Containers/hashset.h"
#include "../../Common/Containers/stack.h"
 
namespace Ponca
{
 
    template <typename _DataPoint>
    struct KnnGraphDefaultTraits
    {
        using DataPoint = _DataPoint;
 
    private:
        using Scalar     = typename DataPoint::Scalar;
        using VectorType = typename DataPoint::VectorType;
 
    public:
        using AabbType = Eigen::AlignedBox<Scalar, DataPoint::Dim>;
 
        // Containers
        using IndexType = int;
        using PointContainer = const std::vector<DataPoint>&;
        using IndexContainer = std::vector<IndexType>;
        using IndexContainerRef = IndexContainer&;
 
        using KnnGraphRangeSet = std::set<int>;
        using KnnGraphRangeStack = std::stack<int>;
 
        PONCA_MULTIARCH static IndexType* getIndexRawPtr(IndexContainer& idx) { return idx.data(); }
        PONCA_MULTIARCH static const IndexType* getIndexRawPtr(const IndexContainer& idx) { return idx.data(); }
    };
    template <typename _DataPoint>
    struct KnnGraphPointerTraits
    {
        enum
        {
            K_MAX_NN = 1000 
        };
        using DataPoint = _DataPoint;
 
    private:
        using Scalar     = typename DataPoint::Scalar;
        using VectorType = typename DataPoint::VectorType;
 
    public:
        using AabbType = Eigen::AlignedBox<Scalar, DataPoint::Dim>;
 
        // Containers
        using IndexType = int;
        using PointContainer = DataPoint*;
        using IndexContainer = IndexType*;
        using IndexContainerRef = IndexContainer;
 
        using KnnGraphRangeSet = HashSet<K_MAX_NN>;
        using KnnGraphRangeStack = Stack<int, K_MAX_NN>;
 
        PONCA_MULTIARCH static IndexType* getIndexRawPtr(IndexContainer& idx) { return idx; }
        PONCA_MULTIARCH static const IndexType* getIndexRawPtr(const IndexContainer& idx) { return idx; }
    };
} // namespace Ponca