diff --git a/control_elements/sodoku_reset_button.jpg b/control_elements/sodoku_reset_button.jpg new file mode 100644 index 0000000..aea3a23 Binary files /dev/null and b/control_elements/sodoku_reset_button.jpg differ diff --git a/sodoku.py b/sodoku.py index 93e0b62..03bd80a 100644 --- a/sodoku.py +++ b/sodoku.py @@ -26,6 +26,8 @@ class Sodoku(GameBase): self.fill_data_coordinates() + self.sd_reset_board = cv.imread("control_elements/sodoku_reset_button.jpg", cv.IMREAD_COLOR) + self.needles = {1: cv.imread("sodoku/1.jpg", cv.IMREAD_COLOR), 2: cv.imread("sodoku/2.jpg", cv.IMREAD_COLOR), 3: cv.imread("sodoku/3.jpg", cv.IMREAD_COLOR), @@ -59,8 +61,8 @@ class Sodoku(GameBase): def assess_playfield_and_make_move(self): new_observation, new_screenshot = self.get_current_board_state() - screenshot = self.capture_window.get_screenshot() - if self.check_for_button_and_execute(screenshot, self.reset_board): + + if self.check_for_button_and_execute(self.capture_window.get_screenshot(), self.sd_reset_board): cv.waitKey(500) self.find_patterns_and_valid_moves(new_observation)