54 lines
2.2 KiB
Python
54 lines
2.2 KiB
Python
import os
|
|
|
|
|
|
def run_loop(runtime, client, queue, *, clock, sleep, iterations=None,
|
|
sync_every=150, heartbeat_every=900, flush_every=300,
|
|
frame_interval=0.0):
|
|
counters = {"frames": 0, "syncs": 0, "beats": 0, "flushes": 0, "errors": 0}
|
|
i = 0
|
|
while iterations is None or i < iterations:
|
|
i += 1
|
|
try:
|
|
runtime.step(clock())
|
|
counters["frames"] += 1
|
|
except Exception:
|
|
counters["errors"] += 1
|
|
if sync_every and i % sync_every == 0:
|
|
cfg = client.fetch_config()
|
|
if cfg is not None:
|
|
from edge.zones import zones_from_config
|
|
runtime.apply_zones(zones_from_config(cfg))
|
|
counters["syncs"] += 1
|
|
if heartbeat_every and i % heartbeat_every == 0:
|
|
if client.send_heartbeat({"fps": 0.0, "frames": counters["frames"]}):
|
|
counters["beats"] += 1
|
|
if flush_every and i % flush_every == 0:
|
|
queue.flush()
|
|
counters["flushes"] += 1
|
|
if frame_interval:
|
|
sleep(frame_interval)
|
|
return counters
|
|
|
|
|
|
def main() -> None: # pragma: no cover (integration entrypoint, exercised on device)
|
|
import time
|
|
from edge.config import load_config
|
|
from edge.cloudclient import CloudClient
|
|
from edge.alert import PostClient
|
|
from edge.queue import OfflineQueue
|
|
from edge.runtime import PipelineRuntime
|
|
from edge.adapters.uvc_source import UvcSource
|
|
from edge.adapters.movenet_pose import MoveNetPoseAdapter
|
|
|
|
cfg = load_config(dict(os.environ))
|
|
nonce = lambda: os.urandom(8).hex()
|
|
now = time.time
|
|
base = cfg.cafe24_url.rstrip("/")
|
|
client = CloudClient(base, cfg.device_id, cfg.secret, nonce, now)
|
|
post = PostClient(base + "/api/events", cfg.device_id, cfg.secret, nonce, now)
|
|
queue = OfflineQueue(os.environ.get("QUEUE_DB", "queue.db"), post)
|
|
source = UvcSource(int(os.environ.get("UVC_DEVICE", "0")))
|
|
pose = MoveNetPoseAdapter(os.environ["MOVENET_ONNX"])
|
|
runtime = PipelineRuntime(cfg, source, pose, queue)
|
|
run_loop(runtime, client, queue, clock=now, sleep=time.sleep, iterations=None,
|
|
frame_interval=1.0 / float(os.environ.get("TARGET_FPS", "12")))
|