Update macros to find corrupted AO2Ds

Author: mhemmer-cern

UTILS/checkCorruptedAO2Ds.C

@@ -16,7 +16,9 @@ void MyErrorHandler(int level, Bool_t abort, const char *location, const char *m
     DefaultErrorHandler(level, abort, location, msg); // Call ROOT’s default handler
 }
 
-int checkCorruptedAO2Ds(TString infileName = "/alice/sim/2024/LHC24h2/535545/AOD/005/AO2D.root", bool fromAlien = true) {
+int checkCorruptedAO2Ds(TString infileName = "AO2D_FZK.root", bool fromAlien = false) {
+    
+    SetErrorHandler(MyErrorHandler);
 
     if (fromAlien) {
         TGrid::Connect("alien://");
@@ -32,7 +34,7 @@ int checkCorruptedAO2Ds(TString infileName = "/alice/sim/2024/LHC24h2/535545/AOD
 
     // all VLA branches in the AO2Ds.root
     std::map<std::string, std::vector<std::string>> branchesToCheck = {
-        {"O2mcparticle_001", std::vector<std::string>{"fIndexArray_Mothers"}},
+        {"O2mcparticle_001", std::vector<std::string>{"fIndexArray_Mothers", "fVx", "fIndexMcCollisions"}},
         {"O2ft0", std::vector<std::string>{"fAmplitudeA", "fChannelA", "fAmplitudeC", "fChannelC"}},
         {"O2fv0a", std::vector<std::string>{"fAmplitude", "fChannel"}},
         {"O2mccalolabel_001", std::vector<std::string>{"fIndexArrayMcParticles", "fAmplitudeA"}},

UTILS/findCorruptedAO2Ds.sh

@@ -2,28 +2,29 @@
 
 # Simple script to find corrupted AO2Ds using the checkCorruptedAO2Ds.C macro
 
-PRODUCTION=LHC24h2
+PRODUCTION=LHC24f3c
 RUN=* # use * for all runs
-NJOBS=90
+NJOBS=20
+PRODUCTIONCYCLE=0
 
 OUTPUTFILE=corrupted_files_$PRODUCTION.txt
 if [ -e "$OUTPUTFILE" ]; then
   rm $OUTPUTFILE
 fi
 
 # find all files in alien
-if [ "$variable" == "*" ]; then
-  alien_find alien:///alice/sim/2024/${PRODUCTION} 5*/AOD/*/AO2D.root > files_to_check.txt
+if [ "$RUN" == "*" ]; then
+  alien_find alien:///alice/sim/2024/${PRODUCTION}/${PRODUCTIONCYCLE}/5*/AOD/*/AO2D.root > files_to_check.txt
 else
-  alien_find alien:///alice/sim/2024/${PRODUCTION} ${RUN}/AOD/*/AO2D.root > files_to_check.txt
+  alien_find alien:///alice/sim/2024/${PRODUCTION}/${PRODUCTIONCYCLE}/${RUN}/AOD/*/AO2D.root > files_to_check.txt
 fi
 mapfile -t FILESTOCHECK < files_to_check.txt
 
-# process AO2Ds
+process AO2Ds
 process_file() {
   IFS='/' read -a num <<< "$1"
   INPUT=$1
-  echo '.x checkCorruptedAO2Ds.C("'${INPUT}'", true)' | root -l -b > log_${num[5]}_${num[7]}
+  echo '.x checkCorruptedAO2Ds.C("'${INPUT}'", true)' | root -l -b > log_${num[6]}_${num[8]}
   echo '.q'
 }
 export -f process_file
@@ -36,12 +37,12 @@ ERRORSTR="Found corrupted file!"
 REPAIRSTR="Found file in need of repair!"
 for FILE in "${FILESTOCHECK[@]}"; do
   IFS='/' read -a num <<< "$FILE"
-  if grep -q "$ERRORSTR" log_${num[5]}_${num[7]}; then
-    echo $FILE >> $OUTPUTFILE
-  else grep -q "$REPAIRSTR" log_${num[5]}_${num[7]}; then
-    echo $FILE >> $OUTPUTFILE
+  if grep -q "$ERRORSTR" log_${num[6]}_${num[8]}; then
+    echo $FILE " is corrupted!" >> $OUTPUTFILE
+  elif grep -q "$REPAIRSTR" log_${num[6]}_${num[8]}; then
+    echo $FILE " is broken!" >> $OUTPUTFILE
   fi
 done
 
 rm files_to_check.txt
-rm log*
+rm log_*