ADD warning if trying to load data from tcp/ws listener without starting the server first

Author: eleniv3d

src/gh/components/DF_tcp_listener/code.py

@@ -114,15 +114,20 @@ def stop_server():
 
         # Load buffered points into Rhino PointCloud
         if i_load and not sc.sticky[f'{prefix}_prev_load']:
-            raw = sc.sticky.get(f'{prefix}_cloud_buffer_raw', [])
-            if raw:
-                pc = rg.PointCloud()
-                for x, y, z, r, g, b in raw:
-                    pc.Add(rg.Point3d(x, y, z), sd.Color.FromArgb(int(r), int(g), int(b)))
-                sc.sticky[f'{prefix}_latest_cloud'] = pc
-                sc.sticky[f'{prefix}_status_message'] = f'Loaded pcd with {pc.Count} pts'
+            if not sc.sticky.get(f'{prefix}_server_started', False):
+                self.AddRuntimeMessage(self.RuntimeMessageLevel.Warning,
+                                       "Please start server here before trying to send data from remote device.")
+                sc.sticky[f'{prefix}_status_message'] = "Server not started"
             else:
-                sc.sticky[f'{prefix}_status_message'] = 'No data buffered'
+                raw = sc.sticky.get(f'{prefix}_cloud_buffer_raw', [])
+                if raw:
+                    pc = rg.PointCloud()
+                    for x, y, z, r, g, b in raw:
+                        pc.Add(rg.Point3d(x, y, z), sd.Color.FromArgb(int(r), int(g), int(b)))
+                    sc.sticky[f'{prefix}_latest_cloud'] = pc
+                    sc.sticky[f'{prefix}_status_message'] = f'Loaded pcd with {pc.Count} pts'
+                else:
+                    sc.sticky[f'{prefix}_status_message'] = 'No data buffered'
 
         # Update previous states
         sc.sticky[f'{prefix}_prev_start'] = i_start

src/gh/components/DF_websocket_listener/code.py

@@ -107,10 +107,15 @@ def run_server():
 
         # LOAD buffered PCD on i_load rising edge
         if i_load and not sc.sticky[f'{prefix}_prev_load']:
-            sc.sticky[f'{prefix}_loaded_pcd'] = sc.sticky.get(f'{prefix}_last_pcd')
-            cnt = len(sc.sticky[f'{prefix}_loaded_pcd']) if sc.sticky[f'{prefix}_loaded_pcd'] else 0
-            logs.append(f"Loaded pcd with {cnt} pts")
-            ghenv.Component.ExpireSolution(True)  # noqa: F821
+            if not sc.sticky.get(f'{prefix}_server'):
+                self.AddRuntimeMessage(self.RuntimeMessageLevel.Warning,
+                    "Please start server here before trying to send data from remote device.")
+                logs.append("Server not started")
+            else:
+                sc.sticky[f'{prefix}_loaded_pcd'] = sc.sticky.get(f'{prefix}_last_pcd')
+                cnt = len(sc.sticky[f'{prefix}_loaded_pcd']) if sc.sticky[f'{prefix}_loaded_pcd'] else 0
+                logs.append(f"Loaded pcd with {cnt} pts")
+                ghenv.Component.ExpireSolution(True)  # noqa: F821
 
         # BUILD output PointCloud
         raw = sc.sticky.get(f'{prefix}_loaded_pcd')
@@ -127,7 +132,7 @@ def run_server():
         # UPDATE UI message & return outputs
         ghenv.Component.Message = logs[-1] if logs else 'Waiting..'  # noqa: F821
         sc.sticky[f'{prefix}_prev_start'] = i_start
-        sc.sticky[f'{prefix}_prev_stop']  = i_stop
-        sc.sticky[f'{prefix}_prev_load']  = i_load
+        sc.sticky[f'{prefix}_prev_stop'] = i_stop
+        sc.sticky[f'{prefix}_prev_load'] = i_load
 
         return [o_cloud]