OverridingStyleTest.java

/**
 * Copyright (c) 2025, 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/.
 * SPDX-License-Identifier: MPL-2.0
 */
package com.powsybl.nad.svg;

import com.powsybl.iidm.network.Network;
import com.powsybl.iidm.network.test.ThreeWindingsTransformerNetworkFactory;
import com.powsybl.nad.AbstractTest;
import com.powsybl.nad.layout.LayoutParameters;
import com.powsybl.nad.model.BranchEdge;
import com.powsybl.nad.model.BusNode;
import com.powsybl.nad.model.ThreeWtEdge;
import com.powsybl.nad.svg.iidm.DefaultLabelProvider;
import com.powsybl.nad.svg.iidm.TopologicalStyleProvider;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;

/**
 * @author Florian Dupuy {@literal <florian.dupuy at rte-france.com>}
 */
class OverridingStyleTest extends AbstractTest {

    @BeforeEach
    void setup() {
        setLayoutParameters(new LayoutParameters());
        setSvgParameters(new SvgParameters()
                .setInsertNameDesc(true)
                .setSvgWidthAndHeightAdded(true)
                .setFixedWidth(800));
    }

    @Override
    protected StyleProvider getStyleProvider(Network network) {
        return new TopologicalStyleProvider(network) {

            @Override
            public String getBusNodeStyle(BusNode busNode) {
                String equipmentId = busNode.getEquipmentId();
                if (equipmentId.startsWith("FFR")) {
                    return busNode.getRingIndex() == 0 ? "fill:blue; background:blue" : "fill:lightblue; background:lightblue";
                } else if (equipmentId.startsWith("NNL")) {
                    return "fill:orange; background:orange";
                } else if (equipmentId.startsWith("BBE")) {
                    return busNode.getRingIndex() == 0 ? "fill:yellow; background:yellow" : "fill:deeppink; background:deeppink";
                } else if (equipmentId.startsWith("DDE")) {
                    return "fill:red; stroke: black; stroke-width: 2px; background:red; border: solid 2px";
                }
                return null;
            }

            @Override
            public String getSideEdgeStyle(BranchEdge edge, BranchEdge.Side side) {
                String equipmentId = edge.getEquipmentId();
                if (!equipmentId.substring(0, 3).equals(equipmentId.substring(9, 12))) {
                    return "stroke-dasharray:15,9;stroke-width:10px;stroke:black";
                }
                return null;
            }

            @Override
            public String getThreeWtEdgeStyle(ThreeWtEdge threeWtEdge) {
                return switch (threeWtEdge.getSide()) {
                    case ONE -> "stroke: lightgray; stroke-dasharray:4,4;";
                    case TWO -> "stroke: gray; stroke-dasharray:8,8";
                    case THREE -> "stroke: black; stroke-dasharray:12,12";
                };
            }
        };
    }

    @Override
    protected LabelProvider getLabelProvider(Network network) {
        return new DefaultLabelProvider(network, getSvgParameters());
    }

    @Test
    void testEurope() {
        Network network = Network.read("simple-eu.uct", getClass().getResourceAsStream("/simple-eu.uct"));
        assertSvgEquals("/simple-eu_overridden_styles.svg", network);
        assertSvgEquals("/simple-eu_overridden_styles.svg", network);
    }

    @Test
    void test3wt() {
        Network network = ThreeWindingsTransformerNetworkFactory.create();
        assertSvgEquals("/3wt_overridden_styles.svg", network);
    }
}