#! /usr/bin/env bash

# $Header: /home/ubuntu/mnt/e9_copy/MITgcm/tools/example_scripts/ACESgrid/Attic/aces_test_all,v 1.17 2010/05/10 14:03:18 jmc Exp $
# $Name:  $

# action =2 : submit test jobs ; =1 : get jobs output ; =3 : do both
case $1 in
 '-subOnly') action=2 ; shift ;;
 '-getOnly') action=1 ; shift ;;
 '-sub+get') action=3 ; shift ;;
          *) action=3 ;;
esac
#echo "action= $action , Arg= $# "

today=`date +%Y%m%d`

if test $# = 0
then
  tst_list='gnu ifc pgi adm mp2 mth tuv'
else
  tst_list=$*
fi

# QSUB="/usr/local/bin/qsub"
# QSTAT="/usr/local/bin/qstat"
QSUB=qsub
QSTAT=qstat
TST_DIR="/home/jmc/test_ACES"
MPACK="MITgcm_tools/mpack-1.6/mpack"
SUB_DIR="MITgcm_tools/example_scripts/ACESgrid"

TMP_FIL="$TST_DIR/output/TTT.$$"
LOG_FIL="$TST_DIR/output/tst_all."`date +%m%d`".log"

rm -f $LOG_FIL ; touch $LOG_FIL
echo -n '-- Starting: ' | tee -a $LOG_FIL
date | tee -a $LOG_FIL

#-- clean up old log files:
cd $TST_DIR/output
n=$(( `ls tst_all.*.log | wc -l` - 10 ))
if test $n -gt 0 ; then
  echo ' remove old log files:' | tee -a $LOG_FIL
    ls -lt tst_all.*.log | tail -"$n" | tee -a $LOG_FIL
    ls -t  tst_all.*.log | tail -"$n" | xargs rm -f
fi

. /etc/profile.d/modules.sh
module list >> $LOG_FIL 2>&1

#-- now really do something:

cd $TST_DIR

nbtst=0
for i in $tst_list
do
  sfx=`echo ${i}"_mpi" | sed 's/mth_mpi/ifc_mth/' | sed 's/tuv_mpi/op64_tuv/' | sed 's/mp2_mpi/mp2_mth/'`
  if test -f $SUB_DIR/aces_test_$sfx ; then
    JOB="tst_"$i
    job_exist=`$QSTAT -a | grep $USER | grep $JOB | wc -l`
    if [ $action -ge 2 ] ; then
    #-- to submit job
      if test "x_$job_exist" = x_0 ; then
        echo -n $JOB | tee -a $LOG_FIL
        $QSUB $SUB_DIR/aces_test_$sfx | tee -a $LOG_FIL
        eval M_$i='submitted'
        nbtst=`expr $nbtst + 1`
      else
        echo $JOB | tee -a $LOG_FIL
        $QSTAT -a | grep $USER | grep $JOB | tee -a $LOG_FIL
        echo 'job already exist => skip this test' | tee -a $LOG_FIL
        eval M_$i='skipped'
      fi
    else
    #-- to get outp back:
      if test "x_$job_exist" = x_0 ; then
        echo -n 'did not find any job: ' | tee -a $LOG_FIL
        echo $JOB | tee -a $LOG_FIL
        eval M_$i='skipped'
      else
        echo -n 'found a job: ' | tee -a $LOG_FIL
        echo $JOB | tee -a $LOG_FIL
        $QSTAT -a | grep $USER | grep $JOB | tee -a $LOG_FIL
        eval M_$i='submitted'
        nbtst=`expr $nbtst + 1`
      fi
    fi
  else
    echo 'no file:' aces_test_$sfx 'to submit' | tee -a $LOG_FIL
    eval M_$i='skipped'
  fi
done

if test $action = 2 ; then
  count=0
else
#date_str=`date +%Y%m%d`"_0"

  MPACKDIR=`dirname $MPACK`
#- build the mpack utility (from build_mpack in testreport):
  ( cd $MPACKDIR && ./configure && make ) > TTT.build_mpack.$$ 2>&1
  RETVAL=$?
  if test "x$RETVAL" != x0 ; then
    echo
    echo "Error building the mpack tools at: $MPACK_DIR" | tee -a $LOG_FIL
    if test -x $MPACK ; then
       HAVE_MPACK=t
       echo "  use (old ?) executable:" | tee -a $LOG_FIL 
       ls -l $MPACK | tee -a $LOG_FIL
    else
       HAVE_MPACK=f
    fi
  else
    if test -x $MPACK ; then
       rm -f TTT.build_mpack.$$
       HAVE_MPACK=t
       echo " building mpack: OK" | tee -a $LOG_FIL
    else
       echo " $MPACK not executable" | tee -a $LOG_FIL
       HAVE_MPACK=f
    fi
  fi
  echo

  count=$nbtst
