kd_node.h

Go to the documentation of this file.

/* ============================================================
 *
 * This file is a part of digiKam
 *
 * Date        : 2019-06-08
 * Description : Node of KD-Tree for vector space partitioning
 *
 * Copyright (C) 2020 by Nghia Duong <minhnghiaduong997 at gmail dot com>
 *
 * This program is free software; you can redistribute it
 * and/or modify it under the terms of the GNU General
 * Public License as published by the Free Software Foundation;
 * either version 2, or (at your option)
 * any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * ============================================================ */
 
#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:
 
    KDNode* insert(const cv::Mat& nodePos, const int identity);
 
    cv::Mat getPosition() const;
 
    double getClosestNeighbors(QMap<double, QVector<int> >& neighborList,
                               const cv::Mat&               position,
                               float                        sqRange,
                               float                        cosThreshold,
                               int                          maxNbNeighbors) const;
    int getIdentity();
 
    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