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(e, 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         Element *e = solver->applyFunction(function->clone(solver, map), array, inputs.getSize(), overflowstatus->clone(solver, map));
  56         return e;
  57 }
  58
  59 void ElementFunction::updateParents() {
  60         for(uint i=0;i < inputs.getSize(); i++) inputs.get(i)->parents.push(this);
  61 }
  62
  63 Set * ElementFunction::getRange() {
  64         return function->getRange();
  65 }
  66
  67 void ElementSet::serialize(Serializer* serializer){
  68         if(serializer->isSerialized(this))
  69                 return;
  70         serializer->addObject(this);
  71
  72         set->serialize(serializer);
  73
  74         serializer->mywrite(&type, sizeof(ASTNodeType));
  75         ElementSet *This = this;
  76         serializer->mywrite(&This, sizeof(ElementSet*));
  77         serializer->mywrite(&set, sizeof(Set*));
  78 }
  79
  80 void ElementSet::print(){
  81         model_println("{ElementSet:");
  82         set->print();
  83         model_println("}\n");
  84 }
  85
  86 void ElementConst::serialize(Serializer* serializer){
  87         if(serializer->isSerialized(this))
  88                 return;
  89         serializer->addObject(this);
  90
  91         set->serialize(serializer);
  92
  93         serializer->mywrite(&type, sizeof(ASTNodeType));
  94         ElementSet *This = this;
  95         serializer->mywrite(&This, sizeof(ElementSet*));
  96         VarType type = set->getType();
  97         serializer->mywrite(&type, sizeof(VarType));
  98         serializer->mywrite(&value, sizeof(uint64_t));
  99 }
 100
 101 void ElementConst::print(){
 102         model_println("{ElementConst: %lu}", value);
 103 }
 104
 105 void ElementFunction::serialize(Serializer* serializer){
 106         if(serializer->isSerialized(this))
 107                 return;
 108         serializer->addObject(this);
 109
 110         function->serialize(serializer);
 111         uint size = inputs.getSize();
 112         for(uint i=0; i<size; i++){
 113                 Element *input = inputs.get(i);
 114                 input->serialize(serializer);
 115         }
 116         serializeBooleanEdge(serializer, overflowstatus);
 117
 118         serializer->mywrite(&type, sizeof(ASTNodeType));
 119         ElementFunction *This = this;
 120         serializer->mywrite(&This, sizeof(ElementFunction *));
 121         serializer->mywrite(&function, sizeof(Function *));
 122         serializer->mywrite(&size, sizeof(uint));
 123         for(uint i=0; i<size; i++){
 124                 Element* input = inputs.get(i);
 125                 serializer->mywrite(&input, sizeof(Element*));
 126         }
 127         Boolean* overflowstat = overflowstatus.getRaw();
 128         serializer->mywrite(&overflowstat, sizeof(Boolean*));
 129 }
 130
 131 void ElementFunction::print(){
 132         model_println("{ElementFunction:");
 133         function->print();
 134         model_println("Elements:");
 135         uint size = inputs.getSize();
 136         for(uint i=0; i<size; i++){
 137                 Element *input = inputs.get(i);
 138                 input->print();
 139         }
 140         model_println("}\n");
 141 }