95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
"""Contains functions to find significant pieces of a solitaire screenshot"""
|
|
|
|
import itertools
|
|
from dataclasses import dataclass
|
|
from typing import Optional, Tuple
|
|
|
|
import cv2
|
|
import numpy
|
|
import math
|
|
|
|
|
|
@dataclass
|
|
class Adjustment:
|
|
"""Configuration for a grid"""
|
|
|
|
x: int = 0
|
|
y: int = 0
|
|
w: int = 0
|
|
h: int = 0
|
|
dx: float = 0
|
|
dy: float = 0
|
|
|
|
|
|
def get_square(
|
|
adjustment: Adjustment, index_x: int = 0, index_y: int = 0
|
|
) -> Tuple[int, int, int, int]:
|
|
"""Get one square from index and adjustment"""
|
|
return (
|
|
math.floor(adjustment.x + adjustment.dx * index_x),
|
|
math.floor(adjustment.y + adjustment.dy * index_y),
|
|
math.floor(adjustment.x + adjustment.w + adjustment.dx * index_x),
|
|
math.floor(adjustment.y + adjustment.h + adjustment.dy * index_y),
|
|
)
|
|
|
|
|
|
def adjust_squares(
|
|
image: numpy.ndarray,
|
|
count_x: int,
|
|
count_y: int,
|
|
adjustment: Optional[Adjustment] = None,
|
|
) -> Adjustment:
|
|
|
|
if not adjustment:
|
|
adjustment = Adjustment(w=10, h=10)
|
|
speed_mod = "n"
|
|
speed_mods = ["n", "s", "h"]
|
|
|
|
def _adjustment_step(keycode: int, speed_mod: str) -> None:
|
|
assert adjustment is not None
|
|
x_keys = {104: -1, 115: +1}
|
|
y_keys = {116: -1, 110: +1}
|
|
w_keys = {97: -1, 117: +1}
|
|
h_keys = {111: -1, 101: +1}
|
|
dx_keys = {59: -1, 112: +1}
|
|
dy_keys = {44: -1, 46: +1}
|
|
speed_facs = {"n": 1, "s": 8, "h": 64}
|
|
cur_high_speed_fac = speed_facs[speed_mod]
|
|
if keycode in x_keys:
|
|
adjustment.x += x_keys[keycode] * cur_high_speed_fac
|
|
elif keycode in y_keys:
|
|
adjustment.y += y_keys[keycode] * cur_high_speed_fac
|
|
elif keycode in w_keys:
|
|
adjustment.w += w_keys[keycode] * cur_high_speed_fac
|
|
elif keycode in h_keys:
|
|
adjustment.h += h_keys[keycode] * cur_high_speed_fac
|
|
elif keycode in dx_keys:
|
|
adjustment.dx += dx_keys[keycode] * cur_high_speed_fac * 1 / 8
|
|
elif keycode in dy_keys:
|
|
adjustment.dy += dy_keys[keycode] * cur_high_speed_fac * 1 / 8
|
|
|
|
cv2.namedWindow("Window", flags=cv2.WINDOW_NORMAL)
|
|
|
|
while True:
|
|
working_image = image.copy()
|
|
for index_x, index_y in itertools.product(range(count_x), range(count_y)):
|
|
square = get_square(adjustment, index_x, index_y)
|
|
cv2.rectangle(
|
|
working_image,
|
|
(math.floor(square[0]), math.floor(square[1])),
|
|
(math.floor(square[2]), math.floor(square[3])),
|
|
(0, 0, 0),
|
|
)
|
|
cv2.imshow("Window", working_image)
|
|
keycode = cv2.waitKey(0)
|
|
print(keycode)
|
|
if keycode == 27:
|
|
break
|
|
if keycode == 229:
|
|
speed_mod = speed_mods[(speed_mods.index(speed_mod) + 1) % len(speed_mods)]
|
|
continue
|
|
_adjustment_step(keycode, speed_mod)
|
|
|
|
cv2.destroyWindow("Window")
|
|
return adjustment
|