Files
cvc5/examples/api/python/sygus-inv.py
yoni206 1ac06eb879 Changing curly braces to brackets in python sygus examples (#9279)
The examples used {...} for lists rather than [...].
2022-11-21 21:46:04 +00:00

68 lines
2.1 KiB
Python

#!/usr/bin/env python
###############################################################################
# Top contributors (to current version):
# Yoni Zohar, Aina Niemetz, Mudathir Mohamed
#
# This file is part of the cvc5 project.
#
# Copyright (c) 2009-2022 by the authors listed in the file AUTHORS
# in the top-level source directory and their institutional affiliations.
# All rights reserved. See the file COPYING in the top-level source
# directory for licensing information.
# #############################################################################
#
# A simple demonstration of the solving capabilities of the cvc5
# sygus solver through the Python API. This is a direct
# translation of sygus-inv.cpp .
##
import utils
import cvc5
from cvc5 import Kind
if __name__ == "__main__":
slv = cvc5.Solver()
# required options
slv.setOption("sygus", "true")
slv.setOption("incremental", "false")
# set the logic
slv.setLogic("LIA")
integer = slv.getIntegerSort()
boolean = slv.getBooleanSort()
zero = slv.mkInteger(0)
one = slv.mkInteger(1)
ten = slv.mkInteger(10)
# declare input variables for functions
x = slv.mkVar(integer, "x")
xp = slv.mkVar(integer, "xp")
# (ite (< x 10) (= xp (+ x 1)) (= xp x))
ite = slv.mkTerm(Kind.ITE,
slv.mkTerm(Kind.LT, x, ten),
slv.mkTerm(Kind.EQUAL, xp, slv.mkTerm(Kind.ADD, x, one)),
slv.mkTerm(Kind.EQUAL, xp, x))
# define the pre-conditions, transition relations, and post-conditions
pre_f = slv.defineFun("pre-f", [x], boolean, slv.mkTerm(Kind.EQUAL, x, zero))
trans_f = slv.defineFun("trans-f", [x, xp], boolean, ite)
post_f = slv.defineFun("post-f", [x], boolean, slv.mkTerm(Kind.LEQ, x, ten))
# declare the invariant-to-synthesize
inv_f = slv.synthInv("inv-f", [x])
slv.addSygusInvConstraint(inv_f, pre_f, trans_f, post_f)
# print solutions if available
if slv.checkSynth().hasSolution():
# Output should be equivalent to:
# (define-fun inv-f ((x Int)) Bool (not (>= x 11)))
terms = [inv_f]
utils.print_synth_solutions(terms, slv.getSynthSolutions(terms))