#!/bin/bash

readonly TEST_DATA="/usr/share/edgetpudemo"
#TODO: Use video_device.mp4 once vcodec is available.
readonly VIDEO_DEVICE_FILE="/home/mendel/video_test_yuy2.yuv"
readonly VIDEO_STREAM_FILE="${TEST_DATA}/video_stream.mp4"
readonly TPU_MODEL_FILE="${TEST_DATA}/mobilenet_ssd_v1_coco_quant_postprocess_edgetpu.tflite@Running MobileNet SSD v1 on Edge TPU"
readonly CPU_MODEL_FILE="${TEST_DATA}/mobilenet_ssd_v1_coco_quant_postprocess.tflite@Running MobileNet SSD v1 on CPU"
readonly LABELS_FILE="${TEST_DATA}/coco_labels.txt"

if [[ "$1" == "--device" ]]; then
  echo "Press 'q' to quit."
  echo "Press 'n' to switch between models."

  edgetpu_detect \
      --source "${VIDEO_DEVICE_FILE}" \
      --model "${TPU_MODEL_FILE},${CPU_MODEL_FILE}" \
      --labels "${LABELS_FILE}" \
      --filter car,truck \
      --max_area 0.1 \
      --color white \
      --loop \
      --displaymode window
# Disable stream demo for now.
# elif [[ "$1" == "--stream" ]]; then
#   echo "Press 'q' to quit."
#   echo "Press 'n' to switch between models."

#   SERVER_INDEX_HTML="${TEST_DATA}/index.html" edgetpu_detect_server \
#       --source "${VIDEO_STREAM_FILE}" \
#       --model "${TPU_MODEL_FILE},${CPU_MODEL_FILE}" \
#       --labels "${LABELS_FILE}" \
#       --filter car,truck \
#       --max_area 0.1 \
#       --color white \
#       --loop
else
  echo "Run on-device inference:"
  echo "  $0 --device"
  # echo "Run streaming server:"
  # echo "  $0 --stream"
fi