KdTree.h

/**********************************************************************
 *
 * GEOS - Geometry Engine Open Source
 * http://geos.osgeo.org
 *
 * Copyright (C) 2020 Paul Ramsey <pramsey@cleverelephant.ca>
 *
 * This is free software; you can redistribute and/or modify it under
 * the terms of the GNU Lesser General Public Licence as published
 * by the Free Software Foundation.
 * See the COPYING file for more information.
 *
 **********************************************************************/
 
#pragma once
 
#include <geos/export.h>
#include <geos/geom/Envelope.h>
#include <geos/index/kdtree/KdNodeVisitor.h>
#include <geos/index/kdtree/KdNode.h>
 
#include <vector>
#include <deque>
 
#ifdef _MSC_VER
#pragma warning(push)
#pragma warning(disable: 4251) // warning C4251: needs to have dll-interface to be used by clients of class
#endif
 
 
namespace geos {
namespace index { // geos::index
namespace kdtree { // geos::index::kdtree
 
class GEOS_DLL KdTree {
 
private:
 
    std::deque<KdNode> nodeQue;
    KdNode *root;
    std::size_t numberOfNodes;
    double tolerance;
 
    KdNode* findBestMatchNode(const geom::Coordinate& p);
    KdNode* insertExact(const geom::Coordinate& p, void* data);
 
    static void queryNode(KdNode* currentNode, const geom::Envelope& queryEnv, bool odd, KdNodeVisitor& visitor);
    static KdNode* queryNodePoint(KdNode* currentNode, const geom::Coordinate& queryPt, bool odd);
 
    KdNode* createNode(const geom::Coordinate& p, void* data);
 
 
    class BestMatchVisitor : public KdNodeVisitor {
    public:
        BestMatchVisitor(const geom::Coordinate& p_p, double p_tolerance);
        geom::Envelope queryEnvelope();
        KdNode* getNode();
        void visit(KdNode* node) override;
 
        // Declare type as noncopyable
        BestMatchVisitor(const BestMatchVisitor& other) = delete;
        BestMatchVisitor& operator=(const BestMatchVisitor& rhs) = delete;
 
    private:
        // Members
        double tolerance;
        KdNode* matchNode;
        double matchDist;
        const geom::Coordinate& p;
    };
 
    class AccumulatingVisitor : public KdNodeVisitor {
    public:
        explicit AccumulatingVisitor(std::vector<KdNode*>& p_nodeList) :
            nodeList(p_nodeList) {};
        void visit(KdNode* node) override { nodeList.push_back(node); }
 
        // Declare type as noncopyable
        AccumulatingVisitor(const AccumulatingVisitor& other) = delete;
        AccumulatingVisitor& operator=(const AccumulatingVisitor& rhs) = delete;
 
    private:
        // Members
        std::vector<KdNode*>& nodeList;
    };
 
 
 
public:
 
    static std::vector<geom::Coordinate> toCoordinates(const std::vector<KdNode*>& kdNodes);
 
    static std::vector<geom::Coordinate> toCoordinates(const std::vector<KdNode*>& kdNodes, bool includeRepeated);
 
    KdTree() :
        root(nullptr),
        numberOfNodes(0),
        tolerance(0.0)
        {};
 
    explicit KdTree(double p_tolerance) :
        root(nullptr),
        numberOfNodes(0),
        tolerance(p_tolerance)
        {};
 
    bool isEmpty() { return root == nullptr; }
 
    KdNode* insert(const geom::Coordinate& p);
    KdNode* insert(const geom::Coordinate& p, void* data);
 
    void query(const geom::Envelope& queryEnv, KdNodeVisitor& visitor);
 
    std::vector<KdNode*> query(const geom::Envelope& queryEnv);
 
    void query(const geom::Envelope& queryEnv, std::vector<KdNode*>& result);
 
    KdNode* query(const geom::Coordinate& queryPt);
 
};
 
} // namespace geos::index::kdtree
} // namespace geos::index
} // namespace geos
 
#ifdef _MSC_VER
#pragma warning(pop)
#endif