#! /bin/bash
#
# Runs a test from the test directory and compare the resulting VTU files.
#
# Usage:
#
# runTest.sh REFERENCE_RESULT_FILE TEST_RESULT_FILE TEST_BINARY TEST_ARGS
#

function usage() {
    echo "Usage:"
    echo
    echo "runTest.sh TEST_TYPE TEST_BINARY [TEST_ARGS]"
    echo "where TEST_TYPE can either be --plain or --simulation"
};

TEST_TYPE="$1"
TEST_NAME="$2"
TEST_ARGS="${@:3:100}"

# make sure we have at least 2 parameters
if test "$#" -lt 2; then
    echo "Wrong number of parameters"
    echo
    usage
    exit 1
fi

# find the binary in the its folder
TEST_BINARY=$(find -type f -executable -name "$TEST_NAME")
    
# make sure the binary is of the test is present
if ! test -x "$TEST_BINARY"; then
    echo "$TEST_NAME does not exist or is not executable"
    echo
    usage
    exit 1
fi

#run the test
echo "######################"
echo "# Running test '$TEST_NAME'"
echo "######################"


case "$TEST_TYPE" in
    "--simulation")
        RND="$RANDOM"
        "$TEST_BINARY" $TEST_ARGS | tee "test-$RND.log"
        RET="${PIPESTATUS[0]}"
        if test "$RET" != "0"; then
            echo "Executing the binary failed!"
            rm "test-$RND.log"
            exit 1
        fi

        # compare the results
        echo "######################"
        echo "# Comparing results"
        echo "######################"
        SIM_NAME=$(grep "Initializing problem" test-$RND.log | sed "s/.*\"\(.*\)\".*/\1/" | head -n1)
        TEST_RESULT=$(ls $SIM_NAME-0*.vtu $SIM_NAME-0*.vtp 2> /dev/null | sort | tail -n 1)
        rm "test-$RND.log"
        if ! test -r "$TEST_RESULT"; then
            echo "File $TEST_RESULT does not exist or is not readable"
            exit 1
        fi

        REFERENCE_RESULT="referencesolutions/$TEST_RESULT"
        if ! test -r "$REFERENCE_RESULT"; then
            echo "File $REFERENCE_RESULT does not exist or is not readable"
            exit 1
        fi
        

        if ! python bin/fuzzycomparevtu.py "$REFERENCE_RESULT" "$TEST_RESULT"; then
            echo "The files \"$TEST_RESULT\" and \"$REFERENCE_RESULT\" are different."
            echo "Make sure the contents of \"$TEST_RESULT\" are still valid and "
            echo "make it the reference result if necessary."
            exit 1
        fi
        
        # SUCCESS!!!!!!
        echo "Result and reference result are identical" 
        exit 0

        ;;

    "--simulation-diffusion")
        if ! "$TEST_BINARY" $TEST_ARGS ; then
            echo "Executing the binary failed!"
            exit 1
        fi
        if ! python bin/fuzzycomparevtu.py referencesolutions/mimeticdiffusion-00001.vtu mimeticdiffusion-00001.vtu; then
            echo "The files \"mimeticdiffusion-00001.vtu\" and \"referencesolutions/mimeticdiffusion-00001.vtu\" are different."
            echo "Make sure the contents of \"mimeticdiffusion-00001.vtu\" are still valid and "
            echo "make it the reference result if necessary."
            exit 1
        fi

        exit 0
        ;;
    
    "--parallel-simulation")
        RND="$RANDOM"
        mpirun -np 4 "$TEST_BINARY" $TEST_ARGS | tee "test-$RND.log"
        RET="${PIPESTATUS[0]}"
        if test "$RET" != "0"; then
            echo "Executing the binary failed!"
            rm "test-$RND.log"
            exit 1
        fi
        rm "test-$RND.log"

        # # compare the results
        # echo "######################"
        # echo "# Comparing results"
        # echo "######################"
        # SIM_NAME=$(grep "Writing result file for" test-$RND.log | sed "s/.*\"\(.*\)\".*/\1/" | head -n1)
        # TEST_RESULT=$(ls $SIM_NAME*.vtu $SIM_NAME*.vtp 2> /dev/null | sort | tail -n 1)
        # rm "test-$RND.log"
        # if ! test -r "$TEST_RESULT"; then
        #     echo "File $TEST_RESULT does not exist or is not readable"
        #     exit 1
        # fi

        # REFERENCE_RESULT="referencesolutions/$TEST_RESULT"
        # if ! test -r "$REFERENCE_RESULT"; then
        #     echo "File $REFERENCE_RESULT does not exist or is not readable"
        #     exit 1
        # fi
        

        # if ! python bin/fuzzycomparevtu.py "$REFERENCE_RESULT" "$TEST_RESULT"; then
        #     echo "The files \"$TEST_RESULT\" and \"$REFERENCE_RESULT\" are different."
        #     echo "Make sure the contents of \"$TEST_RESULT\" are still valid and "
        #     echo "make it the reference result if necessary."
        #     exit 1
        # fi
        
        # # SUCCESS!!!!!!
        # echo "Result and reference result are identical" 
        exit 0

        ;;

    "--parameters")
        HELP_MSG="$($TEST_BINARY --help)"
        if test "$(echo $HELP_MSG | grep -i usage == '')"; then
            echo "$TEST_BINARY did not accept '--help' parameter"
            exit 1
        fi
        HELP_MSG2="$($TEST_BINARY -h)"
        if test "$HELP_MSG" != "$HELP_MSG2"; then
            echo "Output of $TEST_BINARY different when passing '--help' and '-h'"
            exit 1
        fi

        RND="$RANDOM"
        cat > "paramfile-$RND.ini" <<EOF
EndTime=100
InitialTimeStepSize=100
UndefinedParam="blubb"
EOF
        $TEST_BINARY --parameter-file="paramfile-$RND.ini"
        ;;

    "--plain")
        if ! "$TEST_BINARY" $TEST_ARGS; then
            exit 1
        fi

        exit 0
        ;;
esac