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