[fix] : when approaching the frame edges points the laser to other side

4SIR · May 16, 2024 · 8a8442e · 8a8442e
1 parent 67576ec
commit 8a8442e
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/__main__.py b/src/__main__.py
@@ -112,12 +112,22 @@
                 next_position = (current_center[0] + movement_vector[0], current_center[1] + movement_vector[1])
 
                 if next_position[0] > current_center[0]:
+                    if (cat_right_edge + 75) > right_edge:
+                        laser_x = cat_left_edge - (laser_offset_x * movement_vector[0]) - 50
+                        laser_y = cat_bottom_edge + movement_vector[1]
+                        cv2.circle(frame, (int(laser_x), int(laser_y)), 5, (255, 255, 255), -1)
+
                     laser_offset_x = 2
 
                     laser_x = cat_right_edge + (laser_offset_x * movement_vector[0]) + 50
                     laser_y = cat_bottom_edge + movement_vector[1]
                     cv2.circle(frame, (int(laser_x), int(laser_y)), 5, (255, 255, 255), -1)
-                else: 
+                else:
+                    if (cat_left_edge - 75) < left_edge:
+                        laser_x = cat_right_edge + (laser_offset_x * movement_vector[0]) + 50
+                        laser_y = cat_bottom_edge + movement_vector[1]
+                        cv2.circle(frame, (int(laser_x), int(laser_y)), 5, (255, 255, 255), -1)
+
                     laser_offset_x = -2
 
                     laser_x = cat_left_edge - (laser_offset_x * movement_vector[0]) - 50