[PWGLF] Changes to accomodate online mult-integrated efficiency correction + electron rejection (AliceO2Group#15770)

Author: scannito

PWGLF/DataModel/LFPhiStrangeCorrelationTables.h

@@ -151,11 +151,11 @@ DECLARE_SOA_COLUMN(Phi, phi, float);
 DECLARE_SOA_COLUMN(HasTOF, hasTOF, bool);
 
 DECLARE_SOA_DYNAMIC_COLUMN(InNSigmaRegion, inNSigmaRegion,
-                           [](float nSigmaTPC, float pt, bool hasTOF, float nSigmaTOF, float pidTPCMax, float tofPIDThreshold, float pidTOFMax) -> bool {
+                           [](float nSigmaTPC, bool hasTOF, float nSigmaTOF, float pidTPCMax, float pidTOFMax) -> bool {
                              if (std::abs(nSigmaTPC) >= pidTPCMax) {
                                return false; // TPC check failed
                              }
-                             if (pt >= tofPIDThreshold && hasTOF && std::abs(nSigmaTOF) >= pidTOFMax) {
+                             if (hasTOF && std::abs(nSigmaTOF) >= pidTOFMax) {
                                return false; // TOF check failed
                              }
                              return true;
@@ -171,8 +171,9 @@ DECLARE_SOA_TABLE(PionTracksData, "AOD", "PITRACKSDATA",
                   lf_selection_pion_track::Y,
                   lf_selection_pion_track::Phi,
                   lf_selection_pion_track::HasTOF,
-                  lf_selection_pion_track::InNSigmaRegion<lf_selection_pion_track::NSigmaTPC, lf_selection_pion_track::Pt,
-                                                          lf_selection_pion_track::HasTOF, lf_selection_pion_track::NSigmaTOF>);
+                  lf_selection_pion_track::InNSigmaRegion<lf_selection_pion_track::NSigmaTPC,
+                                                          lf_selection_pion_track::HasTOF,
+                                                          lf_selection_pion_track::NSigmaTOF>);
 
 DECLARE_SOA_TABLE(PionTracksMcReco, "AOD", "PITRACKSMCRECO",
                   soa::Index<>,
@@ -183,8 +184,9 @@ DECLARE_SOA_TABLE(PionTracksMcReco, "AOD", "PITRACKSMCRECO",
                   lf_selection_pion_track::Y,
                   lf_selection_pion_track::Phi,
                   lf_selection_pion_track::HasTOF,
-                  lf_selection_pion_track::InNSigmaRegion<lf_selection_pion_track::NSigmaTPC, lf_selection_pion_track::Pt,
-                                                          lf_selection_pion_track::HasTOF, lf_selection_pion_track::NSigmaTOF>);
+                  lf_selection_pion_track::InNSigmaRegion<lf_selection_pion_track::NSigmaTPC,
+                                                          lf_selection_pion_track::HasTOF,
+                                                          lf_selection_pion_track::NSigmaTOF>);
 } // namespace o2::aod
 
 #endif // PWGLF_DATAMODEL_LFPHISTRANGECORRELATIONTABLES_H_

PWGLF/TableProducer/Strangeness/phiStrangeCorrelator.cxx

@@ -573,7 +573,7 @@ struct PionTrackProducer {
   using FilteredSimCollisions = soa::Filtered<SimCollisions>;
 
   // Defining the type of the tracks for data and MC
-  using FullTracks = soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::TrackSelection, aod::pidTPCFullPi, aod::pidTPCFullKa, aod::pidTPCFullPr, aod::pidTOFFullPi, aod::pidTOFFullKa, aod::pidTOFFullPr>;
+  using FullTracks = soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::TrackSelection, aod::pidTPCFullEl, aod::pidTPCFullPi, aod::pidTPCFullKa, aod::pidTPCFullPr, aod::pidTOFFullPi, aod::pidTOFFullKa, aod::pidTOFFullPr>;
   using FullMCTracks = soa::Join<FullTracks, aod::McTrackLabels>;
 
   void init(InitContext&)
@@ -633,6 +633,22 @@ struct PionTrackProducer {
   template <typename T>
   bool pidHypothesesRejection(const T& track)
   {
+    // Electron rejection
+    auto nSigmaTPCEl = aod::pidutils::tpcNSigma(o2::track::PID::Electron, track);
+
+    if (nSigmaTPCEl > -3.0f && nSigmaTPCEl < 5.0f) {
+      auto nSigmaTPCPi = aod::pidutils::tpcNSigma(o2::track::PID::Pion, track);
+      auto nSigmaTPCKa = aod::pidutils::tpcNSigma(o2::track::PID::Kaon, track);
+      auto nSigmaTPCPr = aod::pidutils::tpcNSigma(o2::track::PID::Proton, track);
+
+      if (std::abs(nSigmaTPCPi) > 3.0f &&
+          std::abs(nSigmaTPCKa) > 3.0f &&
+          std::abs(nSigmaTPCPr) > 3.0f) {
+        return false;
+      }
+    }
+
+    // Other hadron species rejection
     for (size_t speciesIndex = 0; speciesIndex < trackConfigs.trkPIDspecies->size(); ++speciesIndex) {
       auto const& pid = trackConfigs.trkPIDspecies->at(speciesIndex);
       auto nSigmaTPC = aod::pidutils::tpcNSigma(pid, track);