rs package

rs/extended_context_automaton.py

@@ -0,0 +1,76 @@
+from sys import exit
+from colour import *
+
+from rs.context_automaton import ContextAutomaton
+
+class ExtendedContextAutomaton(ContextAutomaton):
+    """Extended Context Automaton"""
+    def __init__(self, reaction_system):
+        self._actions = []
+        super(ExtendedContextAutomaton, self).__init__(reaction_system)
+    
+    def add_transition(self, src, action, ctx_reaction, dst):
+        
+        ctx_reactants, ctx_inhibitors, ctx_products = ctx_reaction 
+        
+        if not type(ctx_products) is set and not type(ctx_products) is list:
+            print("Contexts set (context products) must be of type set or list")
+        
+        if not self.is_valid_rs_set(ctx_reactants):
+            raise RuntimeError("one of the entities in the reactants set is unknown (undefined)!")
+        
+        if not self.is_valid_rs_set(ctx_inhibitors):
+            raise RuntimeError("one of the entities in the inhibitors set is unknown (undefined)!")
+        
+        if not self.is_valid_rs_set(ctx_products):
+            raise RuntimeError("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")
+
+        if not self.is_state(dst):
+            raise RuntimeError("\"" + dst + "\" is an unknown (undefined) state")
+
+        src_id = self.get_state_id(src)
+        dst_id = self.get_state_id(dst)
+        act_id = self.get_action_id(action)
+        r_ids = self.get_set_of_ids(ctx_reactants)
+        i_ids = self.get_set_of_ids(ctx_inhibitors)
+        p_ids = self.get_set_of_ids(ctx_products)
+        
+        self._transitions.append((src_id, act_id, (r_ids, i_ids, p_ids), dst_id))
+    
+    def show_transitions(self):
+        print(C_MARK_INFO + " Context automaton transitions:")
+        for src_id, act_id, reaction, dst_id in self._transitions:
+            str_transition = self.get_state_name(src_id) + " --( " 
+            str_transition += self.get_action_name(act_id) + " | "
+            str_transition += "( " + self.rsset2str(reaction[0]) + "," + self.rsset2str(reaction[1]) + "," + self.rsset2str(reaction[2]) + " )"
+            str_transition += " )--> " + self.get_state_name(dst_id)
+            print(" - " + str_transition)
+    
+    def add_action(self, action_name):
+        if action_name not in self._actions:
+            self._actions.append(action_name)
+        else:
+            print("\'%s\' already added. skipping..." % (action_name,))
+    
+    def get_action_id(self, action_name):
+        try:
+            return self._actions.index(action_name)
+        except ValueError:
+            print_error("Undefined context automaton action: " + repr(action_name))
+            exit(1)
+    
+    def get_action_name(self, action_id):
+        return self._actions[action_id]
+                        
+    def show_actions(self):
+        print(C_MARK_INFO + " Context automaton actions:")
+        for act in self._actions:
+            print(" - " + act)
+    
+    def show(self):
+        self.show_states()
+        self.show_actions()
+        self.show_transitions()