StarRailCopilot/module/base/template.py

import cv2
import numpy as np
from PIL import Image
import module.config.server as server


class Template:
    def __init__(self, file):
        """
        Args:
            file (dict[str], str): Filepath of template file.
        """
        self.server = server.server
        self.file = file[self.server] if isinstance(file, dict) else file
        self.image = np.array(Image.open(self.file))

    def match(self, image, similarity=0.85):
        """
        Args:
            image:
            similarity (float): 0 to 1.

        Returns:
            bool: If matches.
        """
        res = cv2.matchTemplate(np.array(image), self.image, cv2.TM_CCOEFF_NORMED)
        _, sim, _, _ = cv2.minMaxLoc(res)
        # print(self.file, sim)
        return sim > similarity

    def match_multi(self, image, similarity=0.85):
        """
        Args:
            image:
            similarity (float): 0 to 1.

        Returns:
            np.ndarray: np.array([[x0, y0], [x1, y1])
        """
        result = cv2.matchTemplate(np.array(image), self.image, cv2.TM_CCOEFF_NORMED)
        result = np.array(np.where(result > similarity)).T

        return result