File indexing completed on 2025-03-09 03:55:01

 /* ============================================================
  *
  * This file is a part of digiKam
  *
  * Date        : 2019-06-08
  * Description : Node of KD-Tree for vector space partitioning
  *
  * SPDX-FileCopyrightText: 2020 by Nghia Duong <minhnghiaduong997 at gmail dot com>
  *
  * SPDX-License-Identifier: GPL-2.0-or-later
  *
  * ============================================================ */
 
 #ifndef DIGIKAM_KD_NODE_H
 #define DIGIKAM_KD_NODE_H
 
 // C++ includes
 
 #include <vector>
 
 // Qt includes
 
 #include <QVector>
 #include <QMap>
 
 // Local include
 
 #include "digikam_opencv.h"
 
 namespace Digikam
 {
 
 class KDNode
 {
 public:
 
     explicit KDNode(const cv::Mat& nodePos,
                     const int      identity,
                     int            splitAxis,
                     int            dimension);
     ~KDNode();
 
 public:
 
     static float sqrDistance(const float* const pos1, const float* const pos2, int dimension);
     static float cosDistance(const float* const pos1, const float* const pos2, int dimension);
 
 public:
 
     /**
      * Insert a new node to the sub-tree
      */
     KDNode* insert(const cv::Mat& nodePos, const int identity);
 
     /**
      * Return position vector of a node
      */
     cv::Mat getPosition() const;
 
     /**
      * Return a list of closest neighbors, limited by maxNbNeighbors and sqRange
      */
     double getClosestNeighbors(QMap<double, QVector<int> >& neighborList,
                                const cv::Mat&               position,
                                float                        sqRange,
                                float                        cosThreshold,
                                int                          maxNbNeighbors) const;
     /**
      * Return identity of the node
      */
     int getIdentity();
 
     /**
      * Set database entry ID of the node
      */
     void setNodeId(int id);
 
 private:
 
     void updateRange(const cv::Mat&);
 
     KDNode* findParent(const cv::Mat& nodePos);
 
 private:
 
     // Disable
     KDNode(const KDNode&)            = delete;
     KDNode& operator=(const KDNode&) = delete;
 
 private:
 
     class Private;
     Private* d;
 };
 
 } // namespace Digikam
 
 #endif // DIGIKAM_KD_NODE_H