Geo: replace intermediate geo objects with libs/geo (#37721)

Replaces intermediate geo objects built by ShapeBuilders with
objects from the libs/geo hierarchy. This should allow us to build
all geo functionality around a single hierarchy.

Follow up for #35320

Author: imotov

build.gradle

@@ -212,6 +212,7 @@ allprojects {
     "org.elasticsearch:elasticsearch-core:${version}": ':libs:core',
     "org.elasticsearch:elasticsearch-nio:${version}": ':libs:nio',
     "org.elasticsearch:elasticsearch-x-content:${version}": ':libs:x-content',
+    "org.elasticsearch:elasticsearch-geo:${version}": ':libs:elasticsearch-geo',
     "org.elasticsearch:elasticsearch-secure-sm:${version}": ':libs:secure-sm',
     "org.elasticsearch.client:elasticsearch-rest-client:${version}": ':client:rest',
     "org.elasticsearch.client:elasticsearch-rest-client-sniffer:${version}": ':client:sniffer',

libs/geo/src/main/java/org/elasticsearch/geo/geometry/Line.java

@@ -67,6 +67,14 @@ public double getLon(int i) {
         return lons[i];
     }
 
+    public double[] getLats() {
+        return lats.clone();
+    }
+
+    public double[] getLons() {
+        return lons.clone();
+    }
+
     @Override
     public ShapeType type() {
         return ShapeType.LINESTRING;

libs/geo/src/main/java/org/elasticsearch/geo/geometry/Point.java

@@ -48,11 +48,11 @@ public ShapeType type() {
         return ShapeType.POINT;
     }
 
-    public double lat() {
+    public double getLat() {
         return lat;
     }
 
-    public double lon() {
+    public double getLon() {
         return lon;
     }
 

libs/geo/src/main/java/org/elasticsearch/geo/utils/WellKnownText.java

@@ -123,12 +123,12 @@ public Void visit(MultiLine multiLine) {
                 public Void visit(MultiPoint multiPoint) {
                     // walk through coordinates:
                     sb.append(LPAREN);
-                    visitPoint(multiPoint.get(0).lon(), multiPoint.get(0).lat());
+                    visitPoint(multiPoint.get(0).getLon(), multiPoint.get(0).getLat());
                     for (int i = 1; i < multiPoint.size(); ++i) {
                         sb.append(COMMA);
                         sb.append(SPACE);
                         Point point = multiPoint.get(i);
-                        visitPoint(point.lon(), point.lat());
+                        visitPoint(point.getLon(), point.getLat());
                     }
                     sb.append(RPAREN);
                     return null;
@@ -146,7 +146,7 @@ public Void visit(Point point) {
                         sb.append(EMPTY);
                     } else {
                         sb.append(LPAREN);
-                        visitPoint(point.lon(), point.lat());
+                        visitPoint(point.getLon(), point.getLat());
                         sb.append(RPAREN);
                     }
                     return null;

server/build.gradle

@@ -77,7 +77,8 @@ dependencies {
   compile "org.elasticsearch:elasticsearch-core:${version}"
   compile "org.elasticsearch:elasticsearch-secure-sm:${version}"
   compile "org.elasticsearch:elasticsearch-x-content:${version}"
-
+  compile "org.elasticsearch:elasticsearch-geo:${version}"
+  
   compileOnly project(':libs:plugin-classloader')
   testRuntime project(':libs:plugin-classloader')
 

server/src/main/java/org/elasticsearch/common/geo/GeoShapeType.java

@@ -258,7 +258,7 @@ CoordinateNode validate(CoordinateNode coordinates, boolean coerce) {
     },
     GEOMETRYCOLLECTION("geometrycollection") {
         @Override
-        public ShapeBuilder<?, ?> getBuilder(CoordinateNode coordinates, DistanceUnit.Distance radius,
+        public ShapeBuilder<?, ?, ?> getBuilder(CoordinateNode coordinates, DistanceUnit.Distance radius,
                                        Orientation orientation, boolean coerce) {
             // noop, handled in parser
             return null;
@@ -298,7 +298,7 @@ public static GeoShapeType forName(String geoshapename) {
         throw new IllegalArgumentException("unknown geo_shape ["+geoshapename+"]");
     }
 
-    public abstract ShapeBuilder<?, ?> getBuilder(CoordinateNode coordinates, DistanceUnit.Distance radius,
+    public abstract ShapeBuilder<?, ?, ?> getBuilder(CoordinateNode coordinates, DistanceUnit.Distance radius,
                                             ShapeBuilder.Orientation orientation, boolean coerce);
     abstract CoordinateNode validate(CoordinateNode coordinates, boolean coerce);
 

server/src/main/java/org/elasticsearch/common/geo/builders/CircleBuilder.java

@@ -34,7 +34,7 @@
 import java.io.IOException;
 import java.util.Objects;
 
-public class CircleBuilder extends ShapeBuilder<Circle, CircleBuilder> {
+public class CircleBuilder extends ShapeBuilder<Circle, org.elasticsearch.geo.geometry.Circle, CircleBuilder> {
 
     public static final ParseField FIELD_RADIUS = new ParseField("radius");
     public static final GeoShapeType TYPE = GeoShapeType.CIRCLE;
@@ -164,7 +164,7 @@ public Circle buildS4J() {
     }
 
     @Override
-    public Object buildLucene() {
+    public org.elasticsearch.geo.geometry.Circle buildGeometry() {
         throw new UnsupportedOperationException("CIRCLE geometry is not supported");
     }
 

server/src/main/java/org/elasticsearch/common/geo/builders/EnvelopeBuilder.java

@@ -32,7 +32,7 @@
 import java.io.IOException;
 import java.util.Objects;
 
-public class EnvelopeBuilder extends ShapeBuilder<Rectangle, EnvelopeBuilder> {
+public class EnvelopeBuilder extends ShapeBuilder<Rectangle, org.elasticsearch.geo.geometry.Rectangle, EnvelopeBuilder> {
 
     public static final GeoShapeType TYPE = GeoShapeType.ENVELOPE;
 
@@ -113,8 +113,8 @@ public Rectangle buildS4J() {
     }
 
     @Override
-    public org.apache.lucene.geo.Rectangle buildLucene() {
-        return new org.apache.lucene.geo.Rectangle(bottomRight.y, topLeft.y, topLeft.x, bottomRight.x);
+    public org.elasticsearch.geo.geometry.Rectangle buildGeometry() {
+        return new org.elasticsearch.geo.geometry.Rectangle(bottomRight.y, topLeft.y, topLeft.x, bottomRight.x);
     }
 
     @Override

server/src/main/java/org/elasticsearch/common/geo/builders/GeometryCollectionBuilder.java

@@ -31,11 +31,11 @@
 
 import java.io.IOException;
 import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.List;
 import java.util.Objects;
 
-public class GeometryCollectionBuilder extends ShapeBuilder<Shape, GeometryCollectionBuilder> {
+public class GeometryCollectionBuilder extends ShapeBuilder<Shape,
+    org.elasticsearch.geo.geometry.GeometryCollection<org.elasticsearch.geo.geometry.Geometry>, GeometryCollectionBuilder> {
 
     public static final GeoShapeType TYPE = GeoShapeType.GEOMETRYCOLLECTION;
 
@@ -185,19 +185,14 @@ public Shape buildS4J() {
     }
 
     @Override
-    public Object buildLucene() {
-        List<Object> shapes = new ArrayList<>(this.shapes.size());
+    public org.elasticsearch.geo.geometry.GeometryCollection<org.elasticsearch.geo.geometry.Geometry> buildGeometry() {
+        List<org.elasticsearch.geo.geometry.Geometry> shapes = new ArrayList<>(this.shapes.size());
 
         for (ShapeBuilder shape : this.shapes) {
-            Object o = shape.buildLucene();
-            if (o.getClass().isArray()) {
-                shapes.addAll(Arrays.asList((Object[])o));
-            } else {
-                shapes.add(o);
-            }
+            shapes.add(shape.buildGeometry());
         }
 
-        return shapes.toArray(new Object[shapes.size()]);
+        return new org.elasticsearch.geo.geometry.GeometryCollection<>(shapes);
     }
 
     @Override

server/src/main/java/org/elasticsearch/common/geo/builders/LineStringBuilder.java

@@ -19,16 +19,16 @@
 
 package org.elasticsearch.common.geo.builders;
 
-import org.apache.lucene.geo.Line;
-import org.locationtech.jts.geom.Coordinate;
-import org.locationtech.jts.geom.Geometry;
-import org.locationtech.jts.geom.GeometryFactory;
-import org.locationtech.jts.geom.LineString;
-
 import org.elasticsearch.common.geo.GeoShapeType;
 import org.elasticsearch.common.geo.parsers.ShapeParser;
 import org.elasticsearch.common.io.stream.StreamInput;
 import org.elasticsearch.common.xcontent.XContentBuilder;
+import org.elasticsearch.geo.geometry.Line;
+import org.elasticsearch.geo.geometry.MultiLine;
+import org.locationtech.jts.geom.Coordinate;
+import org.locationtech.jts.geom.Geometry;
+import org.locationtech.jts.geom.GeometryFactory;
+import org.locationtech.jts.geom.LineString;
 import org.locationtech.spatial4j.shape.jts.JtsGeometry;
 
 import java.io.IOException;
@@ -39,7 +39,7 @@
 import static org.elasticsearch.common.geo.GeoUtils.normalizeLat;
 import static org.elasticsearch.common.geo.GeoUtils.normalizeLon;
 
-public class LineStringBuilder extends ShapeBuilder<JtsGeometry, LineStringBuilder> {
+public class LineStringBuilder extends ShapeBuilder<JtsGeometry, org.elasticsearch.geo.geometry.Geometry, LineStringBuilder> {
     public static final GeoShapeType TYPE = GeoShapeType.LINESTRING;
 
     /**
@@ -125,15 +125,15 @@ public JtsGeometry buildS4J() {
     }
 
     @Override
-    public Object buildLucene() {
+    public org.elasticsearch.geo.geometry.Geometry buildGeometry() {
         // decompose linestrings crossing dateline into array of Lines
         Coordinate[] coordinates = this.coordinates.toArray(new Coordinate[this.coordinates.size()]);
         if (wrapdateline) {
-            ArrayList<Line> linestrings = decomposeLucene(coordinates, new ArrayList<>());
+            List<Line> linestrings = decomposeGeometry(coordinates, new ArrayList<>());
             if (linestrings.size() == 1) {
                 return linestrings.get(0);
             } else {
-                return linestrings.toArray(new Line[linestrings.size()]);
+                return new MultiLine(linestrings);
             }
         }
         return new Line(Arrays.stream(coordinates).mapToDouble(i->normalizeLat(i.y)).toArray(),
@@ -149,7 +149,7 @@ static ArrayList<LineString> decomposeS4J(GeometryFactory factory, Coordinate[]
         return strings;
     }
 
-    static ArrayList<Line> decomposeLucene(Coordinate[] coordinates, ArrayList<Line> lines) {
+    static List<Line> decomposeGeometry(Coordinate[] coordinates, List<Line> lines) {
         for (Coordinate[] part : decompose(+DATELINE, coordinates)) {
             for (Coordinate[] line : decompose(-DATELINE, part)) {
                 lines.add(new Line(Arrays.stream(line).mapToDouble(i->normalizeLat(i.y)).toArray(),