com.powsybl.action.ial.simulator.loadflow.RunningContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of powsybl-action-ial-simulator Show documentation

A tool to run simulations with preventive or curative actions

The newest version!

/**
 * Copyright (c) 2017, 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.action.ial.simulator.loadflow;

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

import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @author Geoffroy Jamgotchian {@literal }
 */
public class RunningContext {

    private int round = 0;

    private final Network network;

    private final Contingency contingency;

    private final TimeLine timeLine;

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

    private final Set testedActionsIds = new HashSet<>();

    private final List workedTests = new ArrayList<>();

    public RunningContext(Network network, Contingency contingency) {
        this.network = network;
        this.contingency = contingency;
        timeLine = new TimeLine(contingency);
    }

    public RunningContext(Network network) {
        this(network, null);
    }

    public int getRound() {
        return round;
    }

    public void setRound(int round) {
        this.round = round;
    }

    public Network getNetwork() {
        return network;
    }

    public Contingency getContingency() {
        return contingency;
    }

    public TimeLine getTimeLine() {
        return timeLine;
    }

    private AtomicInteger getRuleMatchCountInternal(String ruleId) {
        return rulesMatchCount.computeIfAbsent(ruleId, k -> new AtomicInteger());
    }

    public int getRuleMatchCount(String ruleId) {
        return getRuleMatchCountInternal(ruleId).get();
    }

    public void incrementRuleMatchCount(String ruleId) {
        getRuleMatchCountInternal(ruleId).incrementAndGet();
    }

    public boolean isTestWorks() {
        return !workedTests.isEmpty();
    }

    public void addWorkedTest(String actionId) {
        workedTests.add(actionId);
    }

    public List getWorkedTests() {
        return Collections.unmodifiableList(workedTests);
    }

    public boolean isTested(String actionId) {
        return testedActionsIds.contains(actionId);
    }

    public void addTested(String actionId) {
        testedActionsIds.add(actionId);
    }
}