StyleProvidersList.java

/**
 * Copyright (c) 2023, 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.sld.svg.styles;

import com.powsybl.sld.library.ComponentLibrary;
import com.powsybl.sld.model.cells.Cell;
import com.powsybl.sld.model.graphs.Graph;
import com.powsybl.sld.model.graphs.VoltageLevelGraph;
import com.powsybl.sld.model.nodes.BranchEdge;
import com.powsybl.sld.model.nodes.Edge;
import com.powsybl.sld.model.nodes.Node;
import com.powsybl.sld.svg.BusInfo;
import com.powsybl.sld.svg.LabelProvider;
import com.powsybl.sld.svg.FeederInfo;

import java.net.URL;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * @author Florian Dupuy {@literal <florian.dupuy at rte-france.com>}
 */
public class StyleProvidersList implements StyleProvider {

    private final List<StyleProvider> styleProviders;

    public StyleProvidersList(StyleProvider... styleProviders) {
        this(List.of(styleProviders));
    }

    public <E> StyleProvidersList(List<StyleProvider> styleProviders) {
        this.styleProviders = styleProviders;
    }

    @Override
    public List<String> getEdgeStyles(Graph graph, Edge edge) {
        return concatenateLists(sp -> sp.getEdgeStyles(graph, edge));
    }

    @Override
    public List<String> getNodeStyles(VoltageLevelGraph graph, Node node, ComponentLibrary componentLibrary, boolean showInternalNodes) {
        return concatenateLists(sp -> sp.getNodeStyles(graph, node, componentLibrary, showInternalNodes));
    }

    @Override
    public List<String> getNodeDecoratorStyles(LabelProvider.NodeDecorator nodeDecorator, Node node, ComponentLibrary componentLibrary) {
        return concatenateLists(sp -> sp.getNodeDecoratorStyles(nodeDecorator, node, componentLibrary));
    }

    @Override
    public List<String> getBranchEdgeStyles(BranchEdge edge, ComponentLibrary componentLibrary) {
        return concatenateLists(sp -> sp.getBranchEdgeStyles(edge, componentLibrary));
    }

    @Override
    public List<String> getNodeSubcomponentStyles(Graph graph, Node node, String subComponentName) {
        return concatenateLists(sp -> sp.getNodeSubcomponentStyles(graph, node, subComponentName));
    }

    @Override
    public void reset() {
        styleProviders.forEach(StyleProvider::reset);
    }

    @Override
    public List<String> getCssFilenames() {
        return concatenateLists(StyleProvider::getCssFilenames);
    }

    @Override
    public List<URL> getCssUrls() {
        return concatenateLists(StyleProvider::getCssUrls);
    }

    @Override
    public List<String> getCellStyles(Cell cell) {
        return concatenateLists(sp -> sp.getCellStyles(cell));
    }

    @Override
    public List<String> getBusStyles(String busId, VoltageLevelGraph graph) {
        return concatenateLists(sp -> sp.getBusStyles(busId, graph));
    }

    @Override
    public List<String> getBusInfoStyle(BusInfo info) {
        return concatenateLists(sp -> sp.getBusInfoStyle(info));
    }

    @Override
    public List<String> getFeederInfoStyles(FeederInfo info) {
        return concatenateLists(sp -> sp.getFeederInfoStyles(info));
    }

    private <T> List<T> concatenateLists(Function<StyleProvider, List<T>> stylesGetter) {
        return styleProviders.stream()
                .map(stylesGetter)
                .flatMap(List::stream)
                .distinct()
                .collect(Collectors.toList());
    }
}