added new lsl marker for stimulation and fixed small bug

portiloop/capture.py

@@ -1323,8 +1323,7 @@ class Capture:
 
         buffer = []
 
-        if not self.spindle_detection_mode == 'Fast':
-            print('here')
+        if not self.spindle_detection_mode == 'Fast' and stimulator is not None:
             stimulation_delayer = UpStateDelayer(self.frequency, self.spindle_freq, self.spindle_detection_mode == 'Peak')
             stimulator.add_delayer(stimulation_delayer)
         else:
@@ -1370,7 +1369,7 @@ class Capture:
                 
             if stimulation_delayer is not None:
                 stimulation_delayer.add_point(point[channel-1])
-            
+
             with self._pause_detect_lock:
                 pause = self._pause_detect
             if detector is not None and not pause:

portiloop/stimulation.py

@@ -46,7 +46,16 @@ class SleepSpindleRealTimeStimulator(Stimulator):
                                   channel_count=1,
                                   channel_format='string',
                                   source_id='portiloop1')  # TODO: replace this by unique device identifier
+        
+        lsl_markers_info_fast = pylsl.StreamInfo(name='Portiloop_stimuli_fast',
+                                  type='Markers',
+                                  channel_count=1,
+                                  channel_format='string',
+                                  source_id='portiloop1')  # TODO: replace this by unique device identifier
+        
         self.lsl_outlet_markers = pylsl.StreamOutlet(lsl_markers_info)
+        self.lsl_outlet_markers_fast = pylsl.StreamOutlet(lsl_markers_info_fast)
+
         self.delayer = None
         
         # Initialize Alsa stuff
@@ -102,6 +111,7 @@ class SleepSpindleRealTimeStimulator(Stimulator):
             # We detect a stimulation
             elif sig:
                 # Record time of stimulation
+                self.lsl_outlet_markers_fast.push_sample(['FASTSTIM'])
                 ts = time.time()
                 
                 # Prompt delayer to try and get a stimulation