import time
import pyautogui
import pyscreeze
import cv2
# 屏幕縮放系數(shù) mac縮放是2 windows一般是1
screenScale=1
#事先讀取按鈕截圖
target= cv2.imread(r"./image/ssk.png",cv2.IMREAD_GRAYSCALE)
# 先截圖
screenshot=pyscreeze.screenshot('my_screenshot.png')
# 讀取圖片 灰色會(huì)快
temp = cv2.imread(r'my_screenshot.png',cv2.IMREAD_GRAYSCALE)
theight, twidth = target.shape[:2]
tempheight, tempwidth = temp.shape[:2]
print("目標(biāo)圖寬高："+str(twidth)+"-"+str(theight))
print("模板圖寬高："+str(tempwidth)+"-"+str(tempheight))
# 先縮放屏幕截圖 INTER_LINEAR INTER_AREA
scaleTemp=cv2.resize(temp, (int(tempwidth / screenScale), int(tempheight / screenScale)))
stempheight, stempwidth = scaleTemp.shape[:2]
print("縮放后模板圖寬高："+str(stempwidth)+"-"+str(stempheight))
# 匹配圖片
res = cv2.matchTemplate(scaleTemp, target, cv2.TM_CCOEFF_NORMED)
mn_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res)
if(max_val>=0.9):
    # 計(jì)算出中心點(diǎn)
    top_left = max_loc
    bottom_right = (top_left[0] + twidth, top_left[1] + theight)
    tagHalfW=int(twidth/2)
    tagHalfH=int(theight/2)
    tagCenterX=top_left[0]+tagHalfW
    tagCenterY=top_left[1]+tagHalfH
    #左鍵點(diǎn)擊屏幕上的這個(gè)位置
    pyautogui.click(tagCenterX,tagCenterY,button='left') # 點(diǎn)擊
else:
    print ("沒找到")

import cv2
import numpy as np
import pyautogui
import time
class ImageClicker:
    def __init__(self, target_image_path, retry_count=3, retry_interval=1, match_threshold=0.75):
        self.target_image_path = target_image_path
        self.retry_count = retry_count
        self.retry_interval = retry_interval
        self.match_threshold = match_threshold
        # 加載目標(biāo)圖片
        self.target_image = cv2.imread(self.target_image_path)
        # 提取目標(biāo)圖片的 SIFT 特征
        self.sift = cv2.SIFT_create()
        self.kp1, self.des1 = self.sift.detectAndCompute(self.target_image, None)
    def click_image(self):
        for i in range(self.retry_count):
            try:
                # 獲取瀏覽器窗口截圖
                screenshot = pyautogui.screenshot()
                screenshot = cv2.cvtColor(np.array(screenshot), cv2.COLOR_RGB2BGR)
                # 提取截圖的 SIFT 特征
                kp2, des2 = self.sift.detectAndCompute(screenshot, None)
                # 進(jìn)行特征匹配
                bf = cv2.BFMatcher()
                matches = bf.knnMatch(self.des1, des2, k=2)
                # 使用 Lowe's Ratio Test 篩選匹配結(jié)果
                good = []
                for m, n in matches:
                    if m.distance < self.match_threshold * n.distance:  # 使用 match_threshold 閾值
                        good.append([m])
                # 計(jì)算目標(biāo)元素的位置
                if len(good) > 0:
                    src_pts = np.float32([self.kp1[m[0].queryIdx].pt for m in good]).reshape(-1, 1, 2)
                    dst_pts = np.float32([kp2[m[0].trainIdx].pt for m in good]).reshape(-1, 1, 2)
                    M, mask = cv2.findHomography(src_pts, dst_pts, cv2.RANSAC, 5.0)
                    h, w = self.target_image.shape[:2]
                    pts = np.float32([[0, 0], [0, h - 1], [w - 1, h - 1], [w - 1, 0]]).reshape(-1, 1, 2)
                    dst = cv2.perspectiveTransform(pts, M)
                    # 計(jì)算點(diǎn)擊坐標(biāo)
                    x = int((dst[0][0][0] + dst[2][0][0]) / 2)  # 計(jì)算水平方向的中間位置
                    y = int((dst[0][0][1] + dst[2][0][1]) / 2)  # 計(jì)算垂直方向的中間位置
                    # 點(diǎn)擊目標(biāo)元素
                    pyautogui.click(x, y)
                    return True  # 點(diǎn)擊成功
            except Exception as e:
                print(f"點(diǎn)擊失?。簕e}")
                time.sleep(self.retry_interval)
        return False  # 點(diǎn)擊失敗
# 使用示例
image_clicker = ImageClicker('4.png', retry_count=5,
                             retry_interval=2,
                             match_threshold=0.8)  # 設(shè)置 match_threshold 為 0.8
if image_clicker.click_image():
    print("點(diǎn)擊成功！")
else:
    print("點(diǎn)擊失??！")

image_clicker = ImageClicker('4.png', retry_count=5, retry_interval=2, match_threshold=0.8)
if image_clicker.click_image():
    print("點(diǎn)擊成功！")
else:
    print("點(diǎn)擊失敗！")