hu.bme.mit.theta.solver.smtlib.impl.z3.Z3OldSmtLibItpSolver Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of theta-solver-smtlib Show documentation
Show all versions of theta-solver-smtlib Show documentation
Solver Smtlib subproject in the Theta model checking framework
/*
* 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.z3;
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.BoolExprs;
import hu.bme.mit.theta.core.type.booltype.BoolType;
import hu.bme.mit.theta.solver.Interpolant;
import hu.bme.mit.theta.solver.ItpMarker;
import hu.bme.mit.theta.solver.ItpMarkerTree;
import hu.bme.mit.theta.solver.ItpPattern;
import hu.bme.mit.theta.solver.SolverStatus;
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.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;
public final class Z3OldSmtLibItpSolver extends SmtLibItpSolver {
private boolean topMostContainsAssertions = false;
public Z3OldSmtLibItpSolver(
final SmtLibSymbolTable symbolTable,
final SmtLibTransformationManager transformationManager,
final SmtLibTermTransformer termTransformer, final SmtLibSolverBinary solverBinary
) {
super(symbolTable, transformationManager, termTransformer, solverBinary);
}
@Override
public ItpPattern createTreePattern(final ItpMarkerTree root) {
checkNotNull(root);
return SmtLibItpPattern.of(root);
}
@Override
public Z3SmtLibItpMarker createMarker() {
final var marker = new Z3SmtLibItpMarker();
markers.add(marker);
return marker;
}
@Override
public void add(ItpMarker marker, Expr assertion) {
if (topMostContainsAssertions) {
issueGeneralCommand("(pop 1)"); // Topmost frame contains marker assertions
topMostContainsAssertions = false;
}
super.add(marker, assertion);
}
@Override
protected void add(final Z3SmtLibItpMarker marker, final Expr assertion,
final Set> consts, final String term) {
consts.stream().map(symbolTable::getDeclaration).forEach(this::issueGeneralCommand);
}
@Override
public SolverStatus check() {
if (topMostContainsAssertions) {
issueGeneralCommand("(pop 1)"); // Topmost frame contains marker assertions
topMostContainsAssertions = false;
}
Z3SmtLibItpMarker.resetMarkerCount();
issueGeneralCommand("(push 1)"); // Topmost frame contains marker assertions
topMostContainsAssertions = true;
for (final var marker : markers.toCollection()) {
final var terms = marker.getTerms();
if (terms.size() == 0) {
issueGeneralCommand(
String.format("(assert (! true :named %s))", marker.getMarkerName()));
} else {
final var term = String.format("(and %s)", String.join(" ",
marker.getTerms().stream().map(Tuple2::get2)
.collect(Collectors.toUnmodifiableList())));
issueGeneralCommand(
String.format("(assert (! %s :named %s))", term, marker.getMarkerName()));
}
}
return super.check();
}
@Override
public void push() {
if (topMostContainsAssertions) {
issueGeneralCommand("(pop 1)"); // Topmost frame contains marker assertions
topMostContainsAssertions = false;
}
super.push(); // Topmost frame contains marker assertions
}
@Override
public void pop(int n) {
if (topMostContainsAssertions) {
issueGeneralCommand("(pop 1)"); // Topmost frame contains marker assertions
topMostContainsAssertions = false;
}
super.pop(n); // Topmost frame contains marker assertions
}
@Override
public Interpolant getInterpolant(final ItpPattern pattern) {
checkState(getStatus() == SolverStatus.UNSAT,
"Cannot get interpolant if status is not UNSAT.");
checkArgument(pattern instanceof SmtLibItpPattern);
@SuppressWarnings("unchecked") final var z3ItpPattern = (SmtLibItpPattern) pattern;
final var term = patternToTerm(z3ItpPattern.getRoot());
final var markerCount = getMarkerCount(z3ItpPattern.getRoot());
final List> itpList = new LinkedList<>();
solverBinary.issueCommand(String.format("(get-interpolant %s)", term));
for (var i = 0; i < markerCount; i++) {
final var res = parseItpResponse(solverBinary.readResponse());
itpList.add(termTransformer.toExpr(res, BoolExprs.Bool(),
new SmtLibModel(Collections.emptyMap())));
}
// itpList.add(False());
final Map> itpMap = new HashMap<>();
buildItpMapFormList(z3ItpPattern.getRoot(), itpList, itpMap);
return new SmtLibInterpolant(itpMap);
}
private String patternToTerm(final ItpMarkerTree markerTree) {
final Collection opTerms = new LinkedList<>();
final Z3SmtLibItpMarker marker = markerTree.getMarker();
opTerms.add(marker.getMarkerName());
for (final var child : markerTree.getChildren()) {
final var childTerm = patternToTerm(child);
opTerms.add(childTerm);
}
return String.format("(interp (and %s))", String.join(" ", opTerms));
}
private void buildItpMapFormList(final ItpMarkerTree pattern,
final List> itpList,
final Map> itpMap) {
for (final ItpMarkerTree child : pattern.getChildren()) {
buildItpMapFormList(child, itpList, itpMap);
}
final ItpMarker marker = pattern.getMarker();
final Expr itpExpr = itpList.get(0);
itpMap.put(marker, itpExpr);
itpList.remove(0);
}
private int getMarkerCount(final ItpMarkerTree markerTree) {
return 1 + markerTree.getChildren().stream().mapToInt(this::getMarkerCount).sum();
}
@Override
protected void init() {
issueGeneralCommand("(set-option :print-success true)");
issueGeneralCommand("(set-option :produce-interpolants true)");
super.init();
issueGeneralCommand("(push 1)"); // Topmost frame contains marker assertions
}
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.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()));
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy