org.elasticsearch.search.aggregations.metrics.InternalBounds Maven / Gradle / Ivy

Go to download
/*
 * 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.search.aggregations.metrics;

import org.elasticsearch.common.geo.BoundingBox;
import org.elasticsearch.common.geo.GeoBoundingBox;
import org.elasticsearch.common.geo.SpatialPoint;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;
import org.elasticsearch.search.aggregations.InternalAggregation;
import org.elasticsearch.search.aggregations.support.SamplingContext;
import org.elasticsearch.xcontent.XContentBuilder;

import java.io.IOException;
import java.util.List;
import java.util.Map;

public abstract class InternalBounds extends InternalAggregation implements SpatialBounds {
    public final double top;
    public final double bottom;

    public InternalBounds(String name, double top, double bottom, Map metadata) {
        super(name, metadata);
        this.top = top;
        this.bottom = bottom;
    }

    /**
     * Read from a stream.
     */
    public InternalBounds(StreamInput in) throws IOException {
        super(in);
        top = in.readDouble();
        bottom = in.readDouble();
    }

    @Override
    protected void doWriteTo(StreamOutput out) throws IOException {
        out.writeDouble(top);
        out.writeDouble(bottom);
    }

    @Override
    public InternalAggregation finalizeSampling(SamplingContext samplingContext) {
        return this;
    }

    @Override
    protected boolean mustReduceOnSingleInternalAgg() {
        return false;
    }

    @Override
    public Object getProperty(List path) {
        if (path.isEmpty()) {
            return this;
        } else if (path.size() == 1) {
            BoundingBox bbox = resolveBoundingBox();
            String bBoxSide = path.get(0);
            return switch (bBoxSide) {
                case "top" -> bbox.top();
                case "left" -> bbox.left();
                case "bottom" -> bbox.bottom();
                case "right" -> bbox.right();
                default -> throw new IllegalArgumentException("Found unknown path element [" + bBoxSide + "] in [" + getName() + "]");
            };
        } else if (path.size() == 2) {
            BoundingBox bbox = resolveBoundingBox();
            T cornerPoint = null;
            String cornerString = path.get(0);
            cornerPoint = switch (cornerString) {
                case "top_left" -> bbox.topLeft();
                case "bottom_right" -> bbox.bottomRight();
                default -> throw new IllegalArgumentException("Found unknown path element [" + cornerString + "] in [" + getName() + "]");
            };
            return selectCoordinate(path.get(1), cornerPoint);
        } else {
            throw new IllegalArgumentException("path not supported for [" + getName() + "]: " + path);
        }
    }

    protected abstract Object selectCoordinate(String coordinateString, T cornerPoint);

    @Override
    public XContentBuilder doXContentBody(XContentBuilder builder, Params params) throws IOException {
        BoundingBox bbox = resolveBoundingBox();
        if (bbox != null) {
            builder.startObject(GeoBoundingBox.BOUNDS_FIELD.getPreferredName());
            bbox.toXContentFragment(builder);
            builder.endObject();
        }
        return builder;
    }

    protected abstract BoundingBox resolveBoundingBox();

    @Override
    public T topLeft() {
        BoundingBox bbox = resolveBoundingBox();
        if (bbox == null) {
            return null;
        } else {
            return bbox.topLeft();
        }
    }

    @Override
    public T bottomRight() {
        BoundingBox bbox = resolveBoundingBox();
        if (bbox == null) {
            return null;
        } else {
            return bbox.bottomRight();
        }
    }
}