#!/bin/sh

#####################################################################
# Script tpc_psurge_thresh.sh merges the results from the SLOSH model
#        and computes the probability of exceeding certain threshholds
#        the  for the Probability Storm Surge
# usage: tpc_psurge_thresh.sh <ScriptDir> <stormAdvDir> <storm> 
#                            <advTime> <thresh>
#####################################################################
set -x

if [[ -z $5 ]] || [[ ! -z $6 ]]
then
  echo "usage: $0 <ScriptDir> <stormAdvDir> <storm> <advTime> <thresh>"
  exit 1
fi

ScriptDir=${USHpsurge:-$1}
stormAdvDir=$2
storm=$3
advTime=$4
thresh=$5

export PS4='thresh_${thresh}_T$SECONDS + '

if [ $RUN_ENVIR = "NCO" ]
then
   EXECpsurge=${EXECpsurge:-/nwprod/exec}
   SloshbsnDir=$PARMpsurge/tpc_sloshbsn
else
   SloshbsnDir=${ScriptDir}/../parm/tpc_sloshbsn
   EXECpsurge=${ScriptDir}/../exec
fi

if [ ! -d $DATA/thresh_$thresh ]
then
   mkdir $DATA/thresh_$thresh
fi

cd $DATA/thresh_$thresh

/nwprod/util/ush/setup.sh

# Define the location for the SLOSH model

ansDir=${stormAdvDir}/ans
outDir=${stormAdvDir}/output

mkdir -p $outDir

# flag:
#   1 >= thresh,
#   2 <= thresh,
#   3 == thresh is 'confidence' ie % of storms less than X,
#   4 > thresh,
#   5 < thresh
flag=4

# Basin has to have potential of >= 'potential' or no prex file is generated.
potential=0

# skip allows one to speed up thresh by skipping frames (typically frames
# are saved every 10 min).
# Assuming now every 60 min.
skip=0

# following awk script ignores first line, and first row.
bsnList=`cat ${SloshbsnDir}/ocean.txt | awk '{if (NR >= 2) {$1=""; print}}' | sed -e s#,#' '#g`

# Let us know when this starts
date

tideFile=${stormAdvDir}/${storm}_${advTime}_wlevel.dat

#####
# Compute "penv" files based on all basins in the run directory
#####
cd ${ansDir}
/nwprod/util/ush/setup.sh
f_first=1
for bsn in $bsnList
do
  if [ -s ${ansDir}/$bsn/weight.txt ]
  then
     initHt=0.0
     if [ ${bsn} == "eoke" ]
     then
        initHt=`grep OKECHOBEE_LAKE $tideFile | cut -d',' -f2`
     fi
     cd $bsn
     /nwprod/util/ush/setup.sh
     envName=${bsn}_gt${thresh}.env
     weight_file=weight.txt
     outenvFile=${outDir}/${envName}

     export pgm=tpc_psurge_penv
     . prep_step

     $EXECpsurge/tpc_psurge_penv -i ${initHt} weight.txt prob ${thresh} ${outenvFile} >> thresh_$thresh.out
     export err=$?; ../err_chk

     if [[ $f_first -eq 1 ]]
     then
        envArg=${envName}
        bsnArg=${bsn}
        f_first=0
     else
        envArg=$envArg,${envName}
        bsnArg=$bsnArg,${bsn}
     fi
  fi
  cd ${ansDir}
done
echo "Done with penv files"

#####
# Test if we didn't have any basins in the run directory (storm was too weak)
#####
if [[ $f_first -eq 1 ]]
then
  exit 0
fi

#####
# Merge the results together into GRIB files.
#####
year=`echo ${advTime} | cut -c 1-4`
mon=`echo ${advTime} | cut -c 5-6`
day=`echo ${advTime} | cut -c 7-8`
hour=`echo ${advTime} | cut -c 9-10`
time=${year}-${mon}-${day}T${hour}0000

cd ${outDir}

/nwprod/util/ush/setup.sh

export pgm=tpc_psurge_envmerge
. prep_step

grbindir=$SloshbsnDir
grbin=$SloshbsnDir/conus.txt
grbout1=${storm}_${advTime}_gt${thresh}_1.grb 

$EXECpsurge/tpc_psurge_envmerge $envArg $bsnArg $grbindir prob $grbin ${thresh} ${time} ${grbout1} >>thresh_$thresh.out
export err=$?; err_chk

if [ $SENDCOM = YES ]
then
    cp *gt*.env $COMOUT/${storm}/.
    cp $grbout1 $COMOUT/${storm}/.
fi

if [ $SENDDBN = YES ]
then
   $DBNROOT/bin/dbn_alert MODEL TPC_PSURGE_GB2 $job $COMOUT/${storm}/$grbout1
fi

cat thresh_$thresh.out >>$pgmout

echo "End of computing for thresh $thresh"
# Let us know when this finishes
date

exit 0