#!/bin/bash

#-----------------------------------------------------------
# $Id$
#-----------------------------------------------------------

#-----[ variable ]------------------------------------------

declare    test_ko="Test KO";
declare    test_ok="Test OK";
declare    test_unknow="???????";
declare    tmp="${MORPHEO_HOME}/tmp/SelfTest";
declare    path="${MORPHEO_TOPLEVEL}/IPs/systemC/processor/Morpheo/";

declare -a directory=(
    "Common"

    "Behavioural"
    "Behavioural/Configuration"
    "Behavioural/Custom"

    "Behavioural/Generic/Counter"
    "Behavioural/Generic/Priority"
    "Behavioural/Generic/Queue_Control"                                 
    "Behavioural/Generic/Queue"                                 
    "Behavioural/Generic/RegisterFile/RegisterFile_Monolithic"  
    "Behavioural/Generic/RegisterFile/RegisterFile_Multi_Banked"
    "Behavioural/Generic/RegisterFile"                          
    "Behavioural/Generic/Select/Select_Priority_Fixed"           
    "Behavioural/Generic/Select"                                
    "Behavioural/Generic/Shifter"                               
    "Behavioural/Generic/Sort"                                  
    "Behavioural/Generic/Victim/Victim_Pseudo_LRU"              
    "Behavioural/Generic/Victim"                                

    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Execute_loop_Glue"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Execute_unit/Execute_unit/Functionnal_unit/Operation"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Execute_unit/Execute_unit/Functionnal_unit"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Execute_unit/Execute_unit/Load_store_unit"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Execute_unit/Execute_unit"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Execute_unit"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Read_unit/Read_unit/Read_queue"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Read_unit/Read_unit/Reservation_station"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Read_unit/Read_unit"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Read_unit"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Write_unit/Write_unit/Execute_queue"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Write_unit/Write_unit/Write_queue"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Write_unit/Write_unit"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Multi_Write_unit"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Network/Execution_unit_to_Write_unit"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Network/Read_unit_to_Execution_unit"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Network"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Register_unit/Register_unit_Glue"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop/Register_unit"
    "Behavioural/Core/Multi_Execute_loop/Execute_loop"
    "Behavioural/Core/Multi_Execute_loop"

    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Commit_unit"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Issue_queue"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/OOO_Engine_Glue"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Reexecute_unit"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Rename_unit/Load_Store_pointer_unit"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Rename_unit/Register_translation_unit/Dependency_checking_unit"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Rename_unit/Register_translation_unit/Free_List_unit"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Rename_unit/Register_translation_unit/Register_Address_Translation_unit"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Rename_unit/Register_translation_unit/Register_translation_unit_Glue"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Rename_unit/Register_translation_unit/Stat_List_unit"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Rename_unit/Register_translation_unit"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Rename_unit/Rename_unit_Glue"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Rename_unit/Rename_select"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Rename_unit"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine/Special_Register_unit"
    "Behavioural/Core/Multi_OOO_Engine/OOO_Engine"
    "Behavioural/Core/Multi_OOO_Engine"
    
    "Behavioural/Core/Multi_Front_end/Front_end/Context_State"
    "Behavioural/Core/Multi_Front_end/Front_end/Decod_unit/Decod/Instruction"
    "Behavioural/Core/Multi_Front_end/Front_end/Decod_unit/Decod"
    "Behavioural/Core/Multi_Front_end/Front_end/Decod_unit/Decod_queue"
    "Behavioural/Core/Multi_Front_end/Front_end/Decod_unit"
    "Behavioural/Core/Multi_Front_end/Front_end/Front_end_Glue"
    "Behavioural/Core/Multi_Front_end/Front_end/Ifetch_unit/Address_management"
    "Behavioural/Core/Multi_Front_end/Front_end/Ifetch_unit/Ifetch_queue"
    "Behavioural/Core/Multi_Front_end/Front_end/Ifetch_unit/Ifetch_unit_Glue"
    "Behavioural/Core/Multi_Front_end/Front_end/Ifetch_unit"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Branch_Target_Buffer/Branch_Target_Buffer_Glue"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Branch_Target_Buffer/Branch_Target_Buffer_Register"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Branch_Target_Buffer"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Direction/Direction_Glue"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Direction"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Direction/Meta_Predictor/Meta_Predictor_Glue"
#   "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Direction/Meta_Predictor/Two_Level_Branch_Predictor/Two_Level_Branch_Predictor_Glue"
#   "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Direction/Meta_Predictor/Two_Level_Branch_Predictor/Branch_History_Table"
#   "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Direction/Meta_Predictor/Two_Level_Branch_Predictor/Pattern_History_Table"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Direction/Meta_Predictor/Two_Level_Branch_Predictor"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Direction/Meta_Predictor"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Prediction_unit_Glue"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Return_Address_Stack"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit/Update_Prediction_Table"
    "Behavioural/Core/Multi_Front_end/Front_end/Prediction_unit"
    "Behavioural/Core/Multi_Front_end/Front_end"
    "Behavioural/Core/Multi_Front_end"

    "Behavioural/Core/Icache_Access"
    "Behavioural/Core/Dcache_Access"

    "Behavioural/Core/Core_Glue"
    "Behavioural/Core"

    "TopLevel"
    );
                                           

