From 1f31352f020a0d7d4d98052c71d4a4f3d3b281a7 Mon Sep 17 00:00:00 2001
From: Théo de la Hogue
Date: Wed, 26 Apr 2023 12:01:12 +0200
Subject: Preventing from division by zero.

---
 src/argaze/PupilAnalysis/WorkloadIndex.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/argaze/PupilAnalysis/WorkloadIndex.py b/src/argaze/PupilAnalysis/WorkloadIndex.py
index 5ed5bee..6728595 100644
--- a/src/argaze/PupilAnalysis/WorkloadIndex.py
+++ b/src/argaze/PupilAnalysis/WorkloadIndex.py
@@ -36,7 +36,13 @@ class PupilDiameterAnalyzer(PupilFeatures.PupilDiameterAnalyzer):
 
         if ts - self.__last_ts >= self.period:
 
-            workload_index = (self.__variations_sum / self.__variations_number) / self.reference.value
+            if self.__variations_number > 0 and self.reference.value > 0.:
+
+                workload_index = (self.__variations_sum / self.__variations_number) / self.reference.value
+
+            else: 
+
+                workload_index = 0.
 
             self.__variations_sum = abs(pupil_diameter.value - self.reference.value)
             self.__variations_number = 1
-- 
cgit v1.1