ms762wu/EMG_Biometrics_2021
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix: add solution for rounds in store_df func

Browse Source
This commit is contained in:
Skudalen 2021-06-25 10:07:35 +02:00
parent 14ad7da806
commit 8a6f8d6329
2 changed files with 27 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
Handle_emg_data.py
View File

@ -9,7 +9,10 @@ class Data_container:
def __init__(self, subject_nr:int, subject_name:str): def __init__(self, subject_nr:int, subject_name:str):
self.subject_nr = subject_nr self.subject_nr = subject_nr
self.subject_name = subject_name self.subject_name = subject_name
self.data_dict = {'left': [None]*8, 'right': [None]*8} self.data_dict_round1 = {'left': [None]*8, 'right': [None]*8}
self.data_dict_round2 = {'left': [None]*8, 'right': [None]*8}
self.data_dict_round3 = {'left': [None]*8, 'right': [None]*8}
self.data_dict_round4 = {'left': [None]*8, 'right': [None]*8}
class CSV_handler: class CSV_handler:
@ -30,14 +33,34 @@ class CSV_handler:
filtered_df = tot_data_frame[["timestamp", emg_str]] filtered_df = tot_data_frame[["timestamp", emg_str]]
return filtered_df return filtered_df
def store_df(self, filename:str, emg_nr:int, which_arm:str, data_container:Data_container): def store_df(self, filename:str, emg_nr:int, which_arm:str, data_container:Data_container, round:int):
df = self.get_min_max_timestamp(filename, emg_nr) df = self.get_min_max_timestamp(filename, emg_nr)
# Links the retrieved data with the subjects data_container # Links the retrieved data with the subjects data_container
subject_nr = data_container.subject_nr subject_nr = data_container.subject_nr
self.data_container_dict[subject_nr] = data_container self.data_container_dict[subject_nr] = data_container
# Places the data correctly: # Places the data correctly:
if which_arm == 'left': if round == 0:
data_container.data_dict['left'][emg_nr+1] = df if which_arm == 'left':
data_container.data_dict_round1['left'][emg_nr+1] = df
else:
data_container.data_dict_round1['right'][emg_nr+1] = df
elif round == 1:
if which_arm == 'left':
data_container.data_dict_round2['left'][emg_nr+1] = df
else:
data_container.data_dict_round2['right'][emg_nr+1] = df
elif round == 2:
if which_arm == 'left':
data_container.data_dict_round3['left'][emg_nr+1] = df
else:
data_container.data_dict_round3['right'][emg_nr+1] = df
elif round == 3:
if which_arm == 'left':
data_container.data_dict_round4['left'][emg_nr+1] = df
else:
data_container.data_dict_round4['right'][emg_nr+1] = df
else:
raise IndexError('Not a valid index')
# Loads the data from the csv files into a storing system in an CSV_handler object # Loads the data from the csv files into a storing system in an CSV_handler object
def load_hard_PP_emg_data(self): def load_hard_PP_emg_data(self):

0
Visualize_data.py → Present_data.py
View File