From b057712b0f4a818997e7dbf8ae1b0495b1620ba6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?K=C3=B6ll=C5=91d=20Csaba?= <kollod.csaba@itk.ppke.hu>
Date: Wed, 4 Jan 2023 11:32:52 +0100
Subject: [PATCH] Dev: Saving to EDF file
- Now works fine
- data conversion must be checked
---
mcc-flow/merge_records.py | 14 ++++++++------
1 file changed, 8 insertions(+), 6 deletions(-)
diff --git a/mcc-flow/merge_records.py b/mcc-flow/merge_records.py
index 6916fb8..5c87696 100644
--- a/mcc-flow/merge_records.py
+++ b/mcc-flow/merge_records.py
@@ -13,7 +13,7 @@ GYRO_ACC_FS = 50
SOUND_FS = 6000
GSR_FS = 10
-EEG_EMG_SCALE = 0.04470348358154 # (mV)
+EEG_EMG_SCALE = 0.04470348358154 # (uV)
GYRO_SCALE = 0.0152587890625 # (deg/sec)
ACC_SCALE = 0.00006103515625 # (g)
@@ -100,14 +100,14 @@ class FlowPacket:
for pkt in to_save:
for i, ch_dat in enumerate(pkt.data):
print(sig)
- # edf.setPhysicalMaximum(sig, 2**7 * EEG_EMG_SCALE)
- # edf.setPhysicalMinimum(sig, -2**7 * EEG_EMG_SCALE)
+ edf.setPhysicalMaximum(sig, 3000 * EEG_EMG_SCALE)
+ edf.setPhysicalMinimum(sig, -3000 * EEG_EMG_SCALE)
# edf.setDigitalMaximum(sig, 2**7)
# edf.setDigitalMinimum(sig, -2**7)
edf.setPhysicalDimension(sig, pkt.unit)
edf.setSamplefrequency(sig, pkt.fs)
edf.setLabel(sig, f'{pkt.type}{i}')
- data_buffer.append(ch_dat)
+ data_buffer.append(np.array(ch_dat, order='C'))
sig += 1
edf.writeSamples(data_buffer)
@@ -154,6 +154,8 @@ def read_packet(file):
# extra
sound = _read_block_data(file, (6000, 1), 'h', 2).T
gsr = _read_block_data(file, (10, 1), 'h', 2).T
+ if not np.array_equal(sound, np.zeros_like(sound)):
+ sound = (sound - np.mean(sound)) / (np.max(sound) - np.min(sound))
sound = PacketData(sound, 'sound', 'unit', SOUND_FS)
gsr = PacketData(gsr, 'gsr', 'unit', GSR_FS)
# print(sound, gsr)
@@ -182,8 +184,8 @@ def read_data(path):
for file in files:
with open(file, 'rb') as f:
data = read_dat_file(f)
- # data.plot()
- data.to_edf('test.edf')
+ data.plot()
+ # data.to_edf('test.edf')
exit(12)
--
GitLab