button.py

import cv2


class Button:
    def __init__(self, position, width, height, text):
        self.position = position
        self.width = width
        self.height = height
        self.text = text
        self.colorFill = (255, 255, 255)
        self.colorBorder = (0, 0, 0)
        self.colorText = (50, 50, 50)
        self.colorFillHighlight = (150, 150, 150)

    def draw(self, frame):
        cv2.rectangle(frame, self.position,
                      (self.position[0] + self.width,
                       self.position[1] + self.height),
                      self.colorFill, cv2.FILLED)
        cv2.rectangle(frame, self.position,
                      (self.position[0] + self.width,
                       self.position[1] + self.height),
                      self.colorBorder, cv2.LINE_4)
        cv2.putText(frame, self.text, (self.position[0] + 40, self.position[1] + 60),
                    cv2.FONT_HERSHEY_PLAIN, 2, self.colorText, 2)

    def checkClick(self, frame, x, y):
        if (self.isBetween(x, self.position[0], self.position[0] + self.width) and
                self.isBetween(y, self.position[1], self.position[1] + self.height)):
            cv2.rectangle(frame, self.position,
                          (self.position[0] + self.width,
                           self.position[1] + self.height),
                          self.colorFillHighlight, cv2.FILLED)
            cv2.putText(frame, self.text, (self.position[0] + 30, self.position[1] + 70),
                        cv2.FONT_HERSHEY_PLAIN, 4, self.colorText, 4)
            return True
        return False

    def isBetween(self, x, x1, x2):
        return x1 <= x <= x2