Thaloria/Litcraft_Python_B
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

added first draft litris

Browse Source
This commit is contained in:
Thaloria
2023-07-23 17:45:37 +02:00
parent c4f71f469b
commit 6d6081ec57
1 changed files with 25 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
litris.py
View File

@@ -28,17 +28,29 @@ class Litris(GameBase):
#self.sd_reset_board = cv.imread("control_elements/sodoku_reset_button.jpg", cv.IMREAD_COLOR) #self.sd_reset_board = cv.imread("control_elements/sodoku_reset_button.jpg", cv.IMREAD_COLOR)
self.needles = {"I": cv.imread("litris/I.jpg", cv.IMREAD_COLOR), self.needles = {"I": cv.imread("litris/I.jpg", cv.IMREAD_UNCHANGED),
"O": cv.imread("litris/O.jpg", cv.IMREAD_COLOR), "O": cv.imread("litris/O.jpg", cv.IMREAD_UNCHANGED),
"L": cv.imread("litris/L.jpg", cv.IMREAD_COLOR), "L": cv.imread("litris/L.jpg", cv.IMREAD_UNCHANGED),
"J": cv.imread("litris/J.jpg", cv.IMREAD_COLOR), "J": cv.imread("litris/J.jpg", cv.IMREAD_UNCHANGED),
"S": cv.imread("litris/S.jpg", cv.IMREAD_COLOR), "S": cv.imread("litris/S.jpg", cv.IMREAD_UNCHANGED),
"Z": cv.imread("litris/Z.jpg", cv.IMREAD_COLOR), "Z": cv.imread("litris/Z.jpg", cv.IMREAD_UNCHANGED),
"T": cv.imread("litris/T.jpg", cv.IMREAD_COLOR), "T": cv.imread("litris/T.jpg", cv.IMREAD_UNCHANGED),
"D": cv.imread("litris/D.jpg", cv.IMREAD_COLOR), "D": cv.imread("litris/D.jpg", cv.IMREAD_UNCHANGED),
"A": cv.imread("litris/A.jpg", cv.IMREAD_COLOR), "A": cv.imread("litris/A.jpg", cv.IMREAD_UNCHANGED),
"B": cv.imread("litris/B.jpg", cv.IMREAD_COLOR), "B": cv.imread("litris/B.jpg", cv.IMREAD_UNCHANGED),
"C": cv.imread("litris/C.jpg", cv.IMREAD_COLOR) "C": cv.imread("litris/C.jpg", cv.IMREAD_UNCHANGED)
}
self.needle_masks = {"I": cv.imread("litris/I-mask.png", cv.IMREAD_COLOR),
"O": cv.imread("litris/O-mask.png", cv.IMREAD_COLOR),
"L": cv.imread("litris/L-mask.png", cv.IMREAD_COLOR),
"J": cv.imread("litris/J-mask.png", cv.IMREAD_COLOR),
"S": cv.imread("litris/S-mask.png", cv.IMREAD_COLOR),
"Z": cv.imread("litris/Z-mask.png", cv.IMREAD_COLOR),
"T": cv.imread("litris/T-mask.png", cv.IMREAD_COLOR),
"D": cv.imread("litris/D-mask.png", cv.IMREAD_COLOR),
"A": cv.imread("litris/A-mask.png", cv.IMREAD_COLOR),
"B": cv.imread("litris/B-mask.png", cv.IMREAD_COLOR),
"C": cv.imread("litris/C-mask.png", cv.IMREAD_COLOR)
} }
def reset_field(self): def reset_field(self):
@@ -126,7 +138,8 @@ class Litris(GameBase):
#cv.imshow("screenshot", screenshot) #cv.imshow("screenshot", screenshot)
#cv.waitKey(150) #cv.waitKey(150)
for needle_key in self.needles.keys(): for needle_key in self.needles.keys():
rectangles = self.vision_stun.find(screenshot, self.needles[needle_key], 0.85, 1) #rectangles = self.vision_stun.find(screenshot, self.needles[needle_key], 0.85, 1)
rectangles = self.vision_stun.find(screenshot, self.needles[needle_key], 0.85, 1, False, self.needle_masks[needle_key])
if len(rectangles) == 0: if len(rectangles) == 0:
if self.overlay.run_mode == 'stopped' or self.overlay.run_mode == 'paused': if self.overlay.run_mode == 'stopped' or self.overlay.run_mode == 'paused':
return return