blob: 3aa052ca622b9abc68d191f14e92c03f87ef8483 [file] [log] [blame]
import argparse
import logging
import signal
from .camera import make_camera
from .gstreamer import Display, run_gen
from .streaming.server import StreamingServer
from . import svg
EMPTY_SVG = str(svg.Svg())
def run_server(add_render_gen_args, render_gen):
logging.basicConfig(level=logging.INFO)
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--source',
help='/dev/videoN:FMT:WxH:N/D or .mp4 file or image file',
default='/dev/video0:YUY2:1280x720:30/1')
parser.add_argument('--bitrate', type=int, default=1000000,
help='Video streaming bitrate (bit/s)')
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 render_overlay(tensor, layout, command):
overlay = gen.send((tensor, layout, command))
server.send_overlay(overlay if overlay else EMPTY_SVG)
camera.render_overlay = render_overlay
signal.pause()
def run_app(add_render_gen_args, render_gen):
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--source',
help='/dev/videoN:FMT:WxH:N/D or .mp4 file or image file',
default='/dev/video0:YUY2:1280x720:30/1')
parser.add_argument('--downscale', type=float, default=2.0,
help='Downscale factor for .mp4 file rendering')
parser.add_argument('--display', type=Display, choices=Display, default=Display.FULLSCREEN,
help='Display mode')
add_render_gen_args(parser)
args = parser.parse_args()
if not run_gen(render_gen(args),
source=args.source,
downscale=args.downscale,
display=args.display):
print('Invalid source argument:', args.source)