Examples with OGCLineString com.esri.core.geometry.ogc.OGCLineString used on opensource projects

Example 1 with OGCLineString

use of com.esri.core.geometry.ogc.OGCLineString in project pigeon by aseldawy.

the class Connect method exec.

@Override
public DataByteArray exec(Tuple b) throws IOException {
    try {
        // Read information from input
        Iterator<Tuple> firstPointIdIter = ((DataBag) b.get(0)).iterator();
        Iterator<Tuple> lastPointIdIter = ((DataBag) b.get(1)).iterator();
        Iterator<Tuple> shapesIter = ((DataBag) b.get(2)).iterator();
        // Shapes that are created after connected line segments
        Vector<OGCGeometry> createdShapes = new Vector<OGCGeometry>();
        Vector<OGCLineString> linestrings = new Vector<OGCLineString>();
        Vector<Long> firstPointId = new Vector<Long>();
        Vector<Long> lastPointId = new Vector<Long>();
        while (firstPointIdIter.hasNext() && lastPointIdIter.hasNext() && shapesIter.hasNext()) {
            OGCGeometry geom = geometryParser.parseGeom(shapesIter.next().get(0));
            long first_point_id = (Long) firstPointIdIter.next().get(0);
            long last_point_id = (Long) lastPointIdIter.next().get(0);
            if (geom.isEmpty()) {
            // Skip empty geometries
            } else if (geom instanceof OGCPolygon) {
                // Copy to output directly. Polygons cannot be connected to other shapes.
                createdShapes.add(geom);
            } else if (geom instanceof OGCLineString) {
                linestrings.add((OGCLineString) geom);
                firstPointId.add(first_point_id);
                lastPointId.add(last_point_id);
            } else {
                throw new GeoException("Cannot connect shapes of type " + geom.getClass());
            }
        }
        if (firstPointIdIter.hasNext() || lastPointIdIter.hasNext() || shapesIter.hasNext()) {
            throw new ExecException("All parameters should be of the same size (" + firstPointId.size() + "," + lastPointId.size() + "," + linestrings.size() + ")");
        }
        // Stores an ordered list of line segments in current connected block
        Vector<OGCLineString> connected_lines = new Vector<OGCLineString>();
        // Total number of points in all visited linestrings
        int sumPoints = 0;
        // Which linestrings to reverse upon connection
        Vector<Boolean> reverse = new Vector<Boolean>();
        long first_point_id = -1;
        long last_point_id = -1;
        // Reorder linestrings to form a contiguous list of connected linestrings
        while (!linestrings.isEmpty()) {
            // Loop invariant:
            // At the beginning of each iteration, the lines in connected_lines are connected.
            // In each iteration, we move one linestring from linestrings to connected_lines
            // while keeping them connected
            int size_before = connected_lines.size();
            for (int i = 0; i < linestrings.size(); ) {
                if (connected_lines.isEmpty()) {
                    // First linestring
                    first_point_id = firstPointId.remove(i);
                    last_point_id = lastPointId.remove(i);
                    reverse.add(false);
                    sumPoints += linestrings.get(i).numPoints();
                    connected_lines.add(linestrings.remove(i));
                } else if (lastPointId.get(i) == first_point_id) {
                    // This linestring goes to the beginning of the list as-is
                    lastPointId.remove(i);
                    first_point_id = firstPointId.remove(i);
                    sumPoints += linestrings.get(i).numPoints();
                    connected_lines.add(0, linestrings.remove(i));
                    reverse.add(0, false);
                } else if (firstPointId.get(i) == first_point_id) {
                    // Should go to the beginning after being reversed
                    firstPointId.remove(i);
                    first_point_id = lastPointId.remove(i);
                    sumPoints += linestrings.get(i).numPoints();
                    connected_lines.add(0, linestrings.remove(i));
                    reverse.add(0, true);
                } else if (firstPointId.get(i) == last_point_id) {
                    // This linestring goes to the end of the list as-is
                    firstPointId.remove(i);
                    last_point_id = lastPointId.remove(i);
                    sumPoints += linestrings.get(i).numPoints();
                    connected_lines.add(linestrings.remove(i));
                    reverse.add(false);
                } else if (lastPointId.get(i) == last_point_id) {
                    // Should go to the end after being reversed
                    lastPointId.remove(i);
                    last_point_id = firstPointId.remove(i);
                    sumPoints += linestrings.get(i).numPoints();
                    connected_lines.add(linestrings.remove(i));
                    reverse.add(true);
                } else {
                    i++;
                }
            }
            if (connected_lines.size() == size_before || linestrings.isEmpty()) {
                // Cannot connect any more lines to the current block. Emit as a shape
                boolean isPolygon = first_point_id == last_point_id;
                Point[] points = new Point[sumPoints - connected_lines.size() + (isPolygon ? 0 : 1)];
                int n = 0;
                for (int i = 0; i < connected_lines.size(); i++) {
                    OGCLineString linestring = connected_lines.get(i);
                    boolean isReverse = reverse.get(i);
                    int last_i = (isPolygon || i < connected_lines.size() - 1) ? linestring.numPoints() - 1 : linestring.numPoints();
                    for (int i_point = 0; i_point < last_i; i_point++) {
                        points[n++] = (Point) linestring.pointN(isReverse ? linestring.numPoints() - 1 - i_point : i_point).getEsriGeometry();
                    }
                }
                MultiPath multi_path = isPolygon ? new Polygon() : new Polyline();
                for (int i = 1; i < points.length; i++) {
                    Segment segment = new Line();
                    segment.setStart(points[i - 1]);
                    segment.setEnd(points[i]);
                    multi_path.addSegment(segment, false);
                }
                createdShapes.add(isPolygon ? new OGCPolygon((Polygon) multi_path, 0, SpatialReference.create(4326)) : new OGCLineString((Polyline) multi_path, 0, SpatialReference.create(4326)));
                // Re-initialize all data structures to connect remaining lines
                if (!linestrings.isEmpty()) {
                    connected_lines.clear();
                    reverse.clear();
                    sumPoints = 0;
                }
            }
        }
        if (createdShapes.size() == 1) {
            return new DataByteArray(createdShapes.get(0).asBinary().array());
        } else if (createdShapes.size() > 1) {
            OGCGeometryCollection collection = new OGCConcreteGeometryCollection(createdShapes, createdShapes.get(0).getEsriSpatialReference());
            return new DataByteArray(collection.asBinary().array());
        } else {
            throw new GeoException("No shapes to connect");
        }
    } catch (Exception e) {
        throw new GeoException(e);
    }
}

Example 2 with OGCLineString

use of com.esri.core.geometry.ogc.OGCLineString in project presto by prestodb.

the class GeoFunctions method stIsRing.

@SqlNullable
@Description("Returns TRUE if and only if the line is closed and simple")
@ScalarFunction("ST_IsRing")
@SqlType(BOOLEAN)
public static Boolean stIsRing(@SqlType(GEOMETRY_TYPE_NAME) Slice input) {
    OGCGeometry geometry = EsriGeometrySerde.deserialize(input);
    validateType("ST_IsRing", geometry, EnumSet.of(LINE_STRING));
    OGCLineString line = (OGCLineString) geometry;
    return line.isClosed() && line.isSimple();
}