index.py

import time
import cv2
import numpy as np
import pyautogui
import threading
import keyboard
import tkinter as tk
from tkinter import filedialog, messagebox, Menu
from PIL import ImageGrab


class MainWindow:
    def __init__(self, root):
        self.root = root
        self.root.title(".0_0.")
        self.root.geometry("400x320")
        self.image_path = None
        self.runnum = 0  # 运行次数
        self.running = False  # 循环控制器
        self.lock = threading.Lock()  # 创建进程锁
        self.escape_thread = None  # 存储ESC键监听线程
        # 创建一个StringVar变量，并与Radio控件绑定
        self.radio = tk.StringVar(value="1")
        # 创建一个DoubleVar变量，并与Scale控件绑定
        self.scale_var = tk.DoubleVar()
        self.threshold = 0.9  # 设置匹配阈值，根据实际情况调整
        # 缩放比
        self.scaling_ratio_x = 1
        self.scaling_ratio_y = 1
        # 获取固定点位
        self.center_point = None
        self.scale(root)  # 获取屏幕缩放比

        menubar = Menu(root)  # 创建菜单栏
        menubar.add_cascade(label="退出", command=root.quit)  # 创建退出菜单
        menubar.add_cascade(label="帮助", command=self.help_info)  # 创建帮助菜单
        root.config(menu=menubar)  # 配置菜单栏到主窗口

        ## 图片上传frame
        frame_image = tk.Frame(root, pady=10, relief=tk.RAISED)
        frame_image.pack(fill=tk.X, side="top")
        tk.Button(
            frame_image,
            text="选择图片",
            width=15,
            cursor="hand2",
            command=self.select_image,
        ).pack(side="left", padx=15)  # 设置水平和垂直间距
        self.label = tk.Label(frame_image, bg="gray", pady=4, text="未选择图片")
        self.label.pack(padx=15, fill="x")

        ## 进度条frame
        frame_scale = tk.Frame(root, relief=tk.RAISED)
        frame_scale.pack(fill=tk.X, side="top")
        tk.Label(frame_scale, width=15, pady=4, text="调整频率（秒/10）:").pack(side="left", padx=15, pady=18)
        tk.Scale(
            frame_scale,
            from_=1,
            to=10,
            cursor="hand2",
            orient=tk.HORIZONTAL,
            variable=self.scale_var,
        ).pack(padx=15, fill="x") # 进度条调节延时

        ## 模式frame
        frame_radio = tk.Frame(root, padx=30, pady=10, bd=2, relief=tk.RAISED)
        frame_radio.pack(
            side="top", fill="x", padx=15
        )  # 使用pack方法将Frame添加到窗口中，并设置填充方式
        tk.Label(frame_radio, text="模式选择", fg="blue").pack()
        tk.Radiobutton(
            frame_radio,
            text="单次匹配: (每次只会选中第一张)",
            variable=self.radio,
            cursor="hand2",
            value="1",
        ).pack(anchor=tk.W)
        tk.Radiobutton(
            frame_radio,
            text="全匹配: (每次符合条件的全部点击)",
            variable=self.radio,
            cursor="hand2",
            value="2",
        ).pack(anchor=tk.W)
        tk.Radiobutton(
            frame_radio,
            text="坐标连点：(首次匹配的位置连续点击)",
            variable=self.radio,
            cursor="hand2",
            value="3",
        ).pack(anchor=tk.W)
        self.radio.trace_add('write', self.on_radio_changed)
    
        self.button2 = tk.Button(root, text="开始 or 重启", command=self.start)
        self.button2.pack(padx=30, side="right")

        root.protocol("WM_DELETE_WINDOW", self.on_closing)  # 绑定窗口关闭事件

    def on_radio_changed(self, *args):
        print(f"当前模式为：{self.radio.get()}")
        if(self.radio.get() != "3"):
            self.center_point = None
    def help_info(self):
        messagebox.showinfo(
            "帮助信息",
            "使用说明：\n1. 选择需要识别的图片\n2. 点击开始按钮，程序会自动识别并点击图片\n3. ESC、Delete、Backspace会终止进程\n4. 点击退出按钮，程序会退出\n5. 调整进度条可以修改进程刷新率\n6. 模式选择可以修改点击逻辑\n7. 点击帮助按钮，会显示帮助信息\n8. 重新选择图片后，进程必须重新开始\n9. 显示有问题建议拖拽放大窗口",
        )

    def on_key_press(self, event):
        if event.name == "esc" or event.name == "backspace" or event.name == "delete":
            with self.lock:
                self.running = False  # 设置标志变量为False，以中断主循环
                # 显示消息框
                messagebox.showinfo("提示", "已关闭进程")
                print(f"{event.name} 键被按下，已关闭进程")
            # 停止监听（可选，如果只需要一次响应）
            keyboard.unhook_all()

    def on_closing(self):
        # if self.escape_thread:
        #     self.escape_thread.join() # 不能处理，会死循环 是等待当前进程完成才能执行后续代码
        if self.running:  # 如果程序正在运行，则等待程序结束
            if messagebox.askokcancel("Quit", "确定关闭程序?"):
                self.running = False
                root.destroy()
        else:  # 如果程序没有正在运行，则直接关闭窗口
            root.destroy()

    def scale(self, root):
        # 获取屏幕的逻辑宽度和高度（以像素为单位）
        logical_width = root.winfo_screenwidth()
        logical_height = root.winfo_screenheight()
        # 获取屏幕的物理宽度和高度（以毫米为单位）
        physical_width = root.winfo_screenmmwidth()
        physical_height = root.winfo_screenmmheight()
        # 将物理尺寸从毫米转换为英寸，1英寸=25.4毫米
        physical_width_inches = physical_width / 25.4
        physical_height_inches = physical_height / 25.4
        # 计算每英寸点数（DPI）
        dpi_x = logical_width / physical_width_inches
        dpi_y = logical_height / physical_height_inches
        # 假设标准DPI为96，计算缩放比
        self.scaling_ratio_x = dpi_x / 96
        self.scaling_ratio_y = dpi_y / 96

    def select_image(self):
        if self.image_path:
            self.running = False
            messagebox.showwarning("注意", "必须重启进程")
        self.image_path = filedialog.askopenfilename()
        if self.image_path:
            self.label.config(text="已选择图片:" + self.image_path)
        else:
            self.label.config(text="未选择图片")

    def loop_default(self):
        while self.running and self.image_path:
            if self.radio.get() == "3" and self.center_point is not None:
                pyautogui.click(self.center_point[0], self.center_point[1])
                time.sleep(self.scale_var.get() / 10)
                continue # 直接去下一次循环
            # print("loop_default")
            # 加载模板图像（你要匹配的小图）
            template_image_path = self.image_path
            template = cv2.imread(template_image_path, 0)  # 以灰度模式读取
            w, h = template.shape[::-1]  # 获取模板图像的宽度和高度

            # 截取当前屏幕图像
            screen = np.array(ImageGrab.grab())
            screen_gray = cv2.cvtColor(screen, cv2.COLOR_BGR2GRAY)  # 转换为灰度图

            # 使用模板匹配算法
            res = cv2.matchTemplate(screen_gray, template, cv2.TM_CCOEFF_NORMED)
            loc = np.where(res >= self.threshold)
            # print(f"Matched locations: {loc}", loc[0], res[res >= 0.8])
            # 获取匹配区域的中心点坐标
            if loc[0].size > 0:
                self.runnum = 0  # 重置计数器
                for pt in zip(*loc[::-1]):
                    if(self.running == False):
                        break
                    center_x = pt[0] + w // (2 * self.scaling_ratio_x)
                    center_y = pt[1] + h // (2 * self.scaling_ratio_y)

                    # 打印中心点坐标（用于调试）
                    # print(f'Matched center: ({center_x}, {center_y})')

                    # 模拟鼠标点击
                    print(f'Matched center: ({center_x}, {center_y})',self.radio.get())
                    pyautogui.click(center_x, center_y)

                    if(self.radio.get() == "3" and self.center_point == None):
                        self.center_point = (center_x, center_y)
                    # 可选：点击后等待一段时间，以便观察效果
                    if(self.radio.get() != "2"):
                        break  # 找到第一个匹配后就退出循环
                time.sleep(self.scale_var.get() / 10)  # 休眠0.1秒
            else:
                print("No match found")
                self.runnum += 1  # 累加计数器
                time.sleep(self.scale_var.get() / 5)  # 休眠0.2秒
                if self.runnum > 600:  # 如果运行次数超过600次，则停止循环
                    self.running = False
                    messagebox.showinfo("提示", "停止循环")
                    print("未找到符合图片，已停止循环进程")

    def start(self):
        self.runnum = 0
        if not self.image_path:
            return messagebox.showinfo("提示", "未选择图片无法开始")
        keyboard.hook(self.on_key_press)
        if not self.running:
            self.running = True
            loop_thread = threading.Thread(target=self.loop_default)
            loop_thread.start()  # 启动关闭程序的线程
            print("循环已启动")


if __name__ == "__main__":
    root = tk.Tk()
    root.wm_attributes("-topmost", 1)  # 窗口置顶
    app = MainWindow(root)
    root.mainloop()