src/Test/tablefuncencodetest.c

   1 #include "csolver.h"
   2 /**
   3  * e1 = {1, 2}
   4  * e2={3, 5, 7}
   5  * e3= f(e1, e2)
   6  *      1 5 => 7
   7  *      2 3 => 5
   8  *      1 7 => 3
   9  *      2 7 => 5
  10  *      2 5 => 3
  11  *      1 3 => 5
  12  * e4 = {6, 10, 19}
  13  * e4 <= e3
  14  * Result: e1=1, e2=5, e3=7, e4=6, overflow=0
  15  */
  16 int main(int numargs, char **argv) {
  17         CSolver *solver = allocCSolver();
  18         uint64_t set1[] = {1, 2};
  19         uint64_t set2[] = {3, 5, 7};
  20         uint64_t set3[] = {6, 10, 19};
  21         Set *s1 = createSet(solver, 0, set1, 2);
  22         Set *s2 = createSet(solver, 0, set2, 3);
  23         Set *s3 = createSet(solver, 0, set3, 3);
  24         Element *e1 = getElementVar(solver, s1);
  25         Element *e2 = getElementVar(solver, s2);
  26         Element *e4 = getElementVar(solver, s3);
  27         Boolean *overflow = getBooleanVar(solver, 2);
  28         Set *d1[] = {s1, s2};
  29         //change the overflow flag
  30         Table *t1 = createTable(solver, d1, 2, s2);
  31         uint64_t row1[] = {1, 5};
  32         uint64_t row2[] = {2, 3};
  33         uint64_t row3[] = {1, 7};
  34         uint64_t row4[] = {2, 7};
  35         uint64_t row5[] = {2, 5};
  36         uint64_t row6[] = {1, 3};
  37         addTableEntry(solver, t1, row1, 2, 7);
  38         addTableEntry(solver, t1, row2, 2, 5);
  39         addTableEntry(solver, t1, row3, 2, 3);
  40         addTableEntry(solver, t1, row4, 2, 5);
  41         addTableEntry(solver, t1, row5, 2, 3);
  42         addTableEntry(solver, t1, row6, 2, 5);
  43         Function *f1 = completeTable(solver, t1, FLAGIFFUNDEFINED);
  44         Element *e3 = applyFunction(solver, f1, (Element * []) {e1,e2}, 2, overflow);
  45
  46         Set *deq[] = {s3,s2};
  47         Predicate *lte = createPredicateOperator(solver, LTE, deq, 2);
  48         Element *inputs2 [] = {e4, e3};
  49         Boolean *pred = applyPredicate(solver, lte, inputs2, 2);
  50         addConstraint(solver, pred);
  51
  52         if (startEncoding(solver) == 1)
  53                 printf("e1=%llu e2=%llu e3=%llu e4=%llu overFlow:%d\n",
  54                                          getElementValue(solver,e1), getElementValue(solver, e2), getElementValue(solver, e3),
  55                                          getElementValue(solver, e4), getBooleanValue(solver, overflow));
  56         else
  57                 printf("UNSAT\n");
  58         deleteSolver(solver);
  59 }