Python | OpenCV 專案：天堂私服遊戲輔助（一）主程式 GUI 框架與 HP/MP 血量監控

# window_capture.py
from ctypes import windll
import numpy as np
import win32gui
import win32process
import win32ui

GA_ROOT = 2   # GetAncestor flag：取頂層視窗

def window_from_point(x, y):
    """回傳 (x, y) 螢幕座標下的頂層視窗 (hwnd, title, pid)；找不到回傳 (None, '', 0)"""
    hwnd = win32gui.WindowFromPoint((x, y))
    if not hwnd:
        return None, "", 0
    hwnd = win32gui.GetAncestor(hwnd, GA_ROOT)
    _, pid = win32process.GetWindowThreadProcessId(hwnd)
    return hwnd, win32gui.GetWindowText(hwnd), pid

def capture_window(hwnd):
    """用 PrintWindow 擷取視窗畫面：被遮擋也能抓，最小化會拿到黑畫面"""
    left, top, right, bottom = win32gui.GetWindowRect(hwnd)
    w, h = right - left, bottom - top
    if w <= 0 or h <= 0:
        return None

    hwnd_dc  = win32gui.GetWindowDC(hwnd)
    mfc_dc   = win32ui.CreateDCFromHandle(hwnd_dc)
    save_dc  = mfc_dc.CreateCompatibleDC()
    save_bmp = win32ui.CreateBitmap()
    save_bmp.CreateCompatibleBitmap(mfc_dc, w, h)
    save_dc.SelectObject(save_bmp)

    # flag=2 (PW_RENDERFULLCONTENT)：支援 DWM 合成畫面，對多數現代應用必要
    result = windll.user32.PrintWindow(hwnd, save_dc.GetSafeHdc(), 2)
    img = np.frombuffer(save_bmp.GetBitmapBits(True), dtype=np.uint8).reshape((h, w, 4))

    win32gui.DeleteObject(save_bmp.GetHandle())
    save_dc.DeleteDC()
    mfc_dc.DeleteDC()
    win32gui.ReleaseDC(hwnd, hwnd_dc)

    return img[:, :, :3].copy() if result == 1 else None   # BGRA → BGR

# hp_monitor.py
import cv2
import json
import numpy as np
import os

CONFIG_FILE    = "config.json"
DEFAULT_HP_ROI = (10, 10, 200, 20)
DEFAULT_MP_ROI = (10, 35, 200, 20)


class HPMonitor:
    def __init__(self):
        self.hp_roi = DEFAULT_HP_ROI
        self.mp_roi = DEFAULT_MP_ROI
        self._load_config()

    def _load_config(self):
        if not os.path.exists(CONFIG_FILE):
            return
        with open(CONFIG_FILE) as f:
            cfg = json.load(f)
        for key in ("hp_roi", "mp_roi"):
            if key in cfg:
                setattr(self, key, tuple(cfg[key]))

    def _color_ratio(self, frame, roi, lower, upper) -> float:
        """HSV 色相範圍遮罩 + 最右側有色像素欄 / 總寬度"""
        x, y, w, h = roi
        if w == 0 or h == 0:
            return 0.0
        hsv  = cv2.cvtColor(frame[y:y+h, x:x+w], cv2.COLOR_BGR2HSV)
        mask = cv2.inRange(hsv, np.array(lower), np.array(upper))
        cols = np.any(mask > 0, axis=0)
        if not np.any(cols):
            return 0.0
        return min(1.0, (int(np.max(np.where(cols))) + 1) / w)

    def read(self, frame):
        hp = self._color_ratio(frame, self.hp_roi,
                               [0, 100, 80], [10, 255, 255])
        mp = self._color_ratio(frame, self.mp_roi,
                               [100, 100, 80], [130, 255, 255])
        return hp, mp

# main.py（第一篇版本）
import json
import queue
import threading
import time
import tkinter as tk
from tkinter import ttk, scrolledtext

import cv2
import win32api

from window_capture import capture_window, window_from_point
from hp_monitor     import HPMonitor, CONFIG_FILE

VK_LBUTTON = 0x01
VK_ESCAPE  = 0x1B


