python-sat 1.8.dev15__cp313-cp313-win_amd64.whl

python_sat-1.8.dev15.data/scripts/mcsls.py

@@ -0,0 +1,576 @@
+#!python
+#-*- coding:utf-8 -*-
+##
+## mcsls.py
+##
+##  Created on: Jan 9, 2017
+##      Author: Alexey S. Ignatiev
+##      E-mail: aignatiev@ciencias.ulisboa.pt
+##
+
+"""
+    ===============
+    List of classes
+    ===============
+
+    .. autosummary::
+        :nosignatures:
+
+        MCSls
+
+    ==================
+    Module description
+    ==================
+
+    This module implements a prototype of a BLS- and CLD-like algorithm for the
+    computation of a *minimal correction subset* (MCS) and/or MCS enumeration.
+    More concretely, the implementation follows the *basic linear search* (BLS)
+    for MCS exctraction augmented with *clause D* (CLD) oracle calls. As a
+    result, the algorithm is not an implementation of the BLS or CLD algorithms
+    as described in [1]_ but a mixture of both. Note that the corresponding
+    original low-level implementations of both can be found `online
+    <https://reason.di.fc.ul.pt/wiki/doku.php?id=mcsls>`_.
+
+    .. [1] Joao Marques-Silva, Federico Heras, Mikolas Janota, Alessandro
+        Previti, Anton Belov. *On Computing Minimal Correction Subsets*.
+        IJCAI 2013. pp.  615-622
+
+    The implementation can be used as an executable (the list of available
+    command-line options can be shown using ``mcsls.py -h``) in the following
+    way:
+
+    ::
+
+        $ xzcat formula.wcnf.xz
+        p wcnf 3 6 4
+        1 1 0
+        1 2 0
+        1 3 0
+        4 -1 -2 0
+        4 -1 -3 0
+        4 -2 -3 0
+
+        $ mcsls.py -d -e all -s glucose3 -vv formula.wcnf.xz
+        c MCS: 1 3 0
+        c cost: 2
+        c MCS: 2 3 0
+        c cost: 2
+        c MCS: 1 2 0
+        c cost: 2
+        c oracle time: 0.0002
+
+    Alternatively, the algorithm can be accessed and invoked through the
+    standard ``import`` interface of Python, e.g.
+
+    .. code-block:: python
+
+        >>> from pysat.examples.mcsls import MCSls
+        >>> from pysat.formula import WCNF
+        >>>
+        >>> wcnf = WCNF(from_file='formula.wcnf.xz')
+        >>>
+        >>> mcsls = MCSls(wcnf, use_cld=True, solver_name='g3')
+        >>> for mcs in mcsls.enumerate():
+        ...     mcsls.block(mcs)
+        ...     print(mcs)
+        [1, 3]
+        [2, 3]
+        [1, 2]
+
+    ==============
+    Module details
+    ==============
+"""
+
+#
+#==============================================================================
+from __future__ import print_function
+import collections
+import getopt
+from math import copysign
+import os
+from pysat.formula import CNFPlus, WCNFPlus
+from pysat.solvers import Solver, SolverNames
+import re
+import sys
+
+
+#
+#==============================================================================
+class MCSls(object):
+    """
+        Algorithm BLS for computing MCSes, augmented with "clause :math:`D`"
+        calls. Given an unsatisfiable partial CNF formula, i.e.  formula in the
+        :class:`.WCNF` format, this class can be used to compute a given number
+        of MCSes of the formula. The implementation follows the description of
+        the basic linear search (BLS) algorithm description in [1]_. It can use
+        any SAT solver available in PySAT. Additionally, the "clause :math:`D`"
+        heuristic can be used when enumerating MCSes.
+
+        The default SAT solver to use is ``m22`` (see :class:`.SolverNames`).
+        The "clause :math:`D`" heuristic is disabled by default, i.e.
+        ``use_cld`` is set to ``False``. Internal SAT solver's timer is also
+        disabled by default, i.e. ``use_timer`` is ``False``.
+
+        :param formula: unsatisfiable partial CNF formula
+        :param use_cld: whether or not to use "clause :math:`D`"
+        :param solver_name: SAT oracle name
+        :param use_timer: whether or not to use SAT solver's timer
+
+        :type formula: :class:`.WCNF`
+        :type use_cld: bool
+        :type solver_name: str
+        :type use_timer: bool
+    """
+
+    def __init__(self, formula, use_cld=False, solver_name='m22', use_timer=False):
+        """
+            Constructor.
+        """
+
+        # bootstrapping the solver with hard clauses
+        self.oracle = Solver(name=solver_name, bootstrap_with=formula.hard,
+                use_timer=use_timer)
+        self.solver = solver_name
+
+        # adding native cardinality constraints (if any) as hard clauses
+        # this can be done only if the Minicard solver is in use
+        if isinstance(formula, WCNFPlus) and formula.atms:
+            # we are using CaDiCaL195 and it can use external linear engine
+            if solver_name in SolverNames.cadical195:
+                self.oracle.activate_atmost()
+
+            assert self.oracle.supports_atmost(), \
+                    '{0} does not support native cardinality constraints. Make sure you use the right type of formula.'.format(solver_name)
+
+            for atm in formula.atms:
+                self.oracle.add_atmost(*atm)
+
+        self.topv = formula.nv  # top variable id
+        self.sels = []
+        self.ucld = use_cld
+        self.smap = {}
+
+        # mappings between internal and external variables
+        VariableMap = collections.namedtuple('VariableMap', ['e2i', 'i2e'])
+        self.vmap = VariableMap(e2i={}, i2e={})
+
+        # at this point internal and external variables are the same
+        for v in range(1, formula.nv + 1):
+            self.vmap.e2i[v] = v
+            self.vmap.i2e[v] = v
+
+        for cl in formula.soft:
+            new_cl = cl[:]
+            if len(cl) > 1 or cl[0] < 0:
+                self.topv += 1
+                sel = self.topv
+
+                new_cl.append(-sel)  # creating a new selector
+                self.oracle.add_clause(new_cl)
+            else:
+                sel = cl[0]
+
+            self.sels.append(sel)
+            self.smap[sel] = len(self.sels)
+
+    def __del__(self):
+        """
+            Destructor.
+        """
+
+        self.delete()
+
+    def __enter__(self):
+        """
+            'with' constructor.
+        """
+
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        """
+            'with' destructor.
+        """
+
+        self.delete()
+
+    def delete(self):
+        """
+            Explicit destructor of the internal SAT oracle.
+        """
+
+        if self.oracle:
+            self.oracle.delete()
+            self.oracle = None
+
+    def add_clause(self, clause, soft=False):
+        """
+            The method for adding a new hard of soft clause to the problem
+            formula. Although the input formula is to be specified as an
+            argument of the constructor of :class:`MCSls`, adding clauses may
+            be helpful when *enumerating* MCSes of the formula. This way, the
+            clauses are added incrementally, i.e. *on the fly*.
+
+            The clause to add can be any iterable over integer literals. The
+            additional Boolean parameter ``soft`` can be set to ``True``
+            meaning the the clause being added is soft (note that parameter
+            ``soft`` is set to ``False`` by default).
+
+            Also note that besides pure clauses, the method can also expect
+            native cardinality constraints represented as a pair ``(lits,
+            bound)``. Only hard cardinality constraints can be added.
+
+            :param clause: a clause to add
+            :param soft: whether or not the clause is soft
+
+            :type clause: iterable(int)
+            :type soft: bool
+        """
+
+        # first, map external literals to internal literals
+        # introduce new variables if necessary
+        cl = list(map(lambda l: self._map_extlit(l), clause if not len(clause) == 2 or not type(clause[0]) in (list, tuple, set) else clause[0]))
+
+        if not soft:
+            if not len(clause) == 2 or not type(clause[0]) in (list, tuple, set):
+                # the clause is hard, and so we simply add it to the SAT oracle
+                self.oracle.add_clause(cl)
+            else:
+                # this should be a native cardinality constraint,
+                # which can be used only together with Minicard
+                assert self.oracle.supports_atmost(), \
+                        '{0} does not support native cardinality constraints. Make sure you use the right type of formula.'.format(self.solver)
+
+                self.oracle.add_atmost(cl, clause[1])
+        else:
+            # soft clauses should be augmented with a selector
+            sel = cl[0]
+            if len(cl) > 1 or cl[0] < 0:
+                self.topv += 1
+                sel = self.topv
+
+                self.oracle.add_clause(cl + [-sel])
+
+            self.sels.append(sel)
+            self.smap[sel] = len(self.sels)
+
+    def compute(self, enable=[]):
+        """
+            Compute and return one solution. This method checks whether the
+            hard part of the formula is satisfiable, i.e. an MCS can be
+            extracted. If the formula is satisfiable, the model computed by the
+            SAT call is used as an *over-approximation* of the MCS in the
+            method :func:`_compute` invoked here, which implements the BLS
+
+            An MCS is reported as a list of integers, each representing a soft
+            clause index (the smallest index is ``1``).
+
+            An optional input parameter is ``enable``, which represents a
+            sequence (normally a list) of soft clause indices that a user
+            would prefer to enable/satisfy. Note that this may result in an
+            unsatisfiable oracle call, in which case ``None`` will be reported
+            as solution. Also, the smallest clause index is assumed to be
+            ``1``.
+
+            :param enable: a sequence of clause ids to enable
+            :type enable: iterable(int)
+
+            :rtype: list(int)
+        """
+
+        self.setd = []
+        self.solution = None
+        self.bb_assumps = []  # backbone assumptions
+        self.ss_assumps = []  # satisfied soft clause assumptions
+
+        if self.oracle.solve(assumptions=[self.sels[cl_id - 1] for cl_id in enable]):
+            # hard part is satisfiable => there is a solution
+            self._overapprox()
+            self._compute()
+
+            self.solution = [self.smap[-l] for l in self.bb_assumps]
+
+        return self.solution
+
+    def enumerate(self):
+        """
+            This method iterates through MCSes enumerating them until the
+            formula has no more MCSes. The method iteratively invokes
+            :func:`compute`. Note that the method does not block the MCSes
+            computed - this should be explicitly done by a user.
+        """
+
+        done = False
+        while not done:
+            mcs = self.compute()
+
+            if mcs != None:
+                yield mcs
+            else:
+                done = True
+
+    def block(self, mcs):
+        """
+            Block a (previously computed) MCS. The MCS should be given as an
+            iterable of integers. Note that this method is not automatically
+            invoked from :func:`enumerate` because a user may want to block
+            some of the MCSes conditionally depending on the needs. For
+            example, one may want to compute disjoint MCSes only in which case
+            this standard blocking is not appropriate.
+
+            :param mcs: an MCS to block
+            :type mcs: iterable(int)
+        """
+
+        self.oracle.add_clause([self.sels[cl_id - 1] for cl_id in mcs])
+
+    def _overapprox(self):
+        """
+            The method extracts a model corresponding to an over-approximation
+            of an MCS, i.e. it is the model of the hard part of the formula
+            (the corresponding oracle call is made in :func:`compute`).
+
+            Here, the set of selectors is divided into two parts:
+            ``self.ss_assumps``, which is an under-approximation of an MSS
+            (maximal satisfiable subset) and ``self.setd``, which is an
+            over-approximation of the target MCS. Both will be further refined
+            in :func:`_compute`.
+        """
+
+        model = self.oracle.get_model()
+
+        for sel in self.sels:
+            if len(model) < sel or model[sel - 1] > 0:
+                # soft clauses contain positive literals
+                # so if var is true then the clause is satisfied
+                self.ss_assumps.append(sel)
+            else:
+                self.setd.append(sel)
+
+    def _compute(self):
+        """
+            The main method of the class, which computes an MCS given its
+            over-approximation. The over-approximation is defined by a model
+            for the hard part of the formula obtained in :func:`_overapprox`
+            (the corresponding oracle is made in :func:`compute`).
+
+            The method is essentially a simple loop going over all literals
+            unsatisfied by the previous model, i.e. the literals of
+            ``self.setd`` and checking which literals can be satisfied. This
+            process can be seen a refinement of the over-approximation of the
+            MCS. The algorithm follows the pseudo-code of the BLS algorithm
+            presented in [1]_.
+
+            Additionally, if :class:`MCSls` was constructed with the
+            requirement to make "clause :math:`D`" calls, the method calls
+            :func:`do_cld_check` at every iteration of the loop using the
+            literals of ``self.setd`` not yet checked, as the contents of
+            "clause :math:`D`".
+        """
+
+        # unless clause D checks are used, test one literal at a time
+        # and add it either to satisfied of backbone assumptions
+        i = 0
+        while i < len(self.setd):
+            if self.ucld:
+                self.do_cld_check(self.setd[i:])
+                i = 0
+
+            if self.setd:
+                # if may be empty after the clause D check
+
+                self.ss_assumps.append(self.setd[i])
+                if not self.oracle.solve(assumptions=self.ss_assumps + self.bb_assumps):
+                    self.ss_assumps.pop()
+                    self.bb_assumps.append(-self.setd[i])
+
+            i += 1
+
+    def do_cld_check(self, cld):
+        """
+            Do the "clause :math:`D`" check. This method receives a list of
+            literals, which serves a "clause :math:`D`" [1]_, and checks
+            whether the formula conjoined with :math:`D` is satisfiable.
+
+            If clause :math:`D` cannot be satisfied together with the formula,
+            then negations of all of its literals are backbones of the formula
+            and the MCSls algorithm can stop. Otherwise, the literals satisfied
+            by the new model refine the MCS further.
+
+            Every time the method is called, a new fresh selector variable
+            :math:`s` is introduced, which augments the current clause
+            :math:`D`. The SAT oracle then checks if clause :math:`(D \\vee
+            \\neg{s})` can be satisfied together with the internal formula.
+            The :math:`D` clause is then disabled by adding a hard clause
+            :math:`(\\neg{s})`.
+
+            :param cld: clause :math:`D` to check
+            :type cld: list(int)
+        """
+
+        # adding a selector literal to clause D
+        # selector literals for clauses D currently
+        # cannot be reused, but this may change later
+        self.topv += 1
+        sel = self.topv
+        cld.append(-sel)
+
+        # adding clause D
+        self.oracle.add_clause(cld)
+        self.ss_assumps.append(sel)
+
+        self.setd = []
+        self.oracle.solve(assumptions=self.ss_assumps + self.bb_assumps)
+
+        self.ss_assumps.pop()  # removing clause D assumption
+        if self.oracle.get_status() == True:
+            model = self.oracle.get_model()
+
+            for l in cld[:-1]:
+                # filtering all satisfied literals
+                if model[abs(l) - 1] > 0:
+                    self.ss_assumps.append(l)
+                else:
+                    self.setd.append(l)
+        else:
+            # clause D is unsatisfiable => all literals are backbones
+            self.bb_assumps.extend([-l for l in cld[:-1]])
+
+        # deactivating clause D
+        self.oracle.add_clause([-sel])
+
+    def _map_extlit(self, l):
+        """
+            Map an external variable to an internal one if necessary.
+
+            This method is used when new clauses are added to the formula
+            incrementally, which may result in introducing new variables
+            clashing with the previously used *clause selectors*. The method
+            makes sure no clash occurs, i.e. it maps the original variables
+            used in the new problem clauses to the newly introduced auxiliary
+            variables (see :func:`add_clause`).
+
+            Given an integer literal, a fresh literal is returned. The returned
+            integer has the same sign as the input literal.
+
+            :param l: literal to map
+            :type l: int
+
+            :rtype: int
+        """
+
+        v = abs(l)
+
+        if v in self.vmap.e2i:
+            return int(copysign(self.vmap.e2i[v], l))
+        else:
+            self.topv += 1
+
+            self.vmap.e2i[v] = self.topv
+            self.vmap.i2e[self.topv] = v
+
+            return int(copysign(self.topv, l))
+
+    def oracle_time(self):
+        """
+            Report the total SAT solving time.
+        """
+
+        return self.oracle.time_accum()
+
+
+#
+#==============================================================================
+def parse_options():
+    """
+        Parses command-line options.
+    """
+
+    try:
+        opts, args = getopt.getopt(sys.argv[1:],
+                                   'de:hs:v',
+                                   ['dcalls',
+                                    'enum=',
+                                    'help',
+                                    'solver=',
+                                    'verbose'])
+    except getopt.GetoptError as err:
+        sys.stderr.write(str(err).capitalize() + '\n')
+        usage()
+        sys.exit(1)
+
+    dcalls = False
+    to_enum = 1
+    solver = 'm22'
+    verbose = 0
+
+    for opt, arg in opts:
+        if opt in ('-d', '--dcalls'):
+            dcalls = True
+        elif opt in ('-e', '--enum'):
+            to_enum = str(arg)
+            if to_enum != 'all':
+                to_enum = int(to_enum)
+        elif opt in ('-h', '--help'):
+            usage()
+            sys.exit(0)
+        elif opt in ('-s', '--solver'):
+            solver = str(arg)
+        elif opt in ('-v', '--verbose'):
+            verbose += 1
+        else:
+            assert False, 'Unhandled option: {0} {1}'.format(opt, arg)
+
+    return dcalls, to_enum, solver, verbose, args
+
+
+#
+#==============================================================================
+def usage():
+    """
+        Prints help message.
+    """
+
+    print('Usage:', os.path.basename(sys.argv[0]), '[options] file')
+    print('Options:')
+    print('        -d, --dcalls           Apply clause D calls')
+    print('        -e, --enum=<string>    How many solutions to compute')
+    print('                               Available values: [1 .. all] (default: 1)')
+    print('        -h, --help')
+    print('        -s, --solver           SAT solver to use')
+    print('                               Available values: cd15, cd19, g3, g4, lgl, mcb, mcm, mpl, m22, mc, mgh (default = m22)')
+    print('        -v, --verbose          Be verbose')
+
+
+#
+#==============================================================================
+if __name__ == '__main__':
+    dcalls, to_enum, solver, verbose, files = parse_options()
+
+    if type(to_enum) == str:
+        to_enum = 0
+
+    if files:
+        # reading standard CNF, WCNF, or (W)CNF+
+        if re.search('cnf[p|+]?(\.(gz|bz2|lzma|xz))?$', files[0]):
+            if re.search('\.wcnf[p|+]?(\.(gz|bz2|lzma|xz))?$', files[0]):
+                formula = WCNFPlus(from_file=files[0])
+            else:  # expecting '*.cnf[,p,+].*'
+                formula = CNFPlus(from_file=files[0]).weighted()
+
+        with MCSls(formula, use_cld=dcalls, solver_name=solver, use_timer=True) as mcsls:
+            for i, mcs in enumerate(mcsls.enumerate()):
+                if verbose:
+                    print('c MCS:', ' '.join([str(cl_id) for cl_id in mcs]), '0')
+
+                    if verbose > 1:
+                        cost = sum([formula.wght[cl_id - 1] for cl_id in mcs])
+                        print('c cost:', cost)
+
+                if to_enum and i + 1 == to_enum:
+                    break
+
+                mcsls.block(mcs)
+
+            print('c oracle time: {0:.4f}'.format(mcsls.oracle_time()))