Examples with Rectangle io.trino.geospatial.Rectangle used on opensource projects

Example 6 with Rectangle

use of io.trino.geospatial.Rectangle in project trino by trinodb.

the class SpatialPartitioningInternalAggregateFunction method output.

@OutputFunction(StandardTypes.VARCHAR)
public static void output(SpatialPartitioningState state, BlockBuilder out) {
    if (state.getCount() == 0) {
        out.appendNull();
        return;
    }
    List<Rectangle> samples = state.getSamples();
    int partitionCount = state.getPartitionCount();
    int maxItemsPerNode = (samples.size() + partitionCount - 1) / partitionCount;
    Rectangle envelope = state.getExtent();
    // Add a small buffer on the right and upper sides
    Rectangle paddedExtent = new Rectangle(envelope.getXMin(), envelope.getYMin(), Math.nextUp(envelope.getXMax()), Math.nextUp(envelope.getYMax()));
    VARCHAR.writeString(out, KdbTreeUtils.toJson(buildKdbTree(maxItemsPerNode, paddedExtent, samples)));
}

Example 7 with Rectangle

use of io.trino.geospatial.Rectangle in project trino by trinodb.

the class PagesRTreeIndex method testReferencePoint.

private boolean testReferencePoint(Envelope probeEnvelope, OGCGeometry buildGeometry, int partition) {
    Envelope buildEnvelope = getEnvelope(buildGeometry);
    Envelope intersection = buildEnvelope.intersection(probeEnvelope);
    if (intersection.isNull()) {
        return false;
    }
    Rectangle extent = partitions.get(partition);
    double x = intersection.getMinX();
    double y = intersection.getMinY();
    return x >= extent.getXMin() && x < extent.getXMax() && y >= extent.getYMin() && y < extent.getYMax();
}

Example 8 with Rectangle

use of io.trino.geospatial.Rectangle in project trino by trinodb.

the class GeoFunctions method spatialPartitions.

@ScalarFunction
@SqlNullable
@Description("Returns an array of spatial partition IDs for a geometry representing a set of points within specified distance from the input geometry")
@SqlType("array(integer)")
public static Block spatialPartitions(@SqlType(KdbTreeType.NAME) Object kdbTree, @SqlType(GEOMETRY_TYPE_NAME) Slice geometry, @SqlType(DOUBLE) double distance) {
    if (isNaN(distance)) {
        throw new TrinoException(INVALID_FUNCTION_ARGUMENT, "distance is NaN");
    }
    if (isInfinite(distance)) {
        throw new TrinoException(INVALID_FUNCTION_ARGUMENT, "distance is infinite");
    }
    if (distance < 0) {
        throw new TrinoException(INVALID_FUNCTION_ARGUMENT, "distance is negative");
    }
    Envelope envelope = deserializeEnvelope(geometry);
    if (envelope.isEmpty()) {
        return null;
    }
    Rectangle expandedEnvelope2D = new Rectangle(envelope.getXMin() - distance, envelope.getYMin() - distance, envelope.getXMax() + distance, envelope.getYMax() + distance);
    return spatialPartitions((KdbTree) kdbTree, expandedEnvelope2D);
}