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
  10 IntegerEncodingTransform::IntegerEncodingTransform(CSolver *_solver)
  11         : Transform(_solver)
  12 {
  13 }
  14
  15 IntegerEncodingTransform::~IntegerEncodingTransform() {
  16 }
  17
  18 void IntegerEncodingTransform::doTransform() {
  19         HashsetOrder *orders = solver->getActiveOrders()->copy();
  20         SetIteratorOrder * orderit=orders->iterator();
  21         while(orderit->hasNext()) {
  22                 Order *order = orderit->next();
  23                 if (GETVARTUNABLE(solver->getTuner(), order->type, ORDERINTEGERENCODING, &onoff) &&
  24                         order->encoding.resolver == NULL)
  25                         integerEncode(order);
  26         }
  27         delete orders;
  28         delete orderit;
  29 }
  30
  31 void IntegerEncodingTransform::integerEncode(Order *currOrder) {
  32         IntegerEncodingRecord *encodingRecord =  new IntegerEncodingRecord(
  33                 solver->createRangeSet(currOrder->set->getType(), 0, (uint64_t) currOrder->set->getSize() - 1));
  34         currOrder->setOrderEncodingType( INTEGERENCODING );
  35         uint size = currOrder->constraints.getSize();
  36         for (uint i = 0; i < size; i++) {
  37                 orderIntegerEncodingSATEncoder(currOrder->constraints.get(i), encodingRecord);
  38         }
  39         currOrder->setOrderResolver(new IntegerEncOrderResolver(solver, encodingRecord));
  40         solver->getActiveOrders()->remove(currOrder);
  41 }
  42
  43
  44 void IntegerEncodingTransform::orderIntegerEncodingSATEncoder(BooleanOrder *boolOrder, IntegerEncodingRecord *ierec) {
  45         //getting two elements and using LT predicate ...
  46         Element *elem1 = ierec->getOrderIntegerElement(solver, boolOrder->first);
  47         Element *elem2 = ierec->getOrderIntegerElement(solver, boolOrder->second);
  48         Set *sarray[] = {ierec->getSecondarySet(), ierec->getSecondarySet()};
  49         Predicate *predicate = solver->createPredicateOperator(SATC_LT, sarray, 2);
  50         Element *parray[] = {elem1, elem2};
  51         BooleanEdge boolean = solver->applyPredicate(predicate, parray, 2);
  52         solver->replaceBooleanWithBoolean(boolOrder, boolean);
  53 }
  54