Skip to content

[PWGJE] Adding D0 Decay channel origin #16854

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
nzardosh merged 15 commits into from
Jun 29, 2026
Merged

[PWGJE] Adding D0 Decay channel origin #16854

Changes from all commits
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
3f4e401
column name fixes jetCorrelationD0
MattOckleton Apr 13, 2026
25f3617
removing custom index columns
MattOckleton Apr 13, 2026
5044716
Merge remote-tracking branch 'refs/remotes/upstream/master'
MattOckleton May 8, 2026
3f7b78e
Adding reflections to jetCorrelationD0 task
MattOckleton May 12, 2026
ba5a80b
Merge remote-tracking branch 'refs/remotes/upstream/master'
MattOckleton May 24, 2026
ee99bb3
Fixes for MC
MattOckleton Jun 25, 2026
191d38a
Merge remote-tracking branch 'upstream/master'
MattOckleton Jun 25, 2026
dfae714
removing whitespace
MattOckleton Jun 25, 2026
faf4e90
clang-format
MattOckleton Jun 25, 2026
c7c69af
formatting tableD0McDetector
MattOckleton Jun 25, 2026
e33800b
formatting headers
MattOckleton Jun 25, 2026
cdf3f68
Merge remote-tracking branch 'refs/remotes/upstream/master'
MattOckleton Jun 26, 2026
036a8a6
Adding MC D0 decay channel
MattOckleton Jun 29, 2026
f49a882
Merge remote-tracking branch 'refs/remotes/upstream/master'
MattOckleton Jun 29, 2026
1c9d686
Fixing McD decay channel function
MattOckleton Jun 29, 2026
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
24 changes: 16 additions & 8 deletions PWGJE/Tasks/jetCorrelationD0.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -80,6 +80,7 @@ DECLARE_SOA_COLUMN(D0EtaD, d0EtaD, float);
DECLARE_SOA_COLUMN(D0PhiD, d0PhiD, float);
DECLARE_SOA_COLUMN(D0MatchedFrom, d0MatchedFrom, int);
DECLARE_SOA_COLUMN(D0SelectedAs, d0SelectedAs, int);
DECLARE_SOA_COLUMN(D0DecayChannel, d0DecayChannel, int8_t);
} // namespace d0Info

DECLARE_SOA_TABLE(D0Tables, "AOD", "D0TABLE",
Expand All @@ -106,7 +107,8 @@ DECLARE_SOA_TABLE(D0McDTables, "AOD", "D0MCDTABLE",
d0Info::D0Phi,
d0Info::D0Y,
d0Info::D0MatchedFrom,
d0Info::D0SelectedAs);
d0Info::D0SelectedAs,
d0Info::D0DecayChannel);

DECLARE_SOA_TABLE(D0McPTables, "AOD", "D0MCPTABLE",
o2::soa::Index<>,
Expand All @@ -115,7 +117,8 @@ DECLARE_SOA_TABLE(D0McPTables, "AOD", "D0MCPTABLE",
d0Info::D0Pt,
d0Info::D0Eta,
d0Info::D0Phi,
d0Info::D0Y);
d0Info::D0Y,
d0Info::D0DecayChannel);

namespace jetInfo
{
Expand Down Expand Up @@ -192,7 +195,7 @@ struct JetCorrelationD0 {
// Configurables
Configurable<std::string> eventSelections{"eventSelections", "sel8", "choose event selection"};
Configurable<bool> skipMBGapEvents{"skipMBGapEvents", false, "decide to run over MB gap events or not"};
Configurable<bool> applyRCTSelections{"applyRCTSelections", false, "decide to apply RCT selections"};
Configurable<bool> applyRCTSelections{"applyRCTSelections", true, "decide to apply RCT selections"};
Configurable<float> jetPtCutMin{"jetPtCutMin", 5.0, "minimum value of jet pt"};
Configurable<float> d0PtCutMin{"d0PtCutMin", 1.0, "minimum value of d0 pt"};
Configurable<float> jetMcPtCutMin{"jetMcPtCutMin", 3.0, "minimum value of jet pt particle level"};
Expand Down Expand Up @@ -340,13 +343,16 @@ struct JetCorrelationD0 {
const auto scores = d0Candidate.mlScores();
fillD0Histograms(d0Candidate, scores);

// flagMcMatchRec() = sign * DecayChannelMain
// |value| identifies the decay channel (D0ToPiK=1, ..., D0ToKK=5), sign identifies D0(+)/D0bar(-), 0 = no match.
int8_t d0DecayChannel = d0Candidate.flagMcMatchRec();

int matchedFrom = 0;
int decayChannel = o2::hf_decay::hf_cand_2prong::DecayChannelMain::D0ToPiK;
int selectedAs = 0;

if (d0Candidate.flagMcMatchRec() == decayChannel) { // matched to D0 on truth level
if (d0DecayChannel > 0) { // matched to a D0 on truth level (any channel)
matchedFrom = 1;
} else if (d0Candidate.flagMcMatchRec() == -decayChannel) { // matched to D0bar on truth level
} else if (d0DecayChannel < 0) { // matched to a D0bar on truth level (any channel)
matchedFrom = -1;
}
if (d0Candidate.candidateSelFlag() & BIT(0)) { // CandidateSelFlag == BIT(0) -> selected as D0
Expand All @@ -365,7 +371,8 @@ struct JetCorrelationD0 {
d0Candidate.phi(),
d0Candidate.y(),
matchedFrom,
selectedAs);
selectedAs,
d0DecayChannel);
for (const auto& jet : jets) {
if (jet.pt() < jetPtCutMin) {
continue;
Expand Down Expand Up @@ -403,7 +410,8 @@ struct JetCorrelationD0 {
d0McPCandidate.pt(),
d0McPCandidate.eta(),
d0McPCandidate.phi(),
d0McPCandidate.y());
d0McPCandidate.y(),
d0McPCandidate.flagMcMatchGen());

for (const auto& jet : jets) {
if (jet.pt() < jetMcPtCutMin) {
Expand Down
Loading