com.powsybl.gse.map.SubstationLayer Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of powsybl-gse-network-map Show documentation

A GSE plugin to draw a network on a map

There is a newer version: 1.8.0

Show newest version

/**
 * Copyright (c) 2018, RTE (http://www.rte-france.com)
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
package com.powsybl.gse.map;

import com.gluonhq.maps.MapView;
import javafx.geometry.Point2D;
import javafx.scene.canvas.GraphicsContext;
import javafx.scene.paint.Color;
import javafx.scene.shape.ArcType;
import javafx.scene.text.Font;
import org.apache.commons.lang3.time.StopWatch;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Objects;

/**
 * @author Geoffroy Jamgotchian 
 */
public class SubstationLayer extends CanvasBasedLayer {

    private static final Logger LOGGER = LoggerFactory.getLogger(SubstationLayer.class);

    private static final Color UNMAPPED_SUBSTATION_COLOR = Color.GRAY;

    private final SubstationGraphicIndex substationIndex;

    public SubstationLayer(MapView mapView, SubstationGraphicIndex substationIndex) {
        super(mapView);
        this.substationIndex = Objects.requireNonNull(substationIndex);
    }

    @Override
    protected void layoutLayer() {
        super.layoutLayer();
        StopWatch stopWatch = new StopWatch();
        stopWatch.start();

        GraphicsContext gc = canvas.getGraphicsContext2D();
        gc.setLineWidth(1);
        gc.setFont(Font.font(11));

        double zoom = baseMap.zoom().doubleValue();
        double size = zoom < 8 ? zoom / 2 : zoom;

        int[] drawnSubstations = new int[1];

        substationIndex.getTree().search(getMapBounds())
                .toBlocking()
                .toIterable()
                .forEach(e -> {
                    SubstationGraphic substation = e.value();
                    Point2D p = baseMap.getMapPoint(substation.getPosition().getLat(),
                                                    substation.getPosition().getLon());
                    if (Objects.isNull(substation.getModel())) {
                        gc.setFill(UNMAPPED_SUBSTATION_COLOR);
                    } else {
                        gc.setFill(substation.getColor());
                    }
                    gc.fillArc(p.getX() - size / 2, p.getY() - size / 2, size, size, 0, 360, ArcType.ROUND);
                    if (zoom > 9) {
                        gc.fillText(substation.getId(), p.getX() + 10, p.getY() + 10);
                    }
                    drawnSubstations[0]++;
                });

        stopWatch.stop();
        LOGGER.info("{} substations drawn in {} ms ", drawnSubstations[0], stopWatch.getTime());
    }
}