import sys
import time
from multiprocessing import freeze_support

from MCamera.mn_camera import *

if __name__ == '__main__':
    freeze_support()
    detector = MoveNetAlgorithmPlugin()

    Camera.add_source(0)
    cap = Camera()
    cap.start_record()

    while True:
        start_time = time.time()

        # frame = cap.get_frame()

        data = cap.get_cache()
        if not data:
            continue
        frame = data[FRAME_DAT]
        catch_time = data[CATCH_TIME]
        lm_list = data[MN_RESULT]
        # print(lm_list)

        cv2.imshow("frame", frame)
        cv2.imshow("pose_frame", detector.find_pose_with_drawing(frame))
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

        print(time.time() - catch_time)
        
        # print(time.time() - start_time)

    cap.release()
    cv2.destroyAllWindows()
    sys.exit()