STRtree.h

/**********************************************************************
 *
 * GEOS - Geometry Engine Open Source
 * http://geos.osgeo.org
 *
 * Copyright (C) 2006 Refractions Research Inc.
 *
 * 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.
 *
 **********************************************************************
 *
 * Last port: index/strtree/STRtree.java rev. 1.11
 *
 **********************************************************************/
 
#pragma once
 
#include <geos/export.h>
#include <geos/index/strtree/ItemDistance.h>
#include <geos/index/strtree/BoundablePair.h>
#include <geos/index/strtree/AbstractSTRtree.h> // for inheritance
#include <geos/index/SpatialIndex.h> // for inheritance
#include <geos/geom/Envelope.h> // for inlines
 
#include <vector>
 
#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
 
// Forward declarations
namespace geos {
namespace index {
namespace strtree {
class Boundable;
}
}
}
 
namespace geos {
namespace index { // geos::index
namespace strtree { // geos::index::strtree
 
class GEOS_DLL STRtree: public AbstractSTRtree, public SpatialIndex {
    using AbstractSTRtree::insert;
    using AbstractSTRtree::query;
 
private:
    class GEOS_DLL STRIntersectsOp: public AbstractSTRtree::IntersectsOp {
    public:
        bool intersects(const void* aBounds, const void* bBounds) override;
    };
 
    std::unique_ptr<BoundableList> createParentBoundables(BoundableList* childBoundables, int newLevel) override;
 
    std::unique_ptr<BoundableList> createParentBoundablesFromVerticalSlices(std::vector<BoundableList*>* verticalSlices,
            int newLevel);
 
    STRIntersectsOp intersectsOp;
 
    std::unique_ptr<BoundableList> sortBoundablesY(const BoundableList* input);
    std::unique_ptr<BoundableList> sortBoundablesX(const BoundableList* input);
 
    std::unique_ptr<BoundableList> createParentBoundablesFromVerticalSlice(
        BoundableList* childBoundables,
        int newLevel);
 
    std::vector<BoundableList*>* verticalSlices(
        BoundableList* childBoundables,
        std::size_t sliceCount);
 
    bool isWithinDistance(BoundablePair* initBndPair, double maxDistance);
 
protected:
 
    AbstractNode* createNode(int level) override;
 
    IntersectsOp*
    getIntersectsOp() override
    {
        return &intersectsOp;
    }
 
public:
 
    ~STRtree() override = default;
 
    STRtree(std::size_t nodeCapacity = 10);
 
    void insert(const geom::Envelope* itemEnv, void* item) override;
 
    //static double centreX(const geom::Envelope *e);
 
    static double
    avg(double a, double b)
    {
        return (a + b) / 2.0;
    }
 
    static double
    centreY(const geom::Envelope* e)
    {
        return STRtree::avg(e->getMinY(), e->getMaxY());
    }
 
    void
    query(const geom::Envelope* searchEnv, std::vector<void*>& matches) override
    {
        AbstractSTRtree::query(searchEnv, matches);
    }
 
    void
    query(const geom::Envelope* searchEnv, ItemVisitor& visitor) override
    {
        return AbstractSTRtree::query(searchEnv, visitor);
    }
 
    std::pair<const void*, const void*> nearestNeighbour(ItemDistance* itemDist);
    const void* nearestNeighbour(const geom::Envelope* env, const void* item, ItemDistance* itemDist);
    std::pair<const void*, const void*> nearestNeighbour(STRtree* tree, ItemDistance* itemDist);
    std::pair<const void*, const void*> nearestNeighbour(BoundablePair* initBndPair);
    std::pair<const void*, const void*> nearestNeighbour(BoundablePair* initBndPair, double maxDistance);
 
    bool
    remove(const geom::Envelope* itemEnv, void* item) override
    {
        return AbstractSTRtree::remove(itemEnv, item);
    }
 
    bool isWithinDistance(STRtree* tree, ItemDistance* itemDist, double maxDistance);
 
};
 
} // namespace geos::index::strtree
} // namespace geos::index
} // namespace geos
 
 
#ifdef _MSC_VER
#pragma warning(pop)
#endif