import math import cv2 import logging import time import numpy as np import pywintypes import pytesseract from fishy.engine.semifisher import fishing_event from fishy.engine.IEngine import IEngine from fishy.engine.window import Window from pynput import keyboard, mouse from fishy.helper import Config, hotkey from fishy.helper.helper import wait_until from fishy.helper.hotkey import Key mse = mouse.Controller() kb = keyboard.Controller() def sign(x): return -1 if x < 0 else 1 offset = 10 def get_crop_coods(window): Window.loop() img = window.get_capture() img = cv2.inRange(img, 0, 1) Window.loop_end() cnt, h = cv2.findContours(img, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE) """ code from https://stackoverflow.com/a/45770227/4512396 """ for i in range(len(cnt)): area = cv2.contourArea(cnt[i]) if 5000 < area < 100000: mask = np.zeros_like(img) cv2.drawContours(mask, cnt, i, 255, -1) x, y, w, h = cv2.boundingRect(cnt[i]) return x, y + offset, x + w, y + h - offset def image_pre_process(img): scale_percent = 200 # percent of original size width = int(img.shape[1] * scale_percent / 100) height = int(img.shape[0] * scale_percent / 100) dim = (width, height) img = cv2.resize(img, dim, interpolation=cv2.INTER_AREA) img = cv2.bitwise_not(img) return img # noinspection PyBroadException def get_values_from_image(img, tesseract_dir): try: pytesseract.pytesseract.tesseract_cmd = tesseract_dir + '/tesseract.exe' tessdata_dir_config = f'--tessdata-dir "{tesseract_dir}" -c tessedit_char_whitelist=0123456789.' text = pytesseract.image_to_string(img, lang="eng", config=tessdata_dir_config) vals = text.split(":") return float(vals[0]), float(vals[1]), float(vals[2]) except Exception: logging.error("Couldn't read coods") return None def unassign_keys(): keys = [Key.UP, Key.RIGHT, Key.LEFT, Key.RIGHT] for k in keys: hotkey.free_key(k) class FullAuto(IEngine): rotate_by = 30 def __init__(self, config, gui_ref): super().__init__(config, gui_ref) self.factors = self.config.get("full_auto_factors", None) self._tesseract_dir = None self._target = None if self.factors is None: logging.warning("Please callibrate first") self._hole_found_flag = False self._curr_rotate_y = 0 def run(self): logging.info("Loading please wait...") self.initalize_keys() try: Window.init(False) except pywintypes.error: logging.info("Game window not found") self.toggle_start() return self.window = Window(color=cv2.COLOR_RGB2GRAY) self.window.crop = get_crop_coods(self.window) self._tesseract_dir = self.config.get("tesseract_dir", None) if self._tesseract_dir is None: logging.warning("Can't start without Tesseract Directory") self.gui.bot_started(False) self.toggle_start() return if self.get_gui is not None: self.gui.bot_started(True) while self.start: Window.loop() self.window.show("test", func=image_pre_process) Window.loop_end() self.gui.bot_started(False) unassign_keys() def get_coods(self): return get_values_from_image(self.window.processed_image(func=image_pre_process), self._tesseract_dir) def move_to(self, target): if target is None: logging.error("set target first") return if self.factors is None: logging.error("you need to callibrate first") return current = self.get_coods() print(f"Moving from {(current[0], current[1])} to {self._target}") move_vec = target[0] - current[0], target[1] - current[1] target_angle = math.degrees(math.atan2(-move_vec[1], move_vec[0])) from_angle = current[2] self.rotate_to(target_angle, from_angle) walking_time = math.sqrt(move_vec[0] ** 2 + move_vec[1] ** 2) / self.factors[0] print(f"walking for {walking_time}") kb.press('w') time.sleep(walking_time) kb.release('w') print("done") def rotate_to(self, target_angle, from_angle=None): if target_angle is None: _, _, from_angle = self.get_coods() if target_angle < 0: target_angle = 360 + target_angle target_angle += 90 while target_angle > 360: target_angle -= 360 print(f"Rotating from {from_angle} to {target_angle}") angle_diff = target_angle - from_angle if abs(angle_diff) > 180: angle_diff = (360 - abs(angle_diff)) * sign(angle_diff) * -1 rotate_times = int(angle_diff / self.factors[1]) * -1 print(f"rotate_times: {rotate_times}") for _ in range(abs(rotate_times)): mse.move(sign(rotate_times) * FullAuto.rotate_by * -1, 0) time.sleep(0.05) def look_for_hole(self): self._hole_found_flag = False def found_hole(e): if e == "look": self._hole_found_flag = True fishing_event.subscribers.append(found_hole) t = 0 while not self._hole_found_flag or t <= self.factors[2]/2: mse.move(0, FullAuto.rotate_by) time.sleep(0.05) t += 0.05 while not self._hole_found_flag or t > 0: mse.move(0, -FullAuto.rotate_by) time.sleep(0.05) t -= 0.05 self._curr_rotate_y = t fishing_event.subscribers.remove(found_hole) return self._hole_found_flag def initalize_keys(self): hotkey.set_hotkey(Key.RIGHT, lambda: logging.info(self.get_coods())) from fishy.engine.fullautofisher.calibrate import Calibrate hotkey.set_hotkey(Key.UP, Calibrate(self).start) # def down(): # t = self.get_coods()[:-1] # self.config.set("target", t) # print(f"target_coods are {t}") # hotkey.set_hotkey(Key.DOWN, down) # hotkey.set_hotkey(Key.RIGHT, lambda: self.move_to(self.config.get("target", None))) from fishy.engine.fullautofisher.recorder import Recorder from fishy.engine.fullautofisher.player import Player hotkey.set_hotkey(Key.LEFT, Recorder(self).start) hotkey.set_hotkey(Key.DOWN, Player(self).start) logging.info("STARTED") if __name__ == '__main__': logging.getLogger("").setLevel(logging.DEBUG) # noinspection PyTypeChecker bot = FullAuto(Config(), None) hotkey.initalize() bot.toggle_start()