vision.py

import cv2
import numpy as np
import subprocess
import sys
import time

TEMPLATES = sys.argv[1:]  # liste des images à chercher
THRESHOLD = 0.88

def screencap():
    proc = subprocess.Popen(
        ["adb", "exec-out", "screencap", "-p"],
        stdout=subprocess.PIPE
    )
    img = proc.stdout.read()
    return cv2.imdecode(np.frombuffer(img, np.uint8), cv2.IMREAD_COLOR)

while True:
    screen = screencap()

    for template_path in TEMPLATES:
        template = cv2.imread(template_path)
        h, w = template.shape[:2]

        result = cv2.matchTemplate(screen, template, cv2.TM_CCOEFF_NORMED)
        _, max_val, _, max_loc = cv2.minMaxLoc(result)

        if max_val >= THRESHOLD:
            x = max_loc[0] + w // 2
            y = max_loc[1] + h // 2

            print(f"{x},{y}")
            sys.exit(0)

    time.sleep(0.3)