com.powsybl.python.contingency.ContingencyContainerImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pypowsybl Show documentation
A C interface to powsybl, for pypowsybl implementation
The newest version!
/**
 * Copyright (c) 2021, 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.python.contingency;

import com.powsybl.commons.PowsyblException;
import com.powsybl.contingency.*;
import com.powsybl.iidm.network.*;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @author Geoffroy Jamgotchian {@literal }
 */
public class ContingencyContainerImpl implements ContingencyContainer {

    private final Map> elementIdsByContingencyId = new HashMap<>();

    @Override
    public void addContingency(String contingencyId, List elementIds) {
        elementIdsByContingencyId.put(contingencyId, elementIds);
    }

    private static ContingencyElement createContingencyElement(Network network, String elementId) {
        Identifiable identifiable = network.getIdentifiable(elementId);
        if (identifiable == null) {
            throw new PowsyblException("Element '" + elementId + "' not found");
        }
        if (identifiable instanceof Line) {
            return new LineContingency(elementId);
        } else if (identifiable instanceof TwoWindingsTransformer) {
            return new TwoWindingsTransformerContingency(elementId);
        } else if (identifiable instanceof HvdcLine) {
            return new HvdcLineContingency(elementId);
        } else if (identifiable instanceof BusbarSection) {
            return new BusbarSectionContingency(elementId);
        } else if (identifiable instanceof Generator) {
            return new GeneratorContingency(elementId);
        } else if (identifiable instanceof DanglingLine) {
            return new DanglingLineContingency(elementId);
        } else if (identifiable instanceof StaticVarCompensator) {
            return new StaticVarCompensatorContingency(elementId);
        } else if (identifiable instanceof ShuntCompensator) {
            return new ShuntCompensatorContingency(elementId);
        } else if (identifiable instanceof ThreeWindingsTransformer) {
            return new ThreeWindingsTransformerContingency(elementId);
        } else if (identifiable instanceof Load) {
            return new LoadContingency(elementId);
        } else if (identifiable instanceof Battery) {
            return new BatteryContingency(elementId);
        } else if (identifiable instanceof Switch) {
            return new SwitchContingency(elementId);
        } else if (identifiable instanceof TieLine) {
            return new TieLineContingency(elementId);
        } else {
            throw new PowsyblException("Element type not supported: " + identifiable.getClass().getSimpleName());
        }
    }

    protected List createContingencies(Network network) {
        List contingencies = new ArrayList<>(elementIdsByContingencyId.size());
        for (Map.Entry> e : elementIdsByContingencyId.entrySet()) {
            String contingencyId = e.getKey();
            List elementIds = e.getValue();
            List elements = elementIds.stream()
                    .map(elementId -> createContingencyElement(network, elementId))
                    .collect(Collectors.toList());
            contingencies.add(new Contingency(contingencyId, elements));
        }
        return contingencies;
    }
}