Spaces:
Sleeping
Sleeping
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
|