hu.bme.mit.theta.xsts.analysis.autoexpl.XstsNewOperandsAutoExpl 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.autoexpl;

import hu.bme.mit.theta.analysis.expr.refinement.autoexpl.AutoExpl;
import hu.bme.mit.theta.analysis.expr.refinement.autoexpl.NewOperandsAutoExpl;
import hu.bme.mit.theta.common.container.Containers;
import hu.bme.mit.theta.core.decl.Decl;
import hu.bme.mit.theta.core.type.BinaryExpr;
import hu.bme.mit.theta.core.type.Expr;
import hu.bme.mit.theta.core.type.anytype.RefExpr;
import hu.bme.mit.theta.core.type.booltype.BoolType;
import hu.bme.mit.theta.core.utils.StmtAtomCollector;
import hu.bme.mit.theta.core.utils.ExprUtils;
import hu.bme.mit.theta.xsts.XSTS;

import java.util.Map;
import java.util.Set;
import java.util.function.Predicate;
import java.util.stream.Collectors;

public class XstsNewOperandsAutoExpl implements XstsAutoExpl {

    @Override
    public AutoExpl create(XSTS xsts) {
        final Set> atoms = Containers.createSet();
        atoms.addAll(StmtAtomCollector.collectAtoms(xsts.getEnv()));
        atoms.addAll(StmtAtomCollector.collectAtoms(xsts.getInit()));
        atoms.addAll(StmtAtomCollector.collectAtoms(xsts.getTran()));
        atoms.addAll(ExprUtils.getAtoms(xsts.getProp()));
        atoms.addAll(ExprUtils.getAtoms(xsts.getInitFormula()));

        final Set> canonicalAtoms = atoms.stream()
                .map(ExprUtils::canonize)
                .flatMap(atom -> ExprUtils.getAtoms(atom).stream())
                .collect(Collectors.toSet());
        final Map, Set>> declToOps = Containers.createMap();
        canonicalAtoms.stream()
                .filter(atom -> atom.getOps().size() > 1)
                .forEach(
                        atom -> {
                            atom.getOps().stream()
                                    .filter(RefExpr.class::isInstance)
                                    .map(RefExpr.class::cast)
                                    .forEach(
                                            ref -> atom.getOps().stream()
                                                    .filter(Predicate.not(ref::equals))
                                                    .forEach(
                                                            op -> declToOps.computeIfAbsent(ref.getDecl(),
                                                                    k -> Containers.createSet()).add((op))
                                                    )
                                    );
                        }
                );
        return new NewOperandsAutoExpl(xsts.getCtrlVars(), declToOps, 0);
    }
}