From 21a974a4d12ef8e9f7b1e8aec9920a4aa41a5f54 Mon Sep 17 00:00:00 2001 From: scattaru Date: Thu, 5 Feb 2026 10:47:37 +0100 Subject: [PATCH 1/2] Added PID table for derived data correlations --- PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h | 7 +++++++ PWGHF/HFC/TableProducer/correlatorDsHadrons.cxx | 10 ++++++++++ 2 files changed, 17 insertions(+) diff --git a/PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h b/PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h index 555d86f34c8..a12d740b117 100644 --- a/PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h +++ b/PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h @@ -85,6 +85,8 @@ DECLARE_SOA_COLUMN(PhiAssocTrack, phiAssocTrack, float); //! Phi of the track DECLARE_SOA_COLUMN(PtAssocTrack, ptAssocTrack, float); //! Pt of the track DECLARE_SOA_COLUMN(DcaXY, dcaXY, float); //! Impact parameter in XY of the track to the primary vertex DECLARE_SOA_COLUMN(DcaZ, dcaZ, float); //! Impact parameter in Z of the track to the primary vertex +DECLARE_SOA_COLUMN(NSigmaTpc, nSigmaTpc, float); //! Number of sigma TPC +DECLARE_SOA_COLUMN(NSigmaTof, nSigmaTpc, float); //! Number of sigma TOF } // namespace hf_assoc_track_reduced DECLARE_SOA_TABLE(AssocTrackReds, "AOD", "ASSOCTRACKRED", //! Table with associated track info soa::Index<>, @@ -103,6 +105,11 @@ DECLARE_SOA_TABLE(AssocTrackSels, "AOD", "ASSOCTRACKSEL", //! Table with associa aod::hf_assoc_track_reduced::DcaXY, aod::hf_assoc_track_reduced::DcaZ); +DECLARE_SOA_TABLE(AssocTrackPids, "AOD", "ASSOCTRACKPID", //! Table with associated track pid info + soa::Index<>, + aod::hf_assoc_track_reduced::NSigmaTpc, + aod::hf_assoc_track_reduced::NSigmaTof); + // definition of columns and tables for Charm-Hadron and Hadron-Hadron correlation pairs namespace hf_correl_charm_had_reduced { diff --git a/PWGHF/HFC/TableProducer/correlatorDsHadrons.cxx b/PWGHF/HFC/TableProducer/correlatorDsHadrons.cxx index 03949919708..b684ffc1347 100644 --- a/PWGHF/HFC/TableProducer/correlatorDsHadrons.cxx +++ b/PWGHF/HFC/TableProducer/correlatorDsHadrons.cxx @@ -176,6 +176,7 @@ struct HfCorrelatorDsHadrons { Produces candSelInfo; Produces assocTrackReduced; Produces assocTrackSelInfo; + Produces assocTrackPidInfo; Configurable fillHistoData{"fillHistoData", true, "Flag for filling histograms in data processes"}; Configurable fillHistoMcRec{"fillHistoMcRec", true, "Flag for filling histograms in MC Rec processes"}; @@ -889,6 +890,15 @@ struct HfCorrelatorDsHadrons { assocTrackReduced(indexHfcReducedCollision, track.globalIndex(), track.phi(), track.eta(), track.pt() * track.sign()); assocTrackSelInfo(indexHfcReducedCollision, track.tpcNClsCrossedRows(), track.itsClusterMap(), track.itsNCls(), track.dcaXY(), track.dcaZ()); + if (trkPIDspecies->at(0) == o2::track::PID::Kaon) { + assocTrackPidInfo(track.tpcNSigmaKa(), track.tofNSigmaKa()); + } + if (trkPIDspecies->at(0) == o2::track::PID::Pion) { + assocTrackPidInfo(track.tpcNSigmaPi(), track.tofNSigmaPi()); + } + if (trkPIDspecies->at(0) == o2::track::PID::Proton) { + assocTrackPidInfo(track.tpcNSigmaPr(), track.tofNSigmaPr()); + } } collReduced(collision.multFT0M(), collision.numContrib(), collision.posZ()); From a105ec49aed1b9b94d6bd9b6396265543222114c Mon Sep 17 00:00:00 2001 From: scattaru Date: Thu, 5 Feb 2026 10:48:30 +0100 Subject: [PATCH 2/2] Fix column name --- PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h b/PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h index a12d740b117..8f8dbecf390 100644 --- a/PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h +++ b/PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h @@ -86,7 +86,7 @@ DECLARE_SOA_COLUMN(PtAssocTrack, ptAssocTrack, float); //! Pt of the track DECLARE_SOA_COLUMN(DcaXY, dcaXY, float); //! Impact parameter in XY of the track to the primary vertex DECLARE_SOA_COLUMN(DcaZ, dcaZ, float); //! Impact parameter in Z of the track to the primary vertex DECLARE_SOA_COLUMN(NSigmaTpc, nSigmaTpc, float); //! Number of sigma TPC -DECLARE_SOA_COLUMN(NSigmaTof, nSigmaTpc, float); //! Number of sigma TOF +DECLARE_SOA_COLUMN(NSigmaTof, nSigmaTof, float); //! Number of sigma TOF } // namespace hf_assoc_track_reduced DECLARE_SOA_TABLE(AssocTrackReds, "AOD", "ASSOCTRACKRED", //! Table with associated track info soa::Index<>,