Spaces:
Build error
Build error
Commit ·
a5c0535
1
Parent(s): ed407e2
Fixed EDFRecorder issues
Browse files- portiloop/src/capture.py +2 -1
- portiloop/src/utils.py +5 -4
portiloop/src/capture.py
CHANGED
|
@@ -953,7 +953,8 @@ class Capture:
|
|
| 953 |
|
| 954 |
# Initialize recording if requested
|
| 955 |
if record:
|
| 956 |
-
recorder = EDFRecorder(self.
|
|
|
|
| 957 |
|
| 958 |
# Initialize LSL to stream if requested
|
| 959 |
if lsl:
|
|
|
|
| 953 |
|
| 954 |
# Initialize recording if requested
|
| 955 |
if record:
|
| 956 |
+
recorder = EDFRecorder(self.signal_labels, self.filename, self.frequency)
|
| 957 |
+
recorder.open_recording_file()
|
| 958 |
|
| 959 |
# Initialize LSL to stream if requested
|
| 960 |
if lsl:
|
portiloop/src/utils.py
CHANGED
|
@@ -23,13 +23,14 @@ class DummyAlsaMixer:
|
|
| 23 |
|
| 24 |
|
| 25 |
class EDFRecorder:
|
| 26 |
-
def __init__(self, signal_labels):
|
| 27 |
-
self.filename =
|
| 28 |
-
self.nb_signals =
|
| 29 |
-
self.samples_per_datarecord_array =
|
| 30 |
self.physical_max = 5
|
| 31 |
self.physical_min = -5
|
| 32 |
self.signal_labels = signal_labels
|
|
|
|
| 33 |
|
| 34 |
def open_recording_file(self):
|
| 35 |
nb_signals = self.nb_signals
|
|
|
|
| 23 |
|
| 24 |
|
| 25 |
class EDFRecorder:
|
| 26 |
+
def __init__(self, signal_labels, filename, frequency):
|
| 27 |
+
self.filename = filename
|
| 28 |
+
self.nb_signals = len(signal_labels)
|
| 29 |
+
self.samples_per_datarecord_array = frequency
|
| 30 |
self.physical_max = 5
|
| 31 |
self.physical_min = -5
|
| 32 |
self.signal_labels = signal_labels
|
| 33 |
+
self.edf_buffer = []
|
| 34 |
|
| 35 |
def open_recording_file(self):
|
| 36 |
nb_signals = self.nb_signals
|