diff --git a/menti_words.py b/menti_words.py index 8017e79..4222dcb 100644 --- a/menti_words.py +++ b/menti_words.py @@ -1,7 +1,7 @@ import cv2 as cv import numpy as np import pydirectinput - +from nltk.corpus import words from utils import mse from game_base_class import GameBase import random @@ -51,9 +51,30 @@ class MentiWords(GameBase): 'W': cv.imread("menti_words/w.jpg", cv.IMREAD_COLOR) + } + self.masks = { + 'A': cv.imread("menti_words/a-mask.png", cv.IMREAD_COLOR), + 'B': cv.imread("menti_words/b-mask.png", cv.IMREAD_COLOR), + 'C': cv.imread("menti_words/c-mask.png", cv.IMREAD_COLOR), + 'D': cv.imread("menti_words/d-mask.png", cv.IMREAD_COLOR), + 'E': cv.imread("menti_words/e-mask.png", cv.IMREAD_COLOR), + 'F': cv.imread("menti_words/f-mask.png", cv.IMREAD_COLOR), + 'H': cv.imread("menti_words/h-mask.png", cv.IMREAD_COLOR), + 'I': cv.imread("menti_words/i-mask.png", cv.IMREAD_COLOR), + 'L': cv.imread("menti_words/l-mask.png", cv.IMREAD_COLOR), + 'M': cv.imread("menti_words/m-mask.png", cv.IMREAD_COLOR), + 'N': cv.imread("menti_words/n-mask.png", cv.IMREAD_COLOR), + 'O': cv.imread("menti_words/o-mask.png", cv.IMREAD_COLOR), + 'P': cv.imread("menti_words/p-mask.png", cv.IMREAD_COLOR), + 'R': cv.imread("menti_words/r-mask.png", cv.IMREAD_COLOR), + 'S': cv.imread("menti_words/s-mask.png", cv.IMREAD_COLOR), + 'T': cv.imread("menti_words/t-mask.png", cv.IMREAD_COLOR), + 'U': cv.imread("menti_words/u-mask.png", cv.IMREAD_COLOR), + + 'W': cv.imread("menti_words/w-mask.png", cv.IMREAD_COLOR) } @@ -62,6 +83,7 @@ class MentiWords(GameBase): with open("menti_dic") as file: self.word_list = [line.rstrip() for line in file] + self.word_list2 = words.words() def reset_lists(self): self.current_letters = [] @@ -98,17 +120,18 @@ class MentiWords(GameBase): def get_current_board_state(self): - #screenshot = cv.imread("menti_words/screenshot.jpg") - screenshot = self.capture_window.get_screenshot() + screenshot = cv.imread("menti_words/screenshot.jpg") + #screenshot = self.capture_window.get_screenshot() screenshot = screenshot[870:1270, 1080:1480] - #cv.imshow("screenshot", screenshot) - #cv.waitKey(150) + cv.imshow("screenshot", screenshot) + cv.waitKey(150) #continue for needle_key in self.needles.keys(): # gray_needle = cv.cvtColor(self.needles[needle_key], cv.COLOR_BGR2GRAY) # thresh_needle = cv.threshold(gray_needle, 0, 255, cv.THRESH_BINARY_INV + cv.THRESH_OTSU)[1] - rectangles = self.vision_stun.find(screenshot, self.needles[needle_key], 0.85, 56) + #rectangles = self.vision_stun.find(screenshot, self.needles[needle_key], 0.85, 56) + rectangles = self.vision_stun.find(screenshot, self.needles[needle_key], 0.95, 1 ,True, self.masks[needle_key]) if len(rectangles) == 0: continue points = self.vision_stun.get_click_points(rectangles) diff --git a/menti_words/a-mask.png b/menti_words/a-mask.png new file mode 100644 index 0000000..c20db02 Binary files /dev/null and b/menti_words/a-mask.png differ diff --git a/menti_words/a.png b/menti_words/a.png new file mode 100644 index 0000000..464b978 Binary files /dev/null and b/menti_words/a.png differ diff --git a/menti_words/b-mask.png b/menti_words/b-mask.png new file mode 100644 index 0000000..c80815e Binary files /dev/null and b/menti_words/b-mask.png differ diff --git a/menti_words/b.png b/menti_words/b.png new file mode 100644 index 0000000..b357260 Binary files /dev/null and b/menti_words/b.png differ diff --git a/menti_words/c-mask.png b/menti_words/c-mask.png new file mode 100644 index 0000000..0954571 Binary files /dev/null and b/menti_words/c-mask.png differ diff --git a/menti_words/c.png b/menti_words/c.png new file mode 100644 index 0000000..4448707 Binary files /dev/null and b/menti_words/c.png differ diff --git a/menti_words/d-mask.png b/menti_words/d-mask.png new file mode 100644 index 0000000..bbf0b16 Binary files /dev/null and b/menti_words/d-mask.png differ diff --git a/menti_words/d.png b/menti_words/d.png new file mode 100644 index 0000000..02f2115 Binary files /dev/null and b/menti_words/d.png differ diff --git a/menti_words/e-mask.png b/menti_words/e-mask.png new file mode 100644 index 0000000..cba70b1 Binary files /dev/null and b/menti_words/e-mask.png differ diff --git a/menti_words/e.png b/menti_words/e.png new file mode 100644 index 0000000..8515ad2 Binary files /dev/null and b/menti_words/e.png differ diff --git a/menti_words/f-mask.png b/menti_words/f-mask.png new file mode 100644 index 0000000..c946c9b Binary files /dev/null and b/menti_words/f-mask.png differ diff --git a/menti_words/f.png b/menti_words/f.png new file mode 100644 index 0000000..1513108 Binary files /dev/null and b/menti_words/f.png differ diff --git a/menti_words/h-mask.png b/menti_words/h-mask.png new file mode 100644 index 0000000..9e89209 Binary files /dev/null and b/menti_words/h-mask.png differ diff --git a/menti_words/h.png b/menti_words/h.png new file mode 100644 index 0000000..d80ea1c Binary files /dev/null and b/menti_words/h.png differ diff --git a/menti_words/i-mask.png b/menti_words/i-mask.png new file mode 100644 index 0000000..f7b55bf Binary files /dev/null and b/menti_words/i-mask.png differ diff --git a/menti_words/i.png b/menti_words/i.png new file mode 100644 index 0000000..f672401 Binary files /dev/null and b/menti_words/i.png differ diff --git a/menti_words/l-mask.png b/menti_words/l-mask.png new file mode 100644 index 0000000..c0d36ee Binary files /dev/null and b/menti_words/l-mask.png differ diff --git a/menti_words/l.png b/menti_words/l.png new file mode 100644 index 0000000..ff1c3c8 Binary files /dev/null and b/menti_words/l.png differ diff --git a/menti_words/m-mask.png b/menti_words/m-mask.png new file mode 100644 index 0000000..ea86f22 Binary files /dev/null and b/menti_words/m-mask.png differ diff --git a/menti_words/m.png b/menti_words/m.png new file mode 100644 index 0000000..04a8154 Binary files /dev/null and b/menti_words/m.png differ diff --git a/menti_words/n-mask.png b/menti_words/n-mask.png new file mode 100644 index 0000000..9fda997 Binary files /dev/null and b/menti_words/n-mask.png differ diff --git a/menti_words/n.png b/menti_words/n.png new file mode 100644 index 0000000..d3acd19 Binary files /dev/null and b/menti_words/n.png differ diff --git a/menti_words/o-mask.png b/menti_words/o-mask.png new file mode 100644 index 0000000..9e1babc Binary files /dev/null and b/menti_words/o-mask.png differ diff --git a/menti_words/o.png b/menti_words/o.png new file mode 100644 index 0000000..b39749d Binary files /dev/null and b/menti_words/o.png differ diff --git a/menti_words/p-mask.png b/menti_words/p-mask.png new file mode 100644 index 0000000..4cbb6cb Binary files /dev/null and b/menti_words/p-mask.png differ diff --git a/menti_words/p.png b/menti_words/p.png new file mode 100644 index 0000000..756b631 Binary files /dev/null and b/menti_words/p.png differ diff --git a/menti_words/r-mask.png b/menti_words/r-mask.png new file mode 100644 index 0000000..1f27b67 Binary files /dev/null and b/menti_words/r-mask.png differ diff --git a/menti_words/r.png b/menti_words/r.png new file mode 100644 index 0000000..3c4cb32 Binary files /dev/null and b/menti_words/r.png differ diff --git a/menti_words/s-mask.png b/menti_words/s-mask.png new file mode 100644 index 0000000..2c5c060 Binary files /dev/null and b/menti_words/s-mask.png differ diff --git a/menti_words/s.png b/menti_words/s.png new file mode 100644 index 0000000..7bcd13e Binary files /dev/null and b/menti_words/s.png differ diff --git a/menti_words/t-mask.png b/menti_words/t-mask.png new file mode 100644 index 0000000..11bde80 Binary files /dev/null and b/menti_words/t-mask.png differ diff --git a/menti_words/t.png b/menti_words/t.png new file mode 100644 index 0000000..ef55ad1 Binary files /dev/null and b/menti_words/t.png differ diff --git a/menti_words/u-mask.png b/menti_words/u-mask.png new file mode 100644 index 0000000..a6ebd73 Binary files /dev/null and b/menti_words/u-mask.png differ diff --git a/menti_words/u.png b/menti_words/u.png new file mode 100644 index 0000000..9828692 Binary files /dev/null and b/menti_words/u.png differ diff --git a/menti_words/w-mask.png b/menti_words/w-mask.png new file mode 100644 index 0000000..8c8584e Binary files /dev/null and b/menti_words/w-mask.png differ diff --git a/menti_words/w.png b/menti_words/w.png new file mode 100644 index 0000000..8caac39 Binary files /dev/null and b/menti_words/w.png differ