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 #include "boolean.h"
   9
  10 Element::Element(ASTNodeType _type) :
  11         ASTNode(_type),
  12         encoding(this) {
  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         return solver->getElementConst(type, value);
  40 }
  41
  42 Element *ElementSet::clone(CSolver *solver, CloneMap *map) {
  43         Element *e = (Element *) map->get(this);
  44         if (e != NULL)
  45                 return e;
  46         e = solver->getElementVar(set->clone(solver, map));
  47         map->put(e, e);
  48         return e;
  49 }
  50
  51 Element *ElementFunction::clone(CSolver *solver, CloneMap *map) {
  52         Element *array[inputs.getSize()];
  53         for (uint i = 0; i < inputs.getSize(); i++) {
  54                 array[i] = inputs.get(i)->clone(solver, map);
  55         }
  56         Element *e = solver->applyFunction(function->clone(solver, map), array, inputs.getSize(), overflowstatus->clone(solver, map));
  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 }
  63
  64 Set * ElementFunction::getRange() {
  65         return function->getRange();
  66 }
  67
  68 void ElementSet::serialize(Serializer* serializer){
  69         if(serializer->isSerialized(this))
  70                 return;
  71         serializer->addObject(this);
  72
  73         set->serialize(serializer);
  74
  75         serializer->mywrite(&type, sizeof(ASTNodeType));
  76         ElementSet *This = this;
  77         serializer->mywrite(&This, sizeof(ElementSet*));
  78         serializer->mywrite(&set, sizeof(Set*));
  79 }
  80
  81 void ElementConst::serialize(Serializer* serializer){
  82         if(serializer->isSerialized(this))
  83                 return;
  84         serializer->addObject(this);
  85
  86         set->serialize(serializer);
  87
  88         serializer->mywrite(&type, sizeof(ASTNodeType));
  89         ElementSet *This = this;
  90         serializer->mywrite(&This, sizeof(ElementSet*));
  91         VarType type = set->getType();
  92         serializer->mywrite(&type, sizeof(VarType));
  93         serializer->mywrite(&value, sizeof(uint64_t));
  94 }
  95
  96 void ElementFunction::serialize(Serializer* serializer){
  97         if(serializer->isSerialized(this))
  98                 return;
  99         serializer->addObject(this);
 100
 101         function->serialize(serializer);
 102         uint size = inputs.getSize();
 103         for(uint i=0; i<size; i++){
 104                 Element *input = inputs.get(i);
 105                 input->serialize(serializer);
 106         }
 107         serializeBooleanEdge(serializer, overflowstatus);
 108
 109         serializer->mywrite(&type, sizeof(ASTNodeType));
 110         ElementFunction *This = this;
 111         serializer->mywrite(&This, sizeof(ElementFunction *));
 112         serializer->mywrite(&function, sizeof(Function *));
 113         serializer->mywrite(&size, sizeof(uint));
 114         for(uint i=0; i<size; i++){
 115                 Element* input = inputs.get(i);
 116                 serializer->mywrite(&input, sizeof(Element*));
 117         }
 118         Boolean* overflowstat = overflowstatus.getRaw();
 119         serializer->mywrite(&overflowstat, sizeof(Boolean*));
 120 }
 121