import cv2
from predict import Predict

video_path = r'E:\pythonProject\ftc\ftcdata\ftc_data.mp4'


class VideoProcess:
    def __init__(self):
        self.predict = Predict()
        self.frame = None

    def capture(self):
        # 0 表示打开本地摄像头
        # 1 2 3 外接摄像头
        # 路径/播放视频
        # 打开摄像头 视频采集
        cap = cv2.VideoCapture(video_path)
        while cap.isOpened():
            # 读取一帧一帧的图像
            retval, frame = cap.read()
            self.frame = frame
            if not retval:
                print('can not read frame')
                break
            # 翻转图像 1为水平翻转
            # frame = cv2.flip(frame, 1)
            # 调用检测
            self.process()
            # imshow 会默认创建一个窗口
            cv2.imshow('frame', frame)
            key = cv2.waitKey(25)
            if key == ord('z'):
                break
        cap.release()
        cv2.destroyAllWindows()

    def process(self):
        raise NotImplementedError

    def showInfo(self, info, org=(100, 100), fontScale=2.5, color=(0, 255, 0), thickness=3):
        cv2.putText(self.frame, str(info), org=org, fontFace=cv2.FONT_ITALIC,
                    fontScale=fontScale, color=color, thickness=thickness)


if __name__ == '__main__':
    video = VideoProcess()
    video.process()