Small edit
[satune.git] / src / Test / funcencodingtest.cc
1 #include "csolver.h"
2 /**
3  * e1 = {6}
4  * e2={4, 2}
5  * e3=Fsub(e1,e2) {4, 2}
6  * e4= f(e1, e2)
7  *      6 2 => 3
8  *      6 4 => 1
9  * e5 = f(e1)=>e1 {6}
10  * e6 = (e3, e4, e5) {2, 3, 1}
11  *      4 3 6 => 3
12  *      2 1 6 => 1
13  *      2 3 6 => 2
14  *      4 1 6 => 1
15  * e7 = {2, 1, 0}
16  * e7 > e6
17  * Result: e1=6, e2=4, e7=2
18  */
19 int main(int numargs, char **argv) {
20         CSolver *solver = new CSolver();
21         uint64_t set1[] = {6};
22         uint64_t set2[] = {4, 2};
23         uint64_t set3[] = {3, 1};
24         uint64_t set4[] = {2, 3, 1};
25         uint64_t set5[] = {2, 1, 0};
26         Set *s1 = solver->createSet(0, set1, 1);
27         Set *s2 = solver->createSet(0, set2, 2);
28         Set *s3 = solver->createSet(0, set3, 2);
29         Set *s4 = solver->createSet(0, set4, 3);
30         Set *s5 = solver->createSet(0, set5, 3);
31         Element *e1 = solver->getElementVar(s1);
32         Element *e2 = solver->getElementVar(s2);
33         Element *e7 = solver->getElementVar(s5);
34         Boolean *overflow = solver->getBooleanVar(2);
35         Set *d1[] = {s1, s2};
36         //change the overflow flag
37         Function *f1 = solver->createFunctionOperator(SATC_SUB, d1, 2, s2, SATC_IGNORE);
38         Element *in1[] = {e1, e2};
39         Element *e3 = solver->applyFunction(f1, in1, 2, overflow);
40         Table *t1 = solver->createTable(d1, 2, s3);
41         uint64_t row1[] = {6, 2};
42         uint64_t row2[] = {6, 4};
43         solver->addTableEntry(t1, row1, 2, 3);
44         solver->addTableEntry(t1, row2, 2, 1);
45         Function *f2 = solver->completeTable(t1, SATC_IGNOREBEHAVIOR);
46         Element *e4 = solver->applyFunction(f2, in1, 2, overflow);
47
48         Set *d2[] = {s1};
49         Element *in2[] = {e1};
50         Table *t2 = solver->createTable(d2, 1, s1);
51         uint64_t row3[] = {6};
52         solver->addTableEntry(t2, row3, 1, 6);
53         Function *f3 = solver->completeTable(t2, SATC_IGNOREBEHAVIOR);
54         Element *e5 = solver->applyFunction(f3, in2, 1, overflow);
55
56         Set *d3[] = {s2, s3, s1};
57         Element *in3[] = {e3, e4, e5};
58         Table *t3 = solver->createTable(d3, 3, s4);
59         uint64_t row4[] = {4, 3, 6};
60         uint64_t row5[] = {2, 1, 6};
61         uint64_t row6[] = {2, 3, 6};
62         uint64_t row7[] = {4, 1, 6};
63         solver->addTableEntry(t3, row4, 3, 3);
64         solver->addTableEntry(t3, row5, 3, 1);
65         solver->addTableEntry(t3, row6, 3, 2);
66         solver->addTableEntry(t3, row7, 3, 1);
67         Function *f4 = solver->completeTable(t3, SATC_IGNOREBEHAVIOR);
68         Element *e6 = solver->applyFunction(f4, in3, 3, overflow);
69
70         Set *deq[] = {s5,s4};
71         Predicate *gt = solver->createPredicateOperator(SATC_GT, deq, 2);
72         Element *inputs2 [] = {e7, e6};
73         Boolean *pred = solver->applyPredicate(gt, inputs2, 2);
74         solver->addConstraint(pred);
75
76         if (solver->solve() == 1)
77                 printf("e1=%" PRIu64 " e2=%" PRIu64 " e7=%" PRIu64 "\n",
78                                          solver->getElementValue(e1), solver->getElementValue(e2), solver->getElementValue(e7));
79         else
80                 printf("UNSAT\n");
81         delete solver;
82 }