Examples with Description io.prestosql.spi.function.Description used on opensource projects

Example 56 with Description

use of io.prestosql.spi.function.Description in project hetu-core by openlookeng.

the class BingTileFunctions method geometryToBingTiles.

@Description("Given a geometry and a zoom level, returns the minimum set of Bing tiles that fully covers that geometry")
@ScalarFunction("geometry_to_bing_tiles")
@SqlType("array(" + BingTileType.NAME + ")")
public static Block geometryToBingTiles(@SqlType(GEOMETRY_TYPE_NAME) Slice input, @SqlType(StandardTypes.INTEGER) long zoomLevelInput) {
    checkZoomLevel(zoomLevelInput);
    int zoomLevel = toIntExact(zoomLevelInput);
    OGCGeometry ogcGeometry = deserialize(input);
    if (ogcGeometry.isEmpty()) {
        return EMPTY_TILE_ARRAY;
    }
    Envelope envelope = getEnvelope(ogcGeometry);
    checkLatitude(envelope.getYMin(), LATITUDE_SPAN_OUT_OF_RANGE);
    checkLatitude(envelope.getYMax(), LATITUDE_SPAN_OUT_OF_RANGE);
    checkLongitude(envelope.getXMin(), LONGITUDE_SPAN_OUT_OF_RANGE);
    checkLongitude(envelope.getXMax(), LONGITUDE_SPAN_OUT_OF_RANGE);
    boolean pointOrRectangle = isPointOrRectangle(ogcGeometry, envelope);
    BingTile leftUpperTile = latitudeLongitudeToTile(envelope.getYMax(), envelope.getXMin(), zoomLevel);
    BingTile rightLowerTile = getTileCoveringLowerRightCorner(envelope, zoomLevel);
    // XY coordinates start at (0,0) in the left upper corner and increase left to right and top to bottom
    long tileCount = (long) (rightLowerTile.getX() - leftUpperTile.getX() + 1) * (rightLowerTile.getY() - leftUpperTile.getY() + 1);
    checkGeometryToBingTilesLimits(ogcGeometry, envelope, pointOrRectangle, tileCount, zoomLevel);
    BlockBuilder blockBuilder = BIGINT.createBlockBuilder(null, toIntExact(tileCount));
    if (pointOrRectangle || zoomLevel <= OPTIMIZED_TILING_MIN_ZOOM_LEVEL) {
        // all tiles covering the bounding box intersect the geometry.
        for (int x = leftUpperTile.getX(); x <= rightLowerTile.getX(); x++) {
            for (int y = leftUpperTile.getY(); y <= rightLowerTile.getY(); y++) {
                BingTile tile = BingTile.fromCoordinates(x, y, zoomLevel);
                if (pointOrRectangle || !disjoint(tileToEnvelope(tile), ogcGeometry)) {
                    BIGINT.writeLong(blockBuilder, tile.encode());
                }
            }
        }
    } else {
        // Intersection checks above are expensive. The logic below attempts to reduce the number
        // of these checks. The idea is to identify large tiles which are fully covered by the
        // geometry. For each such tile, we can cheaply compute all the containing tiles at
        // the right zoom level and append them to results in bulk. This way we perform a single
        // containment check instead of 2 to the power of level delta intersection checks, where
        // level delta is the difference between the desired zoom level and level of the large
        // tile covered by the geometry.
        BingTile[] tiles = getTilesInBetween(leftUpperTile, rightLowerTile, OPTIMIZED_TILING_MIN_ZOOM_LEVEL);
        for (BingTile tile : tiles) {
            appendIntersectingSubtiles(ogcGeometry, zoomLevel, tile, blockBuilder);
        }
    }
    return blockBuilder.build();
}

Example 57 with Description

use of io.prestosql.spi.function.Description in project hetu-core by openlookeng.

the class GeoFunctions method stInteriorRings.

@SqlNullable
@Description("Returns an array of interior rings of a polygon")
@ScalarFunction("ST_InteriorRings")
@SqlType("array(" + GEOMETRY_TYPE_NAME + ")")
public static Block stInteriorRings(@SqlType(GEOMETRY_TYPE_NAME) Slice input) {
    OGCGeometry geometry = deserialize(input);
    validateType("ST_InteriorRings", geometry, EnumSet.of(POLYGON));
    if (geometry.isEmpty()) {
        return null;
    }
    OGCPolygon polygon = (OGCPolygon) geometry;
    BlockBuilder blockBuilder = GEOMETRY.createBlockBuilder(null, polygon.numInteriorRing());
    for (int i = 0; i < polygon.numInteriorRing(); i++) {
        GEOMETRY.writeSlice(blockBuilder, serialize(polygon.interiorRingN(i)));
    }
    return blockBuilder.build();
}

Example 58 with Description

use of io.prestosql.spi.function.Description in project hetu-core by openlookeng.

the class GeoFunctions method stGeometries.

@SqlNullable
@Description("Returns an array of geometries in the specified collection")
@ScalarFunction("ST_Geometries")
@SqlType("array(" + GEOMETRY_TYPE_NAME + ")")
public static Block stGeometries(@SqlType(GEOMETRY_TYPE_NAME) Slice input) {
    OGCGeometry geometry = deserialize(input);
    if (geometry.isEmpty()) {
        return null;
    }
    GeometryType type = GeometryType.getForEsriGeometryType(geometry.geometryType());
    if (!type.isMultitype()) {
        BlockBuilder blockBuilder = GEOMETRY.createBlockBuilder(null, 1);
        GEOMETRY.writeSlice(blockBuilder, serialize(geometry));
        return blockBuilder.build();
    }
    OGCGeometryCollection collection = (OGCGeometryCollection) geometry;
    BlockBuilder blockBuilder = GEOMETRY.createBlockBuilder(null, collection.numGeometries());
    for (int i = 0; i < collection.numGeometries(); i++) {
        GEOMETRY.writeSlice(blockBuilder, serialize(collection.geometryN(i)));
    }
    return blockBuilder.build();
}

Example 59 with Description

use of io.prestosql.spi.function.Description in project hetu-core by openlookeng.

the class GeoFunctions method stX.

@SqlNullable
@Description("Return the X coordinate of the point")
@ScalarFunction("ST_X")
@SqlType(DOUBLE)
public static Double stX(@SqlType(GEOMETRY_TYPE_NAME) Slice input) {
    OGCGeometry geometry = deserialize(input);
    validateType("ST_X", geometry, EnumSet.of(POINT));
    if (geometry.isEmpty()) {
        return null;
    }
    return ((OGCPoint) geometry).X();
}

Example 60 with Description

use of io.prestosql.spi.function.Description in project hetu-core by openlookeng.

the class GeoFunctions method stPoints.

@SqlNullable
@Description("Returns an array of points in a linestring")
@ScalarFunction("ST_Points")
@SqlType("array(" + GEOMETRY_TYPE_NAME + ")")
public static Block stPoints(@SqlType(GEOMETRY_TYPE_NAME) Slice input) {
    OGCGeometry geometry = deserialize(input);
    validateType("ST_Points", geometry, EnumSet.of(LINE_STRING));
    if (geometry.isEmpty()) {
        return null;
    }
    MultiPath lines = (MultiPath) geometry.getEsriGeometry();
    BlockBuilder blockBuilder = GEOMETRY.createBlockBuilder(null, lines.getPointCount());
    for (int i = 0; i < lines.getPointCount(); i++) {
        GEOMETRY.writeSlice(blockBuilder, serialize(createFromEsriGeometry(lines.getPoint(i), null)));
    }
    return blockBuilder.build();
}