use of org.evosuite.symbolic.solver.smt.SmtIntVariable in project evosuite by EvoSuite.
the class CVC4Solver method buildSmtCheckSatQuery.
private static SmtCheckSatQuery buildSmtCheckSatQuery(Collection<Constraint<?>> constraints) {
ConstraintToCVC4Visitor v = new ConstraintToCVC4Visitor(true);
SmtVariableCollector varCollector = new SmtVariableCollector();
SmtOperatorCollector funCollector = new SmtOperatorCollector();
List<SmtAssertion> smtAssertions = new LinkedList<SmtAssertion>();
for (Constraint<?> c : constraints) {
SmtExpr smtExpr = c.accept(v, null);
if (smtExpr != null) {
SmtAssertion smtAssertion = new SmtAssertion(smtExpr);
smtAssertions.add(smtAssertion);
smtExpr.accept(varCollector, null);
smtExpr.accept(funCollector, null);
}
}
Set<SmtVariable> variables = varCollector.getSmtVariables();
if (variables.isEmpty()) {
// no variables, constraint system is trivial
return null;
}
List<SmtFunctionDefinition> functionDefinitions = new LinkedList<SmtFunctionDefinition>();
final boolean addCharToInt = funCollector.getOperators().contains(Operator.CHAR_TO_INT);
if (addCharToInt) {
String charToIntFunction = buildCharToIntFunction();
SmtFunctionDefinition funcDefinition = new SmtFunctionDefinition(charToIntFunction);
functionDefinitions.add(funcDefinition);
}
final boolean addIntToChar = funCollector.getOperators().contains(Operator.INT_TO_CHAR);
if (addIntToChar) {
String intToCharFunction = buildIntToCharFunction();
SmtFunctionDefinition funcDefinition = new SmtFunctionDefinition(intToCharFunction);
functionDefinitions.add(funcDefinition);
}
List<SmtFunctionDeclaration> functionDeclarations = new LinkedList<SmtFunctionDeclaration>();
for (SmtVariable var : variables) {
String varName = var.getName();
if (var instanceof SmtIntVariable) {
SmtFunctionDeclaration intVar = SmtExprBuilder.mkIntFunctionDeclaration(varName);
functionDeclarations.add(intVar);
} else if (var instanceof SmtRealVariable) {
SmtFunctionDeclaration realVar = SmtExprBuilder.mkRealFunctionDeclaration(varName);
functionDeclarations.add(realVar);
} else if (var instanceof SmtStringVariable) {
SmtFunctionDeclaration stringVar = SmtExprBuilder.mkStringFunctionDeclaration(varName);
functionDeclarations.add(stringVar);
} else {
throw new RuntimeException("Unknown variable type " + var.getClass().getCanonicalName());
}
}
SmtCheckSatQuery smtQuery = new SmtCheckSatQuery(new LinkedList<SmtConstantDeclaration>(), functionDeclarations, functionDefinitions, smtAssertions);
return smtQuery;
}
use of org.evosuite.symbolic.solver.smt.SmtIntVariable in project evosuite by EvoSuite.
the class Z3Str2Solver method buildSmtQuerty.
private static SmtCheckSatQuery buildSmtQuerty(Collection<Constraint<?>> constraints) {
ConstraintToZ3Str2Visitor v = new ConstraintToZ3Str2Visitor();
List<SmtAssertion> assertions = new LinkedList<SmtAssertion>();
SmtVariableCollector varCollector = new SmtVariableCollector();
SmtOperatorCollector opCollector = new SmtOperatorCollector();
for (Constraint<?> c : constraints) {
SmtExpr smtExpr = c.accept(v, null);
if (smtExpr != null) {
SmtAssertion newAssertion = new SmtAssertion(smtExpr);
assertions.add(newAssertion);
smtExpr.accept(varCollector, null);
smtExpr.accept(opCollector, null);
}
}
Set<SmtVariable> smtVariables = varCollector.getSmtVariables();
Set<Operator> smtOperators = opCollector.getOperators();
boolean addCharToIntFunction;
if (smtOperators.contains(SmtOperation.Operator.CHAR_TO_INT)) {
addCharToIntFunction = true;
} else {
addCharToIntFunction = false;
}
Set<SmtVariable> smtVariablesToDeclare = new HashSet<SmtVariable>(smtVariables);
if (addCharToIntFunction) {
Set<SmtStringVariable> charVariables = buildCharVariables();
smtVariablesToDeclare.addAll(charVariables);
}
List<SmtConstantDeclaration> constantDeclarations = new LinkedList<SmtConstantDeclaration>();
for (SmtVariable v1 : smtVariablesToDeclare) {
String varName = v1.getName();
if (v1 instanceof SmtIntVariable) {
SmtConstantDeclaration constantDecl = SmtExprBuilder.mkIntConstantDeclaration(varName);
constantDeclarations.add(constantDecl);
} else if (v1 instanceof SmtRealVariable) {
SmtConstantDeclaration constantDecl = SmtExprBuilder.mkRealConstantDeclaration(varName);
constantDeclarations.add(constantDecl);
} else if (v1 instanceof SmtStringVariable) {
SmtConstantDeclaration constantDecl = SmtExprBuilder.mkStringConstantDeclaration(varName);
constantDeclarations.add(constantDecl);
} else {
throw new RuntimeException("Unknown variable type " + v1.getClass().getCanonicalName());
}
}
List<SmtFunctionDefinition> functionDefinitions = new LinkedList<SmtFunctionDefinition>();
if (addCharToIntFunction) {
String charToInt = buildCharToIntFunction();
SmtFunctionDefinition newFunctionDef = new SmtFunctionDefinition(charToInt);
functionDefinitions.add(newFunctionDef);
}
SmtCheckSatQuery smtCheckSatQuery = new SmtCheckSatQuery(constantDeclarations, functionDefinitions, assertions);
return smtCheckSatQuery;
}
use of org.evosuite.symbolic.solver.smt.SmtIntVariable in project evosuite by EvoSuite.
the class ExprToCVC4Visitor method visit.
@Override
public SmtExpr visit(IntegerVariable e, Void v) {
String varName = e.getName();
SmtIntVariable var = SmtExprBuilder.mkIntVariable(varName);
return var;
}
Aggregations