NetworkState.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.network;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.List;
import java.util.Objects;
import java.util.Set;
/**
* @author Geoffroy Jamgotchian {@literal <geoffroy.jamgotchian at rte-france.com>}
*/
public class NetworkState {
private static final Logger LOGGER = LoggerFactory.getLogger(NetworkState.class);
private final LfNetwork network;
private final List<BusState> busStates;
private final List<BranchState> branchStates;
private final List<HvdcState> hvdcStates;
private final Set<LfBus> excludedSlackBuses;
private final List<AreaState> areaStates;
protected NetworkState(LfNetwork network, List<BusState> busStates, List<BranchState> branchStates, List<HvdcState> hvdcStates,
Set<LfBus> excludedSlackBuses, List<AreaState> areaStates) {
this.network = Objects.requireNonNull(network);
this.busStates = Objects.requireNonNull(busStates);
this.branchStates = Objects.requireNonNull(branchStates);
this.hvdcStates = Objects.requireNonNull(hvdcStates);
this.excludedSlackBuses = Objects.requireNonNull(excludedSlackBuses);
this.areaStates = Objects.requireNonNull(areaStates);
}
public static NetworkState save(LfNetwork network) {
Objects.requireNonNull(network);
LOGGER.trace("Saving network state");
network.setGeneratorsInitialTargetPToTargetP();
List<BusState> busStates = ElementState.save(network.getBuses(), BusState::save);
List<BranchState> branchStates = ElementState.save(network.getBranches(), BranchState::save);
List<HvdcState> hvdcStates = ElementState.save(network.getHvdcs(), HvdcState::save);
List<AreaState> areaStates = ElementState.save(network.getAreas(), AreaState::save);
return new NetworkState(network, busStates, branchStates, hvdcStates, network.getExcludedSlackBuses(), areaStates);
}
public void restore() {
LOGGER.trace("Restoring network state");
ElementState.restore(busStates);
ElementState.restore(branchStates);
ElementState.restore(hvdcStates);
ElementState.restore(areaStates);
network.setExcludedSlackBuses(excludedSlackBuses);
}
}