Files
cvc5/examples/api/cpp/sygus-fun.cpp
Aina Niemetz 7a4a17b193 c api: Introduce C/C++ specific handling of public enums. (#9915)
This introduces a general way to use public enums both in the C++ and
the (upcoming) C API. For the C++ case, we now use enum classes rather
than enums for public enums. This also includes definitions for
C API to_string conversions for public enums. C API definitions in
cvc5_types.h are only included from the C API, guarded via a macro
(thus, for now, not included yet when the header is included).
2023-08-22 18:17:16 +00:00

114 lines
3.5 KiB
C++

/******************************************************************************
* Top contributors (to current version):
* Abdalrhman Mohamed, Mathias Preiner, Andrew Reynolds
*
* 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 Sygus API.
*
* A simple demonstration of how to use the Sygus API to synthesize max and min
* functions.
*/
#include <cvc5/cvc5.h>
#include <iostream>
#include "utils.h"
using namespace cvc5;
int main()
{
Solver slv;
// required options
slv.setOption("sygus", "true");
slv.setOption("incremental", "false");
// set the logic
slv.setLogic("LIA");
Sort integer = slv.getIntegerSort();
Sort boolean = slv.getBooleanSort();
// declare input variables for the functions-to-synthesize
Term x = slv.mkVar(integer, "x");
Term y = slv.mkVar(integer, "y");
// declare the grammar non-terminals
Term start = slv.mkVar(integer, "Start");
Term start_bool = slv.mkVar(boolean, "StartBool");
// define the rules
Term zero = slv.mkInteger(0);
Term one = slv.mkInteger(1);
Term plus = slv.mkTerm(Kind::ADD, {start, start});
Term minus = slv.mkTerm(Kind::SUB, {start, start});
Term ite = slv.mkTerm(Kind::ITE, {start_bool, start, start});
Term And = slv.mkTerm(Kind::AND, {start_bool, start_bool});
Term Not = slv.mkTerm(Kind::NOT, {start_bool});
Term leq = slv.mkTerm(Kind::LEQ, {start, start});
// create the grammar object
Grammar g = slv.mkGrammar({x, y}, {start, start_bool});
// bind each non-terminal to its rules
g.addRules(start, {zero, one, x, y, plus, minus, ite});
g.addRules(start_bool, {And, Not, leq});
// declare the functions-to-synthesize. Optionally, provide the grammar
// constraints
Term max = slv.synthFun("max", {x, y}, integer, g);
Term min = slv.synthFun("min", {x, y}, integer);
// declare universal variables.
Term varX = slv.declareSygusVar("x", integer);
Term varY = slv.declareSygusVar("y", integer);
Term max_x_y = slv.mkTerm(Kind::APPLY_UF, {max, varX, varY});
Term min_x_y = slv.mkTerm(Kind::APPLY_UF, {min, varX, varY});
// add semantic constraints
// (constraint (>= (max x y) x))
slv.addSygusConstraint(slv.mkTerm(Kind::GEQ, {max_x_y, varX}));
// (constraint (>= (max x y) y))
slv.addSygusConstraint(slv.mkTerm(Kind::GEQ, {max_x_y, varY}));
// (constraint (or (= x (max x y))
// (= y (max x y))))
slv.addSygusConstraint(
slv.mkTerm(Kind::OR,
{slv.mkTerm(Kind::EQUAL, {max_x_y, varX}),
slv.mkTerm(Kind::EQUAL, {max_x_y, varY})}));
// (constraint (= (+ (max x y) (min x y))
// (+ x y)))
slv.addSygusConstraint(slv.mkTerm(Kind::EQUAL,
{slv.mkTerm(Kind::ADD, {max_x_y, min_x_y}),
slv.mkTerm(Kind::ADD, {varX, varY})}));
// print solutions if available
if (slv.checkSynth().hasSolution())
{
// Output should be equivalent to:
// (
// (define-fun max ((x Int) (y Int)) Int (ite (<= x y) y x))
// (define-fun min ((x Int) (y Int)) Int (ite (<= x y) x y))
// )
std::vector<Term> terms = {max, min};
utils::printSynthSolutions(terms, slv.getSynthSolutions(terms));
}
return 0;
}