| /* |
| * 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; |
| import org.apache.lucene.geo.Component2D; |
| import org.apache.lucene.geo.XYPolygon; |
| import org.apache.lucene.util.LuceneTestCase; |
| |
| /** random cartesian bounding box, line, and polygon query tests for random indexed arrays of cartesian {@link XYPolygon} types */ |
| @LuceneTestCase.SuppressCodecs("SimpleText") |
| public class TestXYMultiPolygonShapeQueries extends BaseXYShapeTestCase { |
| @Override |
| protected ShapeType getShapeType() { |
| return ShapeType.POLYGON; |
| } |
| |
| @Override |
| protected XYPolygon[] nextShape() { |
| int n = random().nextInt(4) + 1; |
| XYPolygon[] polygons = new XYPolygon[n]; |
| for (int i =0; i < n; i++) { |
| polygons[i] = (XYPolygon) getShapeType().nextShape(); |
| } |
| return polygons; |
| } |
| |
| @Override |
| protected Field[] createIndexableFields(String name, Object o) { |
| XYPolygon[] polygons = (XYPolygon[]) o; |
| List<Field> allFields = new ArrayList<>(); |
| for (XYPolygon polygon : polygons) { |
| Field[] fields = XYShape.createIndexableFields(name, polygon); |
| for (Field field : fields) { |
| allFields.add(field); |
| } |
| } |
| return allFields.toArray(new Field[allFields.size()]); |
| } |
| |
| @Override |
| protected Validator getValidator() { |
| return new MultiPolygonValidator(ENCODER); |
| } |
| |
| protected class MultiPolygonValidator extends Validator { |
| TestXYPolygonShapeQueries.PolygonValidator POLYGONVALIDATOR; |
| |
| MultiPolygonValidator(Encoder encoder) { |
| super(encoder); |
| POLYGONVALIDATOR = new TestXYPolygonShapeQueries.PolygonValidator(encoder); |
| } |
| |
| @Override |
| public Validator setRelation(QueryRelation relation) { |
| super.setRelation(relation); |
| POLYGONVALIDATOR.queryRelation = relation; |
| return this; |
| } |
| |
| @Override |
| public boolean testComponentQuery(Component2D query, Object shape) { |
| XYPolygon[] polygons = (XYPolygon[]) shape; |
| if (queryRelation == QueryRelation.CONTAINS) { |
| return testWithinPolygon(query, polygons); |
| } |
| for (XYPolygon p : polygons) { |
| boolean b = POLYGONVALIDATOR.testComponentQuery(query, p); |
| if (b == true && queryRelation == QueryRelation.INTERSECTS) { |
| return true; |
| } else if (b == false && queryRelation == QueryRelation.DISJOINT) { |
| return false; |
| } else if (b == false && queryRelation == QueryRelation.WITHIN) { |
| return false; |
| } |
| } |
| return queryRelation != QueryRelation.INTERSECTS && queryRelation != QueryRelation.CONTAINS; |
| } |
| |
| private boolean testWithinPolygon(Component2D query, XYPolygon[] polygons) { |
| Component2D.WithinRelation answer = Component2D.WithinRelation.DISJOINT; |
| for (XYPolygon p : polygons) { |
| Component2D.WithinRelation relation = POLYGONVALIDATOR.testWithinQuery(query, XYShape.createIndexableFields("dummy", p)); |
| if (relation == Component2D.WithinRelation.NOTWITHIN) { |
| return false; |
| } else if (relation == Component2D.WithinRelation.CANDIDATE) { |
| answer = relation; |
| } |
| } |
| return answer == Component2D.WithinRelation.CANDIDATE; |
| } |
| } |
| |
| @Slow |
| @Nightly |
| @Override |
| public void testRandomBig() throws Exception { |
| doTestRandom(10000); |
| } |
| } |