Files
wow/auto_bot.py

600 lines
25 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import json
import os
import random
import sys
import time
import math
import ctypes
import cv2
import numpy as np
import win32gui
import win32con
# 开启 DPI 意识
try:
ctypes.windll.shcore.SetProcessDpiAwareness(1)
except Exception:
ctypes.windll.user32.SetProcessDPIAware()
from hardware_control import hw_ctrl
from game_state import parse_game_state, load_layout_config
from stuck_handler import StuckHandler
# 定义按键常量
KEY_TAB = '3'
KEY_LOOT = '4' # 假设你在游戏里设置了互动按键为 F
KEY_ATTACK = '2' # 假设你的主攻击技能是 1
WIN_TITLE = "魔兽世界"
def _config_base():
if getattr(sys, 'frozen', False):
return os.path.dirname(sys.executable)
return os.path.dirname(os.path.abspath(__file__))
def get_config_path(filename):
base = _config_base()
p = os.path.join(base, filename)
if os.path.exists(p):
return p
if getattr(sys, 'frozen', False):
meipass = getattr(sys, '_MEIPASS', '')
if meipass:
p2 = os.path.join(meipass, filename)
if os.path.exists(p2):
return p2
return p
def move_cursor_hw(x, y, settle_sec=0.02):
hw_ctrl.move_to(int(x), int(y))
if settle_sec > 0:
time.sleep(settle_sec)
def get_wow_client_rect(hwnd):
client_left, client_top, client_right, client_bottom = win32gui.GetClientRect(hwnd)
screen_left, screen_top = win32gui.ClientToScreen(hwnd, (client_left, client_top))
screen_right, screen_bottom = win32gui.ClientToScreen(hwnd, (client_right, client_bottom))
return screen_left, screen_top, screen_right, screen_bottom
class CursorManager:
"""通过图像识别判断鼠标图标类型"""
def __init__(self):
self.templates = {}
self.handle_cache = {}
self.log_path = get_config_path('cursor_recognition.log')
self._load_templates()
def _load_templates(self):
# 强制使用纯相对路径,由 Python 自动处理 CWD完美避开中文路径编码问题
cursor_dir = get_config_path(os.path.join('images', 'cursor'))
files = {'Point': 'Point.PNG', 'Attack': 'Attack.PNG', 'LootAll': 'LootAll.PNG', 'Skin': 'Skin.PNG'}
for name, fname in files.items():
path = os.path.join(cursor_dir, fname)
if os.path.exists(path):
try:
# 使用 numpy 读取二进制流再解码
img_array = np.fromfile(path, dtype=np.uint8)
img = cv2.imdecode(img_array, cv2.IMREAD_UNCHANGED)
if img is not None:
if img.shape[2] == 3:
img = cv2.cvtColor(img, cv2.COLOR_BGR2BGRA)
self.templates[name] = img
except Exception as e:
print(f">>> [CursorMgr] 加载 {fname} 失败: {e}")
else:
print(f">>> [CursorMgr] 找不到文件: {path}")
def get_type(self, hcursor):
if not hcursor or hcursor == 0: return 'Other', 0.0
if hcursor in self.handle_cache: return self.handle_cache[hcursor]
res = self._identify(hcursor)
self.handle_cache[hcursor] = res
return res
def _debug_log(self, message):
try:
with open(self.log_path, 'a', encoding='utf-8') as f:
f.write(f"{time.strftime('%Y-%m-%d %H:%M:%S')} {message}\n")
except Exception:
pass
def _normalize_for_match(self, img):
if img is None:
return None
if len(img.shape) == 2:
gray = img
elif img.shape[2] == 4:
alpha = img[:, :, 3].astype(np.float32) / 255.0
bgr = img[:, :, :3].astype(np.float32)
bg = np.full_like(bgr, 255.0)
composed = (bgr * alpha[..., None]) + (bg * (1.0 - alpha[..., None]))
gray = cv2.cvtColor(composed.astype(np.uint8), cv2.COLOR_BGR2GRAY)
else:
gray = cv2.cvtColor(img[:, :, :3], cv2.COLOR_BGR2GRAY)
return cv2.GaussianBlur(gray, (3, 3), 0)
def _edge_map(self, gray):
if gray is None:
return None
return cv2.Canny(gray, 32, 96)
def _identify(self, hcursor):
import win32ui
try:
width, height = 48, 48
hdc = win32gui.GetDC(0)
dc = win32ui.CreateDCFromHandle(hdc)
memdc = dc.CreateCompatibleDC()
bitmap = win32ui.CreateBitmap()
bitmap.CreateCompatibleBitmap(dc, width, height)
memdc.SelectObject(bitmap)
win32gui.DrawIconEx(memdc.GetSafeHdc(), 0, 0, hcursor, width, height, 0, 0, win32con.DI_NORMAL)
bits = bitmap.GetBitmapBits(True)
target = np.frombuffer(bits, dtype='uint8').reshape(height, width, 4)
memdc.DeleteDC()
win32gui.ReleaseDC(0, hdc)
best_name = 'Other'
max_score = 0.0
best_scale = 1.0
target_gray = self._normalize_for_match(target)
target_edge = self._edge_map(target_gray)
target_has_edge = target_edge is not None and np.count_nonzero(target_edge) > 0
scales = [0.55, 0.65, 0.75, 0.85, 0.95, 1.0, 1.1, 1.2, 1.35, 1.5]
for name, temp in self.templates.items():
t_h, t_w = temp.shape[:2]
for s in scales:
s_w, s_h = int(t_w * s), int(t_h * s)
if s_w > width or s_h > height:
continue
interp = cv2.INTER_AREA if s < 1.0 else cv2.INTER_CUBIC
res_temp = cv2.resize(temp, (s_w, s_h), interpolation=interp)
temp_gray = self._normalize_for_match(res_temp)
if temp_gray is None:
continue
gray_res = cv2.matchTemplate(target_gray, temp_gray, cv2.TM_CCOEFF_NORMED)
_, gray_score, _, _ = cv2.minMaxLoc(gray_res)
score = gray_score
temp_edge = self._edge_map(temp_gray)
if target_has_edge and temp_edge is not None and np.count_nonzero(temp_edge) > 0:
edge_res = cv2.matchTemplate(target_edge, temp_edge, cv2.TM_CCOEFF_NORMED)
_, edge_score, _, _ = cv2.minMaxLoc(edge_res)
score = max(score, edge_score)
if score > max_score:
max_score = score
best_name = name
best_scale = s
if max_score > 0.2:
self._debug_log(f"best={best_name} score={max_score:.3f} scale={best_scale:.2f}")
print(f">>>> [雷达识别] 目标: {best_name} | 最高分: {max_score:.3f}")
return best_name, max_score
except Exception:
return 'Other', 0.0
def load_attack_loop(path):
"""从 JSON 加载攻击循环配置。支持 hp_below_percent/hp_key、mp_below_percent/mp_key0=不启用)"""
if not path or not os.path.exists(path):
return None
try:
with open(path, 'r', encoding='utf-8') as f:
cfg = json.load(f)
steps = cfg.get('steps') or []
trigger = float(cfg.get('trigger_chance', 0.3))
hp_below = int(cfg.get('hp_below_percent', 0)) or 0
mp_below = int(cfg.get('mp_below_percent', 0)) or 0
return {
'trigger_chance': trigger,
'steps': steps,
'hp_below_percent': hp_below,
'hp_key': (cfg.get('hp_key') or '').strip() or None,
'mp_below_percent': mp_below,
'mp_key': (cfg.get('mp_key') or '').strip() or None,
}
except Exception:
return None
class AutoBot:
def __init__(
self,
waypoints=None,
attack_loop_path=None,
skinning_wait_sec=None,
enable_mouse_loot=True,
turn_error_key=None,
turn_error_hold_sec=None,
):
self.last_tab_time = 0
self.last_interaction_time = 0
self.last_target_hp = 0
self._has_braked_for_target = False
self._was_in_combat_or_target = False
self.is_running = True
self.attack_loop_config = load_attack_loop(attack_loop_path)
self.tab_no_target_count = 0 # 连续按 Tab 仍无目标的次数,满 5 次则转向
self.stuck_handler = StuckHandler()
self.target_acquired_time = None # 本次获得目标的时间(目前仅用于调试/保留接口)
self.last_turn_end_time = 0 # 最近一次结束 A/D 转向的时间,供卡死检测排除
self.skinning_wait_sec = float(skinning_wait_sec) if skinning_wait_sec is not None else 1.5
self.enable_mouse_loot = enable_mouse_loot
self.cursor_mgr = CursorManager()
self.last_turn_signal_time = 0.0
self.turn_error_key = (turn_error_key or 's').strip().lower() or 's'
self.combat_signal_retry_sec = 1.0
self.turn_error_hold_sec = float(turn_error_hold_sec) if turn_error_hold_sec is not None else 0.8
self._last_mouse_path_scale_signature = None
def execute_disengage_loot(self):
"""从有战斗/目标切换到完全脱战的瞬间,执行拾取 + 剥皮。"""
try:
# 1. 拟人化鼠标扫雷补漏拾取
if self.enable_mouse_loot:
self.mouse_sweep_loot()
else:
hw_ctrl.press(KEY_LOOT)
time.sleep(0.5)
# 关闭扫雷时,保留原有的补按交互和固定等待
hw_ctrl.press(KEY_LOOT)
time.sleep(self.skinning_wait_sec + 0.5)
except Exception:
pass
def _load_mouse_path_points(self, client_width, client_height):
path_points = []
layout = load_layout_config()
base_width = int(layout.get('mouse_path_base_window_width', 2560) or 2560)
base_height = int(layout.get('mouse_path_base_window_height', 1600) or 1600)
path_file = get_config_path("loot_path.json")
if os.path.exists(path_file):
with open(path_file, 'r', encoding='utf-8') as f:
raw_path = json.load(f)
if isinstance(raw_path, dict):
path_points = raw_path.get('points') or raw_path.get('path_points') or raw_path.get('offsets') or []
base_width = int(
raw_path.get('base_window_width')
or raw_path.get('window_width')
or raw_path.get('base_client_width')
or base_width
)
base_height = int(
raw_path.get('base_window_height')
or raw_path.get('window_height')
or raw_path.get('base_client_height')
or base_height
)
else:
path_points = raw_path
if path_points:
scale_x = client_width / max(base_width, 1)
scale_y = client_height / max(base_height, 1)
signature = (client_width, client_height, base_width, base_height)
if signature != self._last_mouse_path_scale_signature:
print(
f">>> [扫雷路径] 当前窗口 {client_width}x{client_height}"
f"基准 {base_width}x{base_height},缩放 x={scale_x:.3f} y={scale_y:.3f}"
)
self._last_mouse_path_scale_signature = signature
scaled_points = []
for point in path_points:
if not isinstance(point, (list, tuple)) or len(point) < 2:
continue
dx = int(round(float(point[0]) * scale_x))
dy = int(round(float(point[1]) * scale_y))
scaled_points.append((dx, dy))
if scaled_points:
return scaled_points
x_scale, y_scale = 1.8, 0.8
for r in range(50, client_height // 2, 40):
angles = range(180, 360, 5) if (r // 40) % 2 == 0 else range(360, 180, -5)
for a in angles:
rad = math.radians(a)
path_points.append((int(r * math.cos(rad) * x_scale), int(r * math.sin(rad) * y_scale)))
return path_points
def mouse_sweep_scan(
self,
target_cursor_types,
click_wait_map=None,
max_scan_sec=15.0,
score_threshold=0.7,
return_on_first_click=False,
):
if random.random() < 0.1:
return False
hwnd = win32gui.FindWindow(None, WIN_TITLE)
if not hwnd:
return False
click_wait_map = click_wait_map or {}
target_cursor_types = set(target_cursor_types or [])
try:
left, top, right, bottom = get_wow_client_rect(hwnd)
client_width = max(right - left, 1)
client_height = max(bottom - top, 1)
center_x = left + (right - left) // 2
center_y = top + (bottom - top) // 2
move_cursor_hw(left + 50, top + 50, settle_sec=0.2)
_, default_hcursor, _ = win32gui.GetCursorInfo()
path_points = self._load_mouse_path_points(client_width, client_height)
start_time = time.time()
clicked_positions = []
for dx, dy in path_points:
if time.time() - start_time > max_scan_sec:
break
target_x = center_x + dx + random.randint(-5, 5)
target_y = center_y + dy + random.randint(-5, 5)
if not (left + 10 < target_x < right - 10 and top + 10 < target_y < bottom - 10):
continue
if any(math.dist((target_x, target_y), pos) < 30 for pos in clicked_positions):
continue
move_cursor_hw(target_x, target_y, settle_sec=0.02)
_, hcursor, _ = win32gui.GetCursorInfo()
if hcursor == 0 or hcursor == default_hcursor:
continue
ctype_name, score = self.cursor_mgr.get_type(hcursor)
if score > score_threshold and ctype_name in target_cursor_types:
print(f">>> [扫雷] 识别成功: {ctype_name} (得分: {score:.3f}), 执行右键点击")
hw_ctrl.right_click()
clicked_positions.append((target_x, target_y))
wait_sec = float(click_wait_map.get(ctype_name, 0.3))
if wait_sec > 0:
time.sleep(wait_sec)
if return_on_first_click:
move_cursor_hw(center_x, center_y, settle_sec=0.02)
return True
wait_start = time.time()
while time.time() - wait_start < 0.8:
_, curr_h, _ = win32gui.GetCursorInfo()
check_name, _ = self.cursor_mgr.get_type(curr_h)
if check_name == 'Point':
break
time.sleep(0.1)
time.sleep(random.uniform(0.1, 0.2))
elif score > 0.4:
print(f">>> [扫雷] 疑似图标: {ctype_name} (得分: {score:.3f} < 阈值 {score_threshold})")
move_cursor_hw(center_x, center_y, settle_sec=0.02)
return bool(clicked_positions)
except Exception as e:
print(f">>> [扫雷扫描] 出错: {e}")
return False
def mouse_scan_attack_target(self):
return self.mouse_sweep_scan(
['Attack'],
click_wait_map={'Attack': 0.3},
max_scan_sec=4.0,
score_threshold=0.6,
return_on_first_click=True,
)
def mouse_sweep_loot(self):
"""支持图标识别的高精度扫雷拾取。"""
return self.mouse_sweep_scan(
['LootAll', 'Skin'],
click_wait_map={'LootAll': 1.3, 'Skin': self.skinning_wait_sec},
max_scan_sec=15.0,
score_threshold=0.7,
return_on_first_click=False,
)
if random.random() < 0.1: return False
hwnd = win32gui.FindWindow(None, WIN_TITLE)
if not hwnd: return False
try:
rect = win32gui.GetWindowRect(hwnd)
left, top, right, bottom = rect
center_x = left + (right - left) // 2
center_y = top + (bottom - top) // 2
# 1. 强制“角落校准”采样
move_cursor_hw(left + 50, top + 50, settle_sec=0.2)
_, default_hcursor, _ = win32gui.GetCursorInfo()
# 2. 获取扫瞄路径点位
path_points = []
path_file = get_config_path("loot_path.json")
if os.path.exists(path_file):
with open(path_file, 'r') as f:
path_points = json.load(f)
else:
# 降级方案:生成拟人化半椭圆
x_scale, y_scale = 1.8, 0.8
for r in range(50, (bottom-top)//2, 40):
angles = range(180, 360, 5) if (r//40)%2==0 else range(360, 180, -5)
for a in angles:
rad = math.radians(a)
path_points.append((int(r * math.cos(rad) * x_scale), int(r * math.sin(rad) * y_scale)))
# 3. 开始沿路径扫瞄
start_time = time.time()
looted_positions = []
for dx, dy in path_points:
if time.time() - start_time > 15.0: break
target_x = center_x + dx + random.randint(-5, 5)
target_y = center_y + dy + random.randint(-5, 5)
if not (left+10 < target_x < right-10 and top+10 < target_y < bottom-10): continue
if any(math.dist((target_x, target_y), pos) < 30 for pos in looted_positions): continue
move_cursor_hw(target_x, target_y, settle_sec=0.02)
_, hcursor, _ = win32gui.GetCursorInfo()
if hcursor != 0 and hcursor != default_hcursor:
# 识别图标
ctype_name, score = self.cursor_mgr.get_type(hcursor)
if score > 0.7 and ctype_name in ['LootAll', 'Skin']:
print(f">>> [扫雷] 识别成功: {ctype_name} (得分: {score:.3f}), 执行右键点击")
hw_ctrl.right_click()
looted_positions.append((target_x, target_y))
# 根据类型等待
ws = 1.3 if ctype_name == 'LootAll' else self.skinning_wait_sec
time.sleep(ws)
# 等待指针恢复
w_start = time.time()
while time.time() - w_start < 0.8:
_, ch, _ = win32gui.GetCursorInfo()
check_name, _ = self.cursor_mgr.get_type(ch)
if check_name == 'Point': break
time.sleep(0.1)
time.sleep(random.uniform(0.1, 0.2))
elif score > 0.4:
print(f">>> [扫雷] 疑似图标: {ctype_name} (得分: {score:.3f} < 门槛 0.7)")
move_cursor_hw(center_x, center_y, settle_sec=0.02)
return True
except Exception as e:
print(f">>> [扫雷拾取] 出错: {e}")
return False
def execute_combat_logic(self, state):
if self.attack_loop_config:
cfg = self.attack_loop_config
if random.random() >= cfg['trigger_chance']:
return
hp = state.get('hp')
if hp is not None and cfg.get('hp_below_percent') and cfg.get('hp_key') and hp < cfg['hp_below_percent']:
hw_ctrl.press(cfg['hp_key'])
time.sleep(0.2)
mp = state.get('mp')
if mp is not None and cfg.get('mp_below_percent') and cfg.get('mp_key') and mp < cfg['mp_below_percent']:
hw_ctrl.press(cfg['mp_key'])
time.sleep(0.2)
# 每次攻击前选中目标
# hw_ctrl.press(KEY_LOOT)
for step in cfg['steps']:
key = step.get('key') or KEY_ATTACK
delay = float(step.get('delay', 0.5))
hw_ctrl.press(key)
time.sleep(delay)
return
# 默认:模拟手动按键的节奏
if random.random() < 0.3:
hw_ctrl.press(KEY_ATTACK)
time.sleep(0.5)
def execute_logic(self, state):
current_time = time.time()
target_hp = state.get('target_hp', 0)
combat_error_signal = int(state.get('combat_error_signal', 255) or 255)
effective_target = bool(state['target'] and target_hp > 0)
in_combat_or_target = bool(state['combat'] or effective_target)
if in_combat_or_target:
self.target_acquired_time = (self.target_acquired_time or current_time)
self.tab_no_target_count = 0
if effective_target:
is_new_target = (target_hp > self.last_target_hp + 5)
if is_new_target:
self._has_braked_for_target = False
# 1. 刹车逻辑
if not self._has_braked_for_target and 0 < target_hp < self.last_target_hp:
hw_ctrl.keyDown('s')
time.sleep(0.05)
hw_ctrl.keyUp('s')
self._has_braked_for_target = True
# 2. 根据战斗信号执行纠偏动作
if (
combat_error_signal == 100
and (current_time - self.last_interaction_time) >= self.combat_signal_retry_sec
):
hw_ctrl.press(KEY_LOOT)
self.last_interaction_time = current_time
if (
combat_error_signal == 50
and (current_time - self.last_turn_signal_time) >= self.combat_signal_retry_sec
):
hw_ctrl.keyDown(self.turn_error_key)
time.sleep(self.turn_error_hold_sec)
hw_ctrl.keyUp(self.turn_error_key)
self.last_turn_signal_time = current_time
self.last_target_hp = target_hp
if state['combat']:
self.execute_combat_logic(state)
self._was_in_combat_or_target = True
return
else:
if self._was_in_combat_or_target:
self.execute_disengage_loot()
self._was_in_combat_or_target = False
self.last_tab_time = current_time + 1.0
self.last_target_hp = 0
self.last_turn_signal_time = 0.0
self.last_interaction_time = 0.0
self._has_braked_for_target = False
return
self._was_in_combat_or_target = False
self.last_target_hp = 0
self.last_turn_signal_time = 0.0
self.last_interaction_time = 0.0
self._has_braked_for_target = False
self.tab_no_target_count = min(self.tab_no_target_count, 5)
if self.tab_no_target_count >= 5:
turn_key = random.choice(["a", "d"])
hw_ctrl.keyDown(turn_key)
time.sleep(random.uniform(0.3, 0.6))
hw_ctrl.keyUp(turn_key)
self.tab_no_target_count = 0
self.last_tab_time = current_time
self.last_turn_end_time = current_time
elif current_time - self.last_tab_time > random.uniform(0.8, 1.5):
hw_ctrl.press(KEY_TAB)
self.last_tab_time = current_time
self.tab_no_target_count += 1
# 在 main 循环中使用:从 game_state 获取 state
if __name__ == "__main__":
bot = AutoBot()
try:
while True:
state = parse_game_state()
if state:
target_hp = state.get('target_hp')
if target_hp is not None:
hp_part = f"血量:{state['hp']}% 目标血:{target_hp}% | 法力:{state['mp']}% | "
else:
hp_part = f"血量:{state['hp']}% | 法力:{state['mp']}% | "
print(
f"\r[状态] {hp_part}"
f"战斗:{'YES' if state['combat'] else 'NO '} | "
f"目标:{'YES' if state['target'] else 'NO '}",
end=""
)
bot.execute_logic(state)
time.sleep(0.1)
except KeyboardInterrupt:
print("\n已停止。")