5d8865e466717f6625fba9242c5dc15c9b7355e2
[satune.git] / src / AST / boolean.cc
1 #include "boolean.h"
2 #include "structs.h"
3 #include "csolver.h"
4 #include "element.h"
5 #include "order.h"
6 #include "predicate.h"
7
8 Boolean::Boolean(ASTNodeType _type) :
9         ASTNode(_type),
10         polarity(P_UNDEFINED),
11         boolVal(BV_UNDEFINED),
12         parents() {
13 }
14
15 BooleanConst::BooleanConst(bool _isTrue) :
16         Boolean(BOOLCONST),
17         istrue(_isTrue) {
18 }
19
20 BooleanVar::BooleanVar(VarType t) :
21         Boolean(BOOLEANVAR),
22         vtype(t),
23         var(E_NULL) {
24 }
25
26 BooleanOrder::BooleanOrder(Order *_order, uint64_t _first, uint64_t _second) :
27         Boolean(ORDERCONST),
28         order(_order),
29         first(_first),
30         second(_second) {
31         order->constraints.push(this);
32 }
33
34 BooleanPredicate::BooleanPredicate(Predicate *_predicate, Element **_inputs, uint _numInputs, BooleanEdge _undefinedStatus) :
35         Boolean(PREDICATEOP),
36         predicate(_predicate),
37         encoding(this),
38         inputs(_inputs, _numInputs),
39         undefStatus(_undefinedStatus) {
40         for (uint i = 0; i < _numInputs; i++) {
41                 _inputs[i]->parents.push(this);
42         }
43 }
44
45 BooleanLogic::BooleanLogic(CSolver *solver, LogicOp _op, BooleanEdge *array, uint asize) :
46         Boolean(LOGICOP),
47         op(_op),
48         replaced(false),
49         inputs(array, asize) {
50 }
51
52 BooleanEdge cloneEdge(CSolver *solver, CloneMap *map, BooleanEdge e) {
53         bool isnegated=e.isNegated();
54         Boolean *b=e->clone(solver, map);
55         BooleanEdge be=BooleanEdge(b);
56         return isnegated ? be.negate() : be;
57 }
58
59 Boolean *BooleanConst::clone(CSolver *solver, CloneMap *map) {
60         return solver->getBooleanTrue().getRaw();
61 }
62
63 Boolean *BooleanVar::clone(CSolver *solver, CloneMap *map) {
64         Boolean *b = (Boolean *) map->get(this);
65         if (b != NULL)
66                 return b;
67         BooleanEdge bvar = solver->getBooleanVar(type);
68         Boolean * base=bvar.getRaw();
69         map->put(this, base);
70         return base;
71 }
72
73 Boolean *BooleanOrder::clone(CSolver *solver, CloneMap *map) {
74         Order *ordercopy = order->clone(solver, map);
75         return solver->orderConstraint(ordercopy, first, second).getRaw();
76 }
77
78 Boolean *BooleanLogic::clone(CSolver *solver, CloneMap *map) {
79         BooleanEdge array[inputs.getSize()];
80         for (uint i = 0; i < inputs.getSize(); i++) {
81                 array[i] = cloneEdge(solver, map, inputs.get(i));
82         }
83         return solver->applyLogicalOperation(op, array, inputs.getSize()).getRaw();
84 }
85
86 Boolean *BooleanPredicate::clone(CSolver *solver, CloneMap *map) {
87         Element *array[inputs.getSize()];
88         for (uint i = 0; i < inputs.getSize(); i++) {
89                 array[i] = inputs.get(i)->clone(solver, map);
90         }
91         Predicate *pred = predicate->clone(solver, map);
92         BooleanEdge defstatus = undefStatus ? cloneEdge(solver, map, undefStatus) : BooleanEdge();
93
94         return solver->applyPredicateTable(pred, array, inputs.getSize(), defstatus).getRaw();
95 }