97 lines
2.2 KiB
C
97 lines
2.2 KiB
C
|
/**********************************************************************
|
||
|
|
*
|
||
|
|
* 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/Geometry.h>
|
||
|
|
#include <string>
|
||
|
|
#include <vector>
|
||
|
|
#include <cstdint>
|
||
|
|
|
||
|
|
// Forward declarations
|
||
|
|
namespace geos {
|
||
|
|
namespace geom {
|
||
|
|
class Coordinate;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
namespace geos {
|
||
|
|
namespace shape { // geos.shape
|
||
|
|
namespace fractal { // geos.shape.fractal
|
||
|
|
|
||
|
|
|
||
|
|
class GEOS_DLL HilbertEncoder {
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
HilbertEncoder(uint32_t p_level, geom::Envelope& extent);
|
||
|
|
uint32_t encode(const geom::Envelope* env);
|
||
|
|
static void sort(std::vector<geom::Geometry*>& geoms);
|
||
|
|
|
||
|
|
template<typename T>
|
||
|
|
static geom::Envelope getEnvelope(T begin, T end) {
|
||
|
|
geom::Envelope extent;
|
||
|
|
for (auto it = begin; it != end; ++it) {
|
||
|
|
const auto* g = *it;
|
||
|
|
if (extent.isNull())
|
||
|
|
extent = *(g->getEnvelopeInternal());
|
||
|
|
else
|
||
|
|
extent.expandToInclude(*(g->getEnvelopeInternal()));
|
||
|
|
}
|
||
|
|
|
||
|
|
return extent;
|
||
|
|
}
|
||
|
|
|
||
|
|
template<typename T>
|
||
|
|
static void sort(T begin, T end) {
|
||
|
|
auto extent = getEnvelope(begin, end);
|
||
|
|
HilbertEncoder encoder(12, extent);
|
||
|
|
HilbertComparator hilbertCompare(encoder);
|
||
|
|
std::sort(begin, end, hilbertCompare);
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
uint32_t level;
|
||
|
|
double minx;
|
||
|
|
double miny;
|
||
|
|
double strideX;
|
||
|
|
double strideY;
|
||
|
|
|
||
|
|
struct HilbertComparator {
|
||
|
|
|
||
|
|
HilbertEncoder& enc;
|
||
|
|
|
||
|
|
HilbertComparator(HilbertEncoder& e)
|
||
|
|
: enc(e) {};
|
||
|
|
|
||
|
|
bool
|
||
|
|
operator()(const geom::Geometry* a, const geom::Geometry* b)
|
||
|
|
{
|
||
|
|
return enc.encode(a->getEnvelopeInternal()) > enc.encode(b->getEnvelopeInternal());
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
} // namespace geos.shape.fractal
|
||
|
|
} // namespace geos.shape
|
||
|
|
} // namespace geos
|
||
|
|
|
||
|
|
|
||
|
|
|