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         if (undefStatus)
 100                 undefStatus->parents.push(this);
 101 }
 102
 103 void BooleanLogic::updateParents() {
 104         for (uint i = 0; i < inputs.getSize(); i++) inputs.get(i)->parents.push(this);
 105 }
 106
 107 void BooleanVar::serialize(Serializer *serializer) {
 108         if (serializer->isSerialized(this))
 109                 return;
 110         serializer->addObject(this);
 111         serializer->mywrite(&type, sizeof(ASTNodeType));
 112         BooleanVar *This = this;
 113         serializer->mywrite(&This, sizeof(BooleanVar *));
 114         serializer->mywrite(&vtype, sizeof(VarType));
 115 }
 116
 117 void BooleanVar::print() {
 118         model_print("BooleanVar:%lu\n", (uintptr_t)this);
 119 }
 120
 121 void BooleanConst::print() {
 122         model_print("BooleanConst:%s\n", istrue ? "TRUE" : "FALSE");
 123 }
 124
 125 void BooleanOrder::serialize(Serializer *serializer) {
 126         if (serializer->isSerialized(this))
 127                 return;
 128         serializer->addObject(this);
 129         order->serialize(serializer);
 130
 131         serializer->mywrite(&type, sizeof(ASTNodeType));
 132         BooleanOrder *This = this;
 133         serializer->mywrite(&This, sizeof(BooleanOrder *));
 134         serializer->mywrite(&order, sizeof(Order *));
 135         serializer->mywrite(&first, sizeof(uint64_t));
 136         serializer->mywrite(&second, sizeof(uint64_t));
 137 }
 138
 139 void BooleanOrder::print() {
 140         model_print("{BooleanOrder: First= %lu, Second = %lu on Order:\n", first, second);
 141         order->print();
 142         model_print("}\n");
 143 }
 144
 145 void BooleanPredicate::serialize(Serializer *serializer) {
 146         if (serializer->isSerialized(this))
 147                 return;
 148         serializer->addObject(this);
 149
 150         predicate->serialize(serializer);
 151         uint size = inputs.getSize();
 152         for (uint i = 0; i < size; i++) {
 153                 Element *input = inputs.get(i);
 154                 input->serialize(serializer);
 155         }
 156         serializeBooleanEdge(serializer, undefStatus);
 157
 158         serializer->mywrite(&type, sizeof(ASTNodeType));
 159         BooleanPredicate *This = this;
 160         serializer->mywrite(&This, sizeof(BooleanPredicate *));
 161         serializer->mywrite(&predicate, sizeof(Predicate *));
 162         serializer->mywrite(&size, sizeof(uint));
 163         for (uint i = 0; i < size; i++) {
 164                 Element *input = inputs.get(i);
 165                 serializer->mywrite(&input, sizeof(Element *));
 166         }
 167         Boolean *undefStat = undefStatus != BooleanEdge(NULL) ? undefStatus.getRaw() : NULL;
 168         serializer->mywrite(&undefStat, sizeof(Boolean *));
 169 }
 170
 171 void BooleanPredicate::print() {
 172         model_print("{BooleanPredicate:\n");
 173         predicate->print();
 174         model_print("elements:\n");
 175         uint size = inputs.getSize();
 176         for (uint i = 0; i < size; i++) {
 177                 Element *input = inputs.get(i);
 178                 input->print();
 179         }
 180         model_print("}\n");
 181 }
 182
 183 void BooleanLogic::serialize(Serializer *serializer) {
 184         if (serializer->isSerialized(this))
 185                 return;
 186         serializer->addObject(this);
 187         uint size = inputs.getSize();
 188         for (uint i = 0; i < size; i++) {
 189                 BooleanEdge input = inputs.get(i);
 190                 serializeBooleanEdge(serializer, input);
 191         }
 192         serializer->mywrite(&type, sizeof(ASTNodeType));
 193         BooleanLogic *This = this;
 194         serializer->mywrite(&This, sizeof(BooleanLogic *));
 195         serializer->mywrite(&op, sizeof(LogicOp));
 196         serializer->mywrite(&size, sizeof(uint));
 197         for (uint i = 0; i < size; i++) {
 198                 Boolean *input = inputs.get(i).getRaw();
 199                 serializer->mywrite(&input, sizeof(Boolean *));
 200         }
 201 }
 202
 203 void BooleanLogic::print() {
 204         model_print("{BooleanLogic: %s\n",
 205                                                         op == SATC_AND ? "AND" : op == SATC_OR ? "OR" : op == SATC_NOT ? "NOT" :
 206                                                         op == SATC_XOR ? "XOR" : op == SATC_IFF ? "IFF" : "IMPLIES");
 207         uint size = inputs.getSize();
 208         for (uint i = 0; i < size; i++) {
 209                 BooleanEdge input = inputs.get(i);
 210                 if (input.isNegated())
 211                         model_print("!");
 212                 input.getBoolean()->print();
 213         }
 214         model_print("}\n");
 215 }
 216