ElementState.java

/**
 * 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.openloadflow.network;

import java.util.Collection;
import java.util.List;
import java.util.Objects;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * @author Geoffroy Jamgotchian {@literal <geoffroy.jamgotchian at rte-france.com>}
 */
public class ElementState<T extends LfElement> {

    protected final T element;

    protected final boolean disabled;

    public ElementState(T element) {
        this.element = Objects.requireNonNull(element);
        disabled = element.isDisabled();
    }

    public void restore() {
        element.setDisabled(disabled);
    }

    public static <T extends LfElement, U extends ElementState<T>> List<U> save(Collection<T> elements, Function<T, U> save) {
        Objects.requireNonNull(elements);
        Objects.requireNonNull(save);
        return elements.stream().map(save).collect(Collectors.toList());
    }

    public static <T extends LfElement, U extends ElementState<T>> void restore(Collection<U> states) {
        Objects.requireNonNull(states);
        states.forEach(ElementState::restore);
    }
}