import argparse
import logging
import signal

from .camera import make_camera
from .streaming.server import StreamingServer

def run(add_render_gen_args, render_gen):
    logging.basicConfig(level=logging.INFO)

    parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument('--bitrate', type=int, default=1000000,
                        help='Video streaming bitrate (bit/s)')

    parser.add_argument('--source',
                        help='/dev/videoN:FMT:WxH:N/D or .mp4 file or image file',
                        default='/dev/video0:YUY2:1280x720:30/1')
    add_render_gen_args(parser)
    args = parser.parse_args()

    gen = render_gen(args)
    camera = make_camera(args.source, next(gen))
    assert camera is not None

    with StreamingServer(camera, args.bitrate) as server:
        def on_image(tensor, layout):
            overlay = gen.send((tensor, layout, None))
            server.send_overlay(overlay)

        camera.on_image = on_image
        signal.pause()