mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-24 01:21:25 +00:00
72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
from module.base.base import ModuleBase
|
|
from module.base.timer import Timer
|
|
from module.base.utils import color_bar_percentage
|
|
from module.logger import logger
|
|
|
|
|
|
ATTACKER_HP_AREA = (271, 43, 586, 58)
|
|
DEFENDER_HP_AREA = (691, 43, 1005, 58)
|
|
|
|
|
|
class HpDaemon(ModuleBase):
|
|
attacker_hp = 1.0
|
|
defender_hp = 1.0
|
|
# _last_secure_time = 0
|
|
low_hp_confirm_timer: Timer
|
|
|
|
@staticmethod
|
|
def _calculate_hp(image, area, reverse=False, starter=2, prev_color=(239, 32, 33)):
|
|
"""
|
|
Args:
|
|
image:
|
|
area:
|
|
reverse: True if HP is left align.
|
|
starter:
|
|
prev_color:
|
|
|
|
Returns:
|
|
float: HP. 0 to 1.
|
|
"""
|
|
# bar = np.array(image.crop(area))
|
|
# length = bar.shape[1]
|
|
# bar = np.swapaxes(bar, 0, 1)
|
|
# bar = bar[::-1, :, :] if reverse else bar
|
|
# prev_index = 0
|
|
# for index, color in enumerate(bar):
|
|
# if index < starter:
|
|
# continue
|
|
# mask = color_similar_1d(color, prev_color, threshold=30)
|
|
# if np.any(mask):
|
|
# prev_color = color[mask].mean(axis=0)
|
|
# prev_index = index
|
|
#
|
|
# return prev_index / length
|
|
return color_bar_percentage(image, area, prev_color=prev_color, starter=starter, reverse=reverse)
|
|
|
|
def _show_hp(self, low_hp_time=0):
|
|
"""
|
|
Examples:
|
|
[ 80% - 70%]
|
|
[ 80% - 70%]
|
|
[ 80% - 70%] - Low HP: 3.154s
|
|
"""
|
|
text = '[%s - %s]' % (
|
|
str(int(self.attacker_hp * 100)).rjust(2, '0') + '%',
|
|
str(int(self.defender_hp * 100)).rjust(2, '0') + '%')
|
|
if low_hp_time:
|
|
text += ' - Low HP: %ss' % str(round(low_hp_time, 3)).ljust(5, '0')
|
|
logger.info(text)
|
|
|
|
def _at_low_hp(self, image):
|
|
self.attacker_hp = self._calculate_hp(image, area=ATTACKER_HP_AREA, reverse=True)
|
|
self.defender_hp = self._calculate_hp(image, area=DEFENDER_HP_AREA, reverse=False)
|
|
if 0.01 < self.attacker_hp <= self.config.LOW_HP_THRESHOLD:
|
|
if self.low_hp_confirm_timer.reached() and self.low_hp_confirm_timer.current() < 300:
|
|
self._show_hp(self.low_hp_confirm_timer.current())
|
|
return True
|
|
else:
|
|
return False
|
|
else:
|
|
self.low_hp_confirm_timer.reset()
|
|
return False
|