inc_solver.cc

   1 #include "inc_solver.h"
   2
   3 #define SATSOLVER "sat_solver"
   4
   5 IncrementalSolver::IncrementalSolver() :
   6   buffer((int *)malloc(sizeof(int)*BUFFERSIZE)),
   7   solution(NULL),
   8   solutionsize(0),
   9   offset(0)
  10 {
  11   createSolver();
  12 }
  13
  14 IncrementalSolver::~IncrementalSolver() {
  15   killSolver();
  16   free(buffer);
  17 }
  18
  19 void IncrementalSolver::reset() {
  20   killSolver();
  21   offset = 0;
  22   createSolver();
  23 }
  24
  25 void IncrementalSolver::addClauseLiteral(int literal) {
  26   buffer[offset++]=literal;
  27   if (offset==BUFFERSIZE) {
  28     flushBuffer();
  29   }
  30 }
  31
  32 void IncrementalSolver::finishedClauses() {
  33   addClauseLiteral(0);
  34 }
  35
  36 void IncrementalSolver::freeze(int variable) {
  37   addClauseLiteral(IS_FREEZE);
  38   addClauseLiteral(variable);
  39 }
  40
  41 int IncrementalSolver::solve() {
  42   //add an empty clause
  43   addClauseLiteral(IS_RUNSOLVER);
  44   flushBuffer();
  45   int result=readIntSolver();
  46   if (result == IS_SAT) {
  47     int numVars=readIntSolver();
  48     if (numVars > solutionsize) {
  49       if (solution != NULL)
  50         free(solution);
  51       solution = (int *) malloc(numVars*sizeof(int));
  52     }
  53     readSolver(solution, numVars * sizeof(int));
  54   }
  55   return result;
  56 }
  57
  58 int IncrementalSolver::readIntSolver() {
  59   int value;
  60   readSolver(&value, 4);
  61   return value;
  62 }
  63
  64 void IncrementalSolver::readSolver(void * tmp, ssize_t size) {
  65   char *result = (char *) tmp;
  66   ssize_t bytestoread=size;
  67   ssize_t bytesread=0;
  68   do {
  69     ssize_t n=read(from_solver_fd, &((char *)result)[bytesread], bytestoread);
  70     if (n == -1) {
  71       fprintf(stderr, "Read failure\n");
  72       exit(-1);
  73     }
  74     bytestoread -= n;
  75     bytesread += n;
  76   } while(bytestoread != 0);
  77 }
  78
  79 bool IncrementalSolver::getValue(int variable) {
  80   return solution[variable];
  81 }
  82
  83 void IncrementalSolver::createSolver() {
  84   int to_pipe[2];
  85   int from_pipe[2];
  86   if (pipe(to_pipe) || pipe(from_pipe)) {
  87     fprintf(stderr, "Error creating pipe.\n");
  88     exit(-1);
  89   }
  90   if ((solver_pid = fork()) == -1) {
  91     fprintf(stderr, "Error forking.\n");
  92     exit(-1);
  93   }
  94   if (solver_pid == 0) {
  95     //Solver process
  96     close(to_pipe[0]);
  97     close(from_pipe[1]);
  98     if ((dup2(0, to_pipe[1]) == -1) ||
  99         (dup2(1, from_pipe[0]) == -1)) {
 100       fprintf(stderr, "Error duplicating pipes\n");
 101     }
 102     setsid();
 103     execlp(SATSOLVER, SATSOLVER, NULL);
 104     fprintf(stderr, "execlp Failed\n");
 105   } else {
 106     //Our process
 107     to_solver_fd = to_pipe[0];
 108     from_solver_fd = from_pipe[1];
 109     close(to_pipe[1]);
 110     close(from_pipe[0]);
 111   }
 112 }
 113
 114 void IncrementalSolver::killSolver() {
 115   close(to_solver_fd);
 116   close(from_solver_fd);
 117   //Stop the solver
 118   if (solver_pid > 0)
 119     killpg(solver_pid, SIGKILL);
 120 }
 121
 122 void IncrementalSolver::flushBuffer() {
 123   ssize_t bytestowrite=sizeof(int)*offset;
 124   ssize_t byteswritten=0;
 125   do {
 126     ssize_t n=write(to_solver_fd, &((char *)buffer)[byteswritten], bytestowrite);
 127     if (n == -1) {
 128       fprintf(stderr, "Write failure\n");
 129       exit(-1);
 130     }
 131     bytestowrite -= n;
 132     byteswritten += n;
 133   } while(bytestowrite != 0);
 134   offset = 0;
 135 }