mirror of
https://github.com/fishyboteso/fishyboteso.git
synced 2024-08-30 18:32:13 +00:00
90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
import time
|
|
import typing
|
|
from threading import Thread
|
|
from typing import Callable
|
|
from typing import Optional
|
|
|
|
import cv2
|
|
import logging
|
|
|
|
from fishy.engine.semifisher.fishing_event import FishEvent
|
|
|
|
from fishy.engine.common.window import WindowClient
|
|
from fishy.engine.semifisher.fishing_mode import FishingMode
|
|
|
|
from fishy.engine.common.IEngine import IEngine
|
|
from fishy.engine.semifisher import fishing_mode, fishing_event
|
|
from fishy.engine.semifisher.pixel_loc import PixelLoc
|
|
|
|
if typing.TYPE_CHECKING:
|
|
from fishy.gui import GUI
|
|
|
|
|
|
class SemiFisherEngine(IEngine):
|
|
def __init__(self, gui_ref: Optional['Callable[[], GUI]']):
|
|
super().__init__(gui_ref)
|
|
self.fishPixWindow = None
|
|
|
|
def run(self):
|
|
"""
|
|
Starts the fishing
|
|
code explained in comments in detail
|
|
"""
|
|
fishing_event.init()
|
|
self.fishPixWindow = WindowClient(color=cv2.COLOR_RGB2HSV)
|
|
|
|
# check for game window and stuff
|
|
self.gui.bot_started(True)
|
|
|
|
if self.get_gui:
|
|
logging.info("Starting the bot engine, look at the fishing hole to start fishing")
|
|
Thread(target=self._wait_and_check).start()
|
|
while self.start and WindowClient.running():
|
|
capture = self.fishPixWindow.get_capture()
|
|
|
|
if capture is None:
|
|
# if window server crashed
|
|
self.gui.bot_started(False)
|
|
self.toggle_start()
|
|
continue
|
|
|
|
self.fishPixWindow.crop = PixelLoc.val
|
|
fishing_mode.loop(capture[0][0])
|
|
|
|
logging.info("Fishing engine stopped")
|
|
self.gui.bot_started(False)
|
|
fishing_event.unsubscribe()
|
|
self.fishPixWindow.destory()
|
|
|
|
def _wait_and_check(self):
|
|
time.sleep(10)
|
|
if not FishEvent.FishingStarted and self.start:
|
|
logging.warn("Doesn't look like fishing has started \nCheck out #read-me-first on our discord channel to troubleshoot the issue")
|
|
|
|
def show_pixel_vals(self):
|
|
def show():
|
|
freq = 0.5
|
|
t = 0
|
|
while t < 10.0:
|
|
t += freq
|
|
logging.debug(str(FishingMode.CurrentMode) + ":" + str(self.fishPixWindow.get_capture()[0][0]))
|
|
time.sleep(freq)
|
|
|
|
logging.debug("Will display pixel values for 10 seconds")
|
|
time.sleep(5)
|
|
Thread(target=show, args=()).start()
|
|
|
|
def toggle_start(self):
|
|
self.start = not self.start
|
|
if self.start:
|
|
self.thread = Thread(target=self.run)
|
|
self.thread.start()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
logging.getLogger("").setLevel(logging.DEBUG)
|
|
# noinspection PyTypeChecker
|
|
fisher = SemiFisherEngine(None)
|
|
fisher.toggle_start()
|
|
|