def test_get_matches(self):
location = os.path.join(base, "ok_box.png")
base_location = os.path.join(base, "nox", "vagabond.png")
black_screen = os.path.join(base, "nox", "black_screen.png")
assert os.path.exists(location)
assert os.path.exists(base_location)
assert os.path.exists(black_screen)
base_img = cv2.imread(base_location)
trainer = tm.Trainer(base_img, 480, 50)
self.assertTrue(trainer.get_matches(location, LOW_CORR) is False)
base_img = cv2.imread(black_screen)
trainer = tm.Trainer(base_img, 480, 50)
self.assertTrue(trainer.get_matches(location, LOW_CORR) is False)
def compare_with_cancel_button(self, corr=HIGH_CORR, info=None, img=None):
corrword = 'HIGH' if corr == HIGH_CORR else 'LOW'
self.root.debug("LOOKING FOR CANCEL BUTTON, {} CORRERLATION".format(corrword))
img = self.get_img_from_screen_shot()
t = tm.Trainer(img, 240, 0)
location = os.path.join(self.assets, "cancel_button.png")
return self.__wrapper_kmeans_result__(t, location, corr, info)
def scan_for_download(self, corr=HIGH_CORR, info=None):
corrword = 'HIGH' if corr == HIGH_CORR else 'LOW'
self.root.debug("Looking for Download Button, {} CORRERLATION".format(corrword))
img = self.get_img_from_screen_shot()
t = tm.Trainer(img, 480, 0)
location = os.path.join(self.assets, "download_button.png")
return self.__wrapper_kmeans_result__(t, location, corr, info)
def scan_for_retry(self, corr=HIGH_CORR, info=None, img=None):
corrword = 'HIGH' if corr == HIGH_CORR else 'LOW'
self.root.debug("LOOKING FOR RETRY BUTTON, {} CORRERLATION".format(corrword))
if img is None:
img = self.get_img_from_screen_shot()
t = tm.Trainer(img, 400, 500)
location = os.path.join(self.assets, "retry.png")
return self.__wrapper_kmeans_result__(t, location, corr, info)
def scan_for_ok(self, corr=HIGH_CORR, info=None, img=None):
corrword = look_up_translation_correlation(corr)
self.root.debug("LOOK FOR WORD '{}', {} CORRERLATION".format('OK', corrword))
self.root.debug("scan_for_ok()")
if img is None:
img = self.get_img_from_screen_shot()
t = tm.Trainer(img, 480, 50)
location = os.path.join(self.assets, "ok_box.png")
return self.__wrapper_kmeans_result__(t, location, corr, info)
def possible_battle_points(self):
img = self.get_img_from_screen_shot()
t = tm.Trainer(img)
t.whiteCircles()
current_page = self.get_current_page(img)
logging.debug("Current-Page {}".format(current_page))
for x, y in t.circlePoints:
if not getattr(self.current_thread, "do_run", True):
break
yield x, y, current_page
def possible_battle_points(self):
if self.run_time.stop:
self.root.info("Received Stopping signal")
return
img = self.get_img_from_screen_shot()
t = tm.Trainer(img)
t.capture_white_circles()
current_page = self.get_current_page(img)
logging.debug("Current-Page {}".format(current_page))
for x, y in t.circlePoints:
if self.run_time.stop:
self.root.info("Received Stopping signal")
break
yield x, y, current_page