an.tsouchlos/ba-thesis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Changed GenericMultithreadedSimulator interface to return results in a more usable form

Browse Source
This commit is contained in:
Andreas Tsouchlos 2022-12-05 14:49:06 +01:00
parent 0705fbbcaf
commit a82a7c0ca8
1 changed files with 24 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
sw/utility/simulation/simulators.py
View File

@ -7,7 +7,9 @@ from functools import partial
from multiprocessing import Lock from multiprocessing import Lock
from utility import noise from utility import noise
from cpp_modules.cpp_decoders import ProximalDecoder
# TODO: Fix ProximalDecoder_Dynamic
# from cpp_modules.cpp_decoders import ProximalDecoder_Dynamic as ProximalDecoder
def count_bit_errors(d: np.array, d_hat: np.array) -> int: def count_bit_errors(d: np.array, d_hat: np.array) -> int:
@ -259,6 +261,22 @@ class ProximalDecoderSimulator:
return pd.DataFrame(data) return pd.DataFrame(data)
class HashableDict:
"""Class behaving like an immutable dict. More importantly it is
hashable and thus usable as a key type for another dict."""
def __init__(self, data_dict):
assert (isinstance(data_dict, dict))
for key, val in data_dict.items():
self.__dict__[key] = val
def __getitem__(self, item):
return self.__dict__[item]
def __str__(self):
return str(self.__dict__)
class GenericMultithreadedSimulator: class GenericMultithreadedSimulator:
def __init__(self, max_workers=8): def __init__(self, max_workers=8):
self._task_func = None self._task_func = None
@ -273,9 +291,9 @@ class GenericMultithreadedSimulator:
return self._task_params return self._task_params
@task_params.setter @task_params.setter
def task_params(self, params): def task_params(self, sim_params):
assert isinstance(params, dict) self._task_params = {HashableDict(iteration_params): iteration_params
self._task_params = params for iteration_params in sim_params}
@property @property
def task_func(self): def task_func(self):
@ -314,7 +332,8 @@ class GenericMultithreadedSimulator:
" before it can be stopped" " before it can be stopped"
self._executor.shutdown(wait=False, cancel_futures=True) self._executor.shutdown(wait=False, cancel_futures=True)
def get_current_results(self): @property
def current_results(self):
return self._results return self._results
def __getstate__(self): def __getstate__(self):