From d47944257ead124fdd7b33b1430c8a8e5956da59 Mon Sep 17 00:00:00 2001
From: Chiara Zampolli <Chiara.Zampolli@cern.ch>
Date: Wed, 28 Feb 2024 23:48:07 +0100
Subject: [PATCH] To allow to set the trackQC sampling from the JDL, or have a
 subsample with full sampling

---
 .../configurations/asyncReco/async_pass.sh    |  2 +-
 .../configurations/asyncReco/setenv_extra.sh  | 24 ++++++++++++++++++-
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/DATA/production/configurations/asyncReco/async_pass.sh b/DATA/production/configurations/asyncReco/async_pass.sh
index a49a941b0..ce6d3ce5e 100755
--- a/DATA/production/configurations/asyncReco/async_pass.sh
+++ b/DATA/production/configurations/asyncReco/async_pass.sh
@@ -20,7 +20,7 @@ run_AOD_merging() {
 timeStartFullProcessing=`date +%s`
 
 # to skip positional arg parsing before the randomizing part.
-inputarg="${1}"
+export inputarg="${1}"
 
 if [[ "${1##*.}" == "root" ]]; then
     #echo ${1##*.}
diff --git a/DATA/production/configurations/asyncReco/setenv_extra.sh b/DATA/production/configurations/asyncReco/setenv_extra.sh
index 4f9992a67..c1d82bbca 100644
--- a/DATA/production/configurations/asyncReco/setenv_extra.sh
+++ b/DATA/production/configurations/asyncReco/setenv_extra.sh
@@ -622,7 +622,29 @@ fi
 # ad-hoc settings for AOD
 echo ALIEN_JDL_LPMPRODUCTIONTAG = $ALIEN_JDL_LPMPRODUCTIONTAG
 echo ALIEN_JDL_LPMPASSNAME = $ALIEN_JDL_LPMPASSNAME
-export ARGS_EXTRA_PROCESS_o2_aod_producer_workflow="$ARGS_EXTRA_PROCESS_o2_aod_producer_workflow --aod-writer-maxfilesize $AOD_FILE_SIZE --lpmp-prod-tag $ALIEN_JDL_LPMPRODUCTIONTAG --reco-pass $ALIEN_JDL_LPMPASSNAME"
+# Track QC table sampling
+if [[ -n $ALIEN_JDL_TRACKQCFRACTION ]]; then
+  TRACKQC_FRACTION=$ALIEN_JDL_TRACKQCFRACTION
+else
+  if [[ $ALIEN_JDL_ENABLEPERMILFULLTRACKQC == "1" ]]; then
+    PERMIL_FULLTRACKQC=${ALIEN_JDL_PERMILFULLTRACKQC:-100}
+    INVERSE_PERMIL_FULLTRACKQC=$((1000/PERMIL_FULLTRACKQC))
+    if [[ -f wn.xml ]]; then
+      HASHCODE=`grep alien:// wn.xml | tr ' ' '\n' | grep ^lfn | cut -d\" -f2 | head -1 | cksum | cut -d ' ' -f 1`
+    else
+      HASHCODE=`echo "${inputarg}" | cksum | cut -d ' ' -f 1`
+    fi
+    if [[ "$((HASHCODE%INVERSE_PERMIL_FULLTRACKQC))" -eq "0" ]]; then
+      TRACKQC_FRACTION=1
+    else
+      TRACKQC_FRACTION=0.1
+    fi
+  else
+    TRACKQC_FRACTION=0.1
+  fi
+fi
+echo TRACKQC_FRACTION = $TRACKQC_FRACTION
+export ARGS_EXTRA_PROCESS_o2_aod_producer_workflow="$ARGS_EXTRA_PROCESS_o2_aod_producer_workflow --aod-writer-maxfilesize $AOD_FILE_SIZE --lpmp-prod-tag $ALIEN_JDL_LPMPRODUCTIONTAG --reco-pass $ALIEN_JDL_LPMPASSNAME --trackqc-fraction $TRACKQC_FRACTION"
 if [[ $PERIOD == "LHC22c" ]] || [[ $PERIOD == "LHC22d" ]] || [[ $PERIOD == "LHC22e" ]] || [[ $PERIOD == "JUN" ]] || [[ $PERIOD == "LHC22f" ]] || [[ $PERIOD == "LHC22m" ]] || [[ "$RUNNUMBER" == @(526463|526465|526466|526467|526468|526486|526505|526508|526510|526512|526525|526526|526528|526534|526559|526596|526606|526612|526638|526639|526641|526643|526647|526649|526689|526712|526713|526714|526715|526716|526719|526720|526776|526886|526926|526927|526928|526929|526934|526935|526937|526938|526963|526964|526966|526967|526968|527015|527016|527028|527031|527033|527034|527038|527039|527041|527057|527076|527108|527109|527228|527237|527259|527260|527261|527262|527345|527347|527349|527446|527518|527523|527734) ]] ; then
   export ARGS_EXTRA_PROCESS_o2_aod_producer_workflow="$ARGS_EXTRA_PROCESS_o2_aod_producer_workflow --ctpreadout-create 1"
 fi