106 lines
4.2 KiB
Python
106 lines
4.2 KiB
Python
import argparse
|
|
from pathlib import Path
|
|
import sys
|
|
import time
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
if str(ROOT) not in sys.path:
|
|
sys.path.insert(0, str(ROOT))
|
|
|
|
from localcast.sensor_fusion.spout_output import (
|
|
CultCacheRenderFrameSource,
|
|
OpenGLSpoutPointRenderer,
|
|
RenderFrameFileSource,
|
|
SpoutOutputConfig,
|
|
write_cult_status,
|
|
write_status,
|
|
)
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="Publish render-frame packets as a named Spout texture for OBS.")
|
|
parser.add_argument("--frame-json", default=str(ROOT / "calibration" / "runs" / "synthetic-render-frame.json"))
|
|
parser.add_argument("--frame-cache", default=str(ROOT / "calibration" / "runs" / "visual-state.msgpack"))
|
|
parser.add_argument("--sender-name", default="LocalCastBridge Point Cloud")
|
|
parser.add_argument("--width", type=int, default=1920)
|
|
parser.add_argument("--height", type=int, default=1080)
|
|
parser.add_argument("--fps", type=float, default=60.0)
|
|
parser.add_argument("--point-scale", type=float, default=1.0)
|
|
parser.add_argument("--demo-if-missing", action="store_true")
|
|
parser.add_argument("--demo-point-count", type=int, default=4096)
|
|
parser.add_argument("--status", default=str(ROOT / "calibration" / "runs" / "stream-spout-status.json"))
|
|
parser.add_argument("--status-cache", default=str(ROOT / "calibration" / "runs" / "visual-stream-status.msgpack"))
|
|
parser.add_argument("--source", choices=["cultcache", "json"], default="cultcache")
|
|
parser.add_argument("--duration", type=float, help="Optional run duration in seconds for smoke tests.")
|
|
args = parser.parse_args()
|
|
|
|
frame_path = Path(args.frame_json)
|
|
frame_cache_path = Path(args.frame_cache)
|
|
status_path = Path(args.status)
|
|
status_cache_path = Path(args.status_cache)
|
|
config = SpoutOutputConfig(
|
|
sender_name=args.sender_name,
|
|
width=args.width,
|
|
height=args.height,
|
|
fps=args.fps,
|
|
point_scale=args.point_scale,
|
|
demo_point_count=args.demo_point_count,
|
|
)
|
|
if args.source == "cultcache":
|
|
source = CultCacheRenderFrameSource(frame_cache_path, demo_if_missing=args.demo_if_missing, demo_point_count=args.demo_point_count)
|
|
frame_source_name = str(frame_cache_path)
|
|
else:
|
|
source = RenderFrameFileSource(frame_path, demo_if_missing=args.demo_if_missing, demo_point_count=args.demo_point_count)
|
|
frame_source_name = str(frame_path)
|
|
frame_interval = 1.0 / max(1.0, args.fps)
|
|
start = time.monotonic()
|
|
last_status = 0.0
|
|
|
|
with OpenGLSpoutPointRenderer(config) as renderer:
|
|
while True:
|
|
now = time.monotonic()
|
|
if args.duration is not None and now - start >= args.duration:
|
|
break
|
|
frame = source.current_frame(now, config)
|
|
ok = renderer.render(frame)
|
|
if now - last_status >= 1.0:
|
|
write_status(
|
|
status_path,
|
|
sender_name=config.sender_name,
|
|
frames_sent=renderer.frames_sent,
|
|
point_count=len(frame.points),
|
|
frame_path=Path(frame_source_name),
|
|
last_error=None if ok else "SpoutGL sendTexture returned false",
|
|
)
|
|
write_cult_status(
|
|
status_cache_path,
|
|
sender_name=config.sender_name,
|
|
frames_sent=renderer.frames_sent,
|
|
point_count=len(frame.points),
|
|
frame_source=frame_source_name,
|
|
last_error=None if ok else "SpoutGL sendTexture returned false",
|
|
)
|
|
last_status = now
|
|
sleep_for = frame_interval - (time.monotonic() - now)
|
|
if sleep_for > 0:
|
|
time.sleep(sleep_for)
|
|
write_status(
|
|
status_path,
|
|
sender_name=config.sender_name,
|
|
frames_sent=renderer.frames_sent,
|
|
point_count=0,
|
|
frame_path=Path(frame_source_name),
|
|
last_error="stopped",
|
|
)
|
|
write_cult_status(
|
|
status_cache_path,
|
|
sender_name=config.sender_name,
|
|
frames_sent=renderer.frames_sent,
|
|
point_count=0,
|
|
frame_source=frame_source_name,
|
|
last_error="stopped",
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|