KDTree.h

#pragma once

 
#include "TRect.h"

namespace FLImaging
{
    namespace Base
    {
        template<typename T>
        class CFLArray;
 
        template<template<typename> class T1, typename T2>
        class CInternalKDTree;

        template<template<typename> class T1, typename T2>
        class FL_EXPORT CKDTree : public CFLBase
        {
        public:
            CKDTree();

            CKDTree(const CFLArray<T1<T2>>& flaVertices);

            CKDTree(const CFLArray<T1<T2>>* pFlaVertices);

            virtual ~CKDTree();

            const CResult Clear();

            const CResult Build(const CFLArray<T1<T2>>& flaVertices);

            const CResult Build(const CFLArray<T1<T2>>* pFlaVertices);

            int64_t GetCount() const;

            const T1<T2>& GetNearestPoint(const T1<T2>& tpPoint) const;

            size_t GetNearestIndex(const T1<T2>& tpPoint) const;

            std::pair<T1<T2>, size_t> GetNearestPointAndIndex(const T1<T2>& tpPoint) const;
 
 

            const CResult GetNearestNeighborsPoints(const T1<T2>& tpPoint, int64_t i64NearestNeighborsCount, CFLArray<T1<T2>>& flaResultPoints) const;

            const CResult GetNearestNeighborsIndices(const T1<T2>& tpPoint, int64_t i64NearestNeighborsCount, CFLArray<uint64_t>& flaResultIndices) const;

            const CResult GetNearestNeighborsIndices(const T1<T2>& tpPoint, int64_t i64NearestNeighborsCount, CFLArray<uint32_t>& flaResultIndices) const;

            const CResult GetNearestNeighborsPointsAndIndices(const T1<T2>& tpPoint, int64_t i64NearestNeighborsCount, CFLArray<T1<T2>>& flaResultPoints, CFLArray<uint64_t>& flaResultIndices) const;

            const CResult GetNearestNeighborsPointsAndIndices(const T1<T2>& tpPoint, int64_t i64NearestNeighborsCount, CFLArray<T1<T2>>& flaResultPoints, CFLArray<uint32_t>& flaResultIndices) const;
 

            const CResult GetPointsInRange(const T1<T2>& tpLowerBound, const T1<T2>& tpUpperBound, CFLArray<T1<T2>>& flaResultPoints, int64_t i64Count = -1) const;

            const CResult GetIndicesInRange(const T1<T2>& tpLowerBound, const T1<T2>& tpUpperBound, CFLArray<uint64_t>& flaResultIndices, int64_t i64Count = -1) const;

            const CResult GetIndicesInRange(const T1<T2>& tpLowerBound, const T1<T2>& tpUpperBound, CFLArray<uint32_t>& flaResultIndices, int64_t i64Count = -1) const;

            const CResult GetPointsAndIndicesInRange(const T1<T2>& tpLowerBound, const T1<T2>& tpUpperBound, CFLArray<T1<T2>>& flaResultPoints, CFLArray<uint64_t>& flaResultIndices, int64_t i64Count = -1) const;

            const CResult GetPointsAndIndicesInRange(const T1<T2>& tpLowerBound, const T1<T2>& tpUpperBound, CFLArray<T1<T2>>& flaResultPoints, CFLArray<uint32_t>& flaResultIndices, int64_t i64Count = -1) const;
 
 
 
 

            const CResult GetPointsInRadius(const T1<T2>& tpCenter, T2 tRadius, CFLArray<T1<T2>>& flaResultPoints, int64_t i64Count = -1) const;

            const CResult GetIndicesInRadius(const T1<T2>& tpCenter, T2 tRadius, CFLArray<uint64_t>& flaResultIndices, int64_t i64Count = -1) const;

            const CResult GetIndicesInRadius(const T1<T2>& tpCenter, T2 tRadius, CFLArray<uint32_t>& flaResultIndices, int64_t i64Count = -1) const;

            const CResult GetPointsAndIndicesInRadius(const T1<T2>& tpCenter, T2 tRadius, CFLArray<T1<T2>>& flaResultPoints, CFLArray<uint64_t>& flaResultIndices, int64_t i64Count = -1) const;

            const CResult GetPointsAndIndicesInRadius(const T1<T2>& tpCenter, T2 tRadius, CFLArray<T1<T2>>& flaResultPoints, CFLArray<uint32_t>& flaResultIndices, int64_t i64Count = -1) const;
 
 
 
 
 

            const CResult OperateAdd(const T1<T2>& tpVector);

            const CResult OperateMultiply(const T1<T2>& tpVector);
 
            DeclareGetClassType();
 
 
        protected:
 
            CInternalKDTree<T1, T2>* m_pInternal;
 
            friend class CInternalKDTree<T1, T2>;
        };
    }
}