hu.bme.mit.theta.solver.smtlib.impl.cvc5.CVC5SmtLibItpSolver Maven / Gradle / Ivy

Go to download
/*
 *  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.smtlib.impl.cvc5;

import hu.bme.mit.theta.common.Tuple2;
import hu.bme.mit.theta.core.decl.ConstDecl;
import hu.bme.mit.theta.core.type.Expr;
import hu.bme.mit.theta.core.type.booltype.BoolType;
import hu.bme.mit.theta.solver.*;
import hu.bme.mit.theta.solver.smtlib.dsl.gen.SMTLIBv2Lexer;
import hu.bme.mit.theta.solver.smtlib.dsl.gen.SMTLIBv2Parser;
import hu.bme.mit.theta.solver.smtlib.solver.SmtLibItpSolver;
import hu.bme.mit.theta.solver.smtlib.solver.SmtLibSolverException;
import hu.bme.mit.theta.solver.smtlib.solver.binary.SmtLibSolverBinary;
import hu.bme.mit.theta.solver.smtlib.solver.interpolation.SmtLibInterpolant;
import hu.bme.mit.theta.solver.smtlib.solver.interpolation.SmtLibItpPattern;
import hu.bme.mit.theta.solver.smtlib.solver.model.SmtLibModel;
import hu.bme.mit.theta.solver.smtlib.solver.parser.ThrowExceptionErrorListener;
import hu.bme.mit.theta.solver.smtlib.solver.transformer.SmtLibSymbolTable;
import hu.bme.mit.theta.solver.smtlib.solver.transformer.SmtLibTermTransformer;
import hu.bme.mit.theta.solver.smtlib.solver.transformer.SmtLibTransformationManager;
import org.antlr.v4.runtime.CharStreams;
import org.antlr.v4.runtime.CommonTokenStream;
import org.antlr.v4.runtime.ParserRuleContext;
import org.antlr.v4.runtime.misc.Interval;

import java.util.*;
import java.util.function.Supplier;
import java.util.stream.Collectors;

import static com.google.common.base.Preconditions.*;
import static hu.bme.mit.theta.core.type.booltype.BoolExprs.Bool;
import static hu.bme.mit.theta.core.type.booltype.BoolExprs.False;

public class CVC5SmtLibItpSolver extends SmtLibItpSolver {

    private final Supplier itpSolverBinaryFactory;

    public CVC5SmtLibItpSolver(final SmtLibSymbolTable symbolTable,
                               final SmtLibTransformationManager transformationManager,
                               final SmtLibTermTransformer termTransformer, final SmtLibSolverBinary solverBinary,
                               final Supplier itpSolverBinaryFactory) {
        super(symbolTable, transformationManager, termTransformer, solverBinary);
        this.itpSolverBinaryFactory = itpSolverBinaryFactory;
    }

    @Override
    public ItpPattern createTreePattern(ItpMarkerTree root) {
        checkNotNull(root);
        return SmtLibItpPattern.of(root);
    }

    @Override
    public CVC5SmtLibItpMarker createMarker() {
        final var marker = new CVC5SmtLibItpMarker();
        markers.add(marker);
        return marker;
    }

    @Override
    protected void add(CVC5SmtLibItpMarker marker, Expr assertion,
                       Set> consts, String term) {
        consts.stream().map(symbolTable::getDeclaration).forEach(this::issueGeneralCommand);
        issueGeneralCommand(String.format("(assert %s)", term));
    }

    @Override
    public Interpolant getInterpolant(ItpPattern pattern) {
        checkState(getStatus() == SolverStatus.UNSAT,
                "Cannot get interpolant if status is not UNSAT.");
        checkArgument(pattern instanceof SmtLibItpPattern);

        try (final var itpSolverBinary = itpSolverBinaryFactory.get()) {
            itpSolverBinary.issueCommand("(set-option :produce-interpolants true)");
            itpSolverBinary.issueCommand("(set-logic ALL)");
            declarationStack.forEach(
                    constDecl -> itpSolverBinary.issueCommand(symbolTable.getDeclaration(constDecl)));

            @SuppressWarnings("unchecked") final var cvc5ItpPattern = (SmtLibItpPattern) pattern;
            final List markers = cvc5ItpPattern.getSequence();
            final List A = new ArrayList<>();
            final List B = new ArrayList<>(markers);

            final Map> itpMap = new HashMap<>();
            var interpolantCount = 0;
            for (final var marker : markers) {
                B.remove(marker);
                A.add(marker);

                if (B.size() != 0) {
                    final var aTerm = A.stream()
                            .flatMap(m -> m.getTerms().stream().map(Tuple2::get2));
                    final var bTerm = B.stream()
                            .flatMap(m -> m.getTerms().stream().map(Tuple2::get2));

                    itpSolverBinary.issueCommand(
                            String.format("(assert (and %s))", aTerm.collect(Collectors.joining(" "))));
                    itpSolverBinary.issueCommand(
                            String.format("(get-interpolant _cvc5_interpolant%d (not (and %s)))",
                                    interpolantCount++, bTerm.collect(Collectors.joining(" "))));

                    itpMap.put(marker,
                            termTransformer.toExpr(parseItpResponse(itpSolverBinary.readResponse()),
                                    Bool(), new SmtLibModel(Collections.emptyMap())));
                } else {
                    itpMap.put(marker, False());
                }
            }

            return new SmtLibInterpolant(itpMap);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    private String parseItpResponse(final String response) {
        final var lexer = new SMTLIBv2Lexer(CharStreams.fromString(response));
        final var parser = new SMTLIBv2Parser(new CommonTokenStream(lexer));
        try {
            lexer.removeErrorListeners();
            lexer.addErrorListener(new ThrowExceptionErrorListener());
            parser.removeErrorListeners();
            parser.addErrorListener(new ThrowExceptionErrorListener());
            return extractString(parser.model_response_fun().function_def().term());
        } catch (Exception e) {
            try {
                throw new SmtLibSolverException(
                        parser.response().general_response_error().reason.getText());
            } catch (Exception ex) {
                throw new SmtLibSolverException("Could not parse solver output: " + response, e);
            }
        }
    }

    private static String extractString(final ParserRuleContext ctx) {
        return ctx.start.getInputStream()
                .getText(new Interval(ctx.start.getStartIndex(), ctx.stop.getStopIndex()));
    }
}