hu.bme.mit.theta.solver.z3.Z3SolverManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of theta-solver-z3 Show documentation
Solver Z3 subproject in the Theta model checking framework
There is a newer version: 6.8.5
/*
 *  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.solver.z3;

import hu.bme.mit.theta.solver.*;

import java.util.HashSet;
import java.util.Set;

import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkState;

public final class Z3SolverManager extends SolverManager {

    private static final String NAME = "Z3:4.13";

    private boolean closed = false;
    private final Set instantiatedSolvers = new HashSet<>();

    private Z3SolverManager() {
    }

    public static Z3SolverManager create() {
        return new Z3SolverManager();
    }

    @Override
    public boolean managesSolver(final String name) {
        return NAME.equals(name);
    }

    @Override
    public SolverFactory getSolverFactory(final String name) {
        checkArgument(NAME.equals(name));
        return new ManagedFactory(Z3SolverFactory.getInstance());
    }

    @Override
    public synchronized void close() throws Exception {
        for (final var solver : instantiatedSolvers) {
            solver.close();
        }
        closed = true;
    }

    private final class ManagedFactory implements SolverFactory {

        private final SolverFactory solverFactory;

        private ManagedFactory(final SolverFactory solverFactory) {
            this.solverFactory = solverFactory;
        }

        @Override
        public Solver createSolver() {
            checkState(!closed, "Solver manager was closed");
            final var solver = solverFactory.createSolver();
            instantiatedSolvers.add(solver);
            return solver;
        }

        @Override
        public UCSolver createUCSolver() {
            checkState(!closed, "Solver manager was closed");
            final var solver = solverFactory.createUCSolver();
            instantiatedSolvers.add(solver);
            return solver;
        }

        @Override
        public ItpSolver createItpSolver() {
            checkState(!closed, "Solver manager was closed");
            final var solver = solverFactory.createItpSolver();
            instantiatedSolvers.add(solver);
            return solver;
        }

        @Override
        public HornSolver createHornSolver() {
            checkState(!closed, "Solver manager was closed");
            final var solver = solverFactory.createHornSolver();
            instantiatedSolvers.add(solver);
            return solver;
        }
    }
}