From b570813c5a9f717ceee395a808ef49170bac1efd Mon Sep 17 00:00:00 2001 From: Thaloria Date: Tue, 25 Jul 2023 09:47:52 +0200 Subject: [PATCH] added async stone detection thread --- litris.py | 5 ++--- litris_stone_id_thread.py | 6 +++--- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/litris.py b/litris.py index 275f786..ce26cf9 100644 --- a/litris.py +++ b/litris.py @@ -1,5 +1,3 @@ -from copy import copy - import cv2 as cv import numpy as np from game_base_class import GameBase @@ -11,6 +9,7 @@ from litris_stone_id_thread import NewStoneID from window_capture import WindowCapture from vision import Vision from config_file import UserConfigs + class Litris(GameBase): def __init__(self, overlay): @@ -37,7 +36,7 @@ class Litris(GameBase): self.litris_reset_board = cv.imread("control_elements/sodoku_reset_button.jpg", cv.IMREAD_COLOR) - self.stone_id_thread = NewStoneID() + self.stone_id_thread = NewStoneID(self.capture_window) self.move_mode = 1 diff --git a/litris_stone_id_thread.py b/litris_stone_id_thread.py index 2c31d7e..ab1d9df 100644 --- a/litris_stone_id_thread.py +++ b/litris_stone_id_thread.py @@ -18,11 +18,11 @@ Z_FULL = [[0, 0, 0, 0], [0, 1, 1, 0], [0, 0, 1, 1], [0, 0, 0, 0]] T_FULL = [[0, 0, 0, 0], [0, 1, 1, 1], [0, 0, 1, 0], [0, 0, 0, 0]] class NewStoneID(threading.Thread): - def __init__(self): + def __init__(self, cap_win): threading.Thread.__init__(self) - self.config = UserConfigs() - self.capture_window = WindowCapture(None, None, self.config) + #self.config = UserConfigs() + self.capture_window = cap_win self.vision_stun = Vision() self.stone_coordinates = np.zeros((4, 4), dtype=object)