gapt.provers.sat.PicoSAT.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gapt_3 Show documentation

General Architecture for Proof Theory

The newest version!

package gapt.provers.sat

import java.io.IOException

import gapt.formats.dimacs.{DIMACS, readDIMACS, readDRUP, writeDIMACS}
import gapt.utils.{ExternalProgram, runProcess, withTempFile}
import os._
import gapt.expr.formula.Top
import gapt.proofs.rup.RupProof

private object picoSatHelper {
  def getDefaultExecutableName() =
    try {
      // Debian and Fedora have a special name for PicoSAT with proof support
      val name = "picosat.trace"
      runProcess(Seq(name, "--version"))
      name
    } catch {
      case _: IOException =>
        "picosat"
    }
}

object PicoSAT extends PicoSAT(picoSatHelper.getDefaultExecutableName())
class PicoSAT(command: String*) extends DrupSolver with ExternalProgram {
  def runProgram(dimacsInput: String): Either[String, String] =
    withTempFile.fromString(dimacsInput) { dimacsInputFile =>
      withTempFile { drupOutputFile =>
        withTempFile { modelOutputFile =>
          runProcess.withExitValue(command ++ Seq(
            "-r",
            drupOutputFile.toString,
            "-o",
            modelOutputFile.toString,
            dimacsInputFile.toString
          )) match {
            case (10, _) => /* SAT */ Right(read(modelOutputFile))
            case (20, _) => /* UNSAT */ Left(read(drupOutputFile))
            case (_, str) =>
              throw new Exception(s"Error executing external sat prover $command:\n$str")
          }
        }
      }
    }

  override def solve(cnf: DIMACS.CNF): Option[DIMACS.Model] =
    runProgram(writeDIMACS(cnf)) match {
      case Right(model) => Some(readDIMACS(model))
      case Left(proof)  => None
    }

  override val isInstalled: Boolean =
    try getDrupProof(Top()).isDefined
    catch { case _: IOException => false }

  def getDrupProof(cnf: DIMACS.CNF): Option[RupProof] =
    runProgram(writeDIMACS(cnf)) match {
      case Left(proof)  => Some(readDRUP(cnf, proof))
      case Right(model) => None
    }

}