mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-09-28 06:45:03 +00:00
45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
import re
|
|
|
|
from module.ocr.ocr import Digit, Ocr
|
|
from tasks.base.ui import UI
|
|
from tasks.rogue.assets.assets_rogue_ui import *
|
|
from tasks.rogue.keywords import *
|
|
|
|
|
|
class RogueBonusOcr(Ocr):
|
|
def after_process(self, result):
|
|
result = super().after_process(result)
|
|
if self.lang == 'ch':
|
|
replace_pattern_dict = {
|
|
"[宇宝][宙审]": "宇宙",
|
|
}
|
|
for pat, replace in replace_pattern_dict.items():
|
|
result = re.sub(pat, replace, result)
|
|
return result
|
|
|
|
|
|
class RogueUI(UI):
|
|
path: RoguePath
|
|
|
|
@property
|
|
def cosmic_fragment(self):
|
|
"""
|
|
Return valid result only when template appear
|
|
"""
|
|
if self.appear(COSMIC_FRAGMENT):
|
|
return Digit(OCR_COSMIC_FRAGMENT).ocr_single_line(self.device.image)
|
|
return 0
|
|
|
|
def is_page_choose_blessing(self):
|
|
return (self.image_color_count(PAGE_CHOOSE_BUFF, (245, 245, 245), count=200)
|
|
and self.appear(CHECK_BLESSING))
|
|
|
|
def is_page_choose_curio(self):
|
|
return self.appear(PAGE_CHOOSE_CURIO)
|
|
|
|
def is_page_choose_bonus(self):
|
|
return self.appear(PAGE_CHOOSE_BONUS)
|
|
|
|
def is_page_event(self):
|
|
return self.appear(PAGE_EVENT)
|