Add array with momentum threshold values

Author: Lukaszz99

Tools/PIDML/qaPidML.cxx

@@ -47,6 +47,10 @@ struct pidml {
   // available particles: 211, 2212, 321
   static constexpr int particlesPdgCode[numParticles] = {211, 2212, 321};
 
+  // values of track momentum when to switch from only TPC signal to combined TPC and TOF signal
+  // i-th momentum corresponds to the i-th particle
+  static constexpr float pSwitchValue[numParticles] = {0.5, 0.8, 0.5};
+
   HistogramRegistry histReg{
     "allHistograms",
     {{"MC/211", "MC #pi^{+};p_{T} (GeV/c);Counts", {HistType::kTH1F, {{binsNb, 0, maxP}}}},
@@ -322,14 +326,14 @@ struct pidml {
   void pidML(const T& track, const int pdgCodeMC)
   {
     float pidLogits[3];
-    if (track.p() <= 0.5) {
+    if (track.p() <= pSwitchValue[i]) {
       pidLogits[3] = {model211TPC.applyModel(track), model2212TPC.applyModel(track), model321TPC.applyModel(track)};
     } else {
       pidLogits[3] = {model211All.applyModel(track), model2212All.applyModel(track), model321All.applyModel(track)};
     }
     int pid = getParticlePdg(pidLogits);
     // condition for sign: we want to work only with pi, p and K, without antiparticles
-    if ((pid == particlesPdgCode[i]) & (track.sign() == 1)) {
+    if (pid == particlesPdgCode[i] && track.sign() == 1) {
       if (pdgCodeMC == particlesPdgCode[i]) {
         fillPidHistos<i>(track, pdgCodeMC, true);
       } else {
@@ -342,12 +346,11 @@ struct pidml {
   PidONNXModel model211All;
   PidONNXModel model2212All;
   PidONNXModel model321All;
-  // Model with only TPC model
+  // Model with only TPC signal model
   PidONNXModel model211TPC;
   PidONNXModel model2212TPC;
   PidONNXModel model321TPC;
 
-  // Configurable<bool> cfgUseTOF{"useTOF", true, "Use ML model with TOF signal"};
   Configurable<std::string> cfgModelDir{"model-dir", "http://alice-ccdb.cern.ch/Users/m/mkabus/pidml/onnx_models", "base path to the directory with ONNX models"};
   Configurable<std::string> cfgScalingParamsFile{"scaling-params", "http://alice-ccdb.cern.ch/Users/m/mkabus/pidml/onnx_models/train_208_mc_with_beta_and_sigmas_scaling_params.json", "base path to the ccdb JSON file with scaling parameters from training"};