From b7233d004a22f22460126d2d59d490a1c42602f8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jean-S=C3=A9bastien?= <git@jscaux.org>
Date: Wed, 15 Dec 2021 09:58:13 +0100
Subject: [PATCH] Add simple timers for `_data` scripts

---
 scripts/catalogue/c_fixed_N_prog_data.sh | 10 ++++++++++
 scripts/catalogue/c_scan_N_fixed_data.sh | 21 ++++++++++++++++++---
 2 files changed, 28 insertions(+), 3 deletions(-)

diff --git a/scripts/catalogue/c_fixed_N_prog_data.sh b/scripts/catalogue/c_fixed_N_prog_data.sh
index 4c0e4e0..f094038 100755
--- a/scripts/catalogue/c_fixed_N_prog_data.sh
+++ b/scripts/catalogue/c_fixed_N_prog_data.sh
@@ -42,6 +42,8 @@ if [[ $5 -lt 1 ]]; then
     exit 1
 fi
 
+timestart=$(date +%s)
+
 zmodload zsh/datetime
 
 whichDSF=$1
@@ -77,3 +79,11 @@ do
 done
 
 echo '\n\n++ Successfully completed c_fixed_N_prog_data on '$(date '+%Y-%m-%d-%Hh%M')'.\n' | tee -a $logfile
+
+timestop=$(date +%s)
+
+duration=$(($timestop - $timestart))
+hours=$(($duration/3600))
+minutes=$((($duration - 3600*$hours)/60))
+seconds=$(($duration - 3600*$hours - 60*$minutes)) # don't bother printing
+echo 'Total time used: '$hours'h '$minutes'm.\n' | tee -a $logfile
diff --git a/scripts/catalogue/c_scan_N_fixed_data.sh b/scripts/catalogue/c_scan_N_fixed_data.sh
index 9e26418..d00e9a4 100755
--- a/scripts/catalogue/c_scan_N_fixed_data.sh
+++ b/scripts/catalogue/c_scan_N_fixed_data.sh
@@ -37,6 +37,7 @@ if [[ $5 -lt 0 ]]; then
     exit 1
 fi
 
+timestart=$(date +%s)
 
 whichDSF=$1
 kBT=$2
@@ -83,7 +84,6 @@ basedir="$(pwd)"
 # Safety barrier:
 # if directory exists, don't do anything and request directory deletion before proceeding
 dircheck='Lieb-Liniger/'$correlator'/T_'$kBT'/c_scan_N_fixed/k_0_to_'${(l:2::0:)nkmax}'kFo4/sr_'$target_sumrule'/store/data/N_'$N
-echo $dircheck
 if [[ -d $dircheck ]]; then
     echo 'A c scan with these parameters already exists. Aborting.'
     echo 'If you really want to run this command, please first delete directory:\n'$dircheck
@@ -97,9 +97,24 @@ do
     dir='Lieb-Liniger/'$correlator'/T_'$kBT'/c_scan_N_fixed/k_0_to_'${(l:2::0:)nkmax}'kFo4/sr_'$target_sumrule'/store/data/N_'$N'/c_'$c
     mkdir -p $dir
     cd $dir
+    clock1=$(date +%s)
     LiebLin_DSF $whichDSF $c $N $N 0 $iKmax $kBT $Max_Secs $target_sumrule $refine
+    clock2=$(date +%s)
+    dt=$(($clock2 - $clock1))
+    dt_hrs=$(($dt/3600))
+    dt_min=$((($dt - 3600*$dt_hrs)/60))
+    dt_sec=$(($dt - 3600*$dt_hrs - 60*$dt_min))
     cd $basedir
-    echo '++ Successfully computed DSFs for c =' $c', N = '$N'.\n' | tee -a $logfile
+    echo '++ Successfully computed DSFs for c =' $c', N = '$N'.' | tee -a $logfile
+    echo 'Time required: '$dt_hrs'h '$dt_min'm '$dt_sec's.\n' | tee -a $logfile
 done
 
-echo '\n\n++ Successfully completed c_scan_N_fixed_data for N = '$N' on '$(date '+%Y-%m-%d-%Hh%M')'.\n' | tee -a $logfile
+echo '\n++ Successfully completed c_scan_N_fixed_data for N = '$N' on '$(date '+%Y-%m-%d-%Hh%M')'.\n' | tee -a $logfile
+
+timestop=$(date +%s)
+
+duration=$(($timestop - $timestart))
+hours=$(($duration/3600))
+minutes=$((($duration - 3600*$hours)/60))
+seconds=$(($duration - 3600*$hours - 60*$minutes)) # don't bother printing
+echo 'Total time required: '$hours'h '$minutes'm.\n' | tee -a $logfile