src/Backend/satorderencoder.cc

   1 #include "satencoder.h"
   2 #include "structs.h"
   3 #include "common.h"
   4 #include "order.h"
   5 #include "csolver.h"
   6 #include "orderpair.h"
   7 #include "set.h"
   8 #include "tunable.h"
   9 #include "orderanalysis.h"
  10 #include "ordergraph.h"
  11 #include "orderedge.h"
  12 #include "element.h"
  13 #include "predicate.h"
  14 #include "orderelement.h"
  15 #include "orderpairresolver.h"
  16
  17 Edge SATEncoder::encodeOrderSATEncoder(BooleanOrder *constraint) {
  18         switch ( constraint->order->type) {
  19         case SATC_PARTIAL:
  20                 return encodePartialOrderSATEncoder(constraint);
  21         case SATC_TOTAL:
  22                 return encodeTotalOrderSATEncoder(constraint);
  23         default:
  24                 ASSERT(0);
  25         }
  26         return E_BOGUS;
  27 }
  28
  29 Edge SATEncoder::inferOrderConstraintFromGraph(Order *order, uint64_t _first, uint64_t _second) {
  30         if (order->graph != NULL) {
  31                 OrderGraph *graph = order->graph;
  32                 OrderNode *first = graph->lookupOrderNodeFromOrderGraph(_first);
  33                 OrderNode *second = graph->lookupOrderNodeFromOrderGraph(_second);
  34                 if ((first != NULL) && (second != NULL)) {
  35                         OrderEdge *edge = graph->lookupOrderEdgeFromOrderGraph(first, second);
  36                         if (edge != NULL) {
  37                                 if (edge->mustPos)
  38                                         return E_True;
  39                                 else if (edge->mustNeg)
  40                                         return E_False;
  41                         }
  42                         OrderEdge *invedge = graph->lookupOrderEdgeFromOrderGraph(second, first);
  43                         if (invedge != NULL) {
  44                                 if (invedge->mustPos)
  45                                         return E_False;
  46                                 else if (invedge->mustNeg)
  47                                         return E_True;
  48                         }
  49                 }
  50         }
  51         return E_NULL;
  52 }
  53
  54 Edge SATEncoder::getPairConstraint(Order *order, OrderPair *pair) {
  55         Edge gvalue = inferOrderConstraintFromGraph(order, pair->first, pair->second);
  56         if (!edgeIsNull(gvalue))
  57                 return gvalue;
  58
  59         HashtableOrderPair *table = order->getOrderPairTable();
  60         bool negate = false;
  61         OrderPair flipped;
  62         if (pair->first < pair->second) {
  63                 negate = true;
  64                 flipped.first = pair->second;
  65                 flipped.second = pair->first;
  66                 pair = &flipped;
  67         }
  68         OrderPair* tmp;
  69         if (!(table->contains(pair))) {
  70                 tmp = new OrderPair(pair->first, pair->second, getNewVarSATEncoder());
  71                 table->put(tmp, tmp);
  72         } else {
  73                 tmp = table->get(pair);
  74         }
  75         return negate ? tmp->getNegatedConstraint() : tmp->getConstraint();
  76 }
  77
  78 Edge SATEncoder::getPartialPairConstraint(Order *order, OrderPair *pair) {
  79         Edge gvalue = inferOrderConstraintFromGraph(order, pair->first, pair->second);
  80         if (!edgeIsNull(gvalue))
  81                 return gvalue;
  82
  83         HashtableOrderPair *table = order->getOrderPairTable();
  84
  85         OrderPair* tmp;
  86         if (!(table->contains(pair))) {
  87                 Edge constraint = getNewVarSATEncoder();
  88                 tmp = new OrderPair(pair->first, pair->second, constraint);
  89                 table->put(tmp, tmp);
  90                 Edge constraint2 = getNewVarSATEncoder();
  91                 OrderPair *swap = new OrderPair(pair->second, pair->first, constraint2);
  92                 table->put(swap, swap);
  93                 addConstraintCNF(cnf, constraintNegate(constraintAND2(cnf, constraint, constraint2)));
  94         } else {
  95                 tmp = table->get(pair);
  96         }
  97         return tmp->getConstraint();
  98 }
  99
 100 Edge SATEncoder::encodeTotalOrderSATEncoder(BooleanOrder *boolOrder) {
 101         ASSERT(boolOrder->order->type == SATC_TOTAL);
 102         if (boolOrder->order->encoding.resolver == NULL) {
 103                 //This is pairwised encoding ...
 104                 boolOrder->order->setOrderResolver(new OrderPairResolver(solver, boolOrder->order));
 105                 bool doOptOrderStructure = GETVARTUNABLE(solver->getTuner(), boolOrder->order->type, OPTIMIZEORDERSTRUCTURE, &onoff);
 106                 if (doOptOrderStructure) {
 107                         boolOrder->order->graph = buildMustOrderGraph(boolOrder->order);
 108                         reachMustAnalysis(solver, boolOrder->order->graph, true);
 109                 }
 110                 createAllTotalOrderConstraintsSATEncoder(boolOrder->order);
 111         }
 112         OrderPair pair(boolOrder->first, boolOrder->second, E_NULL);
 113         Edge constraint = getPairConstraint(boolOrder->order, &pair);
 114         return constraint;
 115 }
 116
 117
 118 void SATEncoder::createAllTotalOrderConstraintsSATEncoder(Order *order) {
 119 #ifdef CONFIG_DEBUG
 120         model_print("in total order ...\n");
 121 #endif
 122         ASSERT(order->type == SATC_TOTAL);
 123         Set *set = order->set;
 124         uint size = order->set->getSize();
 125         for (uint i = 0; i < size; i++) {
 126                 uint64_t valueI = set->getMemberAt(i);
 127                 for (uint j = i + 1; j < size; j++) {
 128                         uint64_t valueJ = set->getMemberAt(j);
 129                         OrderPair pairIJ(valueI, valueJ, E_NULL);
 130                         Edge constIJ = getPairConstraint(order, &pairIJ);
 131                         for (uint k = j + 1; k < size; k++) {
 132                                 uint64_t valueK = set->getMemberAt(k);
 133                                 OrderPair pairJK(valueJ, valueK, E_NULL);
 134                                 OrderPair pairIK(valueI, valueK, E_NULL);
 135                                 Edge constIK = getPairConstraint(order, &pairIK);
 136                                 Edge constJK = getPairConstraint(order, &pairJK);
 137                                 addConstraintCNF(cnf, generateTransOrderConstraintSATEncoder(constIJ, constJK, constIK));
 138                         }
 139                 }
 140         }
 141 }
 142
 143 Edge SATEncoder::generateTransOrderConstraintSATEncoder(Edge constIJ,Edge constJK,Edge constIK) {
 144         Edge carray[] = {constIJ, constJK, constraintNegate(constIK)};
 145         Edge loop1 = constraintOR(cnf, 3, carray);
 146         Edge carray2[] = {constraintNegate(constIJ), constraintNegate(constJK), constIK};
 147         Edge loop2 = constraintOR(cnf, 3, carray2 );
 148         return constraintAND2(cnf, loop1, loop2);
 149 }
 150
 151 Edge SATEncoder::generatePartialOrderConstraintsSATEncoder(Edge ij,Edge ji, Edge jk, Edge kj,Edge ik, Edge ki) {
 152         Edge uoIJ = constraintAND2(cnf, constraintNegate(ij), constraintNegate(ji));
 153         Edge uoJK = constraintAND2(cnf, constraintNegate(jk), constraintNegate(kj));
 154         Edge uoIK = constraintAND2(cnf, constraintNegate(ik), constraintNegate(ki));
 155
 156         Edge t1[] = {ij, jk, ik};
 157         Edge t2[] = {ji, jk, ik};
 158         Edge t3[] = {ij, kj, ki};
 159         Edge t4[] = {ij, kj, ik};
 160         Edge t5[] = {ji, jk, ki};
 161         Edge t6[] = {ji, kj, ki};
 162         Edge ct1 = constraintAND(cnf, 3, t1);
 163         Edge ct2 = constraintAND(cnf, 3, t2);
 164         Edge ct3 = constraintAND(cnf, 3, t3);
 165         Edge ct4 = constraintAND(cnf, 3, t4);
 166         Edge ct5 = constraintAND(cnf, 3, t5);
 167         Edge ct6 = constraintAND(cnf, 3, t6);
 168
 169         Edge p1[] = {uoIJ, jk, ik};
 170         Edge p2[] = {ij, kj, uoIK};
 171         Edge p3[] = {ji, uoJK, ki};
 172         Edge p4[] = {uoIJ, kj, ki};
 173         Edge p5[] = {ji, jk, uoIK};
 174         Edge p6[] = {ij, uoJK, ik};
 175         Edge cp1 = constraintAND(cnf, 3, p1);
 176         Edge cp2 = constraintAND(cnf, 3, p2);
 177         Edge cp3 = constraintAND(cnf, 3, p3);
 178         Edge cp4 = constraintAND(cnf, 3, p4);
 179         Edge cp5 = constraintAND(cnf, 3, p5);
 180         Edge cp6 = constraintAND(cnf, 3, p6);
 181
 182         Edge o1[] = {uoIJ, uoJK, ik};
 183         Edge o2[] = {ij, uoJK, uoIK};
 184         Edge o3[] = {uoIK, jk, uoIK};
 185         Edge o4[] = {ji, uoJK, uoIK};
 186         Edge o5[] = {uoIJ, uoJK, ki};
 187         Edge o6[] = {uoIJ, kj, uoIK};
 188         Edge co1 = constraintAND(cnf, 3, o1);
 189         Edge co2 = constraintAND(cnf, 3, o2);
 190         Edge co3 = constraintAND(cnf, 3, o3);
 191         Edge co4 = constraintAND(cnf, 3, o4);
 192         Edge co5 = constraintAND(cnf, 3, o5);
 193         Edge co6 = constraintAND(cnf, 3, o6);
 194
 195         Edge unorder [] = {uoIJ, uoJK, uoIK};
 196         Edge cunorder = constraintAND(cnf, 3, unorder);
 197
 198
 199         Edge res[] = {ct1,ct2,ct3,ct4,ct5,ct6,
 200                         cp1,cp2,cp3,cp4,cp5,cp6,
 201                         co1,co2,co3,co4,co5,co6,
 202                         cunorder};
 203         return constraintOR(cnf, 19, res);
 204 }
 205
 206 Edge SATEncoder::encodePartialOrderSATEncoder(BooleanOrder *boolOrder) {
 207         ASSERT(boolOrder->order->type == SATC_PARTIAL);
 208         if (boolOrder->order->encoding.resolver == NULL) {
 209                 //This is pairwised encoding ...
 210                 boolOrder->order->setOrderResolver(new OrderPairResolver(solver, boolOrder->order));
 211                 bool doOptOrderStructure = GETVARTUNABLE(solver->getTuner(), boolOrder->order->type, OPTIMIZEORDERSTRUCTURE, &onoff);
 212                 if (doOptOrderStructure) {
 213                         boolOrder->order->graph = buildMustOrderGraph(boolOrder->order);
 214                         reachMustAnalysis(solver, boolOrder->order->graph, true);
 215                 }
 216                 createAllPartialOrderConstraintsSATEncoder(boolOrder->order);
 217         }
 218         OrderPair pair(boolOrder->first, boolOrder->second, E_NULL);
 219         Edge constraint = getPartialPairConstraint(boolOrder->order, &pair);
 220         return constraint;
 221 }
 222
 223
 224 void SATEncoder::createAllPartialOrderConstraintsSATEncoder(Order *order) {
 225 #ifdef CONFIG_DEBUG
 226         model_print("in partial order ...\n");
 227 #endif
 228         ASSERT(order->type == SATC_TOTAL);
 229         Set *set = order->set;
 230         uint size = order->set->getSize();
 231         for (uint i = 0; i < size; i++) {
 232                 uint64_t valueI = set->getMemberAt(i);
 233                 for (uint j = i + 1; j < size; j++) {
 234                         uint64_t valueJ = set->getMemberAt(j);
 235                         OrderPair pairIJ(valueI, valueJ, E_NULL);
 236                         OrderPair pairJI(valueJ, valueI, E_NULL);
 237                         Edge constIJ = getPartialPairConstraint(order, &pairIJ);
 238                         Edge constJI = getPartialPairConstraint(order, &pairJI);
 239                         for (uint k = j + 1; k < size; k++) {
 240                                 uint64_t valueK = set->getMemberAt(k);
 241                                 OrderPair pairJK(valueJ, valueK, E_NULL);
 242                                 OrderPair pairIK(valueI, valueK, E_NULL);
 243                                 Edge constIK = getPartialPairConstraint(order, &pairIK);
 244                                 Edge constJK = getPartialPairConstraint(order, &pairJK);
 245                                 OrderPair pairKJ(valueK, valueJ, E_NULL);
 246                                 OrderPair pairKI(valueK, valueI, E_NULL);
 247                                 Edge constKI = getPartialPairConstraint(order, &pairKI);
 248                                 Edge constKJ = getPartialPairConstraint(order, &pairKJ);
 249                                 addConstraintCNF(cnf, generatePartialOrderConstraintsSATEncoder(constIJ, constJI,
 250                                         constJK, constKJ, constIK, constKI));
 251                         }
 252                 }
 253         }
 254 }
 255
 256