diff --git a/litris.py b/litris.py index 38bb524..352b913 100644 --- a/litris.py +++ b/litris.py @@ -29,7 +29,7 @@ class Litris(GameBase): self.field = Field() self.field_mem = Field() - #self.sd_reset_board = cv.imread("control_elements/sodoku_reset_button.jpg", cv.IMREAD_COLOR) + self.litris_reset_board = cv.imread("control_elements/sodoku_reset_button.jpg", cv.IMREAD_COLOR) self.stone_id_thread = NewStoneID() @@ -50,6 +50,10 @@ class Litris(GameBase): while True: if self.stone_id_thread.get_pick_up_status() == False: + screenshot = self.capture_window.get_screenshot() + screenshot = screenshot[880:1060, 1400:1600] + if self.check_for_button_and_execute(screenshot, self.litris_reset_board, 1400, 880): + self.field.reset_field() if self.overlay.run_mode == 'stopped' or self.overlay.run_mode == 'paused': return cv.waitKey(50)