import argparse
import face_recognition.comparison
import cv2


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Visualise face recognition algorithms.')
    parser.add_argument('--camera', '-c', type=int, default=0,
                        help='Numeric id of the camera')
    parser.add_argument('--fullscreen', '-f', action='store_true',
                        help='Display output full screen')
    parser.add_argument('--clockwise', action='store_true',
                        help='Rotate clockwise')
    parser.add_argument('--counter-clockwise', action='store_true',
                        help='Rotate counter clockwise')
    parser.add_argument('--cascade', default='haarcascade_frontalface_alt2.xml',
                        help='Cascade XML file to use (opencv format)')

    args = parser.parse_args()

    rotate = None
    if args.clockwise:
        rotate = cv2.ROTATE_90_CLOCKWISE
    if args.counter_clockwise:
        rotate = cv2.ROTATE_90_COUNTERCLOCKWISE

    face_recognition.comparison.main(args.camera, rotate, args.fullscreen, args.cascade)