From 6367261165623246537d3b361cb4c3e935a203c9 Mon Sep 17 00:00:00 2001
From: Paolo Zaffino <p.zaffino@unicz.it>
Date: Wed, 31 Jan 2024 18:16:12 +0000
Subject: [PATCH] Toward disconnection managment, v3

---
 ArduinoConnect/ArduinoConnect.py | 24 +++++++++++++++++++-----
 1 file changed, 19 insertions(+), 5 deletions(-)

diff --git a/ArduinoConnect/ArduinoConnect.py b/ArduinoConnect/ArduinoConnect.py
index c4e4d56..137398b 100644
--- a/ArduinoConnect/ArduinoConnect.py
+++ b/ArduinoConnect/ArduinoConnect.py
@@ -25,6 +25,19 @@ def deviceError(title, message, error_type="warning"):
     deviceMBox.setText(message)
     deviceMBox.exec()
 
+def connectionError():
+    connectionMBox = qt.QMessageBox()
+    connectionMBox.setIcon(qt.QMessageBox().Critical)
+    connectionMBox.setWindowTitle("Connection dropped!")
+    connectionMBox.setText("Do you want to try reconnecting?")
+    connectionMBox.setStandardButtons(qt.QMessageBox.Yes)
+    connectionMBox.addButton(qt.QMessageBox.No)
+    connectionMBox.setDefaultButton(qt.QMessageBox.Yes)
+    if connectionMBox.exec() == qt.QMessageBox.Yes:
+      return True
+    else:
+      return False
+
 #
 # ArduinoAppTemplate
 #
@@ -363,7 +376,6 @@ def __init__(self):
     slicer.mrmlScene.AddNode(self.parameterNode)
 
     self.arduinoConnection = None
-    self.disconnectedByUser = False
 
   def sendMessage(self, messageToSend):
     if self.arduinoConnection is not None:
@@ -384,6 +396,7 @@ def connect(self, port=None, baud=None, samplesPerSecond=None):
 
     try:
       self.arduinoConnection = serial.Serial(self.port, self.baud)
+      self.disconnectedByUser = False
     except serial.serialutil.SerialException:
       return False
 
@@ -419,11 +432,12 @@ def pollSerialDevice(self):
     except (IOError, AttributeError):
       self.disconnect(disconnectedByUser=False)
 
-      deviceError("Critical error", "Connection has dropped!\nClick OK to try connect again", "critical")
-      reconnected = self.connect()
+      userWantsToReconnect = connectionError()
+      if userWantsToReconnect:
+        reconnected = self.connect()
 
-      if not reconnected:
-        qt.QTimer.singleShot(1000/self.arduinoRefreshRateFps, self.pollSerialDevice)
+        if not reconnected:
+          qt.QTimer.singleShot(1000/self.arduinoRefreshRateFps, self.pollSerialDevice)
 
   def processMessage(self, msg):
     return msg