satsolver.hh

// -*- coding: utf-8 -*-
// Copyright (C) by the Spot authors, see the AUTHORS file for details.
//
// This file is part of Spot, a model checking library.
//
// Spot is free software; you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 3 of the License, or
// (at your option) any later version.
//
// Spot is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
// or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public
// License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
#pragma once
 
#include <spot/misc/common.hh>
#include <spot/misc/tmpfile.hh>
#include <vector>
#include <stdexcept>
#include <iosfwd>
#include <initializer_list>
 
struct PicoSAT; // forward
 
namespace spot
{
  class printable;
 
  class satsolver_command: formater
  {
  private:
    const char* satsolver;
 
  public:
    satsolver_command();
 
    bool command_given();
 
    int run(printable* in, printable* out);
 
  };
 
  class SPOT_API satsolver
  {
  public:
    satsolver();
    ~satsolver();
 
    void adjust_nvars(int nvars);
 
    void set_nassumptions_vars(int nassumptions_vars);
 
    void add(std::initializer_list<int> values);
 
    void add(int v);
 
    int get_nb_clauses() const;
 
    int get_nb_vars() const;
 
    std::pair<int, int> stats() const;
 
    template<typename T>
    void comment_rec(T single);
 
    template<typename T, typename... Args>
    void comment_rec(T first, Args... args);
 
    template<typename T>
    void comment(T single);
 
    template<typename T, typename... Args>
    void comment(T first, Args... args);
 
    void assume(int lit);
 
    typedef std::vector<bool> solution;
    typedef std::pair<int, solution> solution_pair;
 
    solution_pair get_solution();
 
  private:  // methods
    void start();
 
    void end_clause();
 
    satsolver::solution
    picosat_get_sol(int res);
 
    satsolver::solution
    satsolver_get_sol(const char* filename);
 
    bool
    xcnf_mode();
 
  private:  // variables
    satsolver_command cmd_;
 
    // cnf streams and associated clause counter.
    // The next 2 pointers will be != nullptr if SPOT_SATSOLVER is given.
    temporary_file* cnf_tmp_;
    std::ostream* cnf_stream_;
    int nclauses_;
    int nvars_;
    int nassumptions_vars_; // Surplus of vars (for 'assume' algorithm).
 
    int nsols_;
 
    PicoSAT* psat_;
 
    // The next 2 pointers will be initialized if SPOT_XCNF env var
    // is set. This requires SPOT_SATSOLVER to be set as well.
    std::ofstream* xcnf_tmp_;
    std::ofstream* xcnf_stream_;
    std::string path_;
  };
 
  template<typename T>
  void
  satsolver::comment_rec(T single)
  {
    if (!psat_)
      *cnf_stream_ << ' ' << single;
  }
 
  template<typename T, typename... Args>
  void
  satsolver::comment_rec(T first, Args... args)
  {
    if (!psat_)
    {
      *cnf_stream_ << ' ' << first;
      comment_rec(args...);
    }
  }
 
  template<typename T>
  void
  satsolver::comment(T single)
  {
    if (!psat_)
      *cnf_stream_ << "c " << single;
  }
 
  template<typename T, typename... Args>
  void
  satsolver::comment(T first, Args... args)
  {
    if (!psat_)
    {
      *cnf_stream_ << "c " << first;
      comment_rec(args...);
    }
  }
 
}