MiloSobral commited on
Commit
a5c0535
·
1 Parent(s): ed407e2

Fixed EDFRecorder issues

Browse files
portiloop/src/capture.py CHANGED
@@ -953,7 +953,8 @@ class Capture:
953
 
954
  # Initialize recording if requested
955
  if record:
956
- recorder = EDFRecorder(self.signal_label)
 
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 = EDF_PATH / 'recording.edf'
28
- self.nb_signals = 8
29
- self.samples_per_datarecord_array = self.frequency
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