-
Notifications
You must be signed in to change notification settings - Fork 0
/
__init__.py
142 lines (121 loc) · 5.05 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
# Copyright or © or Copr. Loïc Paulevé (2023)
#
# loic.pauleve@cnrs.fr
#
# This software is governed by the CeCILL license under French law and
# abiding by the rules of distribution of free software. You can use,
# modify and/ or redistribute the software under the terms of the CeCILL
# license as circulated by CEA, CNRS and INRIA at the following URL
# "http://www.cecill.info".
#
# As a counterpart to the access to the source code and rights to copy,
# modify and redistribute granted by the license, users are provided only
# with a limited warranty and the software's author, the holder of the
# economic rights, and the successive licensors have only limited
# liability.
#
# In this respect, the user's attention is drawn to the risks associated
# with loading, using, modifying and/or developing or reproducing the
# software by the user in light of its specific status of free software,
# that may mean that it is complicated to manipulate, and that also
# therefore means that it is reserved for developers and experienced
# professionals having in-depth computer knowledge. Users are therefore
# encouraged to load and test the software's suitability as regards their
# requirements in conditions enabling the security of their systems and/or
# data to be ensured and, more generally, to use and operate it in the
# same conditions as regards security.
#
# The fact that you are presently reading this means that you have had
# knowledge of the CeCILL license and that you accept its terms.
#
import copy
from colomoto import minibn
import networkx as nx
from .asp_encoding import ASPModel_DNF
from .debug import *
from .domains import *
from .language import ManagedIface
from .manager import BonesisManager
from .snippets import *
from .utils import OverlayedDict
from .views import *
__language_api__ = ["obs", "cfg"]
settings = {
"parallel": 1,
"clingo_options": (),
"clingo_opt_strategy": "bb",
"solutions": "all",
"quiet": False,
"timeout": 0,
"soft_interrupt": False, # if True, silently end solving
"fail_if_timeout": True, # if timeout raise TimeoutError
"clingo_gil_workaround": 1,
# 0/None: no GIL wrapper for clingo
# 1: run clingo in a single background thread
# 2: separate thread for each solution
}
class BoNesis(object):
def __init__(self, domain, data=None):
if not isinstance(domain, BonesisDomain):
if isinstance(domain, minibn.BooleanNetwork):
domain = BooleanNetwork(domain)
elif isinstance(domain, (nx.DiGraph, nx.MultiDiGraph)):
domain = InfluenceGraph(domain)
else:
raise TypeError(f"Cannot handle domain with type '{type(domain)}'")
self.domain = domain
self.data = data or {}
self.manager = BonesisManager(self)
self.settings = OverlayedDict(settings)
self.aspmodel = ASPModel_DNF(self.domain, self.data, self.manager)
self.iface = ManagedIface(self.manager)
self.iface.install(self)
def fork(self):
fo = self.__class__(self.domain, self.data)
fo.manager.reset_from(self.manager)
return fo
def debug(self, asp_output="/tmp/debug.asp"):
with open(asp_output, "w") as fp:
self.aspmodel.make()
fp.write(str(self.aspmodel))
def set_constant(self, cst, value):
self.aspmodel.constants[cst] = value
def install_language(self, scope):
self.iface.install(scope)
def uninstall_language(self, scope):
self.iface.uninstall(scope)
def has_optimizations(self):
return bool(self.manager.optimizations)
def load_code(self, prog, defs=None, dest_scope=None):
scope = {}
self.install_language(scope)
exec(prog, scope, defs)
self.uninstall_language(scope)
del scope["__builtins__"]
ret = defs if defs else scope
if dest_scope is not None:
dest_scope.update(ret)
return ret
def load(self, script, defs=None, dest_scope=None):
with open(script) as fp:
return self.load_code(fp.read(), defs=defs, dest_scope=None)
def solver(self, *args, **kwargs):
self.aspmodel.make()
if "settings" not in kwargs:
kwargs["settings"] = self.settings
return self.aspmodel.solver(*args, **kwargs)
def is_satisfiable(self):
control = self.solver(1)
return control.solve().satisfiable
def boolean_networks(self, *args, **kwargs):
return BooleanNetworksView(self, *args, **kwargs)
def diverse_boolean_networks(self, *args, **kwargs):
return DiverseBooleanNetworksView(self, *args, **kwargs)
def projected_boolean_networks(self, **kwargs):
return ProjectedBooleanNetworksViews(self, **kwargs)
def local_functions(self, **kwargs):
return LocalFunctionsViews(self, **kwargs)
def influence_graphs(self, **kwargs):
return InfluenceGraphView(self, **kwargs)
def assignments(self, solutions="subset-minimal", **kwargs):
return AllSomeView(self, solutions=solutions, **kwargs)