From 5a127abbae6d2ae3fb96cd993da7b6bf8ee0c917 Mon Sep 17 00:00:00 2001
From: Skudalen <markusskudal@gmail.com>
Date: Wed, 23 Jun 2021 11:17:59 +0200
Subject: [PATCH] feat: add class as a data_container for each subject

---
 Handle_emg_data.py | 18 ++++++++++++++++--
 Signal_prep.py     |  3 ++-
 2 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/Handle_emg_data.py b/Handle_emg_data.py
index caca294..cdbf412 100644
--- a/Handle_emg_data.py
+++ b/Handle_emg_data.py
@@ -1,10 +1,20 @@
 import pandas as pd
 from pathlib import Path
 
+class Data_container:
+      
+    def __init__(self):
+        self.subject_nr
+        self.subject_name
+        self.data_dict
+    
+    
+
 class CSV_handler:
 
     def __init__(self):
         self.working_dir = str(Path.cwd())
+        self.data_container_dict = {i: []}
 
     # Makes dataframe from the csv files in the working directory
     def make_df(self, filename):
@@ -13,16 +23,20 @@ class CSV_handler:
         return df
 
     # Extracts out the timestamp and the selected emg signal into a new dataframe 
-    def get_time_emg_table(self, filename, emg_nr):
+    def get_time_emg_table(self, filename: str, subject_nr: int, which_arm: str, emg_nr: int):
         tot_data_frame = self.make_df(filename)
         emg_str = 'emg' + str(emg_nr)
         filtered_df = tot_data_frame[["timestamp", emg_str]]
+
+        #self.data_dict[subject_nr] = [which_arm, emg1]
+
         return filtered_df
     
     def get_emg_str(emg_nr):
         return 'emg' + str(emg_nr)
 
-    
+    def get_min_max_timestamp():
+
 
 
 
diff --git a/Signal_prep.py b/Signal_prep.py
index 62eeb3e..1ed6391 100644
--- a/Signal_prep.py
+++ b/Signal_prep.py
@@ -1,5 +1,6 @@
 import numpy as np 
-
+from Handle_emg_data import CSV_handler
+import matplotlib.pyplot as plt