Skip to content

[PWGJE] Adding bkg indexing to jetCorrelationD0 task #17318

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

Draft
MattOckleton wants to merge 8 commits into
base: master
Choose a base branch
from
+23 −8
Draft

[PWGJE] Adding bkg indexing to jetCorrelationD0 task #17318

Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
ab17815
Adding reflections to jetCorrelationD0 task
MattOckleton May 12, 2026
0f6d5b6
Fixes for MC
MattOckleton Jun 25, 2026
70bea2d
Adding MC D0 decay channel
MattOckleton Jun 29, 2026
98821ab
changing reflection and corr bkg handling
MattOckleton Jul 15, 2026
19d6f78
removing magic numbers
MattOckleton Aug 4, 2026
cf053b5
resolving conflicts and fixing bug in D0 bar selection
MattOckleton Aug 4, 2026
02e2b9c
clang formatting
MattOckleton Aug 4, 2026
3d5c4cc
fixing formatting for megalinter
MattOckleton Aug 4, 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
31 changes: 23 additions & 8 deletions PWGJE/Tasks/jetCorrelationD0.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@
#include <cstdint>
#include <cstdlib>
#include <string>
#include <type_traits>
#include <vector>

#include <math.h>
Expand Down Expand Up @@ -86,8 +87,7 @@ DECLARE_SOA_COLUMN(D0MD, d0MD, float);
DECLARE_SOA_COLUMN(D0PtD, d0PtD, float);
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(D0Category, d0Category, int);
DECLARE_SOA_COLUMN(D0DecayChannel, d0DecayChannel, int8_t);
} // namespace d0Info

Expand All @@ -114,9 +114,7 @@ DECLARE_SOA_TABLE(D0McDTables, "AOD", "D0MCDTABLE",
d0Info::D0Eta,
d0Info::D0Phi,
d0Info::D0Y,
d0Info::D0MatchedFrom,
d0Info::D0SelectedAs,
d0Info::D0DecayChannel);
d0Info::D0Category);

DECLARE_SOA_TABLE(D0McPTables, "AOD", "D0MCPTABLE",
o2::soa::Index<>,
Expand Down Expand Up @@ -357,6 +355,12 @@ struct JetCorrelationD0 {

int matchedFrom = 0;
int selectedAs = 0;
int category = -1; // -1 undefined, 0 signal, 1 reflection, 2-5 correlated backgrounds
constexpr int kD0ToKPi = 1;
constexpr int kD0ToKPiPi = 2;
constexpr int kD0ToPiPi = 3;
constexpr int kD0ToPiPiPi = 4;
constexpr int kD0ToKK = 5;

if (d0DecayChannel > 0) { // matched to a D0 on truth level (any channel)
matchedFrom = 1;
Expand All @@ -368,6 +372,19 @@ struct JetCorrelationD0 {
} else if (d0Candidate.candidateSelFlag() & BIT(1)) { // CandidateSelFlag == BIT(1) -> selected as D0bar
selectedAs = -1;
}
if ((std::abs(d0DecayChannel) == kD0ToKPi) && (matchedFrom != 0) && (selectedAs == matchedFrom)) {
category = 0; // signal -> D0 or D0bar, π+ K−
} else if ((d0DecayChannel == kD0ToKPi) && (selectedAs == -1 * matchedFrom)) {
category = 1; // reflection
} else if (d0DecayChannel == kD0ToKPiPi) {
category = 2; // corr bkg: π+ K− π0
} else if (d0DecayChannel == kD0ToPiPi) {
category = 3; // corr bkg: π+ π−
} else if (d0DecayChannel == kD0ToPiPiPi) {
category = 4; // corr bkg: π+ π− π0
} else if (d0DecayChannel == kD0ToKK) {
category = 5; // corr bkg: K+ K−
}

tableD0McDetector(tableCollision.lastIndex(), // might want to add some more detector level D0 quantities like prompt or non prompt info
scores[2],
Expand All @@ -378,9 +395,7 @@ struct JetCorrelationD0 {
d0Candidate.eta(),
d0Candidate.phi(),
d0Candidate.y(),
matchedFrom,
selectedAs,
d0DecayChannel);
category);
for (const auto& jet : jets) {
if (jet.pt() < jetPtCutMin) {
continue;
Expand Down
Loading