/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.lucene.document;
import java.util.ArrayList;
import java.util.List;
import org.apache.lucene.document.ShapeField.QueryRelation; // javadoc
import org.apache.lucene.document.ShapeField.Triangle;
import org.apache.lucene.geo.Circle;
import org.apache.lucene.geo.GeoUtils;
import org.apache.lucene.geo.LatLonGeometry;
import org.apache.lucene.geo.Line;
import org.apache.lucene.geo.Point;
import org.apache.lucene.geo.Polygon;
import org.apache.lucene.geo.Rectangle;
import org.apache.lucene.geo.Tessellator;
import org.apache.lucene.index.PointValues; // javadoc
import org.apache.lucene.search.BooleanClause;
import org.apache.lucene.search.BooleanQuery;
import org.apache.lucene.search.Query;
import static org.apache.lucene.geo.GeoEncodingUtils.encodeLatitude;
import static org.apache.lucene.geo.GeoEncodingUtils.encodeLongitude;
An geo shape utility class for indexing and searching gis geometries
whose vertices are latitude, longitude values (in decimal degrees).
This class defines seven static factory methods for common indexing and search operations:
createIndexableFields(String, Polygon)
for indexing a geo polygon. createIndexableFields(String, Line)
for indexing a geo linestring. createIndexableFields(String, double, double)
for indexing a lat, lon geo point. newBoxQuery()
for matching geo shapes that have some QueryRelation
with a bounding box. newLineQuery()
for matching geo shapes that have some QueryRelation
with a linestring. newPolygonQuery()
for matching geo shapes that have some QueryRelation
with a polygon. newGeometryQuery()
for matching geo shapes that have some QueryRelation
with one or more LatLonGeometry
.
WARNING: Like LatLonPoint
, vertex values are indexed with some loss of precision from the original double
values (4.190951585769653E-8 for the latitude component and 8.381903171539307E-8 for longitude). See Also:
/**
* An geo shape utility class for indexing and searching gis geometries
* whose vertices are latitude, longitude values (in decimal degrees).
* <p>
* This class defines seven static factory methods for common indexing and search operations:
* <ul>
* <li>{@link #createIndexableFields(String, Polygon)} for indexing a geo polygon.
* <li>{@link #createIndexableFields(String, Line)} for indexing a geo linestring.
* <li>{@link #createIndexableFields(String, double, double)} for indexing a lat, lon geo point.
* <li>{@link #newBoxQuery newBoxQuery()} for matching geo shapes that have some {@link QueryRelation} with a bounding box.
* <li>{@link #newLineQuery newLineQuery()} for matching geo shapes that have some {@link QueryRelation} with a linestring.
* <li>{@link #newPolygonQuery newPolygonQuery()} for matching geo shapes that have some {@link QueryRelation} with a polygon.
* <li>{@link #newGeometryQuery newGeometryQuery()} for matching geo shapes that have some {@link QueryRelation}
* with one or more {@link LatLonGeometry}.
* </ul>
* <b>WARNING</b>: Like {@link LatLonPoint}, vertex values are indexed with some loss of precision from the
* original {@code double} values (4.190951585769653E-8 for the latitude component
* and 8.381903171539307E-8 for longitude).
* @see PointValues
* @see LatLonDocValuesField
*/
public class LatLonShape {
// no instance:
private LatLonShape() {
}
create indexable fields for polygon geometry /** create indexable fields for polygon geometry */
public static Field[] createIndexableFields(String fieldName, Polygon polygon) {
// the lionshare of the indexing is done by the tessellator
List<Tessellator.Triangle> tessellation = Tessellator.tessellate(polygon);
List<Triangle> fields = new ArrayList<>();
for (Tessellator.Triangle t : tessellation) {
fields.add(new Triangle(fieldName, t));
}
return fields.toArray(new Field[fields.size()]);
}
create indexable fields for line geometry /** create indexable fields for line geometry */
public static Field[] createIndexableFields(String fieldName, Line line) {
int numPoints = line.numPoints();
Field[] fields = new Field[numPoints - 1];
// create "flat" triangles
for (int i = 0, j = 1; j < numPoints; ++i, ++j) {
fields[i] = new Triangle(fieldName,
encodeLongitude(line.getLon(i)), encodeLatitude(line.getLat(i)),
encodeLongitude(line.getLon(j)), encodeLatitude(line.getLat(j)),
encodeLongitude(line.getLon(i)), encodeLatitude(line.getLat(i)));
}
return fields;
}
create indexable fields for point geometry /** create indexable fields for point geometry */
public static Field[] createIndexableFields(String fieldName, double lat, double lon) {
return new Field[] {new Triangle(fieldName,
encodeLongitude(lon), encodeLatitude(lat),
encodeLongitude(lon), encodeLatitude(lat),
encodeLongitude(lon), encodeLatitude(lat))};
}
create a query to find all indexed geo shapes that intersect a defined bounding box /** create a query to find all indexed geo shapes that intersect a defined bounding box **/
public static Query newBoxQuery(String field, QueryRelation queryRelation, double minLatitude, double maxLatitude, double minLongitude, double maxLongitude) {
if (queryRelation == QueryRelation.CONTAINS && minLongitude > maxLongitude) {
BooleanQuery.Builder builder = new BooleanQuery.Builder();
builder.add(newBoxQuery(field, queryRelation, minLatitude, maxLatitude, minLongitude, GeoUtils.MAX_LON_INCL), BooleanClause.Occur.MUST);
builder.add(newBoxQuery(field, queryRelation, minLatitude, maxLatitude, GeoUtils.MIN_LON_INCL, maxLongitude), BooleanClause.Occur.MUST);
return builder.build();
}
Rectangle rectangle = new Rectangle(minLatitude, maxLatitude, minLongitude, maxLongitude);
return new LatLonShapeBoundingBoxQuery(field, queryRelation, rectangle);
}
create a query to find all indexed geo shapes that intersect a provided linestring (or array of linestrings)
note: does not support dateline crossing
/** create a query to find all indexed geo shapes that intersect a provided linestring (or array of linestrings)
* note: does not support dateline crossing
**/
public static Query newLineQuery(String field, QueryRelation queryRelation, Line... lines) {
return newGeometryQuery(field, queryRelation, lines);
}
create a query to find all indexed geo shapes that intersect a provided polygon (or array of polygons)
note: does not support dateline crossing
/** create a query to find all indexed geo shapes that intersect a provided polygon (or array of polygons)
* note: does not support dateline crossing
**/
public static Query newPolygonQuery(String field, QueryRelation queryRelation, Polygon... polygons) {
return newGeometryQuery(field, queryRelation, polygons);
}
create a query to find all indexed shapes that comply the QueryRelation
with the provided points /** create a query to find all indexed shapes that comply the {@link QueryRelation} with the provided points
**/
public static Query newPointQuery(String field, QueryRelation queryRelation, double[]... points) {
Point[] pointArray = new Point[points.length];
for (int i =0; i < points.length; i++) {
pointArray[i] = new Point(points[i][0], points[i][1]);
}
return newGeometryQuery(field, queryRelation, pointArray);
}
create a query to find all polygons that intersect a provided circle. /** create a query to find all polygons that intersect a provided circle. */
public static Query newDistanceQuery(String field, QueryRelation queryRelation, Circle... circle) {
return newGeometryQuery(field, queryRelation, circle);
}
create a query to find all indexed geo shapes that intersect a provided geometry (or array of geometries).
/** create a query to find all indexed geo shapes that intersect a provided geometry (or array of geometries).
**/
public static Query newGeometryQuery(String field, QueryRelation queryRelation, LatLonGeometry... latLonGeometries) {
if (latLonGeometries.length == 1) {
LatLonGeometry geometry = latLonGeometries[0];
if (geometry instanceof Rectangle) {
Rectangle rect = (Rectangle) geometry;
return newBoxQuery(field, queryRelation, rect.minLat, rect.maxLat, rect.minLon, rect.maxLon);
} else {
return new LatLonShapeQuery(field, queryRelation, latLonGeometries);
}
} else {
if (queryRelation == QueryRelation.CONTAINS) {
return makeContainsGeometryQuery(field, latLonGeometries);
} else {
return new LatLonShapeQuery(field, queryRelation, latLonGeometries);
}
}
}
private static Query makeContainsGeometryQuery(String field, LatLonGeometry... latLonGeometries) {
BooleanQuery.Builder builder = new BooleanQuery.Builder();
for (LatLonGeometry geometry : latLonGeometries) {
if (geometry instanceof Rectangle) {
// this handles rectangles across the dateline
Rectangle rect = (Rectangle) geometry;
builder.add(newBoxQuery(field, QueryRelation.CONTAINS, rect.minLat, rect.maxLat, rect.minLon, rect.maxLon), BooleanClause.Occur.MUST);
} else {
builder.add(new LatLonShapeQuery(field, QueryRelation.CONTAINS, geometry), BooleanClause.Occur.MUST);
}
}
return builder.build();
}
}