DPG: to use NN including Hadronic Rate in the training, mitigation of run by run variations

Author: Ana Marin

Common/TableProducer/PID/CMakeLists.txt

@@ -55,7 +55,7 @@ o2physics_add_dpl_workflow(pid-tpc-base
 
 o2physics_add_dpl_workflow(pid-tpc
                     SOURCES pidTPC.cxx
-                    PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2Physics::MLCore
+                    PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2::DetectorsBase O2Physics::AnalysisCCDB O2Physics::MLCore
                     COMPONENT_NAME Analysis)
 
 # HMPID

Common/TableProducer/PID/pidTPC.cxx

@@ -37,6 +37,7 @@
 #include "Common/DataModel/EventSelection.h"
 #include "Common/DataModel/Multiplicity.h"
 #include "Common/DataModel/PIDResponseTPC.h"
+#include "Common/CCDB/ctpRateFetcher.h"
 #include "Tools/ML/model.h"
 
 #include "CCDB/BasicCCDBManager.h"
@@ -151,7 +152,9 @@ struct tpcPid {
   Configurable<int> useNetworkAl{"useNetworkAl", 1, {"Switch for applying neural network on the alpha mass hypothesis (if network enabled) (set to 0 to disable)"}};
   Configurable<float> networkBetaGammaCutoff{"networkBetaGammaCutoff", 0.45, {"Lower value of beta-gamma to override the NN application"}};
   Configurable<float> networkInputBatchedMode{"networkInputBatchedMode", -1, {"-1: Takes all tracks, >0: Takes networkInputBatchedMode number of tracks at once"}};
-
+  Configurable<std::string> irSource{"irSource", "ZNC hadronic", "Estimator of the interaction rate (Recommended: pp --> T0VTX, Pb-Pb --> ZNC hadronic)"};
+  ctpRateFetcher mRateFetcher;
+  
   // Parametrization configuration
   bool useCCDBParam = false;
   std::vector<float> track_properties;
@@ -400,6 +403,16 @@ struct tpcPid {
         if (input_dimensions == 7 && networkVersion == "2") {
           track_properties[counter_track_props + 6] = trk.has_collision() ? collisions.iteratorAt(trk.collisionId()).ft0cOccupancyInTimeRange() / 60000. : 1.;
         }
+        if (input_dimensions == 8 && networkVersion == "3") {
+          if(trk.has_collision()){
+            auto trk_bc = (collisions.iteratorAt(trk.collisionId())).template bc_as<B>();
+            float hadronicRate = mRateFetcher.fetch(ccdb.service, trk_bc.timestamp(), trk_bc.runNumber(), irSource) * 1.e-3;
+            track_properties[counter_track_props + 7] = hadronicRate/50.;
+          }else{
+            track_properties[counter_track_props + 7] =1;
+          }
+        }
+
         counter_track_props += input_dimensions;
         in_batch_counter++;
         total_input_count++;