PreContingencyNetworkResult.java
/**
* 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.openloadflow.sa;
import com.powsybl.openloadflow.network.LfBranch;
import com.powsybl.openloadflow.network.LfNetwork;
import com.powsybl.security.monitor.StateMonitor;
import com.powsybl.security.monitor.StateMonitorIndex;
import com.powsybl.security.results.BranchResult;
import java.util.*;
import java.util.function.Predicate;
/**
* @author Geoffroy Jamgotchian {@literal <geoffroy.jamgotchian at rte-france.com>}
*/
public class PreContingencyNetworkResult extends AbstractNetworkResult {
private final Map<String, BranchResult> branchResults = new HashMap<>();
public PreContingencyNetworkResult(LfNetwork network, StateMonitorIndex monitorIndex, boolean createResultExtension) {
super(network, monitorIndex, createResultExtension);
}
@Override
protected void clear() {
super.clear();
branchResults.clear();
}
private void addResults(StateMonitor monitor, Predicate<LfBranch> isBranchDisabled) {
addResults(monitor, branch -> {
branch.createBranchResult(Double.NaN, Double.NaN, createResultExtension)
.forEach(branchResult -> branchResults.put(branchResult.getBranchId(), branchResult));
}, isBranchDisabled);
}
@Override
public void update() {
update(LfBranch::isDisabled);
}
public void update(Predicate<LfBranch> isBranchDisabled) {
clear();
addResults(monitorIndex.getNoneStateMonitor(), isBranchDisabled);
addResults(monitorIndex.getAllStateMonitor(), isBranchDisabled);
}
public BranchResult getBranchResult(String branchId) {
Objects.requireNonNull(branchId);
return branchResults.get(branchId);
}
@Override
public List<BranchResult> getBranchResults() {
return new ArrayList<>(branchResults.values());
}
}