52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
import tempfile
|
|
import time
|
|
from pathlib import Path
|
|
|
|
import cv2
|
|
import numpy as np
|
|
|
|
import pyautogui
|
|
import shenzhen_solitaire.card_detection.configuration as configuration
|
|
import shenzhen_solitaire.clicker.main as clicker
|
|
import shenzhen_solitaire.solver.solver as solver
|
|
from shenzhen_solitaire.card_detection.board_parser import parse_board
|
|
|
|
OFFSET = (0, 0)
|
|
SIZE = (2560, 1440)
|
|
NEW_BUTTON = (1900, 1100)
|
|
|
|
|
|
def solve() -> None:
|
|
with tempfile.TemporaryDirectory() as screenshot_dir:
|
|
screenshot_file = Path(screenshot_dir) / "screenshot.png"
|
|
screenshot = pyautogui.screenshot(region=(*OFFSET, *SIZE))
|
|
screenshot.save(screenshot_file)
|
|
image = cv2.imread(str(screenshot_file))
|
|
|
|
print("Solving")
|
|
conf = configuration.load("test_config.zip")
|
|
board = parse_board(image, conf)
|
|
assert board.check_correct()
|
|
try:
|
|
solution = list(next(solver.solve(board, timeout=10)))
|
|
except StopIteration:
|
|
clicker.click(NEW_BUTTON, OFFSET)
|
|
time.sleep(10)
|
|
return
|
|
print(f"Solved in {len(solution)} steps")
|
|
for step in solution:
|
|
print(step)
|
|
clicker.handle_action(step, OFFSET, conf)
|
|
clicker.click(NEW_BUTTON, OFFSET)
|
|
time.sleep(10)
|
|
|
|
|
|
def main() -> None:
|
|
time.sleep(3)
|
|
while True:
|
|
solve()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|