#-----[ usage ]---------------------------------------------
function usage ()
{
    echo "Usage            : ${0} action";
    echo "Arguments        : ";
    echo " * action";
    echo "   * systemc     : test all model systemC";
    echo '                   for each component, create file in ${MORPHEO_HOME}/tmp/';
    echo "   * systemc_all : same as test, but don't stop at the first error";
    echo "   * vhdl        : test all model VHDL";
    echo '                   for each component, create file in ${MORPHEO_HOME}/tmp/';
    echo "   * vhdl_all    : same as test, but don't stop at the first error";
    echo "   * result      : list test's resultat";
    echo "   * list        : list test";
    echo "   * clean       : erase file in ${MORPHEO_HOME}/tmp/ directory";
    echo "   * clean_all   : clean all component";
    echo "   * help        : print this message";
    echo "";
    echo "Notes            : ";
    echo " * Morpheo environment must be positionned";
    exit;
}

#-----[ simulation ]----------------------------------------
# arg1 : cmd (execute / sim)
# arg2 : all (stop)
# arg3 : file suffix
function simulation ()
{
    pwd=${PWD};

    nb_cpu=$(cat /proc/cpuinfo |grep -c processor);

    for i in ${directory[@]}; do
        component=$(basename ${i});
        dir="${path}/${i}/SelfTest";

        if test -d "${dir}"; then
            # have previous test ok ?
            declare -i make_test=1;
            
            file="${tmp}/${component}.${3}"

            if test -f ${file}; then
        	res_test=$(cat ${file});
        	if test "${test_ok}" = "${res_test}"; then
        	    make_test=0;
        	fi;
            fi;

            if test ${make_test} -eq 1; then
        	cd ${dir};
        	make config;
        	make -j${nb_cpu} ${1};
        	case ${?} in
        	    "0")
        		echo ${test_ok} > ${file};
        		;;
        	    *)

        		echo ${test_ko} > ${file};
        		echo "";
        		echo "${component} : ${test_ko}";
        		echo "";
        		if test ${2} = "1"; then
        		    exit 1;
        		fi;
        		;;
        	esac;
        	make clean;
            fi;
        else
            echo "${component} have not SelfTest directory.";
        fi;

        cd ${pwd};
    done;
}

#-----[ main ]----------------------------------------------
function main ()
{
    # Test operand
    if test ${#} -ne 1; then
	usage ${*};
    fi;

    if test -z "${MORPHEO_TOPLEVEL}"; then
	usage ${*};
    fi;

    if test ! -d ${tmp}; then
	mkdir -p ${tmp};
    fi;

    case ${1} in
	"systemc" )
            simulation "execute" "1" "systemc";
            ;;

        "systemc_all")
            simulation "execute" "0" "systemc";
	    ;;

	"vhdl" )
            simulation "sim"     "1" "vhdl";
            ;;

        "vhdl_all")
            simulation "sim"     "0" "vhdl";
	    ;;

	"result")
            for x in "systemc" "vhdl"; do
                color=34; 
		echo "[1;${color}m * ${x}";

	        for i in ${tmp}/*.${x}; do
		    if test -f ${i}; then
		        component=$(basename ${i} .${x});
		        res_test=$(cat ${i});
		        
		        case "${res_test}" in
			    "${test_ok}") res_test=${test_ok};     color=32; 
			        ;;
			    "${test_ko}") res_test=${test_ko};     color=31; 
			        ;;
			    *)            res_test=${test_unknow}; color=39; 
			        ;;
		        esac
		        echo "[1;${color}m   * ${res_test} - ${component}";
		    fi;
	        done;
            done;
	    ;;


	"list")
	    for i in ${directory[@]}; do
		component=$(basename ${i});
		echo "${i}";
	    done;
	    ;;

	"clean")
	    rm -r ${tmp};
	    ;;
	    
	"clean_all")
            pwd=${PWD};

	    for i in ${directory[@]}; do
		component=$(basename ${i});

		cd ${path}/${i};

                if test -f "Makefile"; then
		    make clean;
		    dir="SelfTest";
                    
		    if test -d ${dir}; then
		        cd ${dir};
		        make clean;
		    else
		        echo "${component} have not SelfTest directory.";
		    fi;
		    cd ${pwd};
                fi;
	    done;
	    ;;

	"help")
	    usage ${*};
	    ;;
	*)
	    usage ${*};
    esac
}

#-----[ Body ]----------------------------------------------
main ${*};