dpllt_basic.cpp

Go to the documentation of this file.

/*****************************************************************************/
/*!
 *\file dpllt_basic.cpp
 *\brief Basic implementation of dpllt module using generic sat solver
 *
 * Author: Clark Barrett
 *
 * Created: Mon Dec 12 19:09:43 2005
 */
/*****************************************************************************/


#include "dpllt_basic.h"
#include "cnf.h"
#include "sat_api.h"
#include "exception.h"


using namespace std;
using namespace CVCL;
using namespace SAT;


static void SATDLevelHook(void *cookie, int change)
{
  DPLLTBasic* db = static_cast<DPLLTBasic*>(cookie);
  if (!db->popScopes()) return;
  for (; change > 0; change--) {
    db->theoryAPI()->push();
  }
  for (; change < 0; change++) {
    db->theoryAPI()->pop();
  }
}


static SatSolver::Lit SATDecisionHook(void *cookie, bool *done)
{
  DPLLTBasic* db = static_cast<DPLLTBasic*>(cookie);

  if (!db->decider()) {
    *done = true;
    return SatSolver::Lit();
  }

  Lit lit = db->decider()->makeDecision();
  if (!lit.isNull()) {
    *done = false;
    return db->cvcl2SAT(lit);
  }

  *done = true;
  Clause c;
  DPLLT::ConsistentResult result;
  result = db->theoryAPI()->checkConsistent(c, true);

  while (result == DPLLT::MAYBE_CONSISTENT) {
    CNF_Formula_Impl cnf;
    bool added = db->theoryAPI()->getNewClauses(cnf);
    if (added) {
      db->addNewClauses(cnf);
      return SatSolver::Lit();
    }
    result = db->theoryAPI()->checkConsistent(c, true);
  }
  if (result == DPLLT::INCONSISTENT) {
    db->addNewClause(c);
  }

  return SatSolver::Lit();
}


static void SATAssignmentHook(void *cookie, SatSolver::Var var, int value)
{
  DPLLTBasic* db = static_cast<DPLLTBasic*>(cookie);
  if (value == 0)
    db->theoryAPI()->assertLit(Lit(db->satSolver()->GetVarIndex(var), false));
  else if (value == 1)
    db->theoryAPI()->assertLit(Lit(db->satSolver()->GetVarIndex(var), true));
  else return;
  Clause c;
  DPLLT::ConsistentResult result;
  result = db->theoryAPI()->checkConsistent(c, false);
  if (result == DPLLT::INCONSISTENT) {
    db->addNewClause(c);
  }
}


static void SATDeductionHook(void *cookie)
{
  DPLLTBasic* db = static_cast<DPLLTBasic*>(cookie);
  Clause c;
  CNF_Formula_Impl cnf;
  if (db->theoryAPI()->getNewClauses(cnf)) db->addNewClauses(cnf);
  Lit l = db->theoryAPI()->getImplication();
  while (!l.isNull()) {
    db->theoryAPI()->getExplanation(l, c);
    db->addNewClause(c);
    c.clear();
    l = db->theoryAPI()->getImplication();
  }
}


void DPLLTBasic::createManager()
{
  d_mng = SatSolver::Create();
  d_mng->RegisterDLevelHook(SATDLevelHook, this);
  d_mng->RegisterDecisionHook(SATDecisionHook, this);
  d_mng->RegisterAssignmentHook(SATAssignmentHook, this);
  d_mng->RegisterDeductionHook(SATDeductionHook, this);
}


