bug fixes for cloning and encodinggraph
[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 }
32
33 void BooleanOrder::updateParents() {
34         order->constraints.push(this);
35 }
36
37 BooleanPredicate::BooleanPredicate(Predicate *_predicate, Element **_inputs, uint _numInputs, BooleanEdge _undefinedStatus) :
38         Boolean(PREDICATEOP),
39         predicate(_predicate),
40         encoding(this),
41         inputs(_inputs, _numInputs),
42         undefStatus(_undefinedStatus) {
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         Boolean *b = (Boolean *) map->get(this);
75         if (b != NULL)
76                 return b;
77         Order *ordercopy = order->clone(solver, map);
78         b= solver->orderConstraint(ordercopy, first, second).getRaw();
79         map->put(this, b);
80         return b;
81 }
82
83 Boolean *BooleanLogic::clone(CSolver *solver, CloneMap *map) {
84         Boolean *b = (Boolean *) map->get(this);
85         if (b != NULL)
86                 return b;
87         BooleanEdge array[inputs.getSize()];
88         for (uint i = 0; i < inputs.getSize(); i++) {
89                 array[i] = cloneEdge(solver, map, inputs.get(i));
90         }
91         b= solver->applyLogicalOperation(op, array, inputs.getSize()).getRaw();
92         map->put(this, b);
93         return b;
94 }
95
96 Boolean *BooleanPredicate::clone(CSolver *solver, CloneMap *map) {
97         Boolean *b = (Boolean *) map->get(this);
98         if (b != NULL)
99                 return b;
100         Element *array[inputs.getSize()];
101         for (uint i = 0; i < inputs.getSize(); i++) {
102                 array[i] = inputs.get(i)->clone(solver, map);
103         }
104         Predicate *pred = predicate->clone(solver, map);
105         BooleanEdge defstatus = undefStatus ? cloneEdge(solver, map, undefStatus) : BooleanEdge();
106
107         b= solver->applyPredicateTable(pred, array, inputs.getSize(), defstatus).getRaw();
108         map->put(this, b);
109         return b;
110 }
111
112 void BooleanPredicate::updateParents() {
113         for (uint i = 0; i < inputs.getSize(); i++) inputs.get(i)->parents.push(this);
114         if (undefStatus)
115                 undefStatus->parents.push(this);
116 }
117
118 void BooleanLogic::updateParents() {
119         for (uint i = 0; i < inputs.getSize(); i++) inputs.get(i)->parents.push(this);
120 }
121
122 void BooleanVar::serialize(Serializer *serializer) {
123         if (serializer->isSerialized(this))
124                 return;
125         serializer->addObject(this);
126         serializer->mywrite(&type, sizeof(ASTNodeType));
127         BooleanVar *This = this;
128         serializer->mywrite(&This, sizeof(BooleanVar *));
129         serializer->mywrite(&vtype, sizeof(VarType));
130 }
131
132 void BooleanVar::print() {
133         model_print("BooleanVar<%p>\n", this);
134 }
135
136 void BooleanConst::print() {
137         model_print("BooleanConst<%p>:%s\n", this, istrue ? "TRUE" : "FALSE");
138 }
139
140 void BooleanOrder::serialize(Serializer *serializer) {
141         if (serializer->isSerialized(this))
142                 return;
143         serializer->addObject(this);
144         order->serialize(serializer);
145
146         serializer->mywrite(&type, sizeof(ASTNodeType));
147         BooleanOrder *This = this;
148         serializer->mywrite(&This, sizeof(BooleanOrder *));
149         serializer->mywrite(&order, sizeof(Order *));
150         serializer->mywrite(&first, sizeof(uint64_t));
151         serializer->mywrite(&second, sizeof(uint64_t));
152 }
153
154 void BooleanOrder::print() {
155         model_print("{BooleanOrder<%p>: First= %lu, Second = %lu on Order:\n", this, first, second);
156         order->print();
157         model_print("}\n");
158 }
159
160 void BooleanPredicate::serialize(Serializer *serializer) {
161         if (serializer->isSerialized(this))
162                 return;
163         serializer->addObject(this);
164
165         predicate->serialize(serializer);
166         uint size = inputs.getSize();
167         for (uint i = 0; i < size; i++) {
168                 Element *input = inputs.get(i);
169                 input->serialize(serializer);
170         }
171         serializeBooleanEdge(serializer, undefStatus);
172
173         serializer->mywrite(&type, sizeof(ASTNodeType));
174         BooleanPredicate *This = this;
175         serializer->mywrite(&This, sizeof(BooleanPredicate *));
176         serializer->mywrite(&predicate, sizeof(Predicate *));
177         serializer->mywrite(&size, sizeof(uint));
178         for (uint i = 0; i < size; i++) {
179                 Element *input = inputs.get(i);
180                 serializer->mywrite(&input, sizeof(Element *));
181         }
182         Boolean *undefStat = undefStatus != BooleanEdge(NULL) ? undefStatus.getRaw() : NULL;
183         serializer->mywrite(&undefStat, sizeof(Boolean *));
184 }
185
186 void BooleanPredicate::print() {
187         model_print("{BooleanPredicate<%p>:\n", this);
188         predicate->print();
189         model_print("elements:\n");
190         uint size = inputs.getSize();
191         for (uint i = 0; i < size; i++) {
192                 Element *input = inputs.get(i);
193                 input->print();
194         }
195         model_print("}\n");
196 }
197
198 void BooleanLogic::serialize(Serializer *serializer) {
199         if (serializer->isSerialized(this))
200                 return;
201         serializer->addObject(this);
202         uint size = inputs.getSize();
203         for (uint i = 0; i < size; i++) {
204                 BooleanEdge input = inputs.get(i);
205                 serializeBooleanEdge(serializer, input);
206         }
207         serializer->mywrite(&type, sizeof(ASTNodeType));
208         BooleanLogic *This = this;
209         serializer->mywrite(&This, sizeof(BooleanLogic *));
210         serializer->mywrite(&op, sizeof(LogicOp));
211         serializer->mywrite(&size, sizeof(uint));
212         for (uint i = 0; i < size; i++) {
213                 Boolean *input = inputs.get(i).getRaw();
214                 serializer->mywrite(&input, sizeof(Boolean *));
215         }
216 }
217
218 void BooleanLogic::print() {
219         model_print("{BooleanLogic<%p>: %s\n", this,
220                                                         op == SATC_AND ? "AND" : op == SATC_OR ? "OR" : op == SATC_NOT ? "NOT" :
221                                                         op == SATC_XOR ? "XOR" : op == SATC_IFF ? "IFF" : "IMPLIES");
222         uint size = inputs.getSize();
223         for (uint i = 0; i < size; i++) {
224                 BooleanEdge input = inputs.get(i);
225                 if (input.isNegated())
226                         model_print("!");
227                 input.getBoolean()->print();
228         }
229         model_print("}\n");
230 }
231