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.BoundingTrainer(img, bounding_area=self.predefined.main_area)
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 = look_up_translation_correlation(corr)
self.root.debug(
"Looking for Download Button, {} CORRERLATION".format(corrword))
img = self.get_img_from_screen_shot()
t = tm.BoundingTrainer(img, 500, 300, 600, 300)
location = os.path.join(self.assets, "download_button.png")
return self.__wrapper_kmeans_result__(t, location, corr, info)
def test_scan_for_download(self):
img = os.path.join(self.provider.assets, "steam",
"download_update.png")
t = tm.BoundingTrainer(img, 500, 300, 600, 300)
location = os.path.join(self.provider.assets, "download_button.png")
# t.show_area(500, 300, 600, 300, img)
self.assertTrue(
t.get_matches(location, 3) is True, "Expecting a download button")
def test_compare_with_back_button(self):
img = os.path.join(self.provider.assets, "steam", "steam_back.png")
t = tm.BoundingTrainer(
img, bounding_area=self.provider.predefined.main_area)
location = os.path.join(self.provider.assets, "back__.png")
# t.show_area_bounded(self.provider.predefined.main_area, img)
# t._debug = True
self.assertTrue(
t.get_matches(location, 3) is True, "Expecting a back button")
def test_scan_for_close(self):
img = os.path.join(self.provider.assets, "steam", "steam_close.png")
area = self.provider.predefined.main_area
area['width'] = 400
t = tm.BoundingTrainer(img, bounding_area=area)
location = os.path.join(self.provider.assets, "close.png")
# t.show_area_bounded(self.provider.predefined.main_area, img)
# t._debug = True
self.assertTrue(
t.get_matches(location, 3) is True, "Expecting a back button")
img = os.path.join(self.provider.assets, "steam", "steam_ok.png")
t = tm.BoundingTrainer(img, bounding_area=area)
location = os.path.join(self.provider.assets, "close.png")
t.get_matches(location, 3)
# t.show_area_bounded(self.provider.predefined.main_area, img)
# t._debug = True
self.assertTrue(
t.get_matches(location, 3) is False, "Is Ok button not close")
def scan_for_close(self, corr=HIGH_CORR, info=None, img=None):
corrword = look_up_translation_correlation(corr)
self.root.debug("LOOKING FOR CLOSE BUTTON, {} CORRERLATION".format(corrword))
if img is None:
img = self.get_img_from_screen_shot()
area = self.predefined.main_area
area['width'] = 400
t = tm.BoundingTrainer(img, bounding_area=area)
location = os.path.join(self.assets, "close.png")
return self.__wrapper_kmeans_result__(t, location, corr, info)
def possible_battle_points(self):
if self.run_time.stop:
self.root.info("Received Stopping signal")
return
img = self.get_img_from_screen_shot()
area = self.predefined.main_area
area['height'] = 800
t = tm.BoundingTrainer(img, bounding_area=area)
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