# -*- coding: utf-8 -*-
# @Author  : WuFanLong
# @Time    : 2021/10/25 10:27
# @File    : face_recognition.py
# @Software: PyCharm 
# @Function:

import cv2
import numpy
from PIL import Image, ImageDraw, ImageFont


# cv2解决绘制中文乱码
def cv2ImgAddText(img, text, left, top, textColor=(0, 255, 0), textSize=20):
    if (isinstance(img, numpy.ndarray)):  # 判断是否OpenCV图片类型
        img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
    # 创建一个可以在给定图像上绘图的对象
    draw = ImageDraw.Draw(img)
    # 字体的格式
    fontStyle = ImageFont.truetype(
        "font/simsun.ttc", textSize, encoding="utf-8")
    # 绘制文本
    draw.text((left, top), text, textColor, font=fontStyle)
    # 转换回OpenCV格式
    return cv2.cvtColor(numpy.asarray(img), cv2.COLOR_RGB2BGR)


class Call3:
    @staticmethod
    def call_back3():
        recognizer = cv2.face.LBPHFaceRecognizer_create()
        recognizer.read('face_trainer/trainer.yml')
        faceCascade = cv2.CascadeClassifier(r'.\model\haarcascade_frontalface_default.xml')
        font = cv2.FONT_HERSHEY_SIMPLEX

        idnum = 0
        file = 'user.txt'
        with open(file, 'r', encoding='utf-8') as f:
            line = f.readlines()
        for i in range(len(line)):
            line[i] = line[i].strip('\n')
        print(line)
        # names = ['None', 'WuFanLong', 'LiuXin']
        names = ['None']
        names += line
        print(names)

        cam = cv2.VideoCapture(0, cv2.CAP_DSHOW)
        ok = cam.isOpened()
        print(ok)

        minW = 0.1 * cam.get(3)
        minH = 0.1 * cam.get(4)

        while True:
            ret, img = cam.read()
            gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
            faces = faceCascade.detectMultiScale(
                gray,
                scaleFactor=1.2,
                minNeighbors=5,
                minSize=(int(minW), int(minH))
            )
            for (x, y, w, h) in faces:
                cv2.rectangle(img, (x, y), (x + w, y + h), (0, 255, 0), 2)
                idnum, confidence = recognizer.predict(gray[y:y + h, x:x + w])
                print(idnum, confidence)

                if confidence < 45:
                    idnum = names[idnum]
                    confidence = "{0}%".format(round(100 - confidence))
                else:
                    idnum = "unknown"
                    confidence = "{0}%".format(round(100 - confidence))

                # cv2.putText(img, str(idnum), (x + 5, y - 5), font, 1, (0, 0, 255), 1)
                img = cv2ImgAddText(img, str(idnum), x + 5, y - 5, (0, 0, 255), 60)
                cv2.putText(img, str(confidence), (x + 5, y + h - 5), font, 1, (0, 0, 0), 1)

            cv2.imshow('camera', img)
            k = cv2.waitKey(10)
            if k == 27:
                break

        cam.release()
        cv2.destroyAllWindows()