#!/bin/bash

# Pipe output to file
#exec &> 'profiles.log'

# Function to handle the interrupt signal
function handle_interrupt() {
   echo "Caught interrupt signal..."
   exit 0
}
# Set the trap to call the handle_interrupt function when SIGINT is received
trap handle_interrupt SIGINT

# Find executables in cgcam directory
function check_exec() {
   cgcam_dir="$HOME/cgcam/src"
   if [ -z "$1" ]; then
      echo "ERROR: supply executable for check_exec"
      exit 1
   fi
   if [ ! -e "$1" ]; then
      if [ ! -e "$cgcam_dir/$1" ]; then
         echo "ERROR: Unable to find $1 in $cgcam_dir"
         exit 1
      fi
      ln -s $cgcam_dir/$1 .
   fi
}

# Round down UTC times for data sampled on the hour
function round_time() {
   if [ -z "$1" ]; then
      echo "ERROR: supply time for round_time"
      exit 1
   fi
   IFS_OLD=$IFS
   IFS=":"
   read -ra utc <<< "$1"
   if [[ ${#utc[@]} -gt 1 ]]; then
      time=${utc[0]}
      mins=${utc[1]}
#      if [[ $mins -gt 50 ]]; then
#         time=$(( $time + 1 ))
#      fi
   else
      time=$utc
   fi
   IFS=$IFS_OLD
   return $time
}

# Command line arguments -f:force rerun of found directory  -s:skip python/merge_wind
# To replot use (-f -s)
force='false'
skip='false'
argc=$#
argv=("$@")
for (( j=0; j<argc; j++ )); do
   case "${argv[j]}" in
   -f)     force='true'
           ;;
   -s)     skip='true'
           ;;
   esac
done

# Read inputs date/lat/lon/utc
input="profile_list.txt"
if [ ! -f $input ]; then
   echo "Unable to find input file $input"
   exit 1
fi
dates=()
lats=()
lons=()
utcs=()
while IFS= read -r line; do
   read -ra data <<< "$line"
   if [ ${data[0]} == '#' ]; then
      continue
   fi
   if [[ ${#data[@]} -ne 4 ]]; then
      echo "ERROR: input list inncomplete $line"
      exit 1
   fi
   dates+=("${data[0]}")
   lats+=("${data[1]}")
   lons+=("${data[2]}")
   utcs+=("${data[3]}")
done < $input

# Set vertical distribution for merge_wind
z0='0'
z1='140k'
dz='1k'
z_blend='30k'
dz_blend='6k'
n_smooth='1'

# Set tics for wind plot. Plotting range uses floor/ceiling of tics
xtics='20'
ytics='20'

# JAWARA data should be downloaded
jawara_dir='/dataST/adam/assim/data/jawara'

wrk_dir=$(pwd)
for n in $(seq 0 $(( ${#dates[@]} - 1 ))); do
   #echo -e "\n$(date)"

   date=${dates[n]}
   lat=${lats[n]}
   lon=${lons[n]}
   utc=${utcs[n]}
   round_time $utc
   time=$?

   day=$(echo $date | cut -d"/" -f 2)
   month=$(echo $date | cut -d"/" -f 1)
   year=$(echo $date | cut -d"/" -f 3)

   # Convert date to YYYY MM DD
   if (( $year < 50 )); then
      yyyy=$((year+2000))
   elif (( $year < 1000 )); then
      yyyy=$((year+1900))
   else
      yyyy=$year
   fi
   yy=${yyyy: 2}
   dd=$(( 10#$day ))
   mm=$(( 10#$month ))
   dd=$(printf "%02d" $dd)
   mm=$(printf "%02d" $mm)
   tt=$(printf "%02d" $time)

   #echo $yyyy $mm $dd $lat $lon $tt

   # Goto case directory
   month_str=$(date -d "2000/$mm/01" +%b)
   case_dir=$month_str$dd"_"$yyyy"_"$lat"_"$lon"_"$tt
   echo -e "\nGenerating $case_dir"
   if [ -d $case_dir ]; then
      echo "Case exists!"
      if [ $force != 'true' ]; then
         continue
      fi
   else
      mkdir $case_dir
   fi
   case_dir=$(realpath $case_dir)
   cd $case_dir

   # Get files
   cp ../apitmp.py ../input_tmp.py ../wind_tmp.gpt .

   # Gather era data for each case
   era_dir="."
   era_file="era."$yyyy$mm$dd".nc"
   if [ -f "$era_dir/$era_file" ]; then
      echo "Found ERA5 data file"
   else
      cd $era_dir
      sed -e "s/YEAR/$yyyy/g" -e "s/MONTH/$mm/g" -e "s/DAY/$dd/g" -e "s/LON/$lon/g" -e "s/LAT/$lat/g" -e "s/TIME/$tt/g" apitmp.py > apiera.py
      if [ $skip == 'false' ]; then
         echo -e "\nRetrieving ERA5 file for $mm/$dd/$yyyy"
         python3 apiera.py
      fi
      cd $case_dir
   fi
   # JAWARA should already be downloaded
   jawara_file="$yy$mm.nc"
   if [ ! -f "$jawara_dir/Z$jawara_file" ]; then
      echo "ERROR: Unable to locate jawara data $jawara_dir/?$jawara_file"
      continue
   fi

   # Convert data
   sed -e "s|YEAR|$yyyy|g" -e "s|MONTH|$mm|g" -e "s|DAY|$dd|g" -e "s|LON|$lon|g" -e "s|LAT|$lat|g" -e "s|ITIME|$time|g" -e "s|ERA_FILE|$era_file|g" -e "s|ERA_DIR|$era_dir|g" -e "s|JAWARA_FILE|$jawara_file|g" -e "s|JAWARA_DIR|$jawara_dir|g" input_tmp.py > input_convert.py
   check_exec "convert_assim_data.py"
   if [ $skip == 'false' ]; then
      echo -e "\nConverting Data"
      python3 convert_assim_data.py
   fi

   # Merge wind to get profile
   check_exec "merge_wind"
   echo -e "zo0   $z0\nzo1   $z1\ndz0   $dz\nref_lon   $lon\nref_lat   $lat\nref_itime  1\nz_blend1   $z_blend\ndz_blend1  $dz_blend\nn_smooth_z  $n_smooth" > merge_wind.inp
   if [ $skip == 'false' ]; then
      echo -e "\nMerging Sources"
      ./merge_wind
   fi

   # Plot winds
   file=$(ls out*prof.out)
   if (( $(echo "$lon < 0" | bc -l) )); then
      lon=${lon/"-"}
      hem_ew='W'
   else
      lon=$lon
      hem_ew='E'
   fi
   if (( $(echo "$lat < 0" | bc -l) )); then
      lat=${lat/"-"}
      hem_ns='S'
   else
      lat=$lat
      hem_ns='N'
   fi
   sed -e "s|YY|$yy|g" -e "s|MM|$mm|g" -e "s|DD|$dd|g" -e "s|LON|$lon|g" -e "s|HEM_EW|$hem_ew|g" -e "s|LAT|$lat|g" -e "s|HEM_NS|$hem_ns|g" -e "s|TIM|$tt|g" -e "s|YTICS|$ytics|g" -e "s|XTICS|$xtics|g" -e "s|FILE|$file|g" wind_tmp.gpt > wind.gpt
   echo -e "\nPlotting Wind (ignore terminal warning)"
   gnuplot wind.gpt

   cd $wrk_dir

done