AbstractClusterFinder.h

/**********************************************************************
 *
 * GEOS - Geometry Engine Open Source
 * http://geos.osgeo.org
 *
 * Copyright (C) 2020-2022 Daniel Baston
 *
 * 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.
 *
 **********************************************************************/

#ifndef GEOS_OPERATION_CLUSTER_ABSTRACTCLUSTERFINDER
#define GEOS_OPERATION_CLUSTER_ABSTRACTCLUSTERFINDER

#include <functional>
#include <memory>
#include <vector>

#include <geos/export.h>
#include <geos/index/strtree/TemplateSTRtree.h>
#include <geos/operation/cluster/Clusters.h>

// Forward declarations
namespace geos {
namespace geom {
    class Envelope;
    class Geometry;
}
namespace operation {
namespace cluster {
    class UnionFind;
}
}
}

namespace geos {
namespace operation {
namespace cluster {

class GEOS_DLL AbstractClusterFinder {

public:
    Clusters cluster(const std::vector<const geom::Geometry*>& g);

    std::vector<std::unique_ptr<geom::Geometry>> clusterToVector(std::unique_ptr<geom::Geometry> && g);

    std::vector<std::unique_ptr<geom::Geometry>> clusterToVector(const geom::Geometry& g);

    std::unique_ptr<geom::Geometry> clusterToCollection(std::unique_ptr<geom::Geometry> && g);

    std::unique_ptr<geom::Geometry> clusterToCollection(const geom::Geometry & g);

protected:
    virtual bool shouldJoin(const geom::Geometry* a, const geom::Geometry* b) = 0;

    virtual const geom::Envelope& queryEnvelope(const geom::Geometry* a) = 0;

    virtual Clusters process(const std::vector<const geom::Geometry*> & components,
                 index::strtree::TemplateSTRtree<std::size_t> & index,
                 UnionFind & uf);

private:
    static std::vector<std::unique_ptr<geom::Geometry>> getComponents(std::unique_ptr<geom::Geometry>&& g);
};



}
}
}

#endif