mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-16 14:31:16 +00:00
103 lines
3.8 KiB
Python
103 lines
3.8 KiB
Python
import cv2
|
|
import numpy as np
|
|
|
|
from module.base.timer import Timer
|
|
from module.base.utils import color_similarity_2d, get_color
|
|
from module.logger import logger
|
|
from tasks.base.ui import UI
|
|
from tasks.forgotten_hall.assets.assets_forgotten_hall_team import *
|
|
from tasks.forgotten_hall.assets.assets_forgotten_hall_ui import ENTER_FORGOTTEN_HALL_DUNGEON, ENTRANCE_CHECKED, SEAT_1, SEAT_2, SEAT_3, SEAT_4
|
|
|
|
|
|
class ForgottenHallTeam(UI):
|
|
def team_prepared(self):
|
|
# White button, with a color of (214, 214, 214)
|
|
color = get_color(self.device.image, ENTER_FORGOTTEN_HALL_DUNGEON.area)
|
|
return np.mean(color) > 180
|
|
|
|
def team_choose_first(self, skip_first_screenshot=True):
|
|
"""
|
|
A temporary method used to choose the first character only
|
|
"""
|
|
logger.info('Team choose first')
|
|
self.interval_clear(ENTRANCE_CHECKED)
|
|
while 1:
|
|
if skip_first_screenshot:
|
|
skip_first_screenshot = False
|
|
else:
|
|
self.device.screenshot()
|
|
|
|
if self.team_prepared():
|
|
logger.info("First character is chosen")
|
|
break
|
|
if self.appear(ENTRANCE_CHECKED, interval=2):
|
|
self.device.click(CHARACTER_1)
|
|
continue
|
|
|
|
def is_character_chosen(self, button: ButtonWrapper) -> bool:
|
|
image = color_similarity_2d(self.image_crop(button), color=(255, 255, 255))
|
|
color = cv2.mean(image)[0]
|
|
# print(button, color)
|
|
# Chosen:
|
|
# CHARACTER_1 210.0230034722222
|
|
# CHARACTER_2 210.12022569444443
|
|
# CHARACTER_3 211.09244791666666
|
|
# CHARACTER_4 210.48046875
|
|
# Not chosen
|
|
# CHARACTER_1 122.38671875
|
|
# CHARACTER_2 124.72960069444444
|
|
# CHARACTER_3 136.55989583333331
|
|
# CHARACTER_4 129.76432291666666
|
|
return color > 180
|
|
|
|
def team_choose_first_4(self, skip_first_screenshot=True):
|
|
"""
|
|
Choose the first 4 characters in list.
|
|
"""
|
|
logger.info('Team choose first 4')
|
|
self.interval_clear(ENTRANCE_CHECKED)
|
|
characters = [CHARACTER_1, CHARACTER_2, CHARACTER_3, CHARACTER_4]
|
|
seats = [SEAT_1, SEAT_2, SEAT_3, SEAT_4]
|
|
while 1:
|
|
if skip_first_screenshot:
|
|
skip_first_screenshot = False
|
|
else:
|
|
self.device.screenshot()
|
|
|
|
chosen_list = [self.is_character_chosen(c) for c in characters]
|
|
seat_list = [not self.appear(s) for s in seats]
|
|
if all(chosen_list):
|
|
logger.info("First 4 characters are chosen")
|
|
break
|
|
if all(seat_list):
|
|
logger.info("4 characters are chosen")
|
|
break
|
|
if self.appear(ENTRANCE_CHECKED, interval=2):
|
|
for character, chosen in zip(characters, chosen_list):
|
|
if not chosen:
|
|
self.device.click(character)
|
|
# Casual sleep, game may not respond that fast
|
|
self.device.sleep((0.1, 0.2))
|
|
|
|
def team_is_prepared(self, skip_first_screenshot=True) -> bool:
|
|
"""
|
|
Pages:
|
|
in: ENTRANCE_CHECKED, ENTER_FORGOTTEN_HALL_DUNGEON
|
|
"""
|
|
seats = [SEAT_1, SEAT_2, SEAT_3, SEAT_4]
|
|
timeout = Timer(1, count=5).start()
|
|
while 1:
|
|
if skip_first_screenshot:
|
|
skip_first_screenshot = False
|
|
else:
|
|
self.device.screenshot()
|
|
|
|
if timeout.reached():
|
|
logger.info('Team not prepared')
|
|
return False
|
|
if self.team_prepared():
|
|
chosen_list = [not self.appear(s) for s in seats]
|
|
if all(chosen_list):
|
|
logger.info("Team already prepared")
|
|
return True
|