fi

#- when it's done, retrieve output and send e-mail
minutes=0 ; freq=10
fsec=`expr $freq \* 60`
echo "Check every $freq mn for $count test(s) to finish" | tee -a $LOG_FIL
echo "- start at :" `date` | tee -a $LOG_FIL
while test $count != 0 ; do

    sleep $fsec
    minutes=$(( $minutes + $freq ))

    for i in $tst_list ; do

      eval comm=\$M_$i
      if test $comm = 'submitted' ; then
        JOB="tst_"$i
        $QSTAT -a > $TMP_FIL
        RETVAL=$?
        ready_to_send=`grep $USER $TMP_FIL | grep $JOB | wc -l`
        rm -f $TMP_FIL
        if test "x$RETVAL" != x0 ; then
           echo " $QSTAT returned with error code: $RETVAL" | tee -a $LOG_FIL
           continue
        fi

        if test "x_$ready_to_send" = x_0 ; then
          run_dir=${TST_DIR}"/MITgcm_"$i"/verification"
#-      results output:
          tdir=`ls -1 -t $run_dir | grep -v tr_out | grep '^tr_aces' | head -1`
          if test "x$tdir" != x ; then
            #- check this is the right output
            chk=`echo $tdir | grep -c $today`
            if test $chk = '0' ; then
              curday=`date +%Y%m%d`
              chk=`echo $tdir | grep -c $curday`
            fi
            if test $chk = '0' ; then
              echo "tdir='$tdir'" | tee -a $LOG_FIL
              echo "Output do not match, no email sent for $i" | tee -a $LOG_FIL
            else
              rm -f "/tmp/tr_aces-"$i".tar.gz"
              ( cd $run_dir ; tar -czf "/tmp/tr_aces-"$i".tar.gz" ./$tdir )
              if test "x$HAVE_MPACK" = xt ; then
                $MPACK -s MITgcm-test -m 3555000 "/tmp/tr_aces-"$i".tar.gz" jmc@mitgcm.org
                echo "Email sent for $i at:     " `date` | tee -a $LOG_FIL
              else
                echo " no email sent for $i (no mpack)" | tee -a $LOG_FIL
              fi
            fi
          else
              echo " no output found for $i" | tee -a $LOG_FIL
          fi
#-      restart output:
          tdir=`ls -1 -t $run_dir | grep -v tr_out | grep '^rs_aces' | head -1`
          if test "x$tdir" != x ; then
            #- check this is the right output
            chk=`echo $tdir | grep -c $today`
            if test $chk = '0' ; then
              curday=`date +%Y%m%d`
              chk=`echo $tdir | grep -c $curday`
            fi
            if test $chk = '0' ; then
              echo "tdir='$tdir'" | tee -a $LOG_FIL
              echo "Restart output do not match, no email sent for $i" | tee -a $LOG_FIL
            else
              rm -f "/tmp/rs_aces-"$i".tar.gz"
              ( cd $run_dir ; tar -czf "/tmp/rs_aces-"$i".tar.gz" ./$tdir )
              if test "x$HAVE_MPACK" = xt ; then
                $MPACK -s MITgcm-test -m 3555000 "/tmp/rs_aces-"$i".tar.gz" jmc@mitgcm.org
                echo "Email sent for $i restart:" `date` | tee -a $LOG_FIL
              else
                echo " no email sent for $i restart (no mpack)" | tee -a $LOG_FIL
              fi
            fi
          else
              echo " no restart output for $i" | tee -a $LOG_FIL
          fi
#-      record successful sending
          eval M_$i=done
          count=`expr $count - 1`
          chmod 644 output/tst_$i.std*
        fi
      fi
    done

    #  "long" queue is 24hrs = 24*60min = 1440min
    if test $minutes -gt 2160 ; then
        hrs=$(( $minutes / 60 ));
        echo "Time expired after $minutes minutes ($hrs hours)" | tee -a $LOG_FIL
        echo ' ' $count '/' $nbtst 'tests not yet finished' | tee -a $LOG_FIL
        exit 1
    fi

done

echo "Retrieving $nbtst tests finish :" `date` | tee -a $LOG_FIL
exit 0