From c37f63f7bf36147c846514dcf895a591ab1a3787 Mon Sep 17 00:00:00 2001 From: timofej Date: Sat, 19 Aug 2023 21:33:25 +0200 Subject: [PATCH] Add save_states option --- neuropercolation/display.py | 100 ++++++++++++++++++++++++++++++------ 1 file changed, 85 insertions(+), 15 deletions(-) diff --git a/neuropercolation/display.py b/neuropercolation/display.py index 1c39e6b..9f9417e 100644 --- a/neuropercolation/display.py +++ b/neuropercolation/display.py @@ -99,7 +99,8 @@ class Simulate1Layer: def run(self, evolutions_per_second=0, evolutions_per_draw=1, - last_evolution_step=0,): + last_evolution_step=0, + save_states=True): """ Evolves and draws the CellularAutomaton :param evolutions_per_second: 0 = as fast as possible | > 0 to slow down the CellularAutomaton @@ -107,12 +108,18 @@ class Simulate1Layer: :param last_evolution_step: 0 = infinite | > 0 evolution step at which this method will stop Warning: is blocking until finished """ - self._append_state() + if save_states: + self._append_state() + else: + self._append_activation() with contextlib.suppress(KeyboardInterrupt): while self._not_at_the_end(last_evolution_step): time_ca_start = time.time() self._cellular_automaton.evolve(evolutions_per_draw) - self._append_state() + if save_states: + self._append_state() + else: + self._append_activation() if self.__draw_engine is not None: time_ca_end = time.time() self._redraw_dirty_cells() @@ -127,7 +134,9 @@ class Simulate1Layer: self.__draw_engine._pygame.quit() except: print('Failed to quit pygame') - self._save_state_list() + if save_states: + self._save_state_list() + self._save_activation_list() def _append_state(self): automaton_state = [[0 for n in range(self.__dimension)] for m in range(self.__dimension)] @@ -144,12 +153,26 @@ class Simulate1Layer: self.__state_list.append(automaton_state) self.__activation_list.append(activation) + def _append_activation(self): + activation = 0 + for coord, cell in self._cellular_automaton._current_state.items(): + x,y = coord + activation += int(cell.state[0]) + + activation = round(1-2*activation/self.__dimension**2, 6) + self.__activation_list.append(activation) + def _save_state_list(self): if not os.path.exists(self.__path): os.makedirs(self.__path) with open(self.__path+f"eps={round(self.__epsilon,3):.3f}_states.txt", 'w', encoding='utf-8') as f: json.dump(self.__state_list, f, indent=1) + + def _save_activation_list(self): + if not os.path.exists(self.__path): + os.makedirs(self.__path) + with open(self.__path+f"eps={round(self.__epsilon,3):.3f}_activation.txt", 'w', encoding='utf-8') as f: json.dump(self.__activation_list, f, indent=1) @@ -232,7 +255,8 @@ class Simulate2Layers: def run(self, evolutions_per_second=0, evolutions_per_draw=1, - last_evolution_step=0,): + last_evolution_step=0, + save_states=True): """ Evolves and draws the CellularAutomaton :param evolutions_per_second: 0 = as fast as possible | > 0 to slow down the CellularAutomaton @@ -240,12 +264,18 @@ class Simulate2Layers: :param last_evolution_step: 0 = infinite | > 0 evolution step at which this method will stop Warning: is blocking until finished """ - self._append_state() + if save_states: + self._append_state() + else: + self._append_activation() with contextlib.suppress(KeyboardInterrupt): while self._not_at_the_end(last_evolution_step): time_ca_start = time.time() self._cellular_automaton.evolve(evolutions_per_draw) - self._append_state() + if save_states: + self._append_state() + else: + self._append_activation() if self.__draw_engine is not None: time_ca_end = time.time() self._redraw_dirty_cells() @@ -255,12 +285,16 @@ class Simulate2Layers: evolution_step=self._cellular_automaton.evolution_step, runlendig=len(str(last_evolution_step))) self._sleep_to_keep_rate(time.time() - time_ca_start, evolutions_per_second) + if self.__draw_engine is not None: try: self.__draw_engine._pygame.quit() except: print('Failed to quit pygame') - self._save_state_list() + + if save_states: + self._save_state_list() + self._save_activation_list() def _append_state(self): automaton_state = [[[0 for n in range(self.__dimension)] for m in range(self.__dimension)] for l in range(2)] @@ -269,13 +303,21 @@ class Simulate2Layers: x,y,l = coord automaton_state[l][y][x] = int(cell.state[0]) activation[l] += int(cell.state[0]) + automaton_state = [[''.join(str(bits) for bits in rows) for rows in layers] for layers in automaton_state] automaton_state = ['.'.join(str(rows) for rows in layers) for layers in automaton_state] automaton_state = '-'.join(str(layers) for layers in automaton_state) + activation = [round(1-2*act/self.__dimension**2, 6) for act in activation] + self.__state_list.append(automaton_state) + self.__activation_list.append(activation) + + def _append_activation(self): + activation = [0]*2 + for coord, cell in self._cellular_automaton._current_state.items(): + x,y,l = coord + activation[l] += int(cell.state[0]) activation = [round(1-2*act/self.__dimension**2, 6) for act in activation] - - self.__state_list.append(automaton_state) self.__activation_list.append(activation) def _save_state_list(self): @@ -284,6 +326,11 @@ class Simulate2Layers: with open(self.__path+f"eps={round(self.__epsilon,3):.3f}_states.txt", 'w', encoding='utf-8') as f: json.dump(self.__state_list, f, indent=1) + + def _save_activation_list(self): + if not os.path.exists(self.__path): + os.makedirs(self.__path) + with open(self.__path+f"eps={round(self.__epsilon,3):.3f}_activation.txt", 'w', encoding='utf-8') as f: json.dump(self.__activation_list, f, indent=1) @@ -371,7 +418,8 @@ class Simulate4Layers: def run(self, evolutions_per_second=0, evolutions_per_draw=1, - last_evolution_step=0,): + last_evolution_step=0, + save_states=True): """ Evolves and draws the CellularAutomaton :param evolutions_per_second: 0 = as fast as possible | > 0 to slow down the CellularAutomaton @@ -380,11 +428,17 @@ class Simulate4Layers: Warning: is blocking until finished """ with contextlib.suppress(KeyboardInterrupt): - self._append_state() + if save_states: + self._append_state() + else: + self._append_activation() while self._not_at_the_end(last_evolution_step): time_ca_start = time.time() self._cellular_automaton.evolve(evolutions_per_draw) - self._append_state() + if save_states: + self._append_state() + else: + self._append_activation() if self.__draw_engine is not None: time_ca_end = time.time() self._redraw_dirty_cells() @@ -399,8 +453,10 @@ class Simulate4Layers: self.__draw_engine._pygame.quit() except: print('Failed to quit pygame') - self._save_state_list() - + if save_states: + self._save_state_list() + self._save_activation_list() + def _append_state(self): automaton_state = [[[[0 for n in range(self.__dimension)] for m in range(self.__dimension)] for l in range(2)] for k in range(2)] activation = [[0,0],[0,0]] @@ -418,12 +474,26 @@ class Simulate4Layers: self.__state_list.append(automaton_state) self.__activation_list.append(activation) + def _append_activation(self): + activation = [[0,0],[0,0]] + for coord, cell in self._cellular_automaton._current_state.items(): + x,y,l,o = coord + activation[o][l] += int(cell.state[0]) + + activation = [[round(1-2*act/self.__dimension**2, 6) for act in layer] for layer in activation] + self.__activation_list.append(activation) + def _save_state_list(self): if not os.path.exists(self.__path): os.makedirs(self.__path) with open(self.__path+f"eps={round(self.__epsilon,3):.3f}_states.txt", 'w', encoding='utf-8') as f: json.dump(self.__state_list, f, indent=1) + + def _save_activation_list(self): + if not os.path.exists(self.__path): + os.makedirs(self.__path) + with open(self.__path+f"eps={round(self.__epsilon,3):.3f}_activation.txt", 'w', encoding='utf-8') as f: json.dump(self.__activation_list, f, indent=1)