StarRailCopilot/tasks/base/daemon.py

124 lines
4.5 KiB
Python

from module.base.timer import Timer
from module.daemon.daemon_base import DaemonBase
from module.device.method import maatouch
from module.logger import logger
from tasks.base.assets.assets_base_daemon import *
from tasks.base.main_page import MainPage
from tasks.base.page import page_main, page_rogue
from tasks.daily.assets.assets_daily_camera import PICTURE_TAKEN
from tasks.map.assets.assets_map_bigmap import TELEPORT_RIGHT
from tasks.map.interact.aim import AimDetectorMixin
from tasks.rogue.route.base import RouteBase
class SecondaryMaatouchBuilder(maatouch.MaatouchBuilder):
def __init__(self, device, contact=0, handle_orientation=False):
"""
Click on secondary contact to avoid interruption of real-person contact
"""
super().__init__(device, contact=1, handle_orientation=handle_orientation)
maatouch.MaatouchBuilder = SecondaryMaatouchBuilder
class Daemon(RouteBase, DaemonBase, AimDetectorMixin):
aim_interval = Timer(0.3, count=1)
def handle_aim_click(self, item=True, enemy=True):
"""
Args:
item:
enemy:
Returns:
bool: If clicked
"""
if not item and not enemy:
return False
if not self.is_in_main():
return False
if self.aim_interval.reached_and_reset():
self.aim.predict(self.device.image, item=item, enemy=enemy)
if self.aim.aimed_enemy:
if self.handle_map_A():
return True
if self.aim.aimed_item:
if self.handle_map_A():
return True
return False
def run(self):
# Rebind daemon settings along with rogue settings
self.config.bind('Daemon', func_list=['Rogue'])
# Check contact
builder = self.device.maatouch_builder
if builder.contact >= 1:
logger.info(f'Maatouch contact on {builder.contact}')
else:
logger.warning(f'Maatouch contact on {builder.contact}, may cause interruptions')
STORY_OPTION.set_search_offset((-5, -10, 32, 5))
INTERACT_COLLECT.set_search_offset((-5, -5, 32, 5))
INTERACT_INVESTIGATE.set_search_offset((-5, -5, 32, 5))
INTERACT_TREASURE.set_search_offset((-5, -5, 32, 5))
teleport_confirm = Timer(1, count=5)
while 1:
self.device.screenshot()
# Check lang
if not MainPage._lang_checked and self.ui_page_appear(page_main, interval=5):
self.handle_lang_check(page=page_main)
# Check again
if not MainPage._lang_check_success:
MainPage._lang_checked = False
# Story
if self.appear_then_click(STORY_NEXT, interval=0.7):
self.interval_reset(STORY_OPTION)
# self.interval_reset(INTERACT_INVESTIGATE)
continue
if self.appear_then_click(STORY_OPTION, interval=1):
# self.interval_reset(INTERACT_INVESTIGATE)
continue
# Map interact
if self.appear_then_click(INTERACT_TREASURE, interval=1):
continue
# if self.appear_then_click(INTERACT_INVESTIGATE, interval=2):
# continue
if self.appear_then_click(INTERACT_COLLECT, interval=1):
continue
# Story teleport
if self.appear_then_click(TELEPORT_RIGHT, interval=3):
teleport_confirm.reset()
continue
if teleport_confirm.started() and not teleport_confirm.reached():
if self.handle_popup_confirm():
logger.info(f'{TELEPORT_RIGHT} -> popup')
continue
# Chat
if self.appear_then_click(CHAT_OPTION, interval=3):
continue
if self.appear_then_click(CHAT_CLOSE, interval=3):
continue
# Popup
if self.handle_reward(interval=1.5):
continue
if self.handle_ui_close(PICTURE_TAKEN, interval=1):
continue
# Rogue
if self.handle_blessing():
continue
if self.ui_page_appear(page_rogue):
if self.handle_event_continue():
continue
if self.handle_event_option():
continue
# Aim click
if self.handle_aim_click(
item='item' in self.config.Daemon_AimClicker,
enemy='enemy' in self.config.Daemon_AimClicker,
):
continue