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