import cv2

class FaceDetector:
    def __init__(self, cascade_path='haarcascade_frontalface_default.xml'):
        self.face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + cascade_path)

    def find_faces(self, img, draw=True):
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        faces = self.face_cascade.detectMultiScale(gray, 1.3, 5)

        bboxs = []
        for i, (x, y, w, h) in enumerate(faces):
            bbox = (x, y, w, h)
            bboxs.append([i, bbox, None])  # 'None' 用于保持与原代码结构一致，因为OpenCV不提供置信度分数

            if draw:
                img = self.fancy_draw(img, bbox)

        return img, bboxs

    def fancy_draw(self, img, bbox, l=30, t=5, rt=1):
        x, y, w, h = bbox
        x1, y1 = x + w, y + h

        # 绘制矩形
        cv2.rectangle(img, bbox, (0, 255, 255), rt)

        # 绘制角落
        # 左上角
        cv2.line(img, (x, y), (x+l, y), (255, 0, 255), t)
        cv2.line(img, (x, y), (x, y+l), (255, 0, 255), t)
        # 右上角
        cv2.line(img, (x1, y), (x1-l, y), (255, 0, 255), t)
        cv2.line(img, (x1, y), (x1, y+l), (255, 0, 255), t)
        # 左下角
        cv2.line(img, (x, y1), (x+l, y1), (255, 0, 255), t)
        cv2.line(img, (x, y1), (x, y1-l), (255, 0, 255), t)
        # 右下角
        cv2.line(img, (x1, y1), (x1-l, y1), (255, 0, 255), t)
        cv2.line(img, (x1, y1), (x1, y1-l), (255, 0, 255), t)

        return img

# 使用示例
if __name__ == "__main__":
    detector = FaceDetector()
    cap = cv2.VideoCapture(0)  # 使用默认摄像头

    while True:
        success, img = cap.read()
        img, bboxs = detector.find_faces(img)

        cv2.imshow("Image", img)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

    cap.release()
    cv2.destroyAllWindows()