⚡ Optimize matrix drawing with OpenCV

README.md

@@ -580,7 +580,7 @@ All charts are saved to the `charts/` folder.
 
 ### Installation
 ```bash
-pip install torch selenium numpy pandas matplotlib
+pip install torch selenium numpy pandas matplotlib opencv-python-headless
 ```
 
 ### Run training

slither_env.py

@@ -5,6 +5,7 @@
 import time
 import json
 import matplotlib.pyplot as plt
+import cv2
 # Ensure matplotlib uses a non-interactive backend for headless environments
 plt.switch_backend('Agg')
 
@@ -255,22 +256,19 @@ def _draw_circle(self, matrix, channel, cx, cy, r, value):
 
     def _draw_thick_line(self, matrix, channel, x0, y0, x1, y1, width, value):
         """Draws a thick line by interpolating circles along the segment."""
-        radius = width / 2.0
-        dist = math.hypot(x1 - x0, y1 - y0)
-
-        if dist == 0:
-            self._draw_circle(matrix, channel, x0, y0, radius, value)
-            return
-
-        # Interpolate circles
-        # Step size should be radius to ensure coverage
-        steps = int(dist / max(0.5, radius * 0.5)) + 1
-
-        for i in range(steps + 1):
-            t = i / max(1, steps)
-            x = x0 + t * (x1 - x0)
-            y = y0 + t * (y1 - y0)
-            self._draw_circle(matrix, channel, x, y, radius, value)
+        pt1 = (int(round(x0)), int(round(y0)))
+        pt2 = (int(round(x1)), int(round(y1)))
+        thickness = int(round(width))
+        if thickness < 1: thickness = 1
+
+        # Draw main line segment
+        cv2.line(matrix[channel], pt1, pt2, float(value), thickness, cv2.LINE_AA)
+
+        # Draw rounded ends (capsule shape)
+        radius = int(round(width / 2.0))
+        if radius > 0:
+            cv2.circle(matrix[channel], pt1, radius, float(value), -1, cv2.LINE_AA)
+            cv2.circle(matrix[channel], pt2, radius, float(value), -1, cv2.LINE_AA)
 
     # =====================================================
     # DEATH CLASSIFICATION (Forensic approach)