implemented sound stimulation + new quantized model

portiloop/detection.py

@@ -66,6 +66,8 @@ class SleepSpindleRealTimeDetector(Detector):
         for idx in reversed(range(1, len(self.stride_counters))):
             self.stride_counters[idx] -= self.stride_counters[idx-1]
         assert sum(self.stride_counters) == self.seq_stride, f"{self.stride_counters} does not sum to {self.seq_stride}"
+        
+        self.h = [np.zeros((1, 7), dtype=np.int8) for _ in range(self.num_models_parallel)]
             
         self.current_stride_counter = self.stride_counters[0] - 1
         
@@ -104,20 +106,22 @@ class SleepSpindleRealTimeDetector(Detector):
             input = np.asarray(input_float) / input_scale + input_zero_point
             input = input.astype(self.input_details[0]["dtype"])
         input = input.reshape((1, 1, -1))
-
-        # FIXME: bad sequence length: 50 instead of 1:
-        # self.interpreters[idx].set_tensor(self.input_details[0]['index'], input)
-        #
-        # if self.verbose:
-        #     start_time = time.time()
-        #
-        # self.interpreters[idx].invoke()
-        #
-        # if self.verbose:
-        #     end_time = time.time()
-        # output = self.interpreters[idx].get_tensor(self.output_details[0]['index'])
-        # output_scale, output_zero_point = self.input_details[0]["quantization"]
-        # output = float(output - output_zero_point) * output_scale
+        
+        # TODO: Milo please implement this:
+#         self.interpreters[idx].set_tensor(self.input_details[0]['index'], (self.h[idx], input))
+        
+#         if self.verbose:
+#             start_time = time.time()
+        
+#         self.interpreters[idx].invoke()
+        
+#         if self.verbose:
+#             end_time = time.time()
+#         output, self.h[idx] = self.interpreters[idx].get_tensor(self.output_details[0]['index'])
+#         output_scale, output_zero_point = self.input_details[0]["quantization"]
+#         output = float(output - output_zero_point) * output_scale
+        
+        # TODO: remove this line:
         output = np.random.uniform()  # FIXME: remove
 
         if self.verbose:

portiloop/notebooks/tests.ipynb

@@ -15,6 +15,14 @@
     "\n",
     "cap = Capture(detector_cls=SleepSpindleRealTimeDetector, stimulator_cls=SleepSpindleRealTimeStimulator)"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "9bd24d7a",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {

portiloop/sounds/sample1.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:64c7aaf93d583adaae3b6942a8192709a7c342feb3db6a932f1e278a760c6037
+size 889422

portiloop/sounds/stimulus.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8730f1af623d95eacaa32a50ec117a36871cbb26960db62fe5d71b7ab96ba0a8
+size 454576

portiloop/stimulation.py

@@ -1,5 +1,8 @@
 from abc import ABC, abstractmethod
 import time
+from playsound import playsound
+from threading import Thread, Lock
+from pathlib import Path
 
 
 # Abstract interface for developers:
@@ -17,10 +20,14 @@ class Stimulator(ABC):
         raise NotImplementedError
 
         
-# Example implementation for sleep spindles:
+# Example implementation for sleep spindles
 
 class SleepSpindleRealTimeStimulator(Stimulator):
     def __init__(self):
+        self._sound = Path(__file__).parent / 'sounds' / 'stimulus.wav'
+        print(f"DEBUG:{self._sound}")
+        self._thread = None
+        self._lock = Lock()
         self.last_detected_ts = time.time()
         self.wait_t = 0.4  # 400 ms
     
@@ -29,7 +36,13 @@ class SleepSpindleRealTimeStimulator(Stimulator):
             if sig:
                 ts = time.time()
                 if ts - self.last_detected_ts > self.wait_t:
-                    print("stimulation")
-                else:
-                    print("same spindle")
+                    with self._lock:
+                        if self._thread is None:
+                            self._thread = Thread(target=self._t_sound, daemon=True)
+                            self._thread.start()
                 self.last_detected_ts = ts
+                
+    def _t_sound(self):
+        playsound(self._sound)
+        with self._lock:
+            self._thread = None