Log sleep quality indicator

Author: IAmCoder

release/firmware.bin

@@ -192,7 +192,7 @@ def detect_motion(self):
             self.lsd.add_image(self.img, motion)
             self.trigger()
 
-        self.lsd.log(self.variance, motion)
+        self.lsd.log(self.variance, motion, 0)
 
     def detect_face(self):
         if self.config.get('Algorithm') != "Face Detection":
@@ -204,11 +204,13 @@ def detect_face(self):
             self.lsd.add_image(self.img, motion)
             self.trigger()
 
-        self.lsd.log(self.variance, motion)
+        self.lsd.log(self.variance, motion, 0)
 
     def detect_rem(self):
         eye_movements = self.rem.detect(self.variance, self.global_variance)
-        self.lsd.log(self.variance, eye_movements)
+        
+        if self.config.get('Algorithm') == "REM Detection":
+            self.lsd.log(self.variance, self.rem.eye_movements, self.quality.indicator)
 
         if self.eye_movements != eye_movements:
             self.eye_movements = eye_movements
@@ -228,6 +230,7 @@ def detect_nrem(self):
             return
 
         eye_movements = self.nrem.detect(self.variance)
+        self.lsd.log(self.variance, self.nrem.eye_movements, self.quality.indicator)
 
         if self.eye_movements != eye_movements:
             self.eye_movements = eye_movements

software/inspec.py

@@ -49,8 +49,9 @@ def create_lsd(self):
         self.lsd_file.close()
         self.lsd_values = "0"
         self.rem_values = "0"
+        self.sqi_values = "0"
 
-    def log(self, variance, rem):
+    def log(self, variance, rem, sqi):
         if self.config.get('CreateLogs') != 1:
             return
 
@@ -64,6 +65,7 @@ def log(self, variance, rem):
             self.write_log()
             self.lsd_values = str(variance)
             self.rem_values = str(rem)
+            self.sqi_values = str(rem)
             return
 
         if (now - self.lsd_minute_start >= 1000 * 60):
@@ -72,16 +74,19 @@ def log(self, variance, rem):
             self.write_log()
             self.lsd_values = str(variance)
             self.rem_values = str(rem)
+            self.sqi_values = str(rem)
             return
 
         self.lsd_values = f'{self.lsd_values},{str(variance)}'
         self.rem_values = f'{self.rem_values},{str(rem)}'
+        self.sqi_values = f'{self.sqi_values},{str(quality)}'
 
     def write_log(self):
         self.lsd_file = open(self.session_file, 'a')
         formatted_time = self.format_time()
         self.lsd_file.write(f'\r\n{formatted_time}:lsd - {self.lsd_values}')
         self.lsd_file.write(f'\r\n{formatted_time}:rem - {self.rem_values}')
+        self.lsd_file.write(f'\r\n{formatted_time}:sqi - {self.sqi_values}')
         self.lsd_file.close()
 
     def add_image(self, image, eye_movements):

software/lsd.py

@@ -1 +1 @@
-version = '1.2.4'
+version = '1.2.5'