Theory of Bags

examples/api/cpp/bags.cpp

  1/******************************************************************************
  2 * Top contributors (to current version):
  3 *   Mudathir Mohamed
  4 *
  5 * This file is part of the cvc5 project.
  6 *
  7 * Copyright (c) 2009-2022 by the authors listed in the file AUTHORS
  8 * in the top-level source directory and their institutional affiliations.
  9 * All rights reserved.  See the file COPYING in the top-level source
 10 * directory for licensing information.
 11 * ****************************************************************************
 12 *
 13 * A simple demonstration of reasoning about bags.
 14 */
 15
 16#include <cvc5/cvc5.h>
 17
 18#include <iostream>
 19
 20using namespace std;
 21using namespace cvc5;
 22
 23int main()
 24{
 25  Solver slv;
 26  slv.setLogic("ALL");
 27  // Produce models
 28  slv.setOption("produce-models", "true");
 29  slv.setOption("incremental", "true");
 30
 31  Sort bag = slv.mkBagSort(slv.getStringSort());
 32  Term A = slv.mkConst(bag, "A");
 33  Term B = slv.mkConst(bag, "B");
 34  Term C = slv.mkConst(bag, "C");
 35  Term x = slv.mkConst(slv.getStringSort(), "x");
 36
 37  Term intersectionAC = slv.mkTerm(Kind::BAG_INTER_MIN, {A, C});
 38  Term intersectionBC = slv.mkTerm(Kind::BAG_INTER_MIN, {B, C});
 39
 40  // union disjoint does not distribute over intersection
 41  {
 42    Term unionDisjointAB = slv.mkTerm(Kind::BAG_UNION_DISJOINT, {A, B});
 43    Term lhs = slv.mkTerm(Kind::BAG_INTER_MIN, {unionDisjointAB, C});
 44    Term rhs =
 45        slv.mkTerm(Kind::BAG_UNION_DISJOINT, {intersectionAC, intersectionBC});
 46    Term guess = slv.mkTerm(Kind::EQUAL, {lhs, rhs});
 47    cout << "cvc5 reports: " << guess.notTerm() << " is "
 48         << slv.checkSatAssuming(guess.notTerm()) << "." << endl;
 49
 50    cout << A << ": " << slv.getValue(A) << endl;
 51    cout << B << ": " << slv.getValue(B) << endl;
 52    cout << C << ": " << slv.getValue(C) << endl;
 53    cout << lhs << ": " << slv.getValue(lhs) << endl;
 54    cout << rhs << ": " << slv.getValue(rhs) << endl;
 55  }
 56
 57  // union max distributes over intersection
 58  {
 59    Term unionMaxAB = slv.mkTerm(Kind::BAG_UNION_MAX, {A, B});
 60    Term lhs = slv.mkTerm(Kind::BAG_INTER_MIN, {unionMaxAB, C});
 61    Term rhs =
 62        slv.mkTerm(Kind::BAG_UNION_MAX, {intersectionAC, intersectionBC});
 63    Term theorem = slv.mkTerm(Kind::EQUAL, {lhs, rhs});
 64    cout << "cvc5 reports: " << theorem.notTerm() << " is "
 65         << slv.checkSatAssuming(theorem.notTerm()) << "." << endl;
 66  }
 67
 68  // Verify emptbag is a subbag of any bag
 69  {
 70    Term emptybag = slv.mkEmptyBag(bag);
 71    Term theorem = slv.mkTerm(Kind::BAG_SUBBAG, {emptybag, A});
 72
 73    cout << "cvc5 reports: " << theorem.notTerm() << " is "
 74         << slv.checkSatAssuming(theorem.notTerm()) << "." << endl;
 75  }
 76
 77  // find an element with multiplicity 4 in the disjoint union of
 78  // ; {|"a", "a", "b", "b", "b"|} and {|"b", "c", "c"|}
 79
 80  {
 81    Term one = slv.mkInteger(1);
 82    Term two = slv.mkInteger(2);
 83    Term three = slv.mkInteger(3);
 84    Term four = slv.mkInteger(4);
 85    Term a = slv.mkString("a");
 86    Term b = slv.mkString("b");
 87    Term c = slv.mkString("c");
 88
 89    Term bag_a_2 = slv.mkTerm(Kind::BAG_MAKE, {a, two});
 90    Term bag_b_3 = slv.mkTerm(Kind::BAG_MAKE, {b, three});
 91    Term bag_b_1 = slv.mkTerm(Kind::BAG_MAKE, {b, one});
 92    Term bag_c_2 = slv.mkTerm(Kind::BAG_MAKE, {c, two});
 93    Term bag_a_2_b_3 = slv.mkTerm(Kind::BAG_UNION_DISJOINT, {bag_a_2, bag_b_3});
 94    Term bag_b_1_c_2 = slv.mkTerm(Kind::BAG_UNION_DISJOINT, {bag_b_1, bag_c_2});
 95    Term union_disjoint =
 96        slv.mkTerm(Kind::BAG_UNION_DISJOINT, {bag_a_2_b_3, bag_b_1_c_2});
 97
 98    Term count_x = slv.mkTerm(Kind::BAG_COUNT, {x, union_disjoint});
 99    Term e = slv.mkTerm(Kind::EQUAL, {four, count_x});
100    Result result = slv.checkSatAssuming(e);
101
102    cout << "cvc5 reports: " << e << " is " << result << "." << endl;
103    if (result.isSat())
104    {
105      cout << x << ": " << slv.getValue(x) << endl;
106    }
107  }
108}