#!/bin/bash

cd build

RED='\033[1;31m'
GREEN='\033[1;32m'
ORANGE='\033[1;33m'
PINK='\033[1;95m'
NC='\033[0m' # No Color

function print_output {
    if [ $1 -eq 0 ]; then
        echo -e "$2 ${GREEN}OK${NC}"
    elif [ $1 -eq 1 ]; then
        echo -e "$2 ${RED}FATAL ERROR${NC}"
    elif [ $1 -eq 2 ] || [ $1 -eq 10 ]; then
        echo -e "$2 ${PINK}CUDNN ERROR${NC}"
    elif [ $1 -eq 4 ] || [ $1 -eq 12 ]; then
        echo -e "$2 ${PINK}TENSORRT ERROR${NC}"
    elif [ $1 -eq 8 ]; then
        echo -e "$2 ${PINK}CUDNN vs TENSORRT ERROR${NC}"
    elif [ $1 -eq 6 ]; then
        echo -e "$2 ${PINK}CUDNN & TENSORTRT ERROR${NC}"
    elif [ $1 -eq 14 ]; then
        echo -e "$2 ${PINK}ERROR FOR EVERY CHECK${NC}"
    else
        echo -e "$2 ${RED}NOT OKAY (OPENCV maybe)${NC}"
    fi 

} 

out_file=results.log
rm $out_file

modes=( 1 ) # only FP32
# modes=( 1 2 ) # FP32 and FP16
# modes=( 1 2 3 ) # FP32, FP16 and INT8

for i in "${modes[@]}"
do
    rm *rt
    if [ $i -eq 1 ]
    then
        export TKDNN_MODE=FP32
        echo -e "${ORANGE}Test FP32${NC}"
    fi
    if [ $i -eq 2 ]
    then
        export TKDNN_MODE=FP16
        echo -e "${ORANGE}Test FP16${NC}"
    fi
    if [ $i -eq 3 ]
    then
        export TKDNN_MODE=INT8
        export TKDNN_CALIB_LABEL_PATH=../demo/COCO_val2017/all_labels.txt
        export TKDNN_CALIB_IMG_PATH=../demo/COCO_val2017/all_images.txt
        echo -e "${ORANGE}Test INT8${NC}"
	fi

    ./test_imuodom &>> $out_file
    res_imuodom=$?
    print_output $res_imuodom test_imuodom

    ./test_resnet101_cnet &>> $out_file
    res_resnet101_cnet=$?
    print_output $res_resnet101_cnet test_resnet101_cnet

    ./test_yolo3 &>> $out_file
    res_yolo3=$?
    print_output $res_yolo3 test_yolo3

    ./test_yolo3_flir &>> $out_file
    res_yolo3_flir=$?
    print_output $res_yolo3_flir test_yolo3_flir

    ./test_yolo3_512 &>> $out_file
    res_yolo3_512=$?
    print_output $res_yolo3_512 test_yolo3_512

    ./test_yolo3_tiny &>> $out_file
    res_yolo3_tiny=$?
    print_output $res_yolo3_tiny test_yolo3_tiny

    ./test_csresnext50-panet-spp &>> $out_file
    res_csresnext50panetspp=$?
    print_output $res_csresnext50panetspp "test_csresnext50-panet-spp"

    ./test_mobilenetv2ssd &>> $out_file
    res_mobilenetv2ssd=$?
    print_output $res_mobilenetv2ssd test_mobilenetv2ssd

    ./test_yolo3_tiny512 &>> $out_file
    res_yolo3_tiny512=$?
    print_output $res_yolo3_tiny512 test_yolo3_tiny512

    ./test_yolo_tiny &>> $out_file
    res_yolo_tiny=$?
    print_output $res_yolo_tiny test_yolo_tiny

    ./test_mobilenetv2ssd512 &>> $out_file
    res_mobilenetv2ssd512=$?
    print_output $res_mobilenetv2ssd512 test_mobilenetv2ssd512

    ./test_mnist &>> $out_file
    res_mnist=$?
    print_output $res_mnist test_mnist

    ./test_yolo &>> $out_file
    res_yolo=$?
    print_output $res_yolo test_yolo

    ./test_yolo3_berkeley &>> $out_file
    res_yolo3_berkeley=$?
    print_output $res_yolo3_berkeley test_yolo3_berkeley

    ./test_yolo_voc &>> $out_file
    res_yolo_voc=$?
    print_output $res_yolo_voc test_yolo_voc

    ./test_dla34_cnet &>> $out_file
    res_dla34_cnet=$?
    print_output $res_dla34_cnet test_dla34_cnet

    ./test_yolo3_coco4 &>> $out_file
    res_yolo3_coco4=$?
    print_output $res_yolo3_coco4 test_yolo3_coco4

done

echo "If errors occured, check logfile $out_file"