BusDiagramDataImporter.java

/**
 * Copyright (c) 2019-2020, 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.sld.cgmes.dl.conversion.importers;

import java.util.Objects;

import com.powsybl.iidm.network.Substation;
import com.powsybl.sld.cgmes.dl.iidm.extensions.NetworkDiagramData;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.powsybl.sld.cgmes.dl.iidm.extensions.DiagramPoint;
import com.powsybl.sld.cgmes.dl.iidm.extensions.NodeDiagramData;
import com.powsybl.iidm.network.Bus;
import com.powsybl.iidm.network.Network;
import com.powsybl.iidm.network.VoltageLevel;
import com.powsybl.triplestore.api.PropertyBag;

/**
 *
 * @author Massimo Ferraro {@literal <massimo.ferraro@techrain.eu>}
 */
public class BusDiagramDataImporter {

    private static final Logger LOG = LoggerFactory.getLogger(BusDiagramDataImporter.class);

    private Network network;

    public BusDiagramDataImporter(Network network) {
        this.network = Objects.requireNonNull(network);
    }

    public void importDiagramData(PropertyBag busDiagramData) {
        Objects.requireNonNull(busDiagramData);
        String busId = busDiagramData.getId("identifiedObject");
        String vlId = busDiagramData.getId("voltageLevel");
        VoltageLevel vl = network.getVoltageLevel(vlId);
        if (vl != null) {
            Bus bus = vl.getBusBreakerView().getBus(busId);
            if (bus != null) {
                NodeDiagramData<Bus> busIidmDiagramData = bus.getExtension(NodeDiagramData.class);
                if (busIidmDiagramData == null) {
                    busIidmDiagramData = new NodeDiagramData<>(bus);
                }
                String diagramName = busDiagramData.get("diagramName");
                NodeDiagramData<Bus>.NodeDiagramDataDetails diagramDetails = busIidmDiagramData.getData(diagramName);
                if (diagramDetails == null) {
                    diagramDetails = busIidmDiagramData.new NodeDiagramDataDetails();
                }
                if (busDiagramData.asInt("seq") == 1) {
                    diagramDetails.setPoint1(new DiagramPoint(busDiagramData.asDouble("x"), busDiagramData.asDouble("y"), busDiagramData.asInt("seq")));
                } else {
                    diagramDetails.setPoint2(new DiagramPoint(busDiagramData.asDouble("x"), busDiagramData.asDouble("y"), busDiagramData.asInt("seq")));
                }
                busIidmDiagramData.addData(diagramName, diagramDetails);
                bus.addExtension(NodeDiagramData.class, busIidmDiagramData);
                NetworkDiagramData.addDiagramName(network, diagramName, bus.getVoltageLevel().getSubstation().map(Substation::getId).orElse(null));
            } else {
                LOG.warn("Cannot find bus {}, name {} in network {}: skipping bus diagram data", busId, busDiagramData.get("name"), network.getId());
            }
        } else {
            LOG.warn("Cannot find voltage level {}, name {} in network {}: skipping bus diagram data", vlId, busDiagramData.get("vlname"), network.getId());
        }
    }

}