mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-27 02:27:12 +00:00
267 lines
9.1 KiB
Python
267 lines
9.1 KiB
Python
import re
|
|
from collections.abc import Iterator
|
|
from datetime import timedelta
|
|
from enum import Enum
|
|
from functools import cached_property
|
|
|
|
from module.base.timer import Timer
|
|
from module.exception import ScriptError
|
|
from module.logger import logger
|
|
from module.ocr.ocr import DigitCounter, Duration, Ocr
|
|
from tasks.dungeon.ui import DungeonTabSwitch as Switch
|
|
from module.ui.draggable_list import DraggableList
|
|
from tasks.assignment.assets.assets_assignment_claim import CLAIM
|
|
from tasks.assignment.assets.assets_assignment_dispatch import EMPTY_SLOT
|
|
from tasks.assignment.assets.assets_assignment_ui import *
|
|
from tasks.assignment.keywords import *
|
|
from tasks.base.ui import UI
|
|
|
|
|
|
class AssignmentStatus(Enum):
|
|
CLAIMABLE = 0
|
|
DISPATCHED = 1
|
|
DISPATCHABLE = 2
|
|
LOCKED = 3
|
|
|
|
|
|
class AssignmentOcr(Ocr):
|
|
# EN has names in multiple rows
|
|
merge_thres_y = 20
|
|
merge_thres_x = 20
|
|
|
|
OCR_REPLACE = {
|
|
'cn': [
|
|
(KEYWORDS_ASSIGNMENT_ENTRY.Winter_Soldiers.name, '[黑]冬的战士们'),
|
|
(KEYWORDS_ASSIGNMENT_ENTRY.Born_to_Obey.name, '[牛]而服从'),
|
|
(KEYWORDS_ASSIGNMENT_ENTRY.Root_Out_the_Turpitude.name,
|
|
'根除恶[擎薯尊掌鞋]?'),
|
|
(KEYWORDS_ASSIGNMENT_ENTRY.Akashic_Records.name, '阿[未][夏复]记录'),
|
|
(KEYWORDS_ASSIGNMENT_ENTRY.Legend_of_the_Puppet_Master.name, '^师传说'),
|
|
(KEYWORDS_ASSIGNMENT_ENTRY.The_Wages_of_Humanity.name, '[赠]养人类'),
|
|
],
|
|
'en': [
|
|
(KEYWORDS_ASSIGNMENT_EVENT_ENTRY.Food_Improvement_Plan.name,
|
|
'Food\s*[I]{0}mprovement Plan'),
|
|
]
|
|
}
|
|
|
|
@cached_property
|
|
def ocr_regex(self) -> re.Pattern | None:
|
|
rules = AssignmentOcr.OCR_REPLACE.get(self.lang)
|
|
if rules is None:
|
|
return None
|
|
return re.compile('|'.join('(?P<%s>%s)' % pair for pair in rules))
|
|
|
|
def filter_detected(self, result) -> bool:
|
|
# Drop duration rows
|
|
res = Duration.timedelta_regex(self.lang).search(result.ocr_text)
|
|
if res.group('hours') or res.group('seconds'):
|
|
return False
|
|
# Locked event assignments
|
|
locked_pattern = {
|
|
'cn': '解锁$',
|
|
'en': 'Locked$',
|
|
}[self.lang]
|
|
res = re.search(locked_pattern, result.ocr_text)
|
|
return not res
|
|
|
|
def after_process(self, result: str):
|
|
result = super().after_process(result)
|
|
|
|
if self.ocr_regex is None:
|
|
return result
|
|
matched = self.ocr_regex.fullmatch(result)
|
|
if matched is None:
|
|
return result
|
|
keyword_lang = self.lang
|
|
for keyword_class in (
|
|
KEYWORDS_ASSIGNMENT_ENTRY,
|
|
KEYWORDS_ASSIGNMENT_EVENT_ENTRY,
|
|
):
|
|
matched = getattr(keyword_class, matched.lastgroup, None)
|
|
if matched is not None:
|
|
break
|
|
else:
|
|
raise ScriptError(f'No keyword found for {matched.lastgroup}')
|
|
matched = getattr(matched, keyword_lang)
|
|
logger.attr(name=f'{self.name} after_process',
|
|
text=f'{result} -> {matched}')
|
|
return matched
|
|
|
|
|
|
ASSIGNMENT_GROUP_SWITCH = Switch(
|
|
'AssignmentGroupSwitch',
|
|
is_selector=True
|
|
)
|
|
ASSIGNMENT_GROUP_SWITCH.add_state(
|
|
KEYWORDS_ASSIGNMENT_EVENT_GROUP.Space_Station_Task_Force,
|
|
check_button=SPACE_STATION_TASK_FORCE_CHECK,
|
|
click_button=SPACE_STATION_TASK_FORCE_CLICK
|
|
)
|
|
ASSIGNMENT_GROUP_SWITCH.add_state(
|
|
KEYWORDS_ASSIGNMENT_GROUP.Character_Materials,
|
|
check_button=CHARACTER_MATERIALS_CHECK,
|
|
click_button=CHARACTER_MATERIALS_CLICK
|
|
)
|
|
ASSIGNMENT_GROUP_SWITCH.add_state(
|
|
KEYWORDS_ASSIGNMENT_GROUP.EXP_Materials_Credits,
|
|
check_button=EXP_MATERIALS_CREDITS_CHECK,
|
|
click_button=EXP_MATERIALS_CREDITS_CLICK
|
|
)
|
|
ASSIGNMENT_GROUP_SWITCH.add_state(
|
|
KEYWORDS_ASSIGNMENT_GROUP.Synthesis_Materials,
|
|
check_button=SYNTHESIS_MATERIALS_CHECK,
|
|
click_button=SYNTHESIS_MATERIALS_CLICK
|
|
)
|
|
ASSIGNMENT_ENTRY_LIST = DraggableList(
|
|
'AssignmentEntryList',
|
|
keyword_class=[AssignmentEntry, AssignmentEventEntry],
|
|
ocr_class=AssignmentOcr,
|
|
search_button=OCR_ASSIGNMENT_ENTRY_LIST,
|
|
check_row_order=False,
|
|
active_color=(40, 40, 40)
|
|
)
|
|
ASSIGNMENT_ENTRY_LIST.known_rows = [
|
|
kw for kc in ASSIGNMENT_ENTRY_LIST.keyword_class
|
|
for kw in kc.instances.values()
|
|
]
|
|
|
|
|
|
class AssignmentUI(UI):
|
|
def goto_group(self, group: AssignmentGroup):
|
|
"""
|
|
Args:
|
|
group (AssignmentGroup):
|
|
|
|
Examples:
|
|
self = AssignmentUI('src')
|
|
self.device.screenshot()
|
|
self.goto_group(KEYWORDS_ASSIGNMENT_GROUP.Character_Materials)
|
|
"""
|
|
if ASSIGNMENT_GROUP_SWITCH.get(self) == group:
|
|
return
|
|
logger.hr('Assignment group goto', level=3)
|
|
if ASSIGNMENT_GROUP_SWITCH.set(group, self):
|
|
self._wait_until_entry_loaded()
|
|
|
|
def goto_entry(self, entry: AssignmentEntry, insight: bool = True):
|
|
"""
|
|
Args:
|
|
entry (AssignmentEntry):
|
|
insight (bool): skip ocr to save time if insight is False
|
|
|
|
Examples:
|
|
self = AssignmentUI('src')
|
|
self.device.screenshot()
|
|
self.goto_entry(KEYWORDS_ASSIGNMENT_ENTRY.Nameless_Land_Nameless_People)
|
|
"""
|
|
if entry.group is None:
|
|
err_msg = f'{entry} is not in any group, please inform developers if possible'
|
|
logger.warning(err_msg)
|
|
for group in self._iter_groups():
|
|
self.goto_group(group)
|
|
if ASSIGNMENT_ENTRY_LIST.select_row(entry, self):
|
|
return
|
|
raise ScriptError(err_msg)
|
|
else:
|
|
self.goto_group(entry.group)
|
|
ASSIGNMENT_ENTRY_LIST.select_row(entry, self, insight=insight)
|
|
|
|
def _wait_until_group_loaded(self):
|
|
skip_first_screenshot = True
|
|
timeout = Timer(2, count=3).start()
|
|
while 1:
|
|
if skip_first_screenshot:
|
|
skip_first_screenshot = False
|
|
else:
|
|
self.device.screenshot()
|
|
|
|
if timeout.reached():
|
|
logger.warning('Wait group loaded timeout')
|
|
break
|
|
if self.image_color_count(GROUP_SEARCH, (40, 40, 40), count=25000) and \
|
|
self.image_color_count(GROUP_SEARCH, (240, 240, 240), count=8000):
|
|
logger.info('Group loaded')
|
|
break
|
|
|
|
def _wait_until_entry_loaded(self):
|
|
skip_first_screenshot = True
|
|
timeout = Timer(2, count=3).start()
|
|
while 1:
|
|
if skip_first_screenshot:
|
|
skip_first_screenshot = False
|
|
else:
|
|
self.device.screenshot()
|
|
|
|
if timeout.reached():
|
|
logger.warning('Wait entry loaded timeout')
|
|
break
|
|
if self.image_color_count(ENTRY_LOADED, (35, 35, 35), count=800):
|
|
logger.info('Entry loaded')
|
|
break
|
|
|
|
@property
|
|
def _limit_status(self) -> tuple[int, int, int]:
|
|
self.device.screenshot()
|
|
current, remain, total = DigitCounter(
|
|
OCR_ASSIGNMENT_LIMIT).ocr_single_line(self.device.image)
|
|
if total and current <= total:
|
|
logger.attr('Assignment', f'{current}/{total}')
|
|
self.config.stored.Assignment.set(current, total)
|
|
else:
|
|
logger.warning(f'Invalid assignment limit: {current}/{total}')
|
|
self.config.stored.Assignment.set(0, 0)
|
|
return current, remain, total
|
|
|
|
def _check_assignment_status(self) -> AssignmentStatus:
|
|
skip_first_screenshot = True
|
|
timeout = Timer(2, count=3).start()
|
|
ret = AssignmentStatus.LOCKED
|
|
while 1:
|
|
if skip_first_screenshot:
|
|
skip_first_screenshot = False
|
|
else:
|
|
self.device.screenshot()
|
|
|
|
if timeout.reached():
|
|
logger.warning(
|
|
'Check assignment status timeout, assume LOCKED'
|
|
)
|
|
break
|
|
if self.appear(CLAIM):
|
|
ret = AssignmentStatus.CLAIMABLE
|
|
break
|
|
if self.appear(DISPATCHED):
|
|
ret = AssignmentStatus.DISPATCHED
|
|
break
|
|
if self.appear(EMPTY_SLOT):
|
|
ret = AssignmentStatus.DISPATCHABLE
|
|
break
|
|
if self.appear(LOCKED):
|
|
ret = AssignmentStatus.LOCKED
|
|
break
|
|
logger.attr('AssignmentStatus', ret.name)
|
|
return ret
|
|
|
|
def _get_assignment_time(self) -> timedelta:
|
|
return Duration(OCR_ASSIGNMENT_TIME).ocr_single_line(self.device.image)
|
|
|
|
def _iter_groups(self) -> Iterator[AssignmentGroup]:
|
|
self._wait_until_group_loaded()
|
|
for state in ASSIGNMENT_GROUP_SWITCH.state_list:
|
|
check = state['check_button']
|
|
click = state['click_button']
|
|
if self.appear(check) or self.appear(click):
|
|
yield state['state']
|
|
|
|
def _iter_entries(self) -> Iterator[AssignmentEntry]:
|
|
"""
|
|
Iterate entries from top to bottom
|
|
"""
|
|
ASSIGNMENT_ENTRY_LIST.load_rows(main=self)
|
|
# Freeze ocr results here
|
|
yield from [
|
|
button.matched_keyword
|
|
for button in ASSIGNMENT_ENTRY_LIST.cur_buttons
|
|
]
|