71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
"""
|
||
战斗状态检测模块，通过模板匹配判断角色是否处于战斗中
|
||
"""
|
||
import os
|
||
import logging
|
||
import cv2
|
||
import numpy as np
|
||
import pyautogui
|
||
from config import GameConfig
|
||
|
||
|
||
class CombatDetector:
|
||
"""通过截图与模板匹配判断当前是否处于战斗状态"""
|
||
|
||
TEMPLATES = [
|
||
'images/1.png',
|
||
'images/2.png',
|
||
'images/3.png',
|
||
'images/4.png',
|
||
]
|
||
SCREENSHOT_PATH = 'screenshot/combat_screenshot.png'
|
||
|
||
def __init__(self, config: GameConfig):
|
||
self.config = config
|
||
self.screen_width, self.screen_height = pyautogui.size()
|
||
self.logger = logging.getLogger(__name__)
|
||
os.makedirs(os.path.dirname(self.SCREENSHOT_PATH), exist_ok=True)
|
||
|
||
def is_in_combat(self) -> bool:
|
||
"""判断是否处于战斗中。
|
||
|
||
截取屏幕左上半区域，逐一与模板图片做归一化相关系数匹配，
|
||
任一模板匹配值超过阈值即视为处于战斗状态。
|
||
|
||
Returns:
|
||
bool: 处于战斗中返回 True，否则返回 False
|
||
"""
|
||
try:
|
||
region_width = self.screen_width // 2
|
||
region_height = self.screen_height // 2
|
||
screenshot = pyautogui.screenshot(region=(0, 0, region_width, region_height))
|
||
screenshot.save(self.SCREENSHOT_PATH)
|
||
|
||
screenshot_cv = cv2.cvtColor(np.array(screenshot), cv2.COLOR_RGB2BGR)
|
||
|
||
for template_file in self.TEMPLATES:
|
||
if not os.path.exists(template_file):
|
||
self.logger.warning(f"模板文件不存在，跳过: {template_file}")
|
||
continue
|
||
|
||
template = cv2.imread(template_file)
|
||
result = cv2.matchTemplate(
|
||
screenshot_cv,
|
||
template,
|
||
cv2.TM_CCOEFF_NORMED
|
||
)
|
||
_, max_val, _, _ = cv2.minMaxLoc(result)
|
||
|
||
if max_val > self.config.recognition_threshold:
|
||
self.logger.info(
|
||
f"战斗状态: 是（模板: {template_file}，匹配值: {max_val:.2f}）"
|
||
)
|
||
return True
|
||
|
||
self.logger.info("战斗状态: 否（所有模板均未匹配）")
|
||
return False
|
||
|
||
except Exception as e:
|
||
self.logger.error(f"判断战斗状态失败: {str(e)}")
|
||
return False
|