Skip to content

Commit 4304310

Browse files
authored
[PWGLF] feat: Add PDG code of ITS track to non-prompt cascade tables (#8429)
1 parent 155a551 commit 4304310

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

PWGLF/DataModel/LFNonPromptCascadeTables.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ DECLARE_SOA_COLUMN(ITSClusSize, itsClusSize, float);
2929
DECLARE_SOA_COLUMN(IsGoodMatch, isGoodMatch, bool);
3030
DECLARE_SOA_COLUMN(IsGoodCascade, isGoodCascade, bool);
3131
DECLARE_SOA_COLUMN(PdgCodeMom, pdgCodeMom, int);
32+
DECLARE_SOA_COLUMN(PdgCodeITStrack, pdgCodeITStrack, int);
3233
DECLARE_SOA_COLUMN(IsFromBeauty, isFromBeauty, bool);
3334
DECLARE_SOA_COLUMN(IsFromCharm, isFromCharm, bool);
3435

@@ -165,6 +166,7 @@ DECLARE_SOA_TABLE(NPCascTableMC, "AOD", "NPCASCTABLEMC",
165166
NPCascadeTable::IsGoodMatch,
166167
NPCascadeTable::IsGoodCascade,
167168
NPCascadeTable::PdgCodeMom,
169+
NPCascadeTable::PdgCodeITStrack,
168170
NPCascadeTable::IsFromBeauty,
169171
NPCascadeTable::IsFromCharm,
170172
NPCascadeTable::PvX,

PWGLF/Tasks/Strangeness/nonPromptCascade.cxx

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@ struct NPCascCandidate {
5151
bool isGoodMatch;
5252
bool isGoodCascade;
5353
int pdgCodeMom;
54+
int pdgCodeITStrack;
5455
bool isFromBeauty;
5556
bool isFromCharm;
5657
float pvX;
@@ -597,8 +598,9 @@ struct NonPromptCascadeTask {
597598
fromHF = isFromHF(track.mcParticle());
598599
pdgCodeMom = track.mcParticle().has_mothers() ? track.mcParticle().mothers_as<aod::McParticles>()[0].pdgCode() : 0;
599600
}
601+
int itsTrackPDG = ITStrack.has_mcParticle() ? ITStrack.mcParticle().pdgCode() : 0;
600602

601-
candidates.emplace_back(NPCascCandidate{track.globalIndex(), ITStrack.globalIndex(), trackedCascade.collisionId(), trackedCascade.matchingChi2(), trackedCascade.itsClsSize(), isGoodMatch, isGoodCascade, pdgCodeMom, std::get<0>(fromHF), std::get<1>(fromHF),
603+
candidates.emplace_back(NPCascCandidate{track.globalIndex(), ITStrack.globalIndex(), trackedCascade.collisionId(), trackedCascade.matchingChi2(), trackedCascade.itsClsSize(), isGoodMatch, isGoodCascade, pdgCodeMom, itsTrackPDG, std::get<0>(fromHF), std::get<1>(fromHF),
602604
primaryVertex.getX(), primaryVertex.getY(), primaryVertex.getZ(),
603605
track.pt(), track.eta(), track.phi(),
604606
protonTrack.pt(), protonTrack.eta(), pionTrack.pt(), pionTrack.eta(), bachelor.pt(), bachelor.eta(),
@@ -627,7 +629,7 @@ struct NonPromptCascadeTask {
627629
auto mcCollision = particle.mcCollision_as<aod::McCollisions>();
628630
auto label = collisions.iteratorAt(c.collisionID);
629631

630-
NPCTableMC(c.matchingChi2, c.itsClusSize, c.isGoodMatch, c.isGoodCascade, c.pdgCodeMom, c.isFromBeauty, c.isFromCharm,
632+
NPCTableMC(c.matchingChi2, c.itsClusSize, c.isGoodMatch, c.isGoodCascade, c.pdgCodeMom, c.pdgCodeITStrack, c.isFromBeauty, c.isFromCharm,
631633
c.pvX, c.pvY, c.pvZ,
632634
c.cascPt, c.cascEta, c.cascPhi,
633635
c.protonPt, c.protonEta, c.pionPt, c.pionEta, c.bachPt, c.bachEta,
@@ -847,7 +849,7 @@ struct NonPromptCascadeTask {
847849
daughtersDCA dDCA;
848850
fillDauDCA(trackedCascade, bachelor, protonTrack, pionTrack, primaryVertex, isOmega, dDCA);
849851

850-
candidates.emplace_back(NPCascCandidate{track.globalIndex(), ITStrack.globalIndex(), trackedCascade.collisionId(), trackedCascade.matchingChi2(), trackedCascade.itsClsSize(), 0, 0, 0, 0, 0,
852+
candidates.emplace_back(NPCascCandidate{track.globalIndex(), ITStrack.globalIndex(), trackedCascade.collisionId(), trackedCascade.matchingChi2(), trackedCascade.itsClsSize(), 0, 0, 0, 0, 0, 0,
851853
primaryVertex.getX(), primaryVertex.getY(), primaryVertex.getZ(),
852854
track.pt(), track.eta(), track.phi(),
853855
protonTrack.pt(), protonTrack.eta(), pionTrack.pt(), pionTrack.eta(), bachelor.pt(), bachelor.eta(),

0 commit comments

Comments
 (0)