#!/bin/sh

rm -rf work runtime_test.tmp final_test.tmp

SCRIPT_FOLDER="./../scripts"
TEST_BENCH_FOLDER="./../tb"
TEST_BENCH="${1%-*}"
TEST_FOLDER="test_source_code/tb_""$TEST_BENCH"
TEST_FILE=$(echo "${1}" | awk -F'-' '{if (NF>1) {print $NF}}')

BOLD=$(tput bold)
NORMAL=$(tput sgr0)

TEST_BENCH_PATH="$TEST_BENCH_FOLDER""/tb_""$TEST_BENCH"
if [ ! -f "$TEST_BENCH_PATH"".v" ]; then
    echo "testbench: ""$TEST_BENCH_PATH"": does not exit"
    exit 1
fi

run_test ()
{
    TEST_FILE_PATH=$1

    rm -rf work transcript

    echo
    echo "$BOLD==========   $TEST_BENCH - $(basename $TEST_FILE_PATH .S)   ==========$NORMAL"

    if [ ! -f "$TEST_FILE_PATH" ]; then
        echo "test file: ""$TEST_FILE_PATH"": does not exit"
        exit 1
    fi

    ./${SCRIPT_FOLDER}/get_bin.sh $TEST_FILE_PATH
    NB_TEST=$(python3 ./${SCRIPT_FOLDER}/gen_test.py $TEST_FILE_PATH | sed -n 's/^Generated \([[:digit:]]*\) tests$/\1/p')

    if [ -z $2 ]; then
        vsim -c -do "do simu.do; quit -f" >& /dev/null
    else
        vsim -do "do simu.do"
    fi

    print_result "$TEST_BENCH-$(basename $TEST_FILE_PATH .S)" "$NB_TEST"
}

print_result ()
{
    while read line; do
        if [[ $line = *"# Errors: "* ]]; then
            if [ $(echo "$line" | sed 's/^# Errors: \([[:digit:]]*\).*/\1/') -ne 0 ]; then
                cat ./transcript
                return 1
            fi
        fi
    done < ./transcript

    nb_test=0
    nb_pass=0
    nb_fail=0
    while read line; do
        if [[ $line = *"[FAIL]"* ]]; then
            echo "$line" | cut -c 3-
            ((nb_fail++))
            ((nb_test++))
        elif [[ $line == *"[PASS]"* ]]; then
            ((nb_pass++))
            ((nb_test++))
        elif [[ $line == *"** Warning"* ]]; then
            echo "$line" | cut -c 6-
        fi
    done < ./transcript

    if [ ! -z $2 ] && [ $2 -ne 0 ] && [ $nb_test -lt $2 ]; then
        echo -e "\033[0;31m$BOLD"Warning:"$NORMAL\033[0m Only $BOLD$nb_test$NORMAL tests were executed, at least $BOLD$2$NORMAL were expected."
    fi

    echo -e "[\033[0;34m$BOLD$1$NORMAL\033[0m]$BOLD Test: $nb_test | Passed: $NORMAL\033[0;32m$BOLD$nb_pass$NORMAL\033[0m$BOLD | Failed: $NORMAL\033[0;31m$BOLD$nb_fail$NORMAL\033[0m"
}

if [ -z "$TEST_FILE" ] || [ "$TEST_FILE" = "all" ]; then

    echo
    echo "$BOLD==========   $TEST_BENCH - Test Bench   ==========$NORMAL"

    ./${SCRIPT_FOLDER}/gen_simu_do.sh "$TEST_BENCH"

    if [ -z $2 ]; then
        vsim -c -do "do simu.do; quit -f" >& /dev/null
    else
        vsim -do "do simu.do"
    fi

    print_result "$TEST_BENCH-TestBench"
fi

if [ ! -z "$TEST_FILE" ]; then
    TEST_BENCH_PATH="$TEST_BENCH_FOLDER""/tb_""$TEST_BENCH""-dyn"
    if [ ! -f "$TEST_BENCH_PATH"".v" ]; then
        exit 0
    fi

    ./${SCRIPT_FOLDER}/gen_simu_do.sh "$TEST_BENCH""-dyn"

    if [ "$TEST_FILE" == "all" ]; then
        for f in "$TEST_BENCH_FOLDER""/""$TEST_FOLDER"/*; do
            run_test "$f" "$2"
        done
    else
        run_test "$TEST_BENCH_FOLDER""/""$TEST_FOLDER""/"$TEST_FILE".S" "$2"
    fi
fi

echo

exit 0