KDNode.h

// k-D node class for k-D trees
// Written by Sergey Kosov in 2017 for Project X
#pragma once

#include "types.h"

namespace DirectGraphicalModels
{
    // ================================ k-D Node Class ================================
    class CKDNode : public std::enable_shared_from_this<CKDNode>
    {
    public:
        DllExport CKDNode(Mat &key, byte value) 
            : CKDNode(key, value, lvalue_cast(std::make_pair(Mat(), Mat())), 0, 0, nullptr, nullptr) {}
        DllExport CKDNode(pair_mat_t &boundingBox, byte splitVal, int splitDim, std::shared_ptr<CKDNode> left, std::shared_ptr<CKDNode> right)
            : CKDNode(EmptyMat, 0, boundingBox, splitVal, splitDim, left, right) {}
        // Copy constructor
        DllExport CKDNode(const CKDNode &) = delete;
        // Destructor
        DllExport ~CKDNode(void) {};
        // Assignment operator
        DllExport bool                      operator=(const CKDNode)  = delete;

        DllExport void                      save(FILE *pFile) const;
        DllExport bool                      isLeaf(void) const { return (!m_pLeft && !m_pRight); }
        DllExport void                      findNearestNeighbors(const Mat &key, size_t maxNeighbors, pair_mat_t &searchBox, float &searchRadius, std::vector<std::shared_ptr<const CKDNode>> &nearestNeighbors) const;
        DllExport Mat                       getKey(void)            const { return m_key; }
        DllExport byte                      getValue(void)          const { return m_value; }
        DllExport pair_mat_t                getBoundingBox(void)    const { return isLeaf() ? std::make_pair(m_key, m_key) : m_boundingBox; }
        DllExport byte                      getSplitVal(void)       const { return m_splitVal; }
        DllExport int                       getSplitDim(void)       const { return m_splitDim; }
        DllExport std::shared_ptr<CKDNode>  Left(void)              const { return m_pLeft; }
        DllExport std::shared_ptr<CKDNode>  Right(void)             const { return m_pRight; }


    private:
        DllExport CKDNode(Mat &key, byte value, pair_mat_t &boundingBox, byte splitVal, int splitDim, std::shared_ptr<CKDNode> left, std::shared_ptr<CKDNode> right);


    private:
        Mat                                 m_key;  
        byte                                m_value;
        pair_mat_t                          m_boundingBox;
        byte                                m_splitVal; 
        int                                 m_splitDim;
        std::shared_ptr<CKDNode>            m_pLeft;
        std::shared_ptr<CKDNode>            m_pRight;
    };

}