46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
from .context import shenzhen_solitaire
|
|
from shenzhen_solitaire.cv import adjustment
|
|
from shenzhen_solitaire.cv import card_finder
|
|
from typing import Tuple, List, Dict
|
|
import cv2
|
|
import numpy
|
|
import itertools
|
|
|
|
A = cv2.imread("Solitaire.png")
|
|
|
|
|
|
def pixelcount(image: numpy.ndarray) -> List[Tuple[Tuple[int, int, int], int]]:
|
|
p: Dict[Tuple[int, int, int], int] = {(0, 0, 0): 0}
|
|
for pixel in itertools.chain.from_iterable(image):
|
|
x = tuple(pixel)
|
|
if x in p:
|
|
p[x] += 1
|
|
else:
|
|
p[x] = 1
|
|
B = sorted(p.items(), key=lambda x: x[1])
|
|
return B
|
|
|
|
def simplify(image: numpy.ndarray) -> None:
|
|
cv2.imshow("Window", image)
|
|
cv2.waitKey(0)
|
|
cv2.destroyWindow("Window")
|
|
print(*card_finder.simplify(image).items(), sep='\n')
|
|
cv2.imshow("Window", image)
|
|
cv2.waitKey(0)
|
|
cv2.destroyWindow("Window")
|
|
|
|
|
|
def main() -> None:
|
|
adj = adjustment.adjust_field(A)
|
|
image_squares = card_finder.get_field_squares(A, adj)
|
|
for img in image_squares:
|
|
print(*pixelcount(img), sep='\n')
|
|
cv2.imshow("Window", img)
|
|
cv2.waitKey(0)
|
|
cv2.destroyWindow("Window")
|
|
print()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|