Edits
[satune.git] / src / AST / element.cc
1 #include "element.h"
2 #include "structs.h"
3 #include "set.h"
4 #include "constraint.h"
5 #include "function.h"
6 #include "table.h"
7 #include "csolver.h"
8
9 Element::Element(ASTNodeType _type) :
10         ASTNode(_type),
11         encoding(this) {
12 }
13
14 ElementSet::ElementSet(Set *s) :
15         Element(ELEMSET),
16         set(s) {
17 }
18
19 ElementFunction::ElementFunction(Function *_function, Element **array, uint numArrays, BooleanEdge _overflowstatus) :
20         Element(ELEMFUNCRETURN),
21         function(_function),
22         inputs(array, numArrays),
23         overflowstatus(_overflowstatus),
24         functionencoding(this) {
25 }
26
27 ElementConst::ElementConst(uint64_t _value, VarType _type, Set *_set) :
28         Element(ELEMCONST),
29         set(_set),
30         value(_value) {
31 }
32
33 Set *getElementSet(Element *This) {
34         switch (This->type) {
35         case ELEMSET:
36                 return ((ElementSet *)This)->set;
37         case ELEMCONST:
38                 return ((ElementConst *)This)->set;
39         case ELEMFUNCRETURN: {
40                 Function *func = ((ElementFunction *)This)->function;
41                 return func->getRange();
42         }
43         default:
44                 ASSERT(0);
45         }
46         ASSERT(0);
47         return NULL;
48 }
49
50 Element *ElementConst::clone(CSolver *solver, CloneMap *map) {
51         return solver->getElementConst(type, value);
52 }
53
54 Element *ElementSet::clone(CSolver *solver, CloneMap *map) {
55         Element *e = (Element *) map->get(this);
56         if (e != NULL)
57                 return e;
58         e = solver->getElementVar(set->clone(solver, map));
59         map->put(e, e);
60         return e;
61 }
62
63 Element *ElementFunction::clone(CSolver *solver, CloneMap *map) {
64         Element *array[inputs.getSize()];
65         for (uint i = 0; i < inputs.getSize(); i++) {
66                 array[i] = inputs.get(i)->clone(solver, map);
67         }
68         Element *e = solver->applyFunction(function->clone(solver, map), array, inputs.getSize(), overflowstatus->clone(solver, map));
69         return e;
70 }
71
72 void ElementFunction::updateParents() {
73         for(uint i=0;i < inputs.getSize(); i++) inputs.get(i)->parents.push(this);
74 }