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         anyValue(false),
  13         frozen(false){
  14 }
  15
  16 ElementSet::ElementSet(Set *s) :
  17         Element(ELEMSET),
  18         set(s) {
  19 }
  20
  21 ElementSet::ElementSet(ASTNodeType _type, Set *s) :
  22         Element(_type),
  23         set(s) {
  24 }
  25
  26 ElementFunction::ElementFunction(Function *_function, Element **array, uint numArrays, BooleanEdge _overflowstatus) :
  27         Element(ELEMFUNCRETURN),
  28         inputs(array, numArrays),
  29         overflowstatus(_overflowstatus),
  30         functionencoding(this),
  31         function(_function) {
  32 }
  33
  34 ElementConst::ElementConst(uint64_t _value, Set *_set) :
  35         ElementSet(ELEMCONST, _set),
  36         value(_value) {
  37 }
  38
  39 Element *ElementConst::clone(CSolver *solver, CloneMap *map) {
  40         Element *e = solver->getElementConst(type, value);
  41         e->anyValue = anyValue;
  42         return e;
  43 }
  44
  45 Element *ElementSet::clone(CSolver *solver, CloneMap *map) {
  46         Element *e = (Element *) map->get(this);
  47         if (e != NULL)
  48                 return e;
  49         e = solver->getElementVar(set->clone(solver, map));
  50         map->put(this, e);
  51         e->anyValue = anyValue;
  52         return e;
  53 }
  54
  55 Element *ElementFunction::clone(CSolver *solver, CloneMap *map) {
  56         Element *array[inputs.getSize()];
  57         for (uint i = 0; i < inputs.getSize(); i++) {
  58                 array[i] = inputs.get(i)->clone(solver, map);
  59         }
  60         BooleanEdge ofstatus = overflowstatus ? cloneEdge(solver, map, overflowstatus) : BooleanEdge();
  61         Element *e = solver->applyFunction(function->clone(solver, map), array, inputs.getSize(), ofstatus);
  62         e->anyValue = anyValue;
  63         return e;
  64 }
  65
  66 void ElementFunction::updateParents() {
  67         for (uint i = 0; i < inputs.getSize(); i++) inputs.get(i)->parents.push(this);
  68         if (overflowstatus)
  69                 overflowstatus->parents.push(this);
  70 }
  71
  72 Set *ElementFunction::getRange() {
  73         return function->getRange();
  74 }
  75
  76 void ElementSet::serialize(Serializer *serializer) {
  77         if (serializer->isSerialized(this))
  78                 return;
  79         serializer->addObject(this);
  80
  81         set->serialize(serializer);
  82
  83         serializer->mywrite(&type, sizeof(ASTNodeType));
  84         serializer->mywrite(&anyValue, sizeof(bool));
  85         ElementSet *This = this;
  86         serializer->mywrite(&This, sizeof(ElementSet *));
  87         serializer->mywrite(&set, sizeof(Set *));
  88 }
  89
  90 void ElementSet::print() {
  91         model_print("{ElementSet<%p>:", this);
  92         set->print();
  93         model_print(" %p ", this);
  94         getElementEncoding()->print();
  95         model_print("}");
  96 }
  97
  98 void ElementConst::serialize(Serializer *serializer) {
  99         if (serializer->isSerialized(this))
 100                 return;
 101         serializer->addObject(this);
 102
 103         set->serialize(serializer);
 104
 105         serializer->mywrite(&type, sizeof(ASTNodeType));
 106         serializer->mywrite(&anyValue, sizeof(bool));
 107         ElementSet *This = this;
 108         serializer->mywrite(&This, sizeof(ElementSet *));
 109         VarType type = set->getType();
 110         serializer->mywrite(&type, sizeof(VarType));
 111         serializer->mywrite(&value, sizeof(uint64_t));
 112 }
 113
 114 void ElementConst::print() {
 115         model_print("{ElementConst<%p>: %" PRIu64 "}\n", this, value);
 116 }
 117
 118 void ElementFunction::serialize(Serializer *serializer) {
 119         if (serializer->isSerialized(this))
 120                 return;
 121         serializer->addObject(this);
 122
 123         function->serialize(serializer);
 124         uint size = inputs.getSize();
 125         for (uint i = 0; i < size; i++) {
 126                 Element *input = inputs.get(i);
 127                 input->serialize(serializer);
 128         }
 129         serializeBooleanEdge(serializer, overflowstatus);
 130
 131         serializer->mywrite(&type, sizeof(ASTNodeType));
 132         serializer->mywrite(&anyValue, sizeof(bool));
 133         ElementFunction *This = this;
 134         serializer->mywrite(&This, sizeof(ElementFunction *));
 135         serializer->mywrite(&function, sizeof(Function *));
 136         serializer->mywrite(&size, sizeof(uint));
 137         for (uint i = 0; i < size; i++) {
 138                 Element *input = inputs.get(i);
 139                 serializer->mywrite(&input, sizeof(Element *));
 140         }
 141         Boolean *overflowstat = overflowstatus.getRaw();
 142         serializer->mywrite(&overflowstat, sizeof(Boolean *));
 143 }
 144
 145 void ElementFunction::print() {
 146         model_print("{ElementFunction<%p>:\n", this);
 147         function->print();
 148         model_print("OverFlow Boolean Flag:\n");
 149         overflowstatus.getBoolean()->print();
 150         model_print("Range:\n");
 151         getRange()->print();
 152         model_print("Elements:\n");
 153         uint size = inputs.getSize();
 154         for (uint i = 0; i < size; i++) {
 155                 Element *input = inputs.get(i);
 156                 input->print();
 157         }
 158         model_print("}\n");
 159 }