mirror of
https://github.com/fishyboteso/fishyboteso.git
synced 2024-08-30 18:32:13 +00:00
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
import logging
|
|
|
|
from fishy.helper import hotkey
|
|
|
|
from fishy.engine.fullautofisher.engine import FullAuto, State
|
|
from fishy.helper.hotkey import Key
|
|
|
|
|
|
def get_controls(engine: FullAuto):
|
|
from fishy.engine.fullautofisher.recorder import Recorder
|
|
from fishy.engine.fullautofisher.player import Player
|
|
controls = [
|
|
("MODE_SELECT", {
|
|
Key.RIGHT: (Recorder(engine).toggle_recording, "start/stop record"),
|
|
Key.UP: (engine.calibrator.calibrate, "calibrate mode"),
|
|
Key.LEFT: (Player(engine).toggle_move, "start/stop play"),
|
|
Key.DOWN: (lambda: engine.controls.select_mode("TEST1"), "test mode"),
|
|
}),
|
|
("TEST1", {
|
|
Key.RIGHT: (engine.test.print_coods, "print coordinates"),
|
|
Key.UP: (engine.test.look_for_hole, "look for hole up down"),
|
|
Key.LEFT: (engine.test.set_target, "set target"),
|
|
Key.DOWN: (engine.test.move_to_target, "move to target")
|
|
})
|
|
]
|
|
|
|
return controls
|
|
|
|
|
|
class Controls:
|
|
def __init__(self, controls, first=0):
|
|
self.current_menu = first - 1
|
|
self.controls = controls
|
|
|
|
def initialize(self):
|
|
self.select_mode(self.controls[0][0])
|
|
|
|
def log_help(self):
|
|
help_str = f"\nCONTROLS: {self.controls[self.current_menu][0]}"
|
|
for key, meta in self.controls[self.current_menu][1].items():
|
|
func, name = meta
|
|
if func:
|
|
hotkey.set_hotkey(key, func)
|
|
help_str += f"\n{key.value}: {name}"
|
|
logging.info(help_str)
|
|
|
|
def select_mode(self, mode):
|
|
if FullAuto.state != State.NONE:
|
|
self.log_help()
|
|
return
|
|
|
|
self.current_menu = 0
|
|
for i, control in enumerate(self.controls):
|
|
if mode == control[0]:
|
|
self.current_menu = i
|
|
self.log_help()
|
|
|
|
def unassign_keys(self):
|
|
keys = []
|
|
for c in self.controls:
|
|
for k in c[1].keys():
|
|
if k not in keys:
|
|
hotkey.free_key(k)
|