Add data collector that saves to csv

Author: PerlinWarp

.gitignore

@@ -7,4 +7,6 @@ dist/*
 .venv
 .env/
 Pipfile
-*.egg-info
+*.egg-info
+
+*.csv

data_collector.py

@@ -0,0 +1,58 @@
+# Simplistic data recording
+import time
+import multiprocessing
+import numpy as np
+import pandas as pd
+
+from pyomyo import Myo, emg_mode
+
+def data_worker(mode=emg_mode.FILTERED, seconds=15, filepath="data_gather.csv"):
+	collect = True
+
+	# ------------ Myo Setup ---------------
+	m = Myo(mode=mode)
+	m.connect()
+
+	myo_data = []
+
+	def add_to_queue(emg, movement):
+		myo_data.append(emg)
+
+	m.add_emg_handler(add_to_queue)
+
+	def print_battery(bat):
+		print("Battery level:", bat)
+
+	m.add_battery_handler(print_battery)
+
+	 # Its go time
+	m.set_leds([0, 128, 0], [0, 128, 0])
+	# Vibrate to know we connected okay
+	m.vibrate(1)
+
+	print("Data Worker started to collect")
+	# Start collecing data.
+	start_time = time.time()
+
+	while collect:
+		if (time.time() - start_time < seconds):
+			m.run()
+		else:
+			collect = False
+			collection_time = time.time() - start_time
+			print("Finished collecting.")
+			print(f"Collection time: {collection_time}")
+			print(len(myo_data), "frames collected")
+
+			# Add columns and save to df
+			myo_cols = ["Channel_1", "Channel_2", "Channel_3", "Channel_4", "Channel_5", "Channel_6", "Channel_7", "Channel_8"]
+			myo_df = pd.DataFrame(myo_data, columns=myo_cols)
+			myo_df.to_csv(filepath, index=False)
+			print("CSV Saved at: ", filepath)
+
+# -------- Main Program Loop -----------
+if __name__ == '__main__':
+	seconds = 10
+	file_name = str(seconds)+"_test_emg.csv"
+	p = multiprocessing.Process(target=data_worker, args=(seconds, file_name))
+	p.start()