run_capture.py

#!/usr/bin/env python3
import time
import argparse
import cv2
import os
from capture_image import CaptureImage
from video_live_generator import VideoLiveGenerator

parser = argparse.ArgumentParser()
parser.add_argument("dir_name", help="directory to save images to")
parser.add_argument("port", help="camera port to read from")
args = parser.parse_args()

generator = VideoLiveGenerator(args.port)
wait_time = 3

if not os.path.isdir(args.dir_name):
    os.mkdir(args.dir_name)

with CaptureImage(generator) as c:

    angle = input("angle: ")
    distance = input("distance: ")

    while True:
        c.update_frame()
        key = cv2.waitKey(wait_time)
        # pos = (generator.get_frame().shape[1] /2, generator.get_frame().shape[0]/2)
        if key == ord('q'):
            break
        if key == ord('c'):
            c.capture(args.dir_name, angle, distance)
            if input("continue? (y/n) ") == "n":
                break

            angle = input("angle: ")
            distance = input("distance: ")