void DPLLTBasic::generate_CDB (CNF_Formula_Impl& cnf)
{
  CNF_Formula::const_iterator i, iend;
  Clause::const_iterator j, jend;
  vector<SatSolver::Lit> clause;
  if (cnf.numVars() > unsigned(d_mng->NumVariables())) {
    d_mng->AddVariables(cnf.numVars() - d_mng->NumVariables());
  }
  cnf.simplify();
  for (i = cnf.end()-1, iend = cnf.begin()-1; i != iend; --i) {
    if ((*i).isSatisfied()) continue;
    for (j = (*i).begin(), jend = (*i).end(); j != jend; ++j) {
      if (!(*j).isFalse()) clause.push_back(cvcl2SAT(*j));
    }
    if (clause.size() != 0) {
      d_mng->AddClause(clause);
      clause.clear();
    }
  }
}


void DPLLTBasic::handle_result(SatSolver::SATStatus outcome)
{
  char * result = "UNKNOWN";
  switch (outcome) {
    case SatSolver::SATISFIABLE:
        if (d_printStats) {
          cout << "Instance satisfiable" << endl;
          for (int i=1, sz = d_mng->NumVariables(); i <= sz; ++i) {
            switch(d_mng->GetVarAssignment(d_mng->GetVar(i))) {
              case -1:
                cout <<"("<< i<<")"; break;
              case 0:
                cout << "-" << i; break;
              case 1:
                cout << i ; break;
              default:
                throw Exception("Unknown variable value state");
            }
            cout << " ";
          }
          cout << endl;
        }
        result = "SAT";
        break;
    case SatSolver::UNSATISFIABLE:
        result  = "UNSAT";
        if (d_printStats) cout << "Instance unsatisfiable" << endl;
        break;
    case SatSolver::BUDGET_EXCEEDED:
        result  = "ABORT : TIME OUT"; 
        cout << "Time out, unable to determine the satisfiablility of the instance";
        cout << endl;
        break;
    case SatSolver::OUT_OF_MEMORY:
        result  = "ABORT : MEM OUT"; 
        cout << "Memory out, unable to determing the satisfiablility of the instance";
        cout << endl;
        break;
    default:
      throw Exception("DPLTBasic::handle_result: Unknown outcome");
  }
  if (d_printStats) d_mng->PrintStatistics();
}


void DPLLTBasic::verify_solution()
{
  // Used to check that all clauses are true, but our decision maker
  // may ignore some clauses that are no longer relevant, so now we check to
  // make sure no clause is false.
  for (SatSolver::Clause cl = d_mng->GetFirstClause(); !cl.IsNull();
       cl = d_mng->GetNextClause(cl)) {
    vector<SatSolver::Lit> lits;
    d_mng->GetClauseLits(cl, &lits);
    for (; lits.size() != 0;) {
      SatSolver::Lit lit = lits.back();
      SatSolver::Var var = d_mng->GetVarFromLit(lit);
      int sign = d_mng->GetPhaseFromLit(lit);
      int var_value = d_mng->GetVarAssignment(var);
      if( (var_value == 1 && sign == 0) ||
          (var_value == 0 && sign == 1) ||
          (var_value == -1) ) break;
      lits.pop_back();
    }
    DebugAssert(lits.size() != 0, "DPLLTBasic::verify_solution failed");
  }
}


DPLLTBasic::DPLLTBasic(TheoryAPI* theoryAPI, Decider* decider, bool printStats)
  : DPLLT(theoryAPI, decider), d_ready(true), d_popScopes(true),
    d_printStats(printStats)
{
  createManager();
  d_cnf = new CNF_Formula_Impl();
}


DPLLTBasic::~DPLLTBasic()
{
  if (d_cnf) delete d_cnf;
  if (d_mng) delete d_mng;
}


void DPLLTBasic::addNewClause(const Clause& c)
{
  DebugAssert(c.size() > 0, "Expected non-empty clause");
  DebugAssert(c.getMaxVar() <= unsigned(d_mng->NumVariables()),
              "Expected no new variables");
  vector<SatSolver::Lit> lits;
  for (Clause::const_iterator i = c.begin(), iend = c.end(); i < iend; ++i) {
    if (!(*i).isFalse()) lits.push_back(cvcl2SAT(*i));
  }
  satSolver()->AddClause(lits);
  (*d_cnf) += c;
}


