mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-23 17:11:42 +00:00
bea05d396e
- 开荒模式移动至出击设置 - 增加开荒模式自动启用, 无脑开就完事了 - 将透视识别参数放到了config里, 这样就可以用地图config覆盖了 - 适配小地图模式, 参数抄A1就行了 - 修复了章节名OCR的识别位置 - 修复了开荒会把BOSS当作精英打问题 - 增加战斗中的剧情跳过 - 注释掉了截图和点击的retry - 增加了捕捉目标点超出移动范围
344 lines
11 KiB
Python
344 lines
11 KiB
Python
import numpy as np
|
|
|
|
from module.base.timer import Timer
|
|
from module.base.utils import color_bar_percentage
|
|
from module.combat.assets import *
|
|
from module.combat.combat_auto import CombatAuto
|
|
from module.combat.combat_manual import CombatManual
|
|
from module.combat.emotion import Emotion
|
|
from module.combat.hp_balancer import HPBalancer
|
|
from module.combat.submarine import SubmarineCall
|
|
from module.handler.enemy_searching import EnemySearchingHandler
|
|
from module.handler.low_emotion import LowEmotionHandler
|
|
from module.handler.story import StoryHandler
|
|
from module.handler.urgent_commission import UrgentCommissionHandler
|
|
from module.logger import logger
|
|
from module.map.assets import MAP_OFFENSIVE
|
|
from module.retire.retirement import Retirement
|
|
from module.ui.assets import BACK_ARROW
|
|
|
|
|
|
class Combat(HPBalancer, UrgentCommissionHandler, EnemySearchingHandler, Retirement, SubmarineCall, LowEmotionHandler,
|
|
CombatAuto, CombatManual, StoryHandler):
|
|
_automation_set_timer = Timer(1)
|
|
_emotion: Emotion
|
|
battle_status_click_interval = 0
|
|
|
|
@property
|
|
def emotion(self):
|
|
if not hasattr(self, '_emotion'):
|
|
self._emotion = Emotion(config=self.config)
|
|
return self._emotion
|
|
|
|
def combat_appear(self):
|
|
"""
|
|
Returns:
|
|
bool: If enter combat.
|
|
"""
|
|
if self.config.ENABLE_MAP_FLEET_LOCK and not self.is_in_map():
|
|
if self.is_combat_loading():
|
|
return True
|
|
if self.handle_retirement():
|
|
self.map_offensive()
|
|
return False
|
|
if self.handle_combat_low_emotion():
|
|
return False
|
|
|
|
if self.appear(BATTLE_PREPARATION):
|
|
return True
|
|
if self.appear(BATTLE_PREPARATION_WITH_OVERLAY) and self.handle_combat_automation_confirm():
|
|
return True
|
|
|
|
return False
|
|
|
|
def map_offensive(self):
|
|
while 1:
|
|
self.device.screenshot()
|
|
|
|
if self.appear_then_click(MAP_OFFENSIVE, interval=1):
|
|
continue
|
|
|
|
# Break
|
|
if self.combat_appear():
|
|
break
|
|
|
|
def is_combat_loading(self):
|
|
"""
|
|
Returns:
|
|
bool:
|
|
"""
|
|
left = color_bar_percentage(self.device.image, area=LOADING_BAR.area, prev_color=(99, 150, 255))
|
|
right = color_bar_percentage(self.device.image, area=LOADING_BAR.area, prev_color=(225, 225, 225), reverse=True)
|
|
if 0.15 < left < 0.95 and right > 0.15 and left + right <= 1.2:
|
|
logger.attr('Loading', f'{int(left * 100)}%({int(right * 100)}%)')
|
|
return True
|
|
|
|
return False
|
|
|
|
def is_combat_executing(self):
|
|
"""
|
|
Returns:
|
|
bool:
|
|
"""
|
|
return self.appear(PAUSE) and np.max(self.device.image.crop(PAUSE_DOUBLE_CHECK.area)) < 153
|
|
|
|
def handle_combat_automation_confirm(self):
|
|
if self.appear(AUTOMATION_CONFIRM_CHECK, interval=1):
|
|
self.appear_then_click(AUTOMATION_CONFIRM, offset=True)
|
|
return True
|
|
|
|
return False
|
|
|
|
def combat_preparation(self, balance_hp=False, emotion_reduce=False, auto=True, fleet_index=1):
|
|
"""
|
|
Args:
|
|
balance_hp (bool):
|
|
emotion_reduce (bool):
|
|
auto (bool):
|
|
fleet_index (int):
|
|
"""
|
|
logger.info('Combat preparation.')
|
|
|
|
if emotion_reduce:
|
|
self.emotion.wait(fleet=fleet_index)
|
|
if balance_hp:
|
|
self.hp_balance()
|
|
# logger.info('start combat')
|
|
|
|
while 1:
|
|
self.device.screenshot()
|
|
|
|
# Automation.
|
|
if self.appear(BATTLE_PREPARATION):
|
|
# if self.handle_combat_automation_confirm():
|
|
# continue
|
|
if self.handle_combat_automation_set(auto=auto):
|
|
continue
|
|
|
|
# Retirement
|
|
if self.handle_retirement():
|
|
continue
|
|
|
|
# Emotion
|
|
if self.handle_combat_low_emotion():
|
|
continue
|
|
|
|
# Combat start
|
|
if self.appear_then_click(BATTLE_PREPARATION, interval=2):
|
|
continue
|
|
|
|
if self.handle_combat_automation_confirm():
|
|
continue
|
|
|
|
# End
|
|
if self.is_combat_executing():
|
|
if emotion_reduce:
|
|
self.emotion.reduce(fleet_index)
|
|
break
|
|
|
|
def handle_combat_automation_set(self, auto):
|
|
"""
|
|
Args:
|
|
auto (bool): If use auto.
|
|
|
|
Returns:
|
|
bool:
|
|
"""
|
|
if not self._automation_set_timer.reached():
|
|
return False
|
|
|
|
if self.appear(AUTOMATION_ON):
|
|
logger.info('[Automation] ON')
|
|
if not auto:
|
|
self.device.click(AUTOMATION_SWITCH)
|
|
self.device.sleep(1)
|
|
self._automation_set_timer.reset()
|
|
return True
|
|
|
|
if self.appear(AUTOMATION_OFF):
|
|
logger.info('[Automation] OFF')
|
|
if auto:
|
|
self.device.click(AUTOMATION_SWITCH)
|
|
self.device.sleep(1)
|
|
self._automation_set_timer.reset()
|
|
return True
|
|
|
|
if self.appear_then_click(AUTOMATION_CONFIRM, offset=True):
|
|
self._automation_set_timer.reset()
|
|
return True
|
|
|
|
return False
|
|
|
|
def combat_execute(self, func=None, call_submarine_at_boss=False, save_get_items=False):
|
|
"""
|
|
Args:
|
|
func: Funtion to run when in combat.
|
|
call_submarine_at_boss (bool):
|
|
save_get_items (bool)
|
|
"""
|
|
logger.info('Combat execute')
|
|
self.submarine_call_reset()
|
|
self.combat_auto_reset()
|
|
self.combat_manual_reset()
|
|
confirm_timer = Timer(10)
|
|
confirm_timer.start()
|
|
|
|
while 1:
|
|
self.device.screenshot()
|
|
|
|
if not confirm_timer.reached() and self.appear_then_click(AUTOMATION_CONFIRM, offset=True):
|
|
continue
|
|
|
|
if self.handle_story_skip():
|
|
continue
|
|
if self.handle_combat_auto():
|
|
continue
|
|
if self.handle_combat_manual():
|
|
continue
|
|
|
|
if call_submarine_at_boss:
|
|
pass
|
|
else:
|
|
if self.handle_submarine_call():
|
|
continue
|
|
|
|
# End
|
|
if self.handle_battle_status(save_get_items=save_get_items):
|
|
break
|
|
|
|
def handle_battle_status(self, save_get_items=False):
|
|
"""
|
|
Args:
|
|
save_get_items (bool):
|
|
|
|
Returns:
|
|
bool:
|
|
"""
|
|
if self.appear_then_click(BATTLE_STATUS_S, screenshot=save_get_items, genre='status',
|
|
interval=self.battle_status_click_interval):
|
|
if not save_get_items:
|
|
self.device.sleep((0.25, 0.5))
|
|
return True
|
|
if self.appear_then_click(BATTLE_STATUS_A, screenshot=save_get_items, genre='status',
|
|
interval=self.battle_status_click_interval):
|
|
logger.warning('Battle status: A')
|
|
if not save_get_items:
|
|
self.device.sleep((0.25, 0.5))
|
|
return True
|
|
|
|
return False
|
|
|
|
def handle_get_items(self, save_get_items=False):
|
|
"""
|
|
Args:
|
|
save_get_items (bool):
|
|
|
|
Returns:
|
|
bool:
|
|
"""
|
|
if self.appear_then_click(GET_ITEMS_1, screenshot=save_get_items, genre='get_items', offset=5,
|
|
interval=self.battle_status_click_interval):
|
|
self.interval_reset(BATTLE_STATUS_S)
|
|
self.interval_reset(BATTLE_STATUS_A)
|
|
return True
|
|
if self.appear_then_click(GET_ITEMS_2, screenshot=save_get_items, genre='get_items', offset=5,
|
|
interval=self.battle_status_click_interval):
|
|
self.interval_reset(BATTLE_STATUS_S)
|
|
self.interval_reset(BATTLE_STATUS_A)
|
|
return True
|
|
|
|
return False
|
|
|
|
def handle_exp_info(self):
|
|
"""
|
|
Returns:
|
|
bool:
|
|
"""
|
|
if self.appear_then_click(EXP_INFO_S):
|
|
self.device.sleep((0.25, 0.5))
|
|
return True
|
|
if self.appear_then_click(EXP_INFO_A):
|
|
self.device.sleep((0.25, 0.5))
|
|
return True
|
|
|
|
return False
|
|
|
|
def handle_get_ship(self, save_get_items=False):
|
|
"""
|
|
Args:
|
|
save_get_items (bool):
|
|
|
|
Returns:
|
|
bool:
|
|
"""
|
|
if self.appear_then_click(GET_SHIP, screenshot=save_get_items, genre='get_ship'):
|
|
return True
|
|
|
|
return False
|
|
|
|
def combat_status(self, save_get_items=False, expected_end=None):
|
|
"""
|
|
Args:
|
|
save_get_items (bool):
|
|
expected_end (str): with_searching, no_searching, in_stage.
|
|
"""
|
|
logger.info('Combat status')
|
|
logger.attr('expected_end', expected_end)
|
|
while 1:
|
|
self.device.screenshot()
|
|
|
|
# Combat status
|
|
if self.handle_get_ship(save_get_items=save_get_items):
|
|
continue
|
|
if self.handle_get_items(save_get_items=save_get_items):
|
|
continue
|
|
if self.handle_battle_status(save_get_items=save_get_items):
|
|
continue
|
|
if self.handle_popup_confirm():
|
|
continue
|
|
if self.handle_exp_info():
|
|
continue
|
|
if self.handle_urgent_commission(save_get_items=save_get_items):
|
|
continue
|
|
if self.handle_story_skip():
|
|
continue
|
|
|
|
# End
|
|
if expected_end is None:
|
|
if self.handle_in_map_with_enemy_searching():
|
|
break
|
|
if isinstance(expected_end, str):
|
|
if expected_end == 'in_stage' and self.handle_in_stage():
|
|
break
|
|
if expected_end == 'with_searching' and self.handle_in_map_with_enemy_searching():
|
|
break
|
|
if expected_end == 'no_searching' and self.handle_in_map():
|
|
break
|
|
if expected_end == 'in_ui' and self.appear(BACK_ARROW, offset=(20, 20)):
|
|
break
|
|
if callable(expected_end):
|
|
if expected_end():
|
|
break
|
|
|
|
def combat(self, balance_hp=None, emotion_reduce=None, func=None, call_submarine_at_boss=None, save_get_items=None,
|
|
expected_end=None, fleet_index=1):
|
|
"""
|
|
Execute a combat.
|
|
"""
|
|
balance_hp = balance_hp if balance_hp is not None else self.config.ENABLE_HP_BALANCE
|
|
emotion_reduce = emotion_reduce if emotion_reduce is not None else self.config.ENABLE_EMOTION_REDUCE
|
|
auto = self.config.COMBAT_AUTO_MODE == 'combat_auto'
|
|
call_submarine_at_boss = call_submarine_at_boss if call_submarine_at_boss is not None else self.config.SUBMARINE_CALL_AT_BOSS
|
|
save_get_items = save_get_items if save_get_items is not None else self.config.ENABLE_SAVE_GET_ITEMS
|
|
self.battle_status_click_interval = 3 if save_get_items else 0
|
|
|
|
# if not hasattr(self, 'emotion'):
|
|
# self.emotion = Emotion(config=self.config)
|
|
|
|
self.combat_preparation(
|
|
balance_hp=balance_hp, emotion_reduce=emotion_reduce, auto=auto, fleet_index=fleet_index)
|
|
self.combat_execute(
|
|
func=func, call_submarine_at_boss=call_submarine_at_boss, save_get_items=save_get_items)
|
|
self.combat_status(
|
|
save_get_items=save_get_items, expected_end=expected_end)
|