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