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