first commit

game_state.py

@@ -0,0 +1,167 @@
+import json
+import os
+import sys
+import time
+import ctypes
+import pyautogui
+import pygetwindow as gw
+from PIL import Image
+
+# 解决 Windows 高 DPI 缩放问题
+try:
+    ctypes.windll.shcore.SetProcessDpiAwareness(1)
+except Exception:
+    pass
+
+# 默认布局参数
+_DEFAULTS = {
+    "pixel_size": 17,
+    "block_start_x": 30,
+    "scan_region_width": 155,
+    "scan_region_height": 15,
+    "offset_left": 20,
+    "offset_top": 45,
+}
+
+SCREENSHOT_DIR = 'screenshot'
+CONFIG_FILE = 'game_state_config.json'
+
+
+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():
+    base = _config_base()
+    p = os.path.join(base, CONFIG_FILE)
+    if os.path.exists(p):
+        return p
+    if getattr(sys, 'frozen', False) and getattr(sys, '_MEIPASS', ''):
+        p2 = os.path.join(sys._MEIPASS, CONFIG_FILE)
+        if os.path.exists(p2):
+            return p2
+    return p
+
+
+def load_layout_config():
+    """加载 game_state_config.json，返回布局参数字典"""
+    path = _get_config_path()
+    cfg = dict(_DEFAULTS)
+    if os.path.exists(path):
+        try:
+            with open(path, 'r', encoding='utf-8') as f:
+                loaded = json.load(f)
+                for k in _DEFAULTS:
+                    if k in loaded:
+                        cfg[k] = loaded[k]
+        except Exception:
+            pass
+    cfg['center_offset'] = cfg['pixel_size'] // 2
+    return cfg
+
+
+def save_layout_config(cfg):
+    """保存布局配置到 game_state_config.json"""
+    path = _get_config_path()
+    out = {k: cfg[k] for k in _DEFAULTS if k in cfg}
+    with open(path, 'w', encoding='utf-8') as f:
+        json.dump(out, f, indent=2, ensure_ascii=False)
+    return path
+
+
+def get_wow_info():
+    windows = gw.getWindowsWithTitle('魔兽世界')
+    if not windows:
+        return None
+    wow = windows[0]
+    cfg = load_layout_config()
+    return {
+        "left": wow.left + cfg['offset_left'],
+        "top": wow.top + cfg['offset_top'],
+        "layout": cfg,
+    }
+
+
+def parse_game_state():
+    pos = get_wow_info()
+    if not pos:
+        return None
+
+    cfg = pos['layout']
+    w = cfg['scan_region_width']
+    h = cfg['scan_region_height']
+    pixel_size = cfg['pixel_size']
+    block_start_x = cfg['block_start_x']
+    center_offset = cfg['center_offset']
+
+    # 截图并保存到 screenshot 文件夹
+    screenshot = pyautogui.screenshot(region=(pos['left'], pos['top'], w, h))
+    os.makedirs(SCREENSHOT_DIR, exist_ok=True)
+    screenshot.save(os.path.join(SCREENSHOT_DIR, 'game_state.png'))
+
+    # 按截图像素分段：每 pixel_size 一格 → hp, mp, combat, target, logistics_death, x, y
+    CHANNEL_NAMES = ('hp', 'mp', 'combat', 'target', 'logistics_death', 'x', 'y')
+    for idx, name in enumerate(CHANNEL_NAMES):
+        left = block_start_x + idx * pixel_size
+        box = (left, 0, left + pixel_size, pixel_size)
+        crop = screenshot.crop(box)
+        crop.save(os.path.join(SCREENSHOT_DIR, f'game_state_{name}.png'))
+
+    # 每格取中心像素，避免裁到边缘混合色
+    def get_val(idx):
+        sample_x = block_start_x + (idx * pixel_size) + center_offset
+        sample_y = center_offset
+        return screenshot.getpixel((sample_x, sample_y))
+
+    state = {}
+    # idx 0=血(红), 1=法(蓝), 2=战斗(绿), 3=目标(红)
+    hp_px = get_val(0)
+    state['hp'] = round(hp_px[0] / 255 * 100)
+    state['target_hp'] = round(hp_px[1] / 255 * 100)
+
+    mp_px = get_val(1)
+    state['mp'] = round(mp_px[2] / 255 * 100)
+
+    cb_px = get_val(2)
+    state['combat'] = cb_px[1] > 150  # 绿色通道
+
+    tg_px = get_val(3)
+    state['target'] = tg_px[0] > 150  # 红色通道
+
+    # 第 5 像素：后勤与死亡 (R=空格数, G=耐久比例, B=死亡状态)
+    p5 = get_val(4)
+    state['free_slots'] = p5[0]
+    state['durability'] = round(p5[1] / 255.0, 3)
+    state['death_state'] = 0 if p5[2] < 50 else (1 if p5[2] < 200 else 2)  # 0存活 1尸体 2灵魂
+
+    # 第 6 像素：x 坐标
+    p6 = get_val(5)
+    state['x'] = round((p6[0] + p6[1]/100), 2)
+    raw_deg =  (p6[2] / 255) * 360.0
+    state['facing'] = (360.0 - raw_deg) % 360.0
+
+    # 第 7 像素：y 坐标
+    p7 = get_val(6)
+    state['y'] = round((p7[0] + p7[1]/100), 2)
+
+    return state
+
+if __name__ == "__main__":
+    print("开始监控魔兽状态... (Ctrl+C 退出)")
+    try:
+        while True:
+            state = parse_game_state()
+            if state:
+                death_txt = ('存活', '尸体', '灵魂')[state.get('death_state', 0)]
+                print(
+                    f"\r[状态] 血:{state['hp']}% 法:{state['mp']}% 目标血:{state.get('target_hp', 0)}% | "
+                    f"战斗:{'Y' if state['combat'] else 'N'} 目标:{'Y' if state['target'] else 'N'} | "
+                    f"空格:{state.get('free_slots', 0)} 耐久:{state.get('durability', 0):.0%} {death_txt} | "
+                    f"x:{state['x']} y:{state['y']} 朝向:{state['facing']:.1f}°",
+                    end=""
+                )
+            time.sleep(0.5)
+    except KeyboardInterrupt:
+        print("\n已停止。")