mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-16 14:31:16 +00:00
29 lines
1004 B
Python
29 lines
1004 B
Python
|
from module.logger import logger
|
||
|
|
from module.ocr.ocr import DigitCounter
|
||
|
|
from tasks.base.ui import UI
|
||
|
|
from tasks.dungeon.assets.assets_dungeon_event import DOUBLE_CALYX_EVENT_TAG, OCR_DOUBLE_EVENT_REMAIN
|
||
|
|
|
||
|
|
|
||
|
|
class DungeonEvent(UI):
|
||
|
|
def has_double_calyx_event(self) -> bool:
|
||
|
|
"""
|
||
|
|
Pages:
|
||
|
|
in: page_guide, Survival_Index, nav at top
|
||
|
|
"""
|
||
|
|
has = self.image_color_count(DOUBLE_CALYX_EVENT_TAG, color=(252, 209, 123), threshold=221, count=50)
|
||
|
|
logger.attr('Double calyx', has)
|
||
|
|
return has
|
||
|
|
|
||
|
|
def get_double_event_remain(self) -> int:
|
||
|
|
"""
|
||
|
|
Pages:
|
||
|
|
in: page_guide, Survival_Index, selected at the nav with double event
|
||
|
|
"""
|
||
|
|
ocr = DigitCounter(OCR_DOUBLE_EVENT_REMAIN)
|
||
|
|
remain, _, total = ocr.ocr_single_line(self.device.image)
|
||
|
|
if total != 12:
|
||
|
|
logger.warning(f'Invalid double event remain')
|
||
|
|
remain = 0
|
||
|
|
logger.attr('Double event remain', remain)
|
||
|
|
return remain
|