From 69fe7cbec4e4cad30338b0980ccc2a55db8f2c72 Mon Sep 17 00:00:00 2001
From: Thaloria <thaloria@web.de>
Date: Sun, 22 May 2022 13:19:36 +0200
Subject: [PATCH] added timed loop exit to equip

---
 equipment_main.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/equipment_main.py b/equipment_main.py
index aad9e4a..feb1052 100644
--- a/equipment_main.py
+++ b/equipment_main.py
@@ -42,7 +42,7 @@ def run():
     # initialize the StunVision class
     vision_stun = Vision()
 
-    loop_time = time()
+    outer_loop_time = time()
 
     needles = []
     hsvs = []
@@ -147,7 +147,10 @@ def run():
 
             cv.waitKey(500)
             for rer in range(0, len(needles), 1):
+                loop_time = time()
                 while True:
+                    if (time() - loop_time) >= 60:
+                        break
                     screenshot = capture_window.get_screenshot_by_area(config.returnOKWindowPos())
                     rectangles = vision_stun.find(screenshot, cv.imread("dig/ok_button.jpg", cv.IMREAD_UNCHANGED), 0.5,
                                                   1)
@@ -209,8 +212,8 @@ def run():
                         pause = True
                         break
             # debug the loop rate
-            print('FPS {}'.format(1 / (time() - loop_time)))
-            loop_time = time()
+            print('FPS {}'.format(1 / (time() - outer_loop_time)))
+            outer_loop_time = time()
             cv.waitKey(150)