timofej/neuropercolation
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add json dump for automaton states

Browse Source
This commit is contained in:
timofej 2023-08-17 23:10:10 +02:00
parent d35beaea81
commit ad73f60e6b
1 changed files with 60 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
neuropercolation/display.py
View File

@ -16,6 +16,9 @@ limitations under the License.
# pylint: disable=all
import os
import json
import pbjson
import time
import operator
import collections
@ -69,6 +72,7 @@ class Simulate2Layers:
eps,
res=4,
draw='pygame',
path='/cloud/Public/_data/neuropercolation/test/',
state_to_color_cb=None,
*args, **kwargs):
"""
@ -83,12 +87,16 @@ class Simulate2Layers:
super().__init__(*args, **kwargs)
self._cellular_automaton = Neuropercolation(dim,eps)
self.__cell_size = [res,res]
self.__dimension = dim
self.__epsilon = eps
self.__gridside = dim*res
self.__samegap = 10
self.__othergap = 20
self.__rect = _Rect(left=0, top=30, width=dim*res, height=2*dim*res+self.__samegap)
self.__draw_engine = PygameEngine((self.__rect.width,self.__rect.height+self.__rect.top)) if draw == 'pygame' else draw
self.__state_to_color = self._get_cell_color if state_to_color_cb is None else state_to_color_cb
self.__path = path
self.__state_list = []
def run(self,
evolutions_per_second=0,
@ -101,10 +109,12 @@ 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()
with contextlib.suppress(KeyboardInterrupt):
while self._is_not_user_terminated() and self._not_at_the_end(last_evolution_step):
time_ca_start = time.time()
self._cellular_automaton.evolve(evolutions_per_draw)
self._append_state()
time_ca_end = time.time()
self._redraw_dirty_cells()
time_ds_end = time.time()
@ -117,6 +127,25 @@ class Simulate2Layers:
self.__draw_engine._pygame.quit()
except:
print('Failed to quit pygame')
self._save_state_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 coord, cell in self._cellular_automaton._current_state.items():
x,y,l = coord
automaton_state[l][y][x] = 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)
self.__state_list.append(automaton_state)
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 _sleep_to_keep_rate(self, time_taken, evolutions_per_second): # pragma: no cover
if evolutions_per_second > 0:
@ -175,6 +204,7 @@ class Simulate4Layers:
coupling=[],
res=4,
draw='pygame',
path='/cloud/Public/_data/neuropercolation/test/',
state_to_color_cb=None,
*args, **kwargs):
"""
@ -189,12 +219,16 @@ class Simulate4Layers:
super().__init__(*args, **kwargs)
self._cellular_automaton = NeuropercolationCoupled(dim,eps,coupling)
self.__cell_size = [res,res]
self.__dimension = dim
self.__epsilon = eps
self.__gridside = dim*res
self.__samegap = 10
self.__othergap = 20
self.__rect = _Rect(left=0, top=30, width=2*dim*res+self.__othergap, height=2*dim*res+self.__samegap)
self.__draw_engine = PygameEngine((self.__rect.width,self.__rect.height+self.__rect.top)) if draw == 'pygame' else draw
self.__state_to_color = self._get_cell_color if state_to_color_cb is None else state_to_color_cb
self.__path = path
self.__state_list = []
def run(self,
evolutions_per_second=0,
@ -208,9 +242,11 @@ class Simulate4Layers:
Warning: is blocking until finished
"""
with contextlib.suppress(KeyboardInterrupt):
self._append_state()
while self._is_not_user_terminated() and self._not_at_the_end(last_evolution_step):
time_ca_start = time.time()
self._cellular_automaton.evolve(evolutions_per_draw)
self._append_state()
time_ca_end = time.time()
self._redraw_dirty_cells()
time_ds_end = time.time()
@ -223,6 +259,26 @@ class Simulate4Layers:
self.__draw_engine._pygame.quit()
except:
print('Failed to quit pygame')
self._save_state_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)]
for coord, cell in self._cellular_automaton._current_state.items():
x,y,l,o = coord
automaton_state[o][l][y][x] = int(cell.state[0])
automaton_state = [[[''.join(str(bits) for bits in rows) for rows in layers] for layers in oscillators] for oscillators in automaton_state]
automaton_state = [['.'.join(str(rows) for rows in layers) for layers in oscillators] for oscillators in automaton_state]
automaton_state = ['-'.join(str(layers) for layers in oscillators) for oscillators in automaton_state]
automaton_state = '='.join(str(oscillators) for oscillators in automaton_state)
self.__state_list.append(automaton_state)
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 _sleep_to_keep_rate(self, time_taken, evolutions_per_second): # pragma: no cover
if evolutions_per_second > 0: