#!/bin/bash

[ -n "$OTF2_PRINT_PATH" ]    || export OTF2_PRINT_PATH=otf2-print
[ -n "$EZTRACE_PATH" ]       || export EZTRACE_PATH=eztrace
[ -n "$EZTRACE_AVAIL_PATH" ] || export EZTRACE_AVAIL_PATH=eztrace_avail
[ -n "$EZTRACE_CC_PATH" ]    || export EZTRACE_CC_PATH=eztrace_cc
[ -n "$MPIRUN_PATH" ]        || export MPIRUN_PATH=mpirun
[ -n "$MPICC_PATH" ]         || export MPICC_PATH=mpicc


C_BLACK='\033[0;30m'
C_DGRAY='\033[1;30m'
C_LGRAY='\033[0;37m'
C_WHITE='\033[1;37m'

C_GREEN='\033[0;32m'
C_LGREEN='\033[1;32m'
C_BLUE='\033[0;34m'
C_LBLUE='\033[1;34m'
C_CYAN='\033[0;36m'
C_LCYAN='\033[1;36m'
C_ORANGE='\033[0;33m'
C_YELLOW='\033[1;33m'
C_RED='\033[0;31m'
C_LRED='\033[1;31m'
C_PURPLE='\033[0;35m'
C_LPURPLE='\033[1;35m'

C_NC='\033[0m'

C_BOLD='\033[1m'


function print_error {
  echo -e "  [${C_LRED}${C_BOLD}ERROR${C_NC}] $@"
}

function print_warning {
  echo -e "[${C_YELLOW}${C_BOLD}WARNING${C_NC}] $@"
}

function print_info {
  echo -e "   [${C_LBLUE}INFO${C_NC}] ${C_BOLD}$@${C_NC}"
}

function print_ok {
  echo -e "     [${C_LGREEN}OK${C_NC}] $@"
}

function print_simple {
  echo "          $@"
}


function check_last_return {
  if [[ $? != 0 ]] ; then
    if [[ "$@" != "" ]]; then
      print_error $@
    fi
    error_occured=true
  elif [[ "$@" != "" ]] ; then
    print_ok $@
  fi
}

function check_dependencies {
    echo "> Checking for dependencies..."
    for d in $@; do
        if ! command -v $d &>/dev/null; then
            print_error "Missing command: $d"
            return 1
        fi	
    done
    print_ok
}

function check_module {
    module=$1
    echo "> Checking for eztrace module '$module'..."
    [ -n "$EZTRACE_AVAIL_PATH" ] || EZTRACE_AVAIL_PATH=eztrace_avail
    if  "$EZTRACE_AVAIL_PATH" | grep "^$module[[:space:]]" > /dev/null ; then
	print_ok
    else
	print_error "Module '$module' not found"
	return 1
    fi
}

function check_compilation {
    echo "> Compiling test programs..."
    if [ -n "$verbose" ]; then
	make
    else
	make > /dev/null
    fi

    if [ "$?" != "0" ]; then
	print_error "Compilation failed"
	return 1
    fi
    print_ok
}

function run_test {
    test=$1
    if ! [ -x "$test" ]; then
	return
    fi
    echo "> Running $test..."
    if [ -n "$verbose" ]; then
	$test
    else
	$test > /dev/null 2>&1
    fi

    if [ "$?" != "0" ]; then
        print_error "Test $test failed"
        return 1
    fi
    print_ok
}

function run_and_check_command {
    cmd=$@
    echo "> Running $cmd"

    ((nb_test++))
    if [ -n "$verbose" ]; then
	$cmd
    else
	$cmd > /dev/null 2>&1
    fi
    if [ "$?" != "0" ]; then
	print_error "command '$cmd' failed"
	return 1
    fi

    print_ok
    ((nb_pass++))
    return 0
}

function trace_get_nb_event_of_type {
    trace_filename=$1
    event_type=$2

    "$OTF2_PRINT_PATH" "$trace_filename" 2>/dev/null |grep "^$event_type[[:space:]]"|wc -l
}

function trace_check_integrity {
    trace_filename=$1

    ((nb_test++))
    echo " > Checking for trace integrity"
    if ! "$OTF2_PRINT_PATH" "$trace_filename" > /dev/null 2>&1 ; then
	print_error "Cannot parse trace '$trace_filename'"
	((nb_failed++))
	return 1
    else
	((nb_pass++))
	print_ok
	return 0
    fi
}

function trace_check_enter_leave_parity {
    trace_filename=$1

    ((nb_test++))
    echo " > Checking for ENTER/LEAVE parity"
    nb_enter=$(trace_get_nb_event_of_type "$trace_filename" "ENTER")
    nb_leave=$(trace_get_nb_event_of_type "$trace_filename" "LEAVE")
    if [ $nb_enter -ne $nb_leave ]; then
	print_error "$nb_enter ENTER events / $nb_leave LEAVE events"
	((nb_failed++))
	return 1
    else
	print_ok "$nb_enter event of each type"
	((nb_pass++))
	return 0
    fi
}

function trace_check_event_type {
    trace_filename=$1
    event_type=$2
    expected_nb=$3
    actual_nb=$(trace_get_nb_event_of_type "$trace_filename" "$event_type")
    
    ((nb_test++))
    echo " > Checking the number of $event_type events"
    if [ $expected_nb -ne $actual_nb ]; then
	print_error "$actual_nb events (expected: $expected_nb)"
	((nb_failed++))
	return 1
    else
	print_ok
	((nb_pass++))
	return 0
    fi
}

function trace_check_nb_enter {
    trace_filename=$1
    event_type="$2"
    expected_nb=$3

    ((nb_test++))
    echo " > Checking the number of ENTER $event_type events"
    actual_nb=$("$OTF2_PRINT_PATH" "$trace_filename" 2>/dev/null |grep "^ENTER[[:space:]]"|awk -vFPAT='([^ ]+)|("[^"]+")' '{print $5}' | grep "\"$event_type\""| wc -l)
    if [ $expected_nb -ne $actual_nb ]; then
	print_error "$actual_nb events (expected: $expected_nb)"
	((nb_failed++))
	return 1
    else
	print_ok
	((nb_pass++))
	return 0
    fi
}

function trace_check_nb_leave {
    trace_filename=$1
    event_type=$2
    expected_nb=$3
    
    ((nb_test++))
    echo " > Checking the number of LEAVE $event_type events"
    actual_nb=$("$OTF2_PRINT_PATH" "$trace_filename" 2>/dev/null |grep "^LEAVE[[:space:]]"|awk  -vFPAT='([^ ]+)|("[^"]+")'  '{print $5}' | grep "\"$event_type\""| wc -l)
    if [ $expected_nb -ne $actual_nb ]; then
	print_error "$actual_nb events (expected: $expected_nb)"
	((nb_failed++))
	return 1
    else
	print_ok
	((nb_pass++))
	return 0
    fi
}