src/AST/boolean.cc

   1 #include "boolean.h"
   2 #include "structs.h"
   3 #include "csolver.h"
   4 #include "element.h"
   5 #include "order.h"
   6 #include "predicate.h"
   7
   8 Boolean::Boolean(ASTNodeType _type) :
   9         ASTNode(_type),
  10         polarity(P_UNDEFINED),
  11         boolVal(BV_UNDEFINED),
  12         parents() {
  13 }
  14
  15 BooleanConst::BooleanConst(bool _isTrue) :
  16         Boolean(BOOLCONST),
  17         istrue(_isTrue) {
  18 }
  19
  20 BooleanVar::BooleanVar(VarType t) :
  21         Boolean(BOOLEANVAR),
  22         vtype(t),
  23         var(E_NULL) {
  24 }
  25
  26 BooleanOrder::BooleanOrder(Order *_order, uint64_t _first, uint64_t _second) :
  27         Boolean(ORDERCONST),
  28         order(_order),
  29         first(_first),
  30         second(_second) {
  31 }
  32
  33 void BooleanOrder::updateParents() {
  34         order->constraints.push(this);
  35 }
  36
  37 BooleanPredicate::BooleanPredicate(Predicate *_predicate, Element **_inputs, uint _numInputs, BooleanEdge _undefinedStatus) :
  38         Boolean(PREDICATEOP),
  39         predicate(_predicate),
  40         encoding(this),
  41         inputs(_inputs, _numInputs),
  42         undefStatus(_undefinedStatus) {
  43 }
  44
  45 BooleanLogic::BooleanLogic(CSolver *solver, LogicOp _op, BooleanEdge *array, uint asize) :
  46         Boolean(LOGICOP),
  47         op(_op),
  48         replaced(false),
  49         inputs(array, asize) {
  50 }
  51
  52 BooleanEdge cloneEdge(CSolver *solver, CloneMap *map, BooleanEdge e) {
  53         bool isnegated = e.isNegated();
  54         Boolean *b = e->clone(solver, map);
  55         BooleanEdge be = BooleanEdge(b);
  56         return isnegated ? be.negate() : be;
  57 }
  58
  59 Boolean *BooleanConst::clone(CSolver *solver, CloneMap *map) {
  60         return solver->getBooleanTrue().getRaw();
  61 }
  62
  63 Boolean *BooleanVar::clone(CSolver *solver, CloneMap *map) {
  64         Boolean *b = (Boolean *) map->get(this);
  65         if (b != NULL)
  66                 return b;
  67         BooleanEdge bvar = solver->getBooleanVar(type);
  68         Boolean *base = bvar.getRaw();
  69         map->put(this, base);
  70         return base;
  71 }
  72
  73 Boolean *BooleanOrder::clone(CSolver *solver, CloneMap *map) {
  74         Order *ordercopy = order->clone(solver, map);
  75         return solver->orderConstraint(ordercopy, first, second).getRaw();
  76 }
  77
  78 Boolean *BooleanLogic::clone(CSolver *solver, CloneMap *map) {
  79         BooleanEdge array[inputs.getSize()];
  80         for (uint i = 0; i < inputs.getSize(); i++) {
  81                 array[i] = cloneEdge(solver, map, inputs.get(i));
  82         }
  83         return solver->applyLogicalOperation(op, array, inputs.getSize()).getRaw();
  84 }
  85
  86 Boolean *BooleanPredicate::clone(CSolver *solver, CloneMap *map) {
  87         Element *array[inputs.getSize()];
  88         for (uint i = 0; i < inputs.getSize(); i++) {
  89                 array[i] = inputs.get(i)->clone(solver, map);
  90         }
  91         Predicate *pred = predicate->clone(solver, map);
  92         BooleanEdge defstatus = undefStatus ? cloneEdge(solver, map, undefStatus) : BooleanEdge();
  93
  94         return solver->applyPredicateTable(pred, array, inputs.getSize(), defstatus).getRaw();
  95 }
  96
  97 void BooleanPredicate::updateParents() {
  98         for (uint i = 0; i < inputs.getSize(); i++) inputs.get(i)->parents.push(this);
  99 }
 100
 101 void BooleanLogic::updateParents() {
 102         for (uint i = 0; i < inputs.getSize(); i++) inputs.get(i)->parents.push(this);
 103 }
 104
 105 void BooleanVar::serialize(Serializer *serializer) {
 106         if (serializer->isSerialized(this))
 107                 return;
 108         serializer->addObject(this);
 109         serializer->mywrite(&type, sizeof(ASTNodeType));
 110         BooleanVar *This = this;
 111         serializer->mywrite(&This, sizeof(BooleanVar *));
 112         serializer->mywrite(&vtype, sizeof(VarType));
 113 }
 114
 115 void BooleanVar::print() {
 116         model_print("BooleanVar:%lu\n", (uintptr_t)this);
 117 }
 118
 119 void BooleanConst::print() {
 120         model_print("BooleanConst:%s\n", istrue ? "TRUE" : "FALSE");
 121 }
 122
 123 void BooleanOrder::serialize(Serializer *serializer) {
 124         if (serializer->isSerialized(this))
 125                 return;
 126         serializer->addObject(this);
 127         order->serialize(serializer);
 128
 129         serializer->mywrite(&type, sizeof(ASTNodeType));
 130         BooleanOrder *This = this;
 131         serializer->mywrite(&This, sizeof(BooleanOrder *));
 132         serializer->mywrite(&order, sizeof(Order *));
 133         serializer->mywrite(&first, sizeof(uint64_t));
 134         serializer->mywrite(&second, sizeof(uint64_t));
 135 }
 136
 137 void BooleanOrder::print() {
 138         model_print("{BooleanOrder: First= %lu, Second = %lu on Order:\n", first, second);
 139         order->print();
 140         model_print("}\n");
 141 }
 142
 143 void BooleanPredicate::serialize(Serializer *serializer) {
 144         if (serializer->isSerialized(this))
 145                 return;
 146         serializer->addObject(this);
 147
 148         predicate->serialize(serializer);
 149         uint size = inputs.getSize();
 150         for (uint i = 0; i < size; i++) {
 151                 Element *input = inputs.get(i);
 152                 input->serialize(serializer);
 153         }
 154         serializeBooleanEdge(serializer, undefStatus);
 155
 156         serializer->mywrite(&type, sizeof(ASTNodeType));
 157         BooleanPredicate *This = this;
 158         serializer->mywrite(&This, sizeof(BooleanPredicate *));
 159         serializer->mywrite(&predicate, sizeof(Predicate *));
 160         serializer->mywrite(&size, sizeof(uint));
 161         for (uint i = 0; i < size; i++) {
 162                 Element *input = inputs.get(i);
 163                 serializer->mywrite(&input, sizeof(Element *));
 164         }
 165         Boolean *undefStat = undefStatus != BooleanEdge(NULL) ? undefStatus.getRaw() : NULL;
 166         serializer->mywrite(&undefStat, sizeof(Boolean *));
 167 }
 168
 169 void BooleanPredicate::print() {
 170         model_print("{BooleanPredicate:\n");
 171         predicate->print();
 172         model_print("elements:\n");
 173         uint size = inputs.getSize();
 174         for (uint i = 0; i < size; i++) {
 175                 Element *input = inputs.get(i);
 176                 input->print();
 177         }
 178         model_print("}\n");
 179 }
 180
 181 void BooleanLogic::serialize(Serializer *serializer) {
 182         if (serializer->isSerialized(this))
 183                 return;
 184         serializer->addObject(this);
 185         uint size = inputs.getSize();
 186         for (uint i = 0; i < size; i++) {
 187                 BooleanEdge input = inputs.get(i);
 188                 serializeBooleanEdge(serializer, input);
 189         }
 190         serializer->mywrite(&type, sizeof(ASTNodeType));
 191         BooleanLogic *This = this;
 192         serializer->mywrite(&This, sizeof(BooleanLogic *));
 193         serializer->mywrite(&op, sizeof(LogicOp));
 194         serializer->mywrite(&size, sizeof(uint));
 195         for (uint i = 0; i < size; i++) {
 196                 Boolean *input = inputs.get(i).getRaw();
 197                 serializer->mywrite(&input, sizeof(Boolean *));
 198         }
 199 }
 200
 201 void BooleanLogic::print() {
 202         model_print("{BooleanLogic: %s\n",
 203                                                         op == SATC_AND ? "AND" : op == SATC_OR ? "OR" : op == SATC_NOT ? "NOT" :
 204                                                         op == SATC_XOR ? "XOR" : op == SATC_IFF ? "IFF" : "IMPLIES");
 205         uint size = inputs.getSize();
 206         for (uint i = 0; i < size; i++) {
 207                 BooleanEdge input = inputs.get(i);
 208                 if (input.isNegated())
 209                         model_print("!");
 210                 input.getBoolean()->print();
 211         }
 212         model_print("}\n");
 213 }
 214