hu.bme.mit.theta.xsts.analysis.XstsLts Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of theta-xsts-analysis Show documentation

Xsts Analysis subproject in the Theta model checking framework

There is a newer version: 6.5.2

/*
 *  Copyright 2024 Budapest University of Technology and Economics
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package hu.bme.mit.theta.xsts.analysis;

import hu.bme.mit.theta.analysis.LTS;
import hu.bme.mit.theta.analysis.expr.ExprState;
import hu.bme.mit.theta.core.stmt.NonDetStmt;
import hu.bme.mit.theta.xsts.XSTS;

import java.util.Collection;
import java.util.stream.Collectors;

public final class XstsLts implements LTS, XstsAction> {

    private final NonDetStmt trans;
    private final NonDetStmt env;
    private final NonDetStmt init;

    private final XstsStmtOptimizer stmtOptimizer;

    private XstsLts(final XSTS xsts, final XstsStmtOptimizer stmtOptimizer) {
        trans = xsts.getTran();
        env = xsts.getEnv();
        init = xsts.getInit();

        this.stmtOptimizer = stmtOptimizer;
    }

    public static  LTS, XstsAction> create(final XSTS xsts,
                                                                             final XstsStmtOptimizer stmtOptimizer) {
        return new XstsLts<>(xsts, stmtOptimizer);
    }

    @Override
    public Collection getEnabledActionsFor(XstsState state) {
        NonDetStmt enabledSet;
        if (!state.isInitialized()) {
            enabledSet = init;
        } else if (state.lastActionWasEnv()) {
            enabledSet = trans;
        } else {
            enabledSet = env;
        }

        return enabledSet.getStmts().stream()
                .map(stmt -> stmtOptimizer.optimizeStmt(state, stmt))
                .map(XstsAction::create)
                .collect(Collectors.toList());
    }
}