#!/bin/bash
#
# DISPENSION CONFIDENTIAL
#
#  [2020] - [2021] Dispension Industries Limited.
#  Portions Copyright © 2014-2020 Atlantean Technical Solutions Limited
#   with full licensing rights granted to Dispension & Successors.
#
#  All Rights Reserved.
#
# NOTICE:  All information contained herein is, and remains
# the property of Dispension Industries  Limited.
# The intellectual and technical concepts contained
# herein are proprietary to Dispension Industries  Limited
# and its suppliers and may be covered by U.S. and Foreign Patents,
# patents in process, and are protected by trade secret or copyright law.
# Dissemination of this information or reproduction of this material
# is strictly forbidden unless prior written permission is obtained
# from Dispension Industries  Limited.
#



HERE=$PWD
DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
cd $DIR/..

PORT=8099
IMGDIR="/tmp/yoloserv_in"
OUTDIR="/tmp/yoloserv_out"
ZZZ=5


export PYTHONPATH="$PYTHONPATH:./yolov5-face_Jan1"
export WEIGHTS="./yolov5-face_Jan1/runs/train/exp/weights/yolov5m6_face.pt"


function f_test(){

    wget http://localhost:$PORT/process/
}

function f_start(){
    mkdir -p $IMGDIR
    mkdir -p $OUTDIR
    echo $$ > var/yoloserv.pid
    while [ -e var/yoloserv.pid ]
    do
        python3 src/yoloserv.py $PORT $IMGDIR $OUTDIR $WEIGHTS
        sleep $ZZZ
    done
}

function f_stop(){
    rm var/yoloserv.pid
    wget http://localhost:$PORT/svc_stop
}



echo "Running $0 with option $1 at $DIR"
case $1 in

 "start")   f_start
            ;;
 "restart")   f_stop
            f_start
            ;;
 "reload")   f_reload
            ;;
 "stop")   f_stop
            ;;
  *)   echo "Error. $1 is not a $0 command."
        ;;
esac



cd $HERE