Upgrade interactivity of the Visualizer.

Author: RobinEnjalbert

src/Core/Rendering/backend/BaseVisualizer.py

@@ -92,6 +92,7 @@ def launch_server(self,
         clients = {}
         for _ in range(nb_clients):
             client, _ = self.server.accept()
+            client.settimeout(0.1)
             idx_client: int = unpack('i', client.recv(4))[0]
             clients[idx_client] = client
 
@@ -194,15 +195,18 @@ def listen_client(self,
         """
 
         while not self.is_done[idx_client]:
-            msg = self.clients[idx_client].recv(4)
-            if len(msg) == 0:
+            try:
+                msg = self.clients[idx_client].recv(4)
+                if len(msg) == 0:
+                    pass
+                elif msg == b'exit':
+                    self.is_done[idx_client] = True
+                    self.exit(force_quit=False)
+                else:
+                    step = unpack('i', msg)[0]
+                    self.requests.append((idx_client, step))
+            except socket.timeout:
                 pass
-            elif msg == b'exit':
-                self.is_done[idx_client] = True
-                self.exit(force_quit=False)
-            else:
-                step = unpack('i', msg)[0]
-                self.requests.append((idx_client, step))
 
     def update_actors(self,
                       step: int,

src/Core/Rendering/backend/Open3d/Open3dVisualizer.py

@@ -1,3 +1,4 @@
+import socket
 from typing import Dict, Optional, Tuple
 from threading import Thread
 from struct import unpack
@@ -115,20 +116,22 @@ def single_client_thread(self):
         """
 
         while not self.is_done[0]:
-
-            msg = self.clients[0].recv(4)
-            if len(msg) == 0:
+            try:
+                msg = self.clients[0].recv(4)
+                if len(msg) == 0:
+                    pass
+                elif msg == b'exit':
+                    self.is_done[0] = True
+                    self.exit(force_quit=False)
+                else:
+                    self.__step = (0, unpack('i', msg)[0])
+                    process_time = time()
+                    gui.Application.instance.post_to_main_thread(self._window,
+                                                                 self.update_visualizer)
+                    dt = max(0., self.fps - (time() - process_time))
+                    sleep(dt)
+            except socket.timeout:
                 pass
-            elif msg == b'exit':
-                self.is_done[0] = True
-                self.exit(force_quit=False)
-            else:
-                self.__step = (0, unpack('i', msg)[0])
-                process_time = time()
-                gui.Application.instance.post_to_main_thread(self._window,
-                                                             self.update_visualizer)
-                dt = max(0., self.fps - (time() - process_time))
-                sleep(dt)
 
         gui.Application.instance.quit()
 

src/Core/Rendering/backend/Vedo/VedoVisualizer.py

@@ -1,3 +1,4 @@
+import socket
 from typing import Dict, Optional
 from struct import unpack
 from vedo import show, Plotter
@@ -112,16 +113,19 @@ def single_client_thread(self, _) -> None:
         Timer callback for a single Factory.
         """
 
-        msg = self.clients[0].recv(4)
-        if len(msg) == 0:
+        try:
+            msg = self.clients[0].recv(4)
+            if len(msg) == 0:
+                pass
+            elif msg == b'exit':
+                self.is_done[0] = True
+                self.exit(force_quit=False)
+            else:
+                step = unpack('i', msg)[0]
+                self.update_visualizer(step=step, idx_factory=0)
+                self.clients[0].send(b'done')
+        except socket.timeout:
             pass
-        elif msg == b'exit':
-            self.is_done[0] = True
-            self.exit(force_quit=False)
-        else:
-            step = unpack('i', msg)[0]
-            self.update_visualizer(step=step, idx_factory=0)
-            self.clients[0].send(b'done')
 
     def multiple_clients_thread(self, _) -> None:
         """