Skip to content

Commit 99559b9

Browse files
[PWGDQ] Adding code to process MC events in Fwd PID task (#8510)
Co-authored-by: Lucamicheletti93 <luca.mike93@gmail.com>
1 parent 2412137 commit 99559b9

File tree

3 files changed

+295
-25
lines changed

3 files changed

+295
-25
lines changed

PWGDQ/DataModel/ReducedInfoTables.h

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -398,6 +398,9 @@ DECLARE_SOA_COLUMN(FwdDcaX, fwdDcaX, float);
398398
DECLARE_SOA_COLUMN(FwdDcaY, fwdDcaY, float); //!
399399
DECLARE_SOA_COLUMN(MftClusterSizesAndTrackFlags, mftClusterSizesAndTrackFlags, uint64_t); //!
400400
DECLARE_SOA_COLUMN(MftNClusters, mftNClusters, int); //!
401+
DECLARE_SOA_INDEX_COLUMN(ReducedMCTrack, reducedMCTrack); //!
402+
DECLARE_SOA_COLUMN(McMask, mcMask, uint16_t); //!
403+
DECLARE_SOA_COLUMN(McReducedFlags, mcReducedFlags, uint16_t); //!
401404
} // namespace reducedmft
402405

403406
// MFT track kinematics
@@ -410,8 +413,13 @@ DECLARE_SOA_TABLE(ReducedMFTsExtra, "AOD", "RMFTEXTRA", //!
410413
reducedmft::MftClusterSizesAndTrackFlags, reducedmft::Sign,
411414
reducedmft::FwdDcaX, reducedmft::FwdDcaY, reducedmft::MftNClusters);
412415

416+
DECLARE_SOA_TABLE(ReducedMFTLabels, "AOD", "RTMFTLABELS", //!
417+
reducedmft::ReducedMCTrackId, reducedmft::McMask, reducedmft::McReducedFlags);
418+
413419
// iterator
414420
using ReducedMFT = ReducedMFTs::iterator;
421+
using ReducedMFTExtra = ReducedMFTsExtra::iterator;
422+
using ReducedMFTLabel = ReducedMFTLabels::iterator;
415423

416424
// muon quantities
417425
namespace reducedmuon
@@ -833,17 +841,18 @@ DECLARE_SOA_TABLE(DitracksExtra, "AOD", "RTDITRKEXTRA", //!
833841
// mft PID reduced data model
834842
namespace fwdpid
835843
{
836-
DECLARE_SOA_COLUMN(Pt, pt, float); //!
837-
DECLARE_SOA_COLUMN(Eta, eta, float); //!
838-
DECLARE_SOA_COLUMN(Phi, phi, float); //!
839-
DECLARE_SOA_COLUMN(Sign, sign, int); //!
844+
DECLARE_SOA_COLUMN(Pt, pt, float); //!
845+
DECLARE_SOA_COLUMN(Eta, eta, float); //!
846+
DECLARE_SOA_COLUMN(Phi, phi, float); //!
847+
DECLARE_SOA_COLUMN(Sign, sign, int); //!
848+
DECLARE_SOA_COLUMN(McDecision, mcDecision, uint32_t); //!
840849
} // namespace fwdpid
841850

842851
DECLARE_SOA_TABLE(FwdPidsAll, "AOD", "RTFWDPIDALL", //!
843852
fwdtrack::TrackType, collision::PosX, collision::PosY, collision::PosZ, collision::NumContrib,
844853
fwdpid::Pt, fwdpid::Eta, fwdpid::Phi, fwdpid::Sign,
845854
reducedmft::MftClusterSizesAndTrackFlags,
846-
reducedmft::FwdDcaX, reducedmft::FwdDcaY, fwdtrack::Chi2MatchMCHMID, fwdtrack::Chi2MatchMCHMFT);
855+
reducedmft::FwdDcaX, reducedmft::FwdDcaY, fwdtrack::Chi2MatchMCHMID, fwdtrack::Chi2MatchMCHMFT, fwdpid::McDecision);
847856

848857
using FwdPidAll = FwdPidsAll::iterator;
849858

0 commit comments

Comments
 (0)