ms762wu/EMG_Biometrics_2021
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity
352b5b1b73
EMG_Biometrics_2021/Handle_emg_data.py

471 lines
32 KiB
Python
Raw Normal View History

feat: initialize files for analysis Add files which lay the framework for the project
2021-06-22 13:32:51 +00:00
import pandas as pd
feat: make a function that gives dataframes from csv, make_df
2021-06-22 18:31:16 +00:00
from pathlib import Path
fix: sort out error in min_max
2021-06-24 08:10:28 +00:00
import numpy as np
feat: make a function that gives dataframes from csv, make_df
2021-06-22 18:31:16 +00:00
feat: connect data_container object to the retrieval of df's
2021-06-23 09:33:23 +00:00
from pandas.core.frame import DataFrame
feat: add class as a data_container for each subject
2021-06-23 09:17:59 +00:00
class Data_container:
feat: connect data_container object to the retrieval of df's
2021-06-23 09:33:23 +00:00
def __init__(self, subject_nr:int, subject_name:str):
self.subject_nr = subject_nr
self.subject_name = subject_name
fix: add solution for rounds in store_df func
2021-06-25 08:07:35 +00:00
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}
feat: complete a denoise func for part of the dataset
2021-06-25 11:41:57 +00:00
self.dict_list = [self.data_dict_round1,
self.data_dict_round2,
self.data_dict_round3,
self.data_dict_round4
]
feat: add class as a data_container for each subject
2021-06-23 09:17:59 +00:00
feat: make a function that gives dataframes from csv, make_df
2021-06-22 18:31:16 +00:00
class CSV_handler:
feat: implement func that gets df from data_dict in handler
2021-06-25 11:57:28 +00:00
def __init__(self, data_type):
feat: connect data_container object to the retrieval of df's
2021-06-23 09:33:23 +00:00
self.working_dir = str(Path.cwd())
feat: complete a denoise func for part of the dataset
2021-06-25 11:41:57 +00:00
self.data_container_dict = {} # Dict with keys equal subject numbers and values equal the relvant datacontainer
feat: implement func that gets df from data_dict in handler
2021-06-25 11:57:28 +00:00
self.data_type = data_type
feat: make a function that gives dataframes from csv, make_df
2021-06-22 18:31:16 +00:00
doc: comments
2021-06-22 19:00:51 +00:00
# Makes dataframe from the csv files in the working directory
test: make file for funcs which tests the functionality
2021-06-22 19:30:55 +00:00
def make_df(self, filename):
feat: make a function that gives dataframes from csv, make_df
2021-06-22 18:31:16 +00:00
filepath = self.working_dir + str(filename)
df = pd.read_csv(filepath)
return df
doc: comments
2021-06-22 19:00:51 +00:00
feat: connect data_container object to the retrieval of df's
2021-06-23 09:33:23 +00:00
# Extracts out the timestamp and the selected emg signal into a new dataframe and stores the data on the subject
chore: split ut store and get mechanism in handler
2021-06-23 11:52:18 +00:00
def get_time_emg_table(self, filename:str, emg_nr:int):
fix: configuration of get_time_emg_table()
2021-06-23 08:32:38 +00:00
tot_data_frame = self.make_df(filename)
test: test plotting data from prev. made fuctions
2021-06-23 09:01:02 +00:00
emg_str = 'emg' + str(emg_nr)
doc: comments
2021-06-22 19:00:51 +00:00
filtered_df = tot_data_frame[["timestamp", emg_str]]
chore: split ut store and get mechanism in handler
2021-06-23 11:52:18 +00:00
return filtered_df
doc: add comments to handle funcs
2021-06-25 09:12:32 +00:00
# Takes in a df and stores the information in a Data_container object
chore: split up storing func to one storing func and one linking func
2021-06-25 08:43:59 +00:00
def store_df_in_container(self, filename:str, emg_nr:int, which_arm:str, data_container:Data_container, round:int):
df = self.get_time_emg_table(filename, emg_nr+1)
feat: get_time_emg_table puts the df into the subjects data_container
2021-06-23 09:41:06 +00:00
# Places the data correctly:
chore: split up storing func to one storing func and one linking func
2021-06-25 08:43:59 +00:00
if round == 1:
fix: add solution for rounds in store_df func
2021-06-25 08:07:35 +00:00
if which_arm == 'left':
chore: split up storing func to one storing func and one linking func
2021-06-25 08:43:59 +00:00
data_container.data_dict_round1['left'][emg_nr] = df # Zero indexed emg_nr in the dict
fix: add solution for rounds in store_df func
2021-06-25 08:07:35 +00:00
else:
chore: split up storing func to one storing func and one linking func
2021-06-25 08:43:59 +00:00
data_container.data_dict_round1['right'][emg_nr] = df
fix: add solution for rounds in store_df func
2021-06-25 08:07:35 +00:00
elif round == 2:
if which_arm == 'left':
chore: split up storing func to one storing func and one linking func
2021-06-25 08:43:59 +00:00
data_container.data_dict_round2['left'][emg_nr] = df
fix: add solution for rounds in store_df func
2021-06-25 08:07:35 +00:00
else:
chore: split up storing func to one storing func and one linking func
2021-06-25 08:43:59 +00:00
data_container.data_dict_round2['right'][emg_nr] = df
fix: add solution for rounds in store_df func
2021-06-25 08:07:35 +00:00
elif round == 3:
if which_arm == 'left':
chore: split up storing func to one storing func and one linking func
2021-06-25 08:43:59 +00:00
data_container.data_dict_round3['left'][emg_nr] = df
fix: add solution for rounds in store_df func
2021-06-25 08:07:35 +00:00
else:
chore: split up storing func to one storing func and one linking func
2021-06-25 08:43:59 +00:00
data_container.data_dict_round3['right'][emg_nr] = df
elif round == 4:
if which_arm == 'left':
data_container.data_dict_round4['left'][emg_nr] = df
else:
data_container.data_dict_round4['right'][emg_nr] = df
fix: add solution for rounds in store_df func
2021-06-25 08:07:35 +00:00
else:
raise IndexError('Not a valid index')
doc: comments
2021-06-22 19:00:51 +00:00
doc: add comments to handle funcs
2021-06-25 09:12:32 +00:00
# Links the data container for a subject to the handler object
chore: split up storing func to one storing func and one linking func
2021-06-25 08:43:59 +00:00
def link_container_to_handler(self, data_container:Data_container):
# Links the retrieved data with the subjects data_container
subject_nr = data_container.subject_nr
self.data_container_dict[subject_nr] = data_container
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
# Loads the data from the csv files into a storing system in an CSV_handler object
doc: add comments to handle funcs
2021-06-25 09:12:32 +00:00
# (hard, hardPP, soft and softPP)
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
def load_hard_PP_emg_data(self):
# CSV data from subject 1
file1_subject1_left = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1810/myoLeftEmg.csv"
file2_subject1_left = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1830/myoLeftEmg.csv"
file3_subject1_left = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1845/myoLeftEmg.csv"
file4_subject1_left = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1855/myoLeftEmg.csv"
subject1_left_files = [file1_subject1_left, file2_subject1_left, file3_subject1_left, file4_subject1_left]
file1_subject1_rigth = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1810/myoRightEmg.csv"
file2_subject1_rigth = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1830/myoRightEmg.csv"
file3_subject1_rigth = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1845/myoRightEmg.csv"
file4_subject1_rigth = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1855/myoRightEmg.csv"
subject1_right_files = [file1_subject1_rigth, file2_subject1_rigth, file3_subject1_rigth, file4_subject1_rigth]
# CSV data from subject 2
file1_subject2_left = "/Exp20201205_2myo_hardTypePP/HaluskaMaros_20201205_2010/myoLeftEmg.csv"
file2_subject2_left = "/Exp20201205_2myo_hardTypePP/HaluskaMaros_20201205_2025/myoLeftEmg.csv"
file3_subject2_left = "/Exp20201205_2myo_hardTypePP/HaluskaMaros_20201205_2035/myoLeftEmg.csv"
file4_subject2_left = "/Exp20201205_2myo_hardTypePP/HaluskaMaros_20201205_2045/myoLeftEmg.csv"
subject2_left_files = [file1_subject2_left, file2_subject2_left, file3_subject2_left, file4_subject2_left]
file1_subject2_rigth = "/Exp20201205_2myo_hardTypePP/HaluskaMaros_20201205_2010/myoRightEmg.csv"
file2_subject2_rigth = "/Exp20201205_2myo_hardTypePP/HaluskaMaros_20201205_2025/myoRightEmg.csv"
file3_subject2_rigth = "/Exp20201205_2myo_hardTypePP/HaluskaMaros_20201205_2035/myoRightEmg.csv"
file4_subject2_rigth = "/Exp20201205_2myo_hardTypePP/HaluskaMaros_20201205_2045/myoRightEmg.csv"
subject2_right_files = [file1_subject2_rigth, file2_subject2_rigth, file3_subject2_rigth, file4_subject2_rigth]
# CSV data from subject 3
file1_subject3_left = "/Exp20201205_2myo_hardTypePP/HaluskovaBeata_20201205_1700/myoLeftEmg.csv"
file2_subject3_left = "/Exp20201205_2myo_hardTypePP/HaluskovaBeata_20201205_1715/myoLeftEmg.csv"
file3_subject3_left = "/Exp20201205_2myo_hardTypePP/HaluskovaBeata_20201205_1725/myoLeftEmg.csv"
file4_subject3_left = "/Exp20201205_2myo_hardTypePP/HaluskovaBeata_20201205_1735/myoLeftEmg.csv"
subject3_left_files = [file1_subject3_left, file2_subject3_left, file3_subject3_left, file4_subject3_left]
file1_subject3_rigth = "/Exp20201205_2myo_hardTypePP/HaluskovaBeata_20201205_1700/myoRightEmg.csv"
file2_subject3_rigth = "/Exp20201205_2myo_hardTypePP/HaluskovaBeata_20201205_1715/myoRightEmg.csv"
file3_subject3_rigth = "/Exp20201205_2myo_hardTypePP/HaluskovaBeata_20201205_1725/myoRightEmg.csv"
file4_subject3_rigth = "/Exp20201205_2myo_hardTypePP/HaluskovaBeata_20201205_1735/myoRightEmg.csv"
subject3_right_files = [file1_subject3_rigth, file2_subject3_rigth, file3_subject3_rigth, file4_subject3_rigth]
# CSV data from subject 4
file1_subject4_left = "/Exp20201205_2myo_hardTypePP/KelisekDavid_20201209_1900/myoLeftEmg.csv"
file2_subject4_left = "/Exp20201205_2myo_hardTypePP/KelisekDavid_20201209_1915/myoLeftEmg.csv"
file3_subject4_left = "/Exp20201205_2myo_hardTypePP/KelisekDavid_20201209_1925/myoLeftEmg.csv"
file4_subject4_left = "/Exp20201205_2myo_hardTypePP/KelisekDavid_20201209_1935/myoLeftEmg.csv"
subject4_left_files = [file1_subject4_left, file2_subject4_left, file3_subject4_left, file4_subject4_left]
file1_subject4_rigth = "/Exp20201205_2myo_hardTypePP/KelisekDavid_20201209_1900/myoRightEmg.csv"
file2_subject4_rigth = "/Exp20201205_2myo_hardTypePP/KelisekDavid_20201209_1915/myoRightEmg.csv"
file3_subject4_rigth = "/Exp20201205_2myo_hardTypePP/KelisekDavid_20201209_1925/myoRightEmg.csv"
file4_subject4_rigth = "/Exp20201205_2myo_hardTypePP/KelisekDavid_20201209_1935/myoRightEmg.csv"
subject4_right_files = [file1_subject4_rigth, file2_subject4_rigth, file3_subject4_rigth, file4_subject4_rigth]
# CSV data from subject 5
file1_subject5_left = "/Exp20201205_2myo_hardTypePP/KelisekRichard_20201209_2030/myoLeftEmg.csv"
file2_subject5_left = "/Exp20201205_2myo_hardTypePP/KelisekRichard_20201209_2040/myoLeftEmg.csv"
file3_subject5_left = "/Exp20201205_2myo_hardTypePP/KelisekRichard_20201209_2050/myoLeftEmg.csv"
file4_subject5_left = "/Exp20201205_2myo_hardTypePP/KelisekRichard_20201209_2100/myoLeftEmg.csv"
subject5_left_files = [file1_subject5_left, file2_subject5_left, file3_subject5_left, file4_subject5_left]
file1_subject5_rigth = "/Exp20201205_2myo_hardTypePP/KelisekRichard_20201209_2030/myoRightEmg.csv"
file2_subject5_rigth = "/Exp20201205_2myo_hardTypePP/KelisekRichard_20201209_2040/myoRightEmg.csv"
file3_subject5_rigth = "/Exp20201205_2myo_hardTypePP/KelisekRichard_20201209_2050/myoRightEmg.csv"
file4_subject5_rigth = "/Exp20201205_2myo_hardTypePP/KelisekRichard_20201209_2100/myoRightEmg.csv"
subject5_right_files = [file1_subject5_rigth, file2_subject5_rigth, file3_subject5_rigth, file4_subject5_rigth]
left_list = [subject1_left_files, subject2_left_files, subject3_left_files, subject4_left_files, subject5_left_files]
right_list = [subject1_right_files, subject2_right_files, subject3_right_files, subject4_right_files, subject5_right_files]
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject1_data_container = Data_container(1, 'HaluskaMarek')
subject2_data_container = Data_container(2, 'HaluskaMaros')
subject3_data_container = Data_container(3, 'HaluskovaBeata')
subject4_data_container = Data_container(4, 'KelisekDavid')
subject5_data_container = Data_container(5, 'KelisekRichard')
subject_data_container_list = [subject1_data_container, subject2_data_container, subject3_data_container,
subject4_data_container, subject5_data_container]
for subject_nr in range(5):
data_container = subject_data_container_list[subject_nr]
# left variant proccessed here
for round in range(4):
for emg_nr in range(8):
filename = left_list[subject_nr][round]
self.store_df_in_container(filename, emg_nr, 'left', data_container, round+1)
# right variant proccessed here
for round in range(4):
for emg_nr in range(8):
filename = right_list[subject_nr][round]
self.store_df_in_container(filename, emg_nr, 'right', data_container, round+1)
# Links the stored data in the data_container to the Handler
self.link_container_to_handler(data_container)
feat: complete a denoise func for part of the dataset
2021-06-25 11:41:57 +00:00
self.data_type = 'hardPP'
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
return self.data_container_dict
def load_soft_PP_emg_data(self):
# CSV data from subject 1
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject1_left = "/Exp20201205_2myo_softTypePP/HaluskaMarek_20201207_1910/myoLeftEmg.csv"
file2_subject1_left = "/Exp20201205_2myo_softTypePP/HaluskaMarek_20201207_1920/myoLeftEmg.csv"
file3_subject1_left = "/Exp20201205_2myo_softTypePP/HaluskaMarek_20201207_1935/myoLeftEmg.csv"
file4_subject1_left = "/Exp20201205_2myo_softTypePP/HaluskaMarek_20201207_1945/myoLeftEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject1_left_files = [file1_subject1_left, file2_subject1_left, file3_subject1_left, file4_subject1_left]
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject1_rigth = "/Exp20201205_2myo_softTypePP/HaluskaMarek_20201207_1910/myoRightEmg.csv"
file2_subject1_rigth = "/Exp20201205_2myo_softTypePP/HaluskaMarek_20201207_1920/myoRightEmg.csv"
file3_subject1_rigth = "/Exp20201205_2myo_softTypePP/HaluskaMarek_20201207_1935/myoRightEmg.csv"
file4_subject1_rigth = "/Exp20201205_2myo_softTypePP/HaluskaMarek_20201207_1945/myoRightEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject1_right_files = [file1_subject1_rigth, file2_subject1_rigth, file3_subject1_rigth, file4_subject1_rigth]
# CSV data from subject 2
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject2_left = "/Exp20201205_2myo_softTypePP/HaluskaMaros_20201205_2055/myoLeftEmg.csv"
file2_subject2_left = "/Exp20201205_2myo_softTypePP/HaluskaMaros_20201205_2110/myoLeftEmg.csv"
file3_subject2_left = "/Exp20201205_2myo_softTypePP/HaluskaMaros_20201205_2125/myoLeftEmg.csv"
file4_subject2_left = "/Exp20201205_2myo_softTypePP/HaluskaMaros_20201205_2145/myoLeftEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject2_left_files = [file1_subject2_left, file2_subject2_left, file3_subject2_left, file4_subject2_left]
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject2_rigth = "/Exp20201205_2myo_softTypePP/HaluskaMaros_20201205_2055/myoRightEmg.csv"
file2_subject2_rigth = "/Exp20201205_2myo_softTypePP/HaluskaMaros_20201205_2110/myoRightEmg.csv"
file3_subject2_rigth = "/Exp20201205_2myo_softTypePP/HaluskaMaros_20201205_2125/myoRightEmg.csv"
file4_subject2_rigth = "/Exp20201205_2myo_softTypePP/HaluskaMaros_20201205_2145/myoRightEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject2_right_files = [file1_subject2_rigth, file2_subject2_rigth, file3_subject2_rigth, file4_subject2_rigth]
# CSV data from subject 3
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject3_left = "/Exp20201205_2myo_softTypePP/HaluskovaBeata_20201205_1745/myoLeftEmg.csv"
file2_subject3_left = "/Exp20201205_2myo_softTypePP/HaluskovaBeata_20201205_1755/myoLeftEmg.csv"
file3_subject3_left = "/Exp20201205_2myo_softTypePP/HaluskovaBeata_20201205_1810/myoLeftEmg.csv"
file4_subject3_left = "/Exp20201205_2myo_softTypePP/HaluskovaBeata_20201205_1825/myoLeftEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject3_left_files = [file1_subject3_left, file2_subject3_left, file3_subject3_left, file4_subject3_left]
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject3_rigth = "/Exp20201205_2myo_softTypePP/HaluskovaBeata_20201205_1745/myoRightEmg.csv"
file2_subject3_rigth = "/Exp20201205_2myo_softTypePP/HaluskovaBeata_20201205_1755/myoRightEmg.csv"
file3_subject3_rigth = "/Exp20201205_2myo_softTypePP/HaluskovaBeata_20201205_1810/myoRightEmg.csv"
file4_subject3_rigth = "/Exp20201205_2myo_softTypePP/HaluskovaBeata_20201205_1825/myoRightEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject3_right_files = [file1_subject3_rigth, file2_subject3_rigth, file3_subject3_rigth, file4_subject3_rigth]
# CSV data from subject 4
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject4_left = "/Exp20201205_2myo_softTypePP/KelisekDavid_20201209_1945/myoLeftEmg.csv"
file2_subject4_left = "/Exp20201205_2myo_softTypePP/KelisekDavid_20201209_1955/myoLeftEmg.csv"
file3_subject4_left = "/Exp20201205_2myo_softTypePP/KelisekDavid_20201209_2010/myoLeftEmg.csv"
file4_subject4_left = "/Exp20201205_2myo_softTypePP/KelisekDavid_20201209_2025/myoLeftEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject4_left_files = [file1_subject4_left, file2_subject4_left, file3_subject4_left, file4_subject4_left]
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject4_rigth = "/Exp20201205_2myo_softTypePP/KelisekDavid_20201209_1945/myoRightEmg.csv"
file2_subject4_rigth = "/Exp20201205_2myo_softTypePP/KelisekDavid_20201209_1955/myoRightEmg.csv"
file3_subject4_rigth = "/Exp20201205_2myo_softTypePP/KelisekDavid_20201209_2010/myoRightEmg.csv"
file4_subject4_rigth = "/Exp20201205_2myo_softTypePP/KelisekDavid_20201209_2025/myoRightEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject4_right_files = [file1_subject4_rigth, file2_subject4_rigth, file3_subject4_rigth, file4_subject4_rigth]
# CSV data from subject 5
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject5_left = "/Exp20201205_2myo_softTypePP/KelisekRichard_20201209_2110/myoLeftEmg.csv"
file2_subject5_left = "/Exp20201205_2myo_softTypePP/KelisekRichard_20201209_2120/myoLeftEmg.csv"
file3_subject5_left = "/Exp20201205_2myo_softTypePP/KelisekRichard_20201209_2130/myoLeftEmg.csv"
file4_subject5_left = "/Exp20201205_2myo_softTypePP/KelisekRichard_20201209_2140/myoLeftEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject5_left_files = [file1_subject5_left, file2_subject5_left, file3_subject5_left, file4_subject5_left]
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject5_rigth = "/Exp20201205_2myo_softTypePP/KelisekRichard_20201209_2110/myoRightEmg.csv"
file2_subject5_rigth = "/Exp20201205_2myo_softTypePP/KelisekRichard_20201209_2120/myoRightEmg.csv"
file3_subject5_rigth = "/Exp20201205_2myo_softTypePP/KelisekRichard_20201209_2130/myoRightEmg.csv"
file4_subject5_rigth = "/Exp20201205_2myo_softTypePP/KelisekRichard_20201209_2140/myoRightEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject5_right_files = [file1_subject5_rigth, file2_subject5_rigth, file3_subject5_rigth, file4_subject5_rigth]
left_list = [subject1_left_files, subject2_left_files, subject3_left_files, subject4_left_files, subject5_left_files]
right_list = [subject1_right_files, subject2_right_files, subject3_right_files, subject4_right_files, subject5_right_files]
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject1_data_container = Data_container(1, 'HaluskaMarek')
fix: fix the load func
2021-06-25 08:52:33 +00:00
subject2_data_container = Data_container(2, 'HaluskaMaros')
subject3_data_container = Data_container(3, 'HaluskovaBeata')
subject4_data_container = Data_container(4, 'KelisekDavid')
subject5_data_container = Data_container(5, 'KelisekRichard')
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject_data_container_list = [subject1_data_container, subject2_data_container, subject3_data_container,
subject4_data_container, subject5_data_container]
for subject_nr in range(5):
fix: fix the load func
2021-06-25 08:52:33 +00:00
data_container = subject_data_container_list[subject_nr]
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
# left variant proccessed here
for round in range(4):
for emg_nr in range(8):
fix: fix the load func
2021-06-25 08:52:33 +00:00
filename = left_list[subject_nr][round]
self.store_df_in_container(filename, emg_nr, 'left', data_container, round+1)
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
# right variant proccessed here
for round in range(4):
for emg_nr in range(8):
fix: fix the load func
2021-06-25 08:52:33 +00:00
filename = right_list[subject_nr][round]
fix: fix the left/rigth config in store func
2021-06-25 09:01:01 +00:00
self.store_df_in_container(filename, emg_nr, 'right', data_container, round+1)
fix: fix the load func
2021-06-25 08:52:33 +00:00
# Links the stored data in the data_container to the Handler
self.link_container_to_handler(data_container)
feat: complete a denoise func for part of the dataset
2021-06-25 11:41:57 +00:00
self.data_type = 'softPP'
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
return self.data_container_dict
def load_hard_original_emg_data(self):
# CSV data from subject 1
fix: fix a load func for original data too
2021-06-25 07:37:49 +00:00
file1_subject1_left = "/Exp20201205_2myo_hardType/HaluskaMarek_20201207_1810/myoLeftEmg.csv"
file2_subject1_left = "/Exp20201205_2myo_hardType/HaluskaMarek_20201207_1830/myoLeftEmg.csv"
file3_subject1_left = "/Exp20201205_2myo_hardType/HaluskaMarek_20201207_1845/myoLeftEmg.csv"
file4_subject1_left = "/Exp20201205_2myo_hardType/HaluskaMarek_20201207_1855/myoLeftEmg.csv"
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject1_left_files = [file1_subject1_left, file2_subject1_left, file3_subject1_left, file4_subject1_left]
fix: fix a load func for original data too
2021-06-25 07:37:49 +00:00
file1_subject1_rigth = "/Exp20201205_2myo_hardType/HaluskaMarek_20201207_1810/myoRightEmg.csv"
file2_subject1_rigth = "/Exp20201205_2myo_hardType/HaluskaMarek_20201207_1830/myoRightEmg.csv"
file3_subject1_rigth = "/Exp20201205_2myo_hardType/HaluskaMarek_20201207_1845/myoRightEmg.csv"
file4_subject1_rigth = "/Exp20201205_2myo_hardType/HaluskaMarek_20201207_1855/myoRightEmg.csv"
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject1_right_files = [file1_subject1_rigth, file2_subject1_rigth, file3_subject1_rigth, file4_subject1_rigth]
# CSV data from subject 2
fix: fix a load func for original data too
2021-06-25 07:37:49 +00:00
file1_subject2_left = "/Exp20201205_2myo_hardType/HaluskaMaros_20201205_2010/myoLeftEmg.csv"
file2_subject2_left = "/Exp20201205_2myo_hardType/HaluskaMaros_20201205_2025/myoLeftEmg.csv"
file3_subject2_left = "/Exp20201205_2myo_hardType/HaluskaMaros_20201205_2035/myoLeftEmg.csv"
file4_subject2_left = "/Exp20201205_2myo_hardType/HaluskaMaros_20201205_2045/myoLeftEmg.csv"
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject2_left_files = [file1_subject2_left, file2_subject2_left, file3_subject2_left, file4_subject2_left]
fix: fix a load func for original data too
2021-06-25 07:37:49 +00:00
file1_subject2_rigth = "/Exp20201205_2myo_hardType/HaluskaMaros_20201205_2010/myoRightEmg.csv"
file2_subject2_rigth = "/Exp20201205_2myo_hardType/HaluskaMaros_20201205_2025/myoRightEmg.csv"
file3_subject2_rigth = "/Exp20201205_2myo_hardType/HaluskaMaros_20201205_2035/myoRightEmg.csv"
file4_subject2_rigth = "/Exp20201205_2myo_hardType/HaluskaMaros_20201205_2045/myoRightEmg.csv"
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject2_right_files = [file1_subject2_rigth, file2_subject2_rigth, file3_subject2_rigth, file4_subject2_rigth]
# CSV data from subject 3
fix: fix a load func for original data too
2021-06-25 07:37:49 +00:00
file1_subject3_left = "/Exp20201205_2myo_hardType/HaluskovaBeata_20201205_1700/myoLeftEmg.csv"
file2_subject3_left = "/Exp20201205_2myo_hardType/HaluskovaBeata_20201205_1715/myoLeftEmg.csv"
file3_subject3_left = "/Exp20201205_2myo_hardType/HaluskovaBeata_20201205_1725/myoLeftEmg.csv"
file4_subject3_left = "/Exp20201205_2myo_hardType/HaluskovaBeata_20201205_1735/myoLeftEmg.csv"
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject3_left_files = [file1_subject3_left, file2_subject3_left, file3_subject3_left, file4_subject3_left]
fix: fix a load func for original data too
2021-06-25 07:37:49 +00:00
file1_subject3_rigth = "/Exp20201205_2myo_hardType/HaluskovaBeata_20201205_1700/myoRightEmg.csv"
file2_subject3_rigth = "/Exp20201205_2myo_hardType/HaluskovaBeata_20201205_1715/myoRightEmg.csv"
file3_subject3_rigth = "/Exp20201205_2myo_hardType/HaluskovaBeata_20201205_1725/myoRightEmg.csv"
file4_subject3_rigth = "/Exp20201205_2myo_hardType/HaluskovaBeata_20201205_1735/myoRightEmg.csv"
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject3_right_files = [file1_subject3_rigth, file2_subject3_rigth, file3_subject3_rigth, file4_subject3_rigth]
# CSV data from subject 4
fix: fix a load func for original data too
2021-06-25 07:37:49 +00:00
file1_subject4_left = "/Exp20201205_2myo_hardType/KelisekDavid_20201209_1900/myoLeftEmg.csv"
file2_subject4_left = "/Exp20201205_2myo_hardType/KelisekDavid_20201209_1915/myoLeftEmg.csv"
file3_subject4_left = "/Exp20201205_2myo_hardType/KelisekDavid_20201209_1925/myoLeftEmg.csv"
file4_subject4_left = "/Exp20201205_2myo_hardType/KelisekDavid_20201209_1935/myoLeftEmg.csv"
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject4_left_files = [file1_subject4_left, file2_subject4_left, file3_subject4_left, file4_subject4_left]
fix: fix a load func for original data too
2021-06-25 07:37:49 +00:00
file1_subject4_rigth = "/Exp20201205_2myo_hardType/KelisekDavid_20201209_1900/myoRightEmg.csv"
file2_subject4_rigth = "/Exp20201205_2myo_hardType/KelisekDavid_20201209_1915/myoRightEmg.csv"
file3_subject4_rigth = "/Exp20201205_2myo_hardType/KelisekDavid_20201209_1925/myoRightEmg.csv"
file4_subject4_rigth = "/Exp20201205_2myo_hardType/KelisekDavid_20201209_1935/myoRightEmg.csv"
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject4_right_files = [file1_subject4_rigth, file2_subject4_rigth, file3_subject4_rigth, file4_subject4_rigth]
# CSV data from subject 5
fix: fix a load func for original data too
2021-06-25 07:37:49 +00:00
file1_subject5_left = "/Exp20201205_2myo_hardType/KelisekRichard_20201209_2030/myoLeftEmg.csv"
file2_subject5_left = "/Exp20201205_2myo_hardType/KelisekRichard_20201209_2040/myoLeftEmg.csv"
file3_subject5_left = "/Exp20201205_2myo_hardType/KelisekRichard_20201209_2050/myoLeftEmg.csv"
file4_subject5_left = "/Exp20201205_2myo_hardType/KelisekRichard_20201209_2100/myoLeftEmg.csv"
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject5_left_files = [file1_subject5_left, file2_subject5_left, file3_subject5_left, file4_subject5_left]
fix: fix a load func for original data too
2021-06-25 07:37:49 +00:00
file1_subject5_rigth = "/Exp20201205_2myo_hardType/KelisekRichard_20201209_2030/myoRightEmg.csv"
file2_subject5_rigth = "/Exp20201205_2myo_hardType/KelisekRichard_20201209_2040/myoRightEmg.csv"
file3_subject5_rigth = "/Exp20201205_2myo_hardType/KelisekRichard_20201209_2050/myoRightEmg.csv"
file4_subject5_rigth = "/Exp20201205_2myo_hardType/KelisekRichard_20201209_2100/myoRightEmg.csv"
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject5_right_files = [file1_subject5_rigth, file2_subject5_rigth, file3_subject5_rigth, file4_subject5_rigth]
left_list = [subject1_left_files, subject2_left_files, subject3_left_files, subject4_left_files, subject5_left_files]
right_list = [subject1_right_files, subject2_right_files, subject3_right_files, subject4_right_files, subject5_right_files]
subject1_data_container = Data_container(1, 'HaluskaMarek')
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject2_data_container = Data_container(2, 'HaluskaMaros')
subject3_data_container = Data_container(3, 'HaluskovaBeata')
subject4_data_container = Data_container(4, 'KelisekDavid')
subject5_data_container = Data_container(5, 'KelisekRichard')
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
subject_data_container_list = [subject1_data_container, subject2_data_container, subject3_data_container,
subject4_data_container, subject5_data_container]
for subject_nr in range(5):
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
data_container = subject_data_container_list[subject_nr]
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
# left variant proccessed here
for round in range(4):
for emg_nr in range(8):
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
filename = left_list[subject_nr][round]
self.store_df_in_container(filename, emg_nr, 'left', data_container, round+1)
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
# right variant proccessed here
for round in range(4):
for emg_nr in range(8):
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
filename = right_list[subject_nr][round]
self.store_df_in_container(filename, emg_nr, 'right', data_container, round+1)
# Links the stored data in the data_container to the Handler
self.link_container_to_handler(data_container)
feat: complete a denoise func for part of the dataset
2021-06-25 11:41:57 +00:00
self.data_type = 'hard'
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
return self.data_container_dict
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
def load_soft_original_emg_data(self):
# CSV data from subject 1
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject1_left = "/Exp20201205_2myo_softType/HaluskaMarek_20201207_1910/myoLeftEmg.csv"
file2_subject1_left = "/Exp20201205_2myo_softType/HaluskaMarek_20201207_1920/myoLeftEmg.csv"
file3_subject1_left = "/Exp20201205_2myo_softType/HaluskaMarek_20201207_1935/myoLeftEmg.csv"
file4_subject1_left = "/Exp20201205_2myo_softType/HaluskaMarek_20201207_1945/myoLeftEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject1_left_files = [file1_subject1_left, file2_subject1_left, file3_subject1_left, file4_subject1_left]
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject1_rigth = "/Exp20201205_2myo_softType/HaluskaMarek_20201207_1910/myoRightEmg.csv"
file2_subject1_rigth = "/Exp20201205_2myo_softType/HaluskaMarek_20201207_1920/myoRightEmg.csv"
file3_subject1_rigth = "/Exp20201205_2myo_softType/HaluskaMarek_20201207_1935/myoRightEmg.csv"
file4_subject1_rigth = "/Exp20201205_2myo_softType/HaluskaMarek_20201207_1945/myoRightEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject1_right_files = [file1_subject1_rigth, file2_subject1_rigth, file3_subject1_rigth, file4_subject1_rigth]
# CSV data from subject 2
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject2_left = "/Exp20201205_2myo_softType/HaluskaMaros_20201205_2055/myoLeftEmg.csv"
file2_subject2_left = "/Exp20201205_2myo_softType/HaluskaMaros_20201205_2110/myoLeftEmg.csv"
file3_subject2_left = "/Exp20201205_2myo_softType/HaluskaMaros_20201205_2125/myoLeftEmg.csv"
file4_subject2_left = "/Exp20201205_2myo_softType/HaluskaMaros_20201205_2145/myoLeftEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject2_left_files = [file1_subject2_left, file2_subject2_left, file3_subject2_left, file4_subject2_left]
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject2_rigth = "/Exp20201205_2myo_softType/HaluskaMaros_20201205_2055/myoRightEmg.csv"
file2_subject2_rigth = "/Exp20201205_2myo_softType/HaluskaMaros_20201205_2110/myoRightEmg.csv"
file3_subject2_rigth = "/Exp20201205_2myo_softType/HaluskaMaros_20201205_2125/myoRightEmg.csv"
file4_subject2_rigth = "/Exp20201205_2myo_softType/HaluskaMaros_20201205_2145/myoRightEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject2_right_files = [file1_subject2_rigth, file2_subject2_rigth, file3_subject2_rigth, file4_subject2_rigth]
# CSV data from subject 3
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject3_left = "/Exp20201205_2myo_softType/HaluskovaBeata_20201205_1745/myoLeftEmg.csv"
file2_subject3_left = "/Exp20201205_2myo_softType/HaluskovaBeata_20201205_1755/myoLeftEmg.csv"
file3_subject3_left = "/Exp20201205_2myo_softType/HaluskovaBeata_20201205_1810/myoLeftEmg.csv"
file4_subject3_left = "/Exp20201205_2myo_softType/HaluskovaBeata_20201205_1825/myoLeftEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject3_left_files = [file1_subject3_left, file2_subject3_left, file3_subject3_left, file4_subject3_left]
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject3_rigth = "/Exp20201205_2myo_softType/HaluskovaBeata_20201205_1745/myoRightEmg.csv"
file2_subject3_rigth = "/Exp20201205_2myo_softType/HaluskovaBeata_20201205_1755/myoRightEmg.csv"
file3_subject3_rigth = "/Exp20201205_2myo_softType/HaluskovaBeata_20201205_1810/myoRightEmg.csv"
file4_subject3_rigth = "/Exp20201205_2myo_softType/HaluskovaBeata_20201205_1825/myoRightEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject3_right_files = [file1_subject3_rigth, file2_subject3_rigth, file3_subject3_rigth, file4_subject3_rigth]
# CSV data from subject 4
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject4_left = "/Exp20201205_2myo_softType/KelisekDavid_20201209_1945/myoLeftEmg.csv"
file2_subject4_left = "/Exp20201205_2myo_softType/KelisekDavid_20201209_1955/myoLeftEmg.csv"
file3_subject4_left = "/Exp20201205_2myo_softType/KelisekDavid_20201209_2010/myoLeftEmg.csv"
file4_subject4_left = "/Exp20201205_2myo_softType/KelisekDavid_20201209_2025/myoLeftEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject4_left_files = [file1_subject4_left, file2_subject4_left, file3_subject4_left, file4_subject4_left]
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject4_rigth = "/Exp20201205_2myo_softType/KelisekDavid_20201209_1945/myoRightEmg.csv"
file2_subject4_rigth = "/Exp20201205_2myo_softType/KelisekDavid_20201209_1955/myoRightEmg.csv"
file3_subject4_rigth = "/Exp20201205_2myo_softType/KelisekDavid_20201209_2010/myoRightEmg.csv"
file4_subject4_rigth = "/Exp20201205_2myo_softType/KelisekDavid_20201209_2025/myoRightEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject4_right_files = [file1_subject4_rigth, file2_subject4_rigth, file3_subject4_rigth, file4_subject4_rigth]
# CSV data from subject 5
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject5_left = "/Exp20201205_2myo_softType/KelisekRichard_20201209_2110/myoLeftEmg.csv"
file2_subject5_left = "/Exp20201205_2myo_softType/KelisekRichard_20201209_2120/myoLeftEmg.csv"
file3_subject5_left = "/Exp20201205_2myo_softType/KelisekRichard_20201209_2130/myoLeftEmg.csv"
file4_subject5_left = "/Exp20201205_2myo_softType/KelisekRichard_20201209_2140/myoLeftEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject5_left_files = [file1_subject5_left, file2_subject5_left, file3_subject5_left, file4_subject5_left]
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
file1_subject5_rigth = "/Exp20201205_2myo_softType/KelisekRichard_20201209_2110/myoRightEmg.csv"
file2_subject5_rigth = "/Exp20201205_2myo_softType/KelisekRichard_20201209_2120/myoRightEmg.csv"
file3_subject5_rigth = "/Exp20201205_2myo_softType/KelisekRichard_20201209_2130/myoRightEmg.csv"
file4_subject5_rigth = "/Exp20201205_2myo_softType/KelisekRichard_20201209_2140/myoRightEmg.csv"
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
subject5_right_files = [file1_subject5_rigth, file2_subject5_rigth, file3_subject5_rigth, file4_subject5_rigth]
left_list = [subject1_left_files, subject2_left_files, subject3_left_files, subject4_left_files, subject5_left_files]
right_list = [subject1_right_files, subject2_right_files, subject3_right_files, subject4_right_files, subject5_right_files]
subject1_data_container = Data_container(1, 'HaluskaMarek')
subject2_data_container = Data_container(2, 'HaluskaMaros')
subject3_data_container = Data_container(3, 'HaluskovaBeata')
subject4_data_container = Data_container(4, 'KelisekDavid')
subject5_data_container = Data_container(5, 'KelisekRichard')
subject_data_container_list = [subject1_data_container, subject2_data_container, subject3_data_container,
subject4_data_container, subject5_data_container]
for subject_nr in range(5):
data_container = subject_data_container_list[subject_nr]
# left variant proccessed here
for round in range(4):
for emg_nr in range(8):
filename = left_list[subject_nr][round]
self.store_df_in_container(filename, emg_nr, 'left', data_container, round+1)
# right variant proccessed here
for round in range(4):
for emg_nr in range(8):
filename = right_list[subject_nr][round]
self.store_df_in_container(filename, emg_nr, 'right', data_container, round+1)
# Links the stored data in the data_container to the Handler
self.link_container_to_handler(data_container)
feat: complete a denoise func for part of the dataset
2021-06-25 11:41:57 +00:00
self.data_type = 'soft'
feat: add load functions for every data version
2021-06-25 09:09:36 +00:00
return self.data_container_dict
chore: move load func to handler and add data in wd
2021-06-25 07:29:23 +00:00
doc: describe the get get_df func
2021-06-25 12:00:25 +00:00
# Retrieves df via the data_dict in the handler object
feat: implement func that gets df from data_dict in handler
2021-06-25 11:57:28 +00:00
def get_df_from_data_dict(self, subject_nr, which_arm, round, emg_nr):
data_type = self.data_type
container = self.data_container_dict.get(subject_nr)
test: test the load and denoise funcs together
2021-06-25 13:28:20 +00:00
df = container.dict_list[round - 1].get(which_arm)[emg_nr - 1]
feat: implement func that gets df from data_dict in handler
2021-06-25 11:57:28 +00:00
return df
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
# Help: gets the str from emg nr
feat: functions for fft
2021-06-23 13:56:35 +00:00
def get_emg_str(emg_nr):
return 'emg' + str(emg_nr)
doc: comments
2021-06-22 19:00:51 +00:00
fix: correct time in filename in load funcs
2021-06-25 09:31:44 +00:00
# Help: gets the min/max of a df
feat: functions for fft
2021-06-23 13:56:35 +00:00
def get_min_max_timestamp(df:DataFrame):
fix: sort out error in min_max
2021-06-24 08:10:28 +00:00
min = int(np.floor(df['timestamp'].min()))
feat: add db4 wavelet func
2021-06-24 10:02:49 +00:00
max = df['timestamp'].max()
feat: functions for fft
2021-06-23 13:56:35 +00:00
return min, max
feat: make a function that gives dataframes from csv, make_df
2021-06-22 18:31:16 +00:00
feat: make df from x and y arrays
2021-06-25 12:27:51 +00:00
# Help: returns df_time_emg
def make_df_from_xandy(x, y, emg_nr):
test: test total denoising func
2021-06-25 12:44:28 +00:00
dict = {'timestamp': x, get_emg_str(emg_nr): y}
df = DataFrame(dict)
#print(df)
feat: make df from x and y arrays
2021-06-25 12:27:51 +00:00
return df
feat: initialize files for analysis Add files which lay the framework for the project
2021-06-22 13:32:51 +00:00
Reference in New Issue Copy Permalink