99cd311757051f10da2a8dccb0a4b6062fda65f3
[satune.git] / src / AST / predicate.cc
1 #include "predicate.h"
2 #include "boolean.h"
3 #include "set.h"
4 #include "table.h"
5 #include "csolver.h"
6
7 PredicateOperator::PredicateOperator(CompOp _op, Set **domain, uint numDomain) : Predicate(OPERATORPRED), op(_op), domains(domain, numDomain) {
8 }
9
10 PredicateTable::PredicateTable(Table *_table, UndefinedBehavior _undefBehavior) : Predicate(TABLEPRED), table(_table), undefinedbehavior(_undefBehavior) {
11 }
12
13 bool PredicateOperator::evalPredicateOperator(uint64_t *inputs) {
14         switch (op) {
15         case EQUALS:
16                 return inputs[0] == inputs[1];
17         case LT:
18                 return inputs[0] < inputs[1];
19         case GT:
20                 return inputs[0] > inputs[1];
21         case LTE:
22                 return inputs[0] <= inputs[1];
23         case GTE:
24                 return inputs[0] >= inputs[1];
25         }
26         ASSERT(0);
27         return false;
28 }
29
30 Predicate *PredicateOperator::clone(CSolver *solver, CloneMap *map) {
31         Predicate *p = (Predicate *) map->get(this);
32         if (p != NULL)
33                 return p;
34
35         Set *array[domains.getSize()];
36         for (uint i = 0; i < domains.getSize(); i++)
37                 array[i] = domains.get(i)->clone(solver, map);
38
39         p = solver->createPredicateOperator(op, array, domains.getSize());
40         map->put(this, p);
41         return p;
42 }
43
44 Predicate *PredicateTable::clone(CSolver *solver, CloneMap *map) {
45         Predicate *p = (Predicate *) map->get(this);
46         if (p != NULL)
47                 return p;
48
49         p = solver->createPredicateTable(table->clone(solver, map), undefinedbehavior);
50         map->put(this, p);
51         return p;
52 }