void DPLLTBasic::addNewClauses(CNF_Formula_Impl& cnf)
{
  generate_CDB(cnf);
  (*d_cnf) += cnf;
}


DPLLTBasic::Result DPLLTBasic::checkSat(const CNF_Formula& cnf)
{
  SatSolver::SATStatus result;

  if (!d_ready) {
    // Copy current formula
    d_cnfStack.push_back(d_cnf);
    d_cnf = new CNF_Formula_Impl(*d_cnf);

    // make unit clauses for current assignment
    int value;
    for (int i = 1; i <= d_mng->NumVariables(); ++i) {
      value = d_mng->GetVarAssignment(d_mng->GetVar(i));
      if (value == 1) {
        d_cnf->newClause();
        d_cnf->addLiteral(Lit(i));
      }
      else if (value == 0) {
        d_cnf->newClause();
        d_cnf->addLiteral(Lit(i, false));
      }
    }

    // Create new manager
    d_mngStack.push_back(d_mng);
    createManager();
  }
  d_ready = false;

  (*d_cnf) += cnf;
  generate_CDB(*d_cnf);

  d_theoryAPI->push();

  result = d_mng->Satisfiable(true);

  // Print result

  if (result == SatSolver::SATISFIABLE)
    verify_solution();
  handle_result (result);

  if (result != SatSolver::SATISFIABLE) {
    d_theoryAPI->pop();
    delete d_mng;
    delete d_cnf;
    if (d_mngStack.size() == 0) {
      createManager();
      d_cnf = new CNF_Formula_Impl();
      d_ready = true;
    }
    else {
      d_mng = d_mngStack.back();
      d_mngStack.pop_back();
      d_cnf = d_cnfStack.back();
      d_cnfStack.pop_back();
    }
  }

  return (result == SatSolver::UNSATISFIABLE ? UNSAT :
          result == SatSolver::SATISFIABLE ? SATISFIABLE :
          ABORT);
}


DPLLTBasic::Result DPLLTBasic::continueCheck(const CNF_Formula& cnf)
{
  SatSolver::SATStatus result;

  if (d_ready) {
    throw Exception
      ("continueCheck should be called after a previous satisfiable result");
  }
  CNF_Formula_Impl cnfImpl(cnf);

  generate_CDB(cnfImpl);
  (*d_cnf) += cnfImpl;

  result = d_mng->Continue();

  // Print result

  if (result == SatSolver::SATISFIABLE)
    verify_solution();
  handle_result (result);

  if (result != SatSolver::SATISFIABLE) {
    d_theoryAPI->pop();
    delete d_mng;
    delete d_cnf;
    if (d_mngStack.size() == 0) {
      createManager();
      d_cnf = new CNF_Formula_Impl();
      d_ready = true;
    }
    else {
      d_mng = d_mngStack.back();
      d_mngStack.pop_back();
      d_cnf = d_cnfStack.back();
      d_cnfStack.pop_back();
    }
  }

  return (result == SatSolver::UNSATISFIABLE ? UNSAT :
          result == SatSolver::SATISFIABLE ? SATISFIABLE :
          ABORT);
}


void DPLLTBasic::returnFromSat()
{
  d_popScopes = false;
  delete d_mng;
  d_popScopes = true;
  delete d_cnf;
  if (d_mngStack.size() == 0) {
    if (d_ready) {
      throw Exception
        ("returnFromSat requires previous SATISFIABLE checkSat call");
    }
    createManager();
    d_cnf = new CNF_Formula_Impl();
    d_ready = true;
  }
  else {
    DebugAssert(!d_ready, "Expected ready to be false");
    d_mng = d_mngStack.back();
    d_mngStack.pop_back();
    d_cnf = d_cnfStack.back();
    d_cnfStack.pop_back();
  }
}

---