src/ASTTransform/integerencoding.cc

   1 #include "integerencoding.h"
   2 #include "set.h"
   3 #include "order.h"
   4 #include "satencoder.h"
   5 #include "csolver.h"
   6 #include "integerencodingrecord.h"
   7 #include "integerencorderresolver.h"
   8 #include "tunable.h"
   9 #include "polarityassignment.h"
  10
  11 IntegerEncodingTransform::IntegerEncodingTransform(CSolver *_solver)
  12         : Transform(_solver)
  13 {
  14 }
  15
  16 IntegerEncodingTransform::~IntegerEncodingTransform() {
  17 }
  18
  19 void IntegerEncodingTransform::doTransform() {
  20         if (solver->isUnSAT()) {
  21                 return;
  22         }
  23         HashsetOrder *orders = solver->getActiveOrders()->copy();
  24         SetIteratorOrder *orderit = orders->iterator();
  25         while (orderit->hasNext()) {
  26                 Order *order = orderit->next();
  27                 if (order->type == SATC_PARTIAL)
  28                         continue;
  29                 if (GETVARTUNABLE(solver->getTuner(), order->set->type, ORDERINTEGERENCODING, &offon))
  30                         integerEncode(order);
  31         }
  32         delete orders;
  33         delete orderit;
  34 }
  35
  36 void IntegerEncodingTransform::integerEncode(Order *currOrder) {
  37         IntegerEncodingRecord *encodingRecord =  new IntegerEncodingRecord(
  38                 solver->createRangeSet(currOrder->set->getType(), 0, (uint64_t) currOrder->getNumUsed() - 1));
  39         currOrder->setOrderEncodingType( INTEGERENCODING );
  40
  41         Vector<BooleanOrder *> *constraints = currOrder->getConstraints();
  42         uint size = constraints->getSize();
  43         for (uint i = 0; i < size; i++) {
  44                 orderIntegerEncodingSATEncoder(constraints->get(i), encodingRecord);
  45         }
  46         currOrder->setOrderResolver(new IntegerEncOrderResolver(solver, encodingRecord));
  47         solver->getActiveOrders()->remove(currOrder);
  48 }
  49
  50
  51 void IntegerEncodingTransform::orderIntegerEncodingSATEncoder(BooleanOrder *boolOrder, IntegerEncodingRecord *ierec) {
  52         //getting two elements and using LT predicate ...
  53         Element *elem1 = ierec->getOrderIntegerElement(solver, boolOrder->first);
  54         Element *elem2 = ierec->getOrderIntegerElement(solver, boolOrder->second);
  55         Predicate *predicate = solver->createPredicateOperator(SATC_LT);
  56         Element *parray[] = {elem1, elem2};
  57         BooleanEdge boolean = solver->applyPredicate(predicate, parray, 2);
  58         updateEdgePolarity(boolean, boolOrder);
  59         solver->replaceBooleanWithBoolean(boolOrder, boolean);
  60 }
  61