com.powsybl.flow_decomposition.VariantManager Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of powsybl-flow-decomposition Show documentation

Implementation of ACER methodology for flow decomposition

The newest version!

/*
 * Copyright (c) 2022, 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.flow_decomposition;

import com.powsybl.contingency.Contingency;
import com.powsybl.iidm.network.Network;

import java.util.List;
import java.util.stream.Collectors;

/**
 * @author Hugo Schindler {@literal }
 */
class VariantManager {
    private final String defaultVariantId;
    private final List contingencies;

    VariantManager(Network network, XnecProvider xnecProvider) {
        this.defaultVariantId = network.getVariantManager().getWorkingVariantId();
        contingencies = xnecProvider.getContingencies(network);
    }

    void createAVariantPerContingency(Network network) {
        if (!contingencies.isEmpty()) {
            List variantIdList = contingencies.stream().map(Contingency::getId).collect(Collectors.toList());
            network.getVariantManager().cloneVariant(defaultVariantId, variantIdList);
            contingencies.forEach(contingency -> {
                setNetworkVariant(network, contingency.getId());
                contingency.toModification().apply(network);
            });
            setDefaultNetworkVariant(network);
        }
    }

    void setDefaultNetworkVariant(Network network) {
        setNetworkVariant(network, defaultVariantId);
    }

    void setNetworkVariant(Network network, String variantId) {
        network.getVariantManager().setWorkingVariant(variantId);
    }

    void deleteAllContingencyVariants(Network network) {
        contingencies.forEach(contingency -> network.getVariantManager().removeVariant(contingency.getId()));
    }
}