# ── Shared State ───────────────────────────────────────────────
class SharedState:
    """執行緒安全的共享資料容器"""
    def __init__(self):
        self._lock   = threading.Lock()
        self.hp      = 1.0
        self.mp      = 1.0
        self.hwnd    = None     # 由「🎯 選取視窗」鎖定的目標視窗 handle
        self.running = False

    def update(self, **kwargs):
        with self._lock:
            for k, v in kwargs.items():
                setattr(self, k, v)

    def get(self, *keys):
        with self._lock:
            if len(keys) == 1:
                return getattr(self, keys[0])
            return tuple(getattr(self, k) for k in keys)


# ── Worker Thread 1：監控迴圈 ──────────────────────────────────
def monitor_loop(state: SharedState, monitor: HPMonitor, log_q: queue.Queue):
    hwnd = state.get("hwnd")
    log_q.put(f"監控執行緒啟動，hwnd={hwnd}")
    # 正常流程下 hwnd 不會為 None（_start 已擋），這裡直接進主迴圈

    while state.get("running"):
        frame = capture_window(hwnd)
        if frame is None:
            time.sleep(0.2)
            continue
        hp, mp = monitor.read(frame)
        state.update(hp=hp, mp=mp)
        time.sleep(0.2)

    log_q.put("監控執行緒結束")


# ── tkinter 主視窗 ─────────────────────────────────────────────
class LineageAssistant:
    BG          = "#1a1a2e"
    FG          = "#e0e0e0"
    RED         = "#e94560"
    BLUE        = "#4fc3f7"
    GREEN       = "#00ff99"
    AMBER       = "#f5a623"   # 圖示靜態色（參考私服輔助面板的琥珀風格）
    AMBER_HOVER = "#ffc857"   # 滑鼠移入時變亮金
    MUTED       = "#555577"   # 禁用狀態

    def __init__(self):
        self.state     = SharedState()
        self.monitor   = HPMonitor()       # 主執行緒 + Worker 共用
        self.log_q     = queue.Queue()
        self.t_monitor = None
        self._build_ui()
        self._poll()
        self.root.mainloop()

    # ── 扁平化圖示按鈕：與面板同色、琥珀色 icon、hover 變亮金 ──
    def _make_icon_btn(self, parent, text, cmd, font):
        btn = tk.Button(
            parent, text=text,
            bg=self.BG, fg=self.AMBER,
            activebackground=self.BG, activeforeground=self.AMBER_HOVER,
            disabledforeground=self.MUTED,
            font=font, relief="flat", bd=0, cursor="hand2",
            command=cmd)

        def on_enter(_):
            if btn["state"] != "disabled":
                btn.config(fg=self.AMBER_HOVER)

        def on_leave(_):
            if btn["state"] != "disabled":
                btn.config(fg=self.AMBER)

        btn.bind("<Enter>", on_enter)
        btn.bind("<Leave>", on_leave)
        return btn

    def _build_ui(self):
        self.root = tk.Tk()
        self.root.title("天堂私服輔助  v0.1")
        self.root.configure(bg=self.BG)
        self.root.resizable(False, False)

        # ── 標題 ──
        tk.Label(self.root, text="⚔  天堂私服輔助",
                 bg=self.BG, fg=self.FG,
                 font=("Consolas", 16, "bold")).pack(pady=10)

        # ── 工具列：🎯 靠左、⚙ 校準 靠右，中間不放任何會撐寬的元件 ──
        toolbar = tk.Frame(self.root, bg=self.BG)
        toolbar.pack(padx=20, pady=(0, 4), fill="x")

        self.btn_pick = self._make_icon_btn(
            toolbar, "🎯 選取視窗", self._pick_window,
            font=("Consolas", 11, "bold"))
        self.btn_pick.pack(side="left")

        self.btn_calib = self._make_icon_btn(
            toolbar, "⚙ 校準血條", self._calibrate,
            font=("Consolas", 11, "bold"))
        self.btn_calib.pack(side="right")

        # ── 目標視窗資訊：獨立一列，長標題不會擠到按鈕 ──
        self.target_var = tk.StringVar(value="目標視窗：未選擇")
        tk.Label(self.root, textvariable=self.target_var,
                 bg=self.BG, fg=self.FG,
                 font=("Consolas", 10), anchor="w"
                 ).pack(padx=20, pady=(0, 6), fill="x")

        # ── HP / MP 血條顯示 ──
        bar_frame = tk.Frame(self.root, bg=self.BG)
        bar_frame.pack(padx=20, fill="x")

        style = ttk.Style()
        style.theme_use("default")
        style.configure("Red.Horizontal.TProgressbar",
                        troughcolor="#330000", background="#e94560")
        style.configure("Blue.Horizontal.TProgressbar",
                        troughcolor="#001133", background="#4fc3f7")

        tk.Label(bar_frame, text="HP", bg=self.BG, fg=self.RED,
                 font=("Consolas", 11, "bold"), width=4
                 ).grid(row=0, column=0, sticky="w")
        self.hp_var = tk.DoubleVar(value=1.0)
        ttk.Progressbar(bar_frame, variable=self.hp_var, maximum=1.0,
                        length=300, style="Red.Horizontal.TProgressbar"
                        ).grid(row=0, column=1, padx=5, pady=4)
        self.hp_lbl = tk.Label(bar_frame, text="100%",
                               bg=self.BG, fg=self.FG,
                               font=("Consolas", 10), width=6)
        self.hp_lbl.grid(row=0, column=2)

        tk.Label(bar_frame, text="MP", bg=self.BG, fg=self.BLUE,
                 font=("Consolas", 11, "bold"), width=4
                 ).grid(row=1, column=0, sticky="w")
        self.mp_var = tk.DoubleVar(value=1.0)
        ttk.Progressbar(bar_frame, variable=self.mp_var, maximum=1.0,
                        length=300, style="Blue.Horizontal.TProgressbar"
                        ).grid(row=1, column=1, padx=5, pady=4)
        self.mp_lbl = tk.Label(bar_frame, text="100%",
                               bg=self.BG, fg=self.FG,
                               font=("Consolas", 10), width=6)
        self.mp_lbl.grid(row=1, column=2)

        # ── 控制按鈕：單顆 ▶ / ■ toggle ──
        btn_frame = tk.Frame(self.root, bg=self.BG)
        btn_frame.pack(pady=10)

        self.btn_toggle = self._make_icon_btn(
            btn_frame, "▶  啟動", self._toggle,
            font=("Consolas", 13, "bold"))
        self.btn_toggle.pack()

        # ── 狀態列 ──
        self.status_var = tk.StringVar(value="就緒")
        tk.Label(self.root, textvariable=self.status_var,
                 bg=self.BG, fg="#888888",
                 font=("Consolas", 9)).pack()

        # ── 日誌視窗 ──
        self.log_box = scrolledtext.ScrolledText(
            self.root, width=52, height=10, state="disabled",
            bg="#0d0d1a", fg=self.GREEN,
            font=("Consolas", 9), insertbackground="white")
        self.log_box.pack(padx=10, pady=(5, 10))

    def _log(self, msg: str):
        self.log_q.put(msg)

    def _flush_log(self):
        while not self.log_q.empty():
            msg = self.log_q.get_nowait()
            ts  = time.strftime("%H:%M:%S")
            self.log_box.config(state="normal")
            self.log_box.insert("end", f"{ts}  {msg}\n")
            self.log_box.see("end")
            self.log_box.config(state="disabled")

    # ── 🎯 滑鼠點擊選窗 ───────────────────────────────────────
    def _pick_window(self):
        """進入選窗模式：滑鼠變十字，下一次左鍵點擊即抓取該位置的頂層視窗"""
        if self.state.get("running"):
            self._log("請先停止監控再重選目標視窗")
            return
        self._log("🎯 請點擊目標遊戲視窗（ESC 可取消）")
        # 選窗期間把整排按鈕都鎖住，避免 crosshair 狀態下誤點其他功能
        for btn in (self.btn_pick, self.btn_calib, self.btn_toggle):
            btn.config(state="disabled")
        self.root.config(cursor="crosshair")
        self._poll_pick()

    def _poll_pick(self):
        """輪詢全域滑鼠狀態：偵測使用者在哪個視窗按下左鍵"""
        # ESC 取消
        if win32api.GetAsyncKeyState(VK_ESCAPE) & 0x8000:
            self._reset_pick()
            self._log("已取消選擇視窗")
            return
        # 左鍵按下
        if win32api.GetAsyncKeyState(VK_LBUTTON) & 0x8000:
            x, y = win32api.GetCursorPos()
            hwnd, title, pid = window_from_point(x, y)
            self._reset_pick()
            if hwnd:
                self.state.update(hwnd=hwnd)
                label = title or "<無標題>"
                short = label if len(label) <= 30 else label[:28] + "…"
                self.target_var.set(f"目標：{short}  (pid={pid})")
                self._log(f"已鎖定：{label}  hwnd={hwnd}  pid={pid}")
            else:
                self._log("未取得有效視窗，請再試一次")
            return
        self.root.after(30, self._poll_pick)

    def _reset_pick(self):
        self.root.config(cursor="")
        for btn in (self.btn_pick, self.btn_calib, self.btn_toggle):
            btn.config(state="normal")

    # ── ⚙ ROI 校準 ───────────────────────────────────────
    def _calibrate(self):
        if self.state.get("running"):
            self._log("請先停止監控再校準")
            return
        hwnd = self.state.get("hwnd")
        if not hwnd:
            self._log("請先按「🎯 選取視窗」選取目標視窗")
            return

        frame = capture_window(hwnd)
        if frame is None:
            self._log("擷取失敗，視窗可能被最小化")
            return

        self._log("框選 HP 血條（紅色），Enter 確認")
        hp = cv2.selectROI("校準 HP 血條（Enter 確認 / C 重選）",
                           frame, fromCenter=False, showCrosshair=True)
        cv2.destroyAllWindows()

        self._log("框選 MP 魔力條（藍色），Enter 確認")
        mp = cv2.selectROI("校準 MP 魔力條（Enter 確認 / C 重選）",
                           frame, fromCenter=False, showCrosshair=True)
        cv2.destroyAllWindows()

        if hp[2] == 0 or mp[2] == 0:
            self._log("校準已取消（未框選有效範圍）")
            return

        self.monitor.hp_roi = tuple(hp)
        self.monitor.mp_roi = tuple(mp)
        self._save_config()
        self._log(f"✅ 校準完成：HP={hp}, MP={mp}")

    def _save_config(self):
        cfg = {
            "hp_roi": list(self.monitor.hp_roi),
            "mp_roi": list(self.monitor.mp_roi),
        }
        with open(CONFIG_FILE, "w") as f:
            json.dump(cfg, f, indent=2)
        self._log(f"已寫入 {CONFIG_FILE}")

    # ── ▶ / ■ 切換：一顆按鈕 toggle ──────────────────────────
    def _toggle(self):
        if self.state.get("running"):
            self._stop()
        else:
            self._start()

    def _start(self):
        if not self.state.get("hwnd"):
            self._log("請先按「🎯 選取視窗」選取目標視窗")
            return
        self.state.update(running=True)
        self.t_monitor = threading.Thread(
            target=monitor_loop,
            args=(self.state, self.monitor, self.log_q),
            daemon=True)
        self.t_monitor.start()
        self.btn_toggle.config(text="■  停止")
        self.status_var.set("運行中...")

    def _stop(self):
        self.state.update(running=False)
        self.btn_toggle.config(text="▶  啟動")
        self.status_var.set("已停止")
        self._log("使用者手動停止")

    def _poll(self):
        """每 200ms 將 SharedState 的值同步到 UI"""
        hp, mp = self.state.get("hp", "mp")
        self.hp_var.set(hp)
        self.mp_var.set(mp)
        self.hp_lbl.config(text=f"{hp * 100:.0f}%")
        self.mp_lbl.config(text=f"{mp * 100:.0f}%")
        self._flush_log()
        self.root.after(200, self._poll)


if __name__ == "__main__":
    LineageAssistant()