#!/usr/bin/env bash
set -euo pipefail

if [[ $# -lt 2 ]]; then
  echo "Usage: $0 <rtsp_url> <host_output_dir>"
  exit 1
fi

RTSP_URL="$1"
HOST_OUTPUT_DIR="$2"

mkdir -p "$HOST_OUTPUT_DIR"

docker run -d \
  --name people-flow-rtsp \
  --restart unless-stopped \
  --network host \
  --gpus all \
  --shm-size 1g \
  -v "$HOST_OUTPUT_DIR:/opt/people-flow/output" \
  people-flow-rtsp:x86-cuda \
  --config /opt/people-flow/configs/docker_x86_config.yaml \
  --output-dir /opt/people-flow/output \
  --device cuda:0 \
  rtsp \
  --input "$RTSP_URL"