Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h
Original file line number Diff line number Diff line change
Expand Up @@ -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, nSigmaTof, float); //! Number of sigma TOF
} // namespace hf_assoc_track_reduced
DECLARE_SOA_TABLE(AssocTrackReds, "AOD", "ASSOCTRACKRED", //! Table with associated track info
soa::Index<>,
Expand All @@ -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
{
Expand Down
10 changes: 10 additions & 0 deletions PWGHF/HFC/TableProducer/correlatorDsHadrons.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -176,6 +176,7 @@ struct HfCorrelatorDsHadrons {
Produces<aod::DsCandSelInfos> candSelInfo;
Produces<aod::AssocTrackReds> assocTrackReduced;
Produces<aod::AssocTrackSels> assocTrackSelInfo;
Produces<aod::AssocTrackPids> assocTrackPidInfo;

Configurable<bool> fillHistoData{"fillHistoData", true, "Flag for filling histograms in data processes"};
Configurable<bool> fillHistoMcRec{"fillHistoMcRec", true, "Flag for filling histograms in MC Rec processes"};
Expand Down Expand Up @@ -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());
Expand Down
Loading