All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.elasticsearch.common.geo.GeometryFormatterFactory Maven / Gradle / Ivy

/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0 and the Server Side Public License, v 1; you may not use this file except
 * in compliance with, at your election, the Elastic License 2.0 or the Server
 * Side Public License, v 1.
 */

package org.elasticsearch.common.geo;

import org.elasticsearch.geometry.Geometry;
import org.elasticsearch.geometry.utils.WellKnownText;

import java.util.ArrayList;
import java.util.List;
import java.util.function.Function;

/**
 * Output formatters supported by geometry fields.
 */
public class GeometryFormatterFactory {

    public static final String GEOJSON = "geojson";
    public static final String WKT = "wkt";

    /**
     * Returns a formatter by name
     */
    public static  Function, List> getFormatter(String name, Function toGeometry) {
        switch (name) {
            case GEOJSON:
                return geometries -> {
                    final List objects = new ArrayList<>(geometries.size());
                    geometries.forEach((shape) -> objects.add(GeoJson.toMap(toGeometry.apply(shape))));
                    return objects;
                };
            case WKT:
                return geometries -> {
                    final List objects = new ArrayList<>(geometries.size());
                    geometries.forEach((shape) -> objects.add(WellKnownText.toWKT(toGeometry.apply(shape))));
                    return objects;
                };
            default:
                throw new IllegalArgumentException("Unrecognized geometry format [" + name + "].");
        }
    }
}