black formatting

This commit is contained in:
Artur Meski
2023-07-11 19:48:37 +01:00
parent 8be615b293
commit a1fb5836a5
24 changed files with 965 additions and 727 deletions

View File

@@ -5,15 +5,13 @@ from rs.context_automaton import ContextAutomaton
class ContextAutomatonWithConcentrations(ContextAutomaton):
def __init__(self, reaction_system):
super(ContextAutomatonWithConcentrations,
self).__init__(reaction_system)
super(ContextAutomatonWithConcentrations, self).__init__(reaction_system)
def is_valid_context(self, context):
if set(
[e for e, lvl in context]).issubset(
self._reaction_system.background_set):
if set([e for e, lvl in context]).issubset(
self._reaction_system.background_set
):
return True
else:
return False
@@ -33,45 +31,38 @@ class ContextAutomatonWithConcentrations(ContextAutomaton):
if not self.is_valid_context(context_set):
raise RuntimeError(
"one of the entities in the context set is unknown (undefined)!")
"one of the entities in the context set is unknown (undefined)!"
)
if not self.is_state(src):
raise RuntimeError(
"\"" + src + "\" is an unknown (undefined) state")
raise RuntimeError('"' + src + '" is an unknown (undefined) state')
if not self.is_state(dst):
raise RuntimeError(
"\"" + dst + "\" is an unknown (undefined) state")
raise RuntimeError('"' + dst + '" is an unknown (undefined) state')
new_context_set = set()
for ent, lvl in set(context_set):
new_context_set.add(
(self._reaction_system.get_entity_id(ent), lvl))
new_context_set.add((self._reaction_system.get_entity_id(ent), lvl))
self._transitions.append(
(self.get_state_id(src),
new_context_set, self.get_state_id(dst)))
(self.get_state_id(src), new_context_set, self.get_state_id(dst))
)
def get_automaton_with_flat_contexts(self, ordinary_reaction_system):
ca = ContextAutomaton(ordinary_reaction_system)
ca._states = self._states
ca._init_state = self._init_state
for src, ctx, dst in self._transitions:
new_ctx = set()
for ent, conc in ctx:
for i in range(1, conc+1):
n = self._reaction_system.get_entity_name(
ent) + "#" + str(i)
for i in range(1, conc + 1):
n = self._reaction_system.get_entity_name(ent) + "#" + str(i)
ca._reaction_system.ensure_bg_set_entity(n)
new_ctx.add(n)
ca.add_transition(
ca.get_state_name(src),
new_ctx, ca.get_state_name(dst))
ca.add_transition(ca.get_state_name(src), new_ctx, ca.get_state_name(dst))
return ca
@@ -80,4 +71,5 @@ class ContextAutomatonWithConcentrations(ContextAutomaton):
self.show_states()
self.show_transitions()
# EOF