mirror of
https://github.com/LmeSzinc/AzurLaneAutoScript.git
synced 2025-01-08 12:27:33 +08:00
130 lines
4.1 KiB
Python
130 lines
4.1 KiB
Python
from module.base.button import Button
|
|
from module.base.utils import *
|
|
from ..campaign_war_archives.campaign_base import CampaignBase as CampaignBase_
|
|
from module.exception import CampaignNameError
|
|
from module.logger import logger
|
|
|
|
# Here manually type coordinates, because the ball appears in event Dreamwaker's Butterfly only.
|
|
BALL = Button(area=(571, 283, 696, 387), color=(), button=(597, 274, 671, 343))
|
|
|
|
|
|
class CampaignBase(CampaignBase_):
|
|
STAGE_INCREASE = [
|
|
'TS1 > T1 > T2 > T3 > T4 > TS2 > T5 > T6',
|
|
'HTS1 > HT1 > HT2 > HT3',
|
|
'HT4 > HTS2 > HT5 > HT6',
|
|
]
|
|
|
|
def campaign_set_chapter(self, name, mode='normal'):
|
|
"""
|
|
Args:
|
|
name (str): Campaign name, such as '7-2', 'd3', 'sp3'.
|
|
mode (str): 'normal' or 'hard'.
|
|
"""
|
|
chapter, stage = self._campaign_separate_name(name)
|
|
name = chapter + stage
|
|
|
|
if chapter.isdigit():
|
|
self.ui_goto_campaign()
|
|
self.campaign_ensure_mode('normal')
|
|
self.campaign_ensure_chapter(index=chapter)
|
|
if mode == 'hard':
|
|
self.campaign_ensure_mode('hard')
|
|
|
|
elif chapter in 'abcd' or chapter == 'ex_sp':
|
|
self.ui_goto_event()
|
|
if chapter in 'ab':
|
|
self.campaign_ensure_mode('normal')
|
|
elif chapter in 'cd':
|
|
self.campaign_ensure_mode('hard')
|
|
elif chapter == 'ex_sp':
|
|
self.campaign_ensure_mode('ex')
|
|
self.campaign_ensure_chapter(index=chapter)
|
|
|
|
elif chapter == 'sp':
|
|
self.ui_goto_sp()
|
|
self.campaign_ensure_chapter(index=chapter)
|
|
|
|
elif chapter in ['t', 'ts', 'ht', 'hts']:
|
|
self.ui_goto_event()
|
|
# Campaign ball
|
|
if stage in ['1', '6']:
|
|
self._campaign_ball_set('blue')
|
|
else:
|
|
self._campaign_ball_set('red')
|
|
# Campaign mode
|
|
if chapter in ['t', 'ts']:
|
|
self.campaign_ensure_mode('normal')
|
|
if chapter in ['ht', 'hts']:
|
|
self.campaign_ensure_mode('hard')
|
|
if chapter == 'ex_sp':
|
|
self.campaign_ensure_mode('ex')
|
|
# Get stage
|
|
self.campaign_ensure_chapter(index=1)
|
|
else:
|
|
logger.warning(f'Unknown campaign chapter: {name}')
|
|
|
|
@staticmethod
|
|
def _campaign_get_chapter_index(name):
|
|
"""
|
|
Args:
|
|
name (str, int):
|
|
|
|
Returns:
|
|
int
|
|
"""
|
|
if isinstance(name, int):
|
|
return name
|
|
else:
|
|
if name.isdigit():
|
|
return int(name)
|
|
elif name in ['a', 'c', 'sp', 'ex_sp', 'ts', 't', 'ht', 'hts']:
|
|
return 1
|
|
elif name in ['b', 'd', 'ex_ex']:
|
|
return 2
|
|
else:
|
|
raise CampaignNameError
|
|
|
|
def _campaign_ball_get(self):
|
|
"""
|
|
Returns:
|
|
str: 'blue' or 'red'.
|
|
"""
|
|
color = get_color(self.device.image, BALL.area)
|
|
# Blue: (93, 127, 182), Red: (186, 116, 124)
|
|
index = np.argmax(color)
|
|
if index == 0:
|
|
return 'red'
|
|
elif index == 2:
|
|
return 'blue'
|
|
else:
|
|
logger.warning(f'Unknown campaign ball color: {color}')
|
|
return 'unknown'
|
|
|
|
def _campaign_ball_set(self, status):
|
|
"""
|
|
Args:
|
|
status (str): 'blue' or 'red'.
|
|
"""
|
|
skip_first_screenshot = True
|
|
while 1:
|
|
if skip_first_screenshot:
|
|
skip_first_screenshot = False
|
|
else:
|
|
self.device.screenshot()
|
|
|
|
current = self._campaign_ball_get()
|
|
logger.attr('Campaign_ball', current)
|
|
|
|
if current == status:
|
|
break
|
|
else:
|
|
if self.is_in_stage():
|
|
self.device.click(BALL)
|
|
self.device.sleep(3)
|
|
# wait until is_in_stage
|
|
while 1:
|
|
self.device.screenshot()
|
|
if self.is_in_stage():
|
|
break
|