Kissat.hpp

#pragma once
 
#include "containers/ClauseBuffer.hpp"
#include "containers/ClauseDatabase.hpp"
#include "containers/ClauseUtils.hpp"
#include "utils/Threading.hpp"
#include <fstream>
#include <iostream>
#include <unordered_map>
 
#include "KissatFamily.hpp"
#include "SolverCdclInterface.hpp"
 
#define KISSAT_
 
// Kissat includes
extern "C"
{
#include "kissat/src/kissat.h"
}
 
class Kissat : public SolverCdclInterface
{
  public:
    Kissat(int id, const std::shared_ptr<ClauseDatabase>& clauseDB);
 
    virtual ~Kissat();
 
    /* Execution */
 
    SatResult solve(const std::vector<int>& cube) override;
 
    void setSolverInterrupt() override;
 
    void unsetSolverInterrupt() override;
 
    void initKissatOptions();
 
    void diversify(const SeedGenerator& getSeed) override;
 
    /* Clause Management */
 
    void loadFormula(const char* filename) override;
 
    void addInitialClauses(const std::vector<simpleClause>& clauses, unsigned int nbVars) override;
 
    void addClause(ClauseExchangePtr clause) override;
 
    void addClauses(const std::vector<ClauseExchangePtr>& clauses) override;
 
    /* Sharing */
 
    bool importClause(const ClauseExchangePtr& clause) override;
 
    void importClauses(const std::vector<ClauseExchangePtr>& clauses) override;
 
    /* Variable Management */
 
    unsigned int getVariablesCount() override;
 
    int getDivisionVariable() override;
 
    void setPhase(const unsigned int var, const bool phase) override;
 
    void bumpVariableActivity(const int var, const int times) override;
 
    /* Statistics And More */
 
    void printStatistics() override;
 
    void printWinningLog() override;
 
    std::vector<int> getFinalAnalysis() override;
 
    std::vector<int> getSatAssumptions() override;
 
    std::vector<int> getModel() override;
 
    std::unordered_map<std::string, int> kissatOptions;
 
    void setFamily(KissatFamily family) { this->family = family; };
 
 
  protected:
    void computeFamily();
    
  protected:
    kissat* solver;
 
    ClauseBuffer clausesToAdd;
 
    std::atomic<bool> stopSolver;
 
    KissatFamily family;
 
    unsigned int originalVars;
 
    friend int kissatTerminate(void* solverPtr);
 
    /* Decided to not use pointers to move because of c++ stl (cannot move an array into a vector, sharedPtr
     * destruction) */
    friend char kissatImportClause(void*, kissat*);
    friend char kissatExportClause(void*, kissat*);
};