mirror of
https://github.com/ok-oldking/ok-wuthering-waves.git
synced 2025-06-07 09:25:34 +00:00
196 lines
7.6 KiB
Python
196 lines
7.6 KiB
Python
import re
|
|
import time
|
|
|
|
import cv2
|
|
|
|
from ok.color.Color import find_color_rectangles, white_color, keep_pixels_in_color_range
|
|
from ok.feature.Box import find_boxes_by_name
|
|
from ok.logging.Logger import get_logger
|
|
from src import text_white_color
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class CombatCheck:
|
|
last_out_of_combat_time = 0
|
|
last_combat_check = 0
|
|
_in_combat = False
|
|
boss_lv_edge = None
|
|
boss_lv_box = None
|
|
in_liberation = False # return True
|
|
has_count_down = False # instant end of combat if count_down goes away
|
|
|
|
def reset_to_false(self):
|
|
self._in_combat = False
|
|
self.boss_lv_edge = None
|
|
self.in_liberation = False # return True
|
|
self.has_count_down = False
|
|
self.last_out_of_combat_time = 0
|
|
self.last_combat_check = 0
|
|
return False
|
|
|
|
def check_count_down(self):
|
|
count_down = self.calculate_color_percentage(text_white_color,
|
|
self.box_of_screen(1820 / 3840, 266 / 2160, 2088 / 3840,
|
|
325 / 2160, name="check_count_down"))
|
|
|
|
if self.has_count_down:
|
|
if count_down < 0.1:
|
|
self.screenshot(f'out of combat because of count_down disappeared {count_down:.2f}%')
|
|
logger.info(f'out of combat because of count_down disappeared {count_down:.2f}%')
|
|
return False
|
|
else:
|
|
return True
|
|
else:
|
|
self.has_count_down = count_down
|
|
return self.has_count_down
|
|
|
|
def check_boss(self):
|
|
current, area = self.keep_boss_text_white()
|
|
max_val = 0
|
|
if current is not None:
|
|
res = cv2.matchTemplate(current, self.boss_lv_edge, cv2.TM_CCOEFF_NORMED)
|
|
min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res)
|
|
if max_val < 0.8:
|
|
self.screenshot_boss_lv(current, f'boss lv not detected by edge {max_val}')
|
|
logger.debug(f'boss lv not detected by edge')
|
|
if not self.find_boss_lv_text(): # double check by text
|
|
if not (self.in_team()[0] and self.check_health_bar()) and not self.check_count_down():
|
|
self.screenshot_boss_lv(current, 'out_of combat boss_health disappeared')
|
|
logger.info(f'out of combat because of boss_health disappeared, res:{max_val} {res}')
|
|
return self.reset_to_false()
|
|
else:
|
|
self.boss_lv_edge = None
|
|
self.boss_lv_box = None
|
|
logger.info(f'boss_health disappeared, but still in combat')
|
|
return True
|
|
else:
|
|
return True
|
|
else:
|
|
logger.debug(f'check boss edge passed {res}')
|
|
return True
|
|
|
|
def screenshot_boss_lv(self, current, name):
|
|
if self.debug:
|
|
if self.boss_lv_box is not None and self.boss_lv_edge is not None:
|
|
self.frame[self.boss_lv_box.y:self.boss_lv_box.y + self.boss_lv_box.height,
|
|
self.boss_lv_box.x:self.boss_lv_box.x + self.boss_lv_box.width] = current
|
|
x, y, w, h = self.boss_lv_box.x, self.boss_lv_box.height + 50 + self.boss_lv_box.y, self.boss_lv_box.width, self.boss_lv_box.height
|
|
self.frame[y:y + h, x:x + w] = self.boss_lv_edge
|
|
self.screenshot(name)
|
|
|
|
def in_combat(self):
|
|
if self.in_liberation:
|
|
logger.debug('in liberation return True')
|
|
return True
|
|
if self._in_combat:
|
|
now = time.time()
|
|
if now - self.last_combat_check > 0.5:
|
|
self.last_combat_check = now
|
|
if not self.in_team()[0]:
|
|
return self.reset_to_false()
|
|
if self.boss_lv_edge is not None:
|
|
return self.check_boss()
|
|
if self.check_count_down():
|
|
return True
|
|
if not self.check_health_bar():
|
|
logger.debug('not in team or no health bar')
|
|
if self.last_out_of_combat_time == 0:
|
|
self.last_out_of_combat_time = now
|
|
logger.debug(
|
|
'first time detected, not in team and no health bar, wait for 4 seconds to double check')
|
|
return True
|
|
elif now - self.last_out_of_combat_time > 4:
|
|
logger.debug('out of combat for 4 secs return False')
|
|
return self.reset_to_false()
|
|
else:
|
|
return True
|
|
else:
|
|
logger.debug(
|
|
'check in combat pass')
|
|
self.last_out_of_combat_time = 0
|
|
return True
|
|
else:
|
|
return True
|
|
else:
|
|
in_combat = self.in_team()[0] and self.check_health_bar()
|
|
if in_combat:
|
|
self._in_combat = True
|
|
return True
|
|
|
|
def check_health_bar(self, find_boss=True):
|
|
if self._in_combat:
|
|
min_height = self.height_of_screen(10 / 2160)
|
|
max_height = min_height * 3
|
|
min_width = self.width_of_screen(15 / 3840)
|
|
else:
|
|
min_height = self.height_of_screen(12 / 2160)
|
|
max_height = min_height * 3
|
|
min_width = self.width_of_screen(100 / 3840)
|
|
|
|
boxes = find_color_rectangles(self.frame, enemy_health_color_red, min_width, min_height, max_height=max_height)
|
|
|
|
if len(boxes) > 0:
|
|
self.draw_boxes('enemy_health_bar_red', boxes, color='blue')
|
|
return True
|
|
elif find_boss:
|
|
return self.find_boss_lv_text()
|
|
|
|
def find_boss_lv_text(self):
|
|
texts = self.ocr(box=self.box_of_screen(1269 / 3840, 10 / 2160, 2533 / 3840, 140 / 2160),
|
|
target_height=720)
|
|
boss_lv_texts = find_boxes_by_name(texts,
|
|
[re.compile(r'(?i)^L[V].*')])
|
|
if len(boss_lv_texts) > 0:
|
|
logger.debug(f'boss_lv_texts: {boss_lv_texts}')
|
|
self.boss_lv_box = boss_lv_texts[0]
|
|
self.boss_lv_edge, area = self.keep_boss_text_white()
|
|
if self.boss_lv_edge is None:
|
|
self.boss_lv_box = None
|
|
return False
|
|
return True
|
|
|
|
def keep_boss_text_white(self):
|
|
corpped = self.boss_lv_box.crop_frame(self.frame)
|
|
image, area = keep_pixels_in_color_range(corpped, white_color)
|
|
if area / image.shape[0] * image.shape[1] < 0.05:
|
|
image, area = keep_pixels_in_color_range(corpped, boss_orange_text_color)
|
|
if area / image.shape[0] * image.shape[1] < 0.05:
|
|
image, area = keep_pixels_in_color_range(corpped,
|
|
boss_red_text_color)
|
|
if area / image.shape[0] * image.shape[1] < 0.05:
|
|
logger.error(f'keep_boss_text_white cant find text with the correct color')
|
|
return None, 0
|
|
return image, area
|
|
|
|
|
|
enemy_health_color_red = {
|
|
'r': (202, 212), # Red range
|
|
'g': (70, 80), # Green range
|
|
'b': (55, 65) # Blue range
|
|
} # 207,75,60
|
|
|
|
enemy_health_color_black = {
|
|
'r': (10, 55), # Red range
|
|
'g': (28, 50), # Green range
|
|
'b': (18, 70) # Blue range
|
|
}
|
|
|
|
boss_orange_text_color = {
|
|
'r': (218, 218), # Red range
|
|
'g': (178, 178), # Green range
|
|
'b': (68, 68) # Blue range
|
|
}
|
|
|
|
boss_red_text_color = {
|
|
'r': (226, 226), # Red range
|
|
'g': (82, 82), # Green range
|
|
'b': (76, 76) # Blue range
|
|
}
|
|
|
|
boss_health_color = {
|
|
'r': (250, 255), # Red range
|
|
'g': (30, 180), # Green range
|
|
'b': (4, 75) # Blue range
|
|
}
|