org.apache.lucene.document.XLatLonShape Maven / Gradle / Ivy

Go to download
/*
 * @notice
 * 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 org.apache.lucene.geo.Polygon;
import org.apache.lucene.geo.XTessellator;
import org.apache.lucene.util.BytesRef;

import java.util.ArrayList;
import java.util.List;

import static org.apache.lucene.document.ShapeField.BYTES;
import static org.apache.lucene.document.ShapeField.TYPE;
import static org.apache.lucene.document.ShapeField.encodeTriangle;

/**
 * This is a copy of the same class from Lucene used here as a way to access a modified version of Tessellator class.
 * Once lucene releases with the bugfix to Tessellator, we can remove both these classes.
 */
public class XLatLonShape {

    // no instance:
    private XLatLonShape() {}

    /** 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 tessellation = XTessellator.tessellate(polygon);
        List fields = new ArrayList<>();
        for (XTessellator.Triangle t : tessellation) {
            fields.add(new Triangle(fieldName, t));
        }
        return fields.toArray(new Field[fields.size()]);
    }

    public static class Triangle extends Field {

        /** xtor from a given Tessellated Triangle object */
        Triangle(String name, XTessellator.Triangle t) {
            super(name, TYPE);
            setTriangleValue(
                t.getEncodedX(0),
                t.getEncodedY(0),
                t.isEdgefromPolygon(0),
                t.getEncodedX(1),
                t.getEncodedY(1),
                t.isEdgefromPolygon(1),
                t.getEncodedX(2),
                t.getEncodedY(2),
                t.isEdgefromPolygon(2)
            );
        }

        /** sets the vertices of the triangle as integer encoded values */
        protected void setTriangleValue(
            int aX,
            int aY,
            boolean abFromShape,
            int bX,
            int bY,
            boolean bcFromShape,
            int cX,
            int cY,
            boolean caFromShape
        ) {
            final byte[] bytes;

            if (fieldsData == null) {
                bytes = new byte[7 * BYTES];
                fieldsData = new BytesRef(bytes);
            } else {
                bytes = ((BytesRef) fieldsData).bytes;
            }
            encodeTriangle(bytes, aY, aX, abFromShape, bY, bX, bcFromShape, cY, cX, caFromShape);
        }
    }
}