Skip to content

[PWGEM/Dilepton] change order of track selection #16575

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
dsekihat merged 1 commit into from
Jun 8, 2026
+13 −13
Merged

[PWGEM/Dilepton] change order of track selection #16575

Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
26 changes: 13 additions & 13 deletions PWGEM/Dilepton/Utils/ElectronModule.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -697,6 +697,10 @@
void fillMapMLPID(TCollision const& collision, TTrack const& track)
{
if (fElectronCut.usePIDML) {
if (!isElectron_TOFif(track, collision, fElectronCut)) { // minimal n sigma cut is taken from the main electron cut.
return;
}

o2::dataformats::DCA mDcaInfoCov;
mDcaInfoCov.set(999, 999, 999, 999, 999);
auto trackParCov = getTrackParCov(track);
Expand All @@ -706,10 +710,6 @@
return;
}

if (!isElectron_TOFif(track, collision, fElectronCut)) { // minimal n sigma cut is taken from the main electron cut.
return;
}

o2::analysis::pwgem::dilepton::mlpid::candidate candidate;
candidate.tpcInnerParam = track.tpcInnerParam();
candidate.tpcNClsFound = track.tpcNClsFound();
Expand All @@ -717,7 +717,7 @@
candidate.tpcNSigmaEl = track.tpcNSigmaEl();
candidate.tofNSigmaEl = fMapTOFNsigmaElReassociated[std::make_pair(collision.globalIndex(), track.globalIndex())];
int total_cluster_size_ob = 0, nl_ob = 0;
for (unsigned int layer = 3; layer < 7; layer++) {

Check failure on line 720 in PWGEM/Dilepton/Utils/ElectronModule.h

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
int cluster_size_per_layer = track.itsClsSizeInLayer(layer);
if (cluster_size_per_layer > 0) {
nl_ob++;
Expand All @@ -726,13 +726,13 @@
}
candidate.meanClusterSizeITSobCosTgl = static_cast<float>(total_cluster_size_ob) / static_cast<float>(nl_ob) * std::cos(std::atan(trackParCov.getTgl()));

std::vector<float> inputFeatures = mlResponsePID.getInputFeatures(candidate);
auto inputFeatures = mlResponsePID.getInputFeatures(candidate);
float binningFeature = mlResponsePID.getBinningFeature(candidate);

int pbin = lower_bound(fElectronCut.binsMl.value.begin(), fElectronCut.binsMl.value.end(), binningFeature) - fElectronCut.binsMl.value.begin() - 1;
if (pbin < 0) {
pbin = 0;
} else if (static_cast<int>(fElectronCut.binsMl.value.size()) - 2 < pbin) {

Check failure on line 735 in PWGEM/Dilepton/Utils/ElectronModule.h

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
pbin = static_cast<int>(fElectronCut.binsMl.value.size()) - 2;
}

Expand All @@ -752,7 +752,7 @@
}
if (fElectronCut.storeOnlyTrueElectronMC) {
const auto& mcParticle = track.template mcParticle_as<aod::McParticles>();
if (std::abs(mcParticle.pdgCode()) != 11) {

Check failure on line 755 in PWGEM/Dilepton/Utils/ElectronModule.h

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.

Check failure on line 755 in PWGEM/Dilepton/Utils/ElectronModule.h

View workflow job for this annotation

GitHub Actions / O2 linter

[pdg/explicit-code] 

Avoid hard-coded PDG codes. Use named values from PDG_t or o2::constants::physics::Pdg instead.
return false;
}
}
Expand Down Expand Up @@ -823,7 +823,7 @@
}

int total_cluster_size = 0, nl = 0;
for (unsigned int layer = 0; layer < 7; layer++) {

Check failure on line 826 in PWGEM/Dilepton/Utils/ElectronModule.h

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
int cluster_size_per_layer = track.itsClsSizeInLayer(layer);
if (cluster_size_per_layer > 0) {
nl++;
Expand Down Expand Up @@ -901,7 +901,7 @@
}

int total_cluster_size = 0, nl = 0;
for (unsigned int layer = 0; layer < 7; layer++) {

Check failure on line 904 in PWGEM/Dilepton/Utils/ElectronModule.h

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
int cluster_size_per_layer = track.itsClsSizeInLayer(layer);
if (cluster_size_per_layer > 0) {
nl++;
Expand All @@ -927,7 +927,7 @@
int pbin = lower_bound(eCut.binsMl.value.begin(), eCut.binsMl.value.end(), binningFeature) - eCut.binsMl.value.begin() - 1;
if (pbin < 0) {
pbin = 0;
} else if (static_cast<int>(eCut.binsMl.value.size()) - 2 < pbin) {

Check failure on line 930 in PWGEM/Dilepton/Utils/ElectronModule.h

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
pbin = static_cast<int>(eCut.binsMl.value.size()) - 2;
}
return fMapProbaEl[std::make_pair(collision.globalIndex(), track.globalIndex())] > eCut.cutsMl.value[pbin];
Expand Down Expand Up @@ -1097,7 +1097,7 @@
trackParCov.getSigma1Pt2());

int total_cluster_size = 0, nl = 0;
for (unsigned int layer = 0; layer < 7; layer++) {

Check failure on line 1100 in PWGEM/Dilepton/Utils/ElectronModule.h

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
int cluster_size_per_layer = track.itsClsSizeInLayer(layer);
if (cluster_size_per_layer > 0) {
nl++;
Expand All @@ -1106,7 +1106,7 @@
}

int total_cluster_size_ib = 0, nl_ib = 0;
for (unsigned int layer = 0; layer < 3; layer++) {

Check failure on line 1109 in PWGEM/Dilepton/Utils/ElectronModule.h

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
int cluster_size_per_layer = track.itsClsSizeInLayer(layer);
if (cluster_size_per_layer > 0) {
nl_ib++;
Expand All @@ -1115,7 +1115,7 @@
}

int total_cluster_size_ob = 0, nl_ob = 0;
for (unsigned int layer = 3; layer < 7; layer++) {

Check failure on line 1118 in PWGEM/Dilepton/Utils/ElectronModule.h

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.
int cluster_size_per_layer = track.itsClsSizeInLayer(layer);
if (cluster_size_per_layer > 0) {
nl_ob++;
Expand Down Expand Up @@ -1671,7 +1671,7 @@
candidate.signedMassLH = electron.sign() * hadron.sign() * eTpair.mass;
candidate.tpcNSigmaKa = hadron.tpcNSigmaKa();

std::vector<float> inputFeatures = mlResponseSCTeT.getInputFeatures(candidate);
auto inputFeatures = mlResponseSCTeT.getInputFeatures(candidate);
float binningFeature = mlResponseSCTeT.getBinningFeature(candidate);

int pbin = lower_bound(fConfigDFeT.binsMl.value.begin(), fConfigDFeT.binsMl.value.end(), binningFeature) - fConfigDFeT.binsMl.value.begin() - 1;
Expand Down Expand Up @@ -1721,7 +1721,7 @@
candidate.ptL = trackParCov.getPt();
candidate.massH = o2::constants::physics::MassK0Short;

std::vector<float> inputFeatures = mlResponseSCTeV0.getInputFeatures(candidate);
auto inputFeatures = mlResponseSCTeV0.getInputFeatures(candidate);
float binningFeature = mlResponseSCTeV0.getBinningFeature(candidate);

int pbin = lower_bound(fConfigDFeV0.binsMl.value.begin(), fConfigDFeV0.binsMl.value.end(), binningFeature) - fConfigDFeV0.binsMl.value.begin() - 1;
Expand Down Expand Up @@ -1772,7 +1772,7 @@
candidate.ptL = trackParCov.getPt();
candidate.massH = o2::constants::physics::MassLambda;

std::vector<float> inputFeatures = mlResponseSCTeV0.getInputFeatures(candidate);
auto inputFeatures = mlResponseSCTeV0.getInputFeatures(candidate);
float binningFeature = mlResponseSCTeV0.getBinningFeature(candidate);

int pbin = lower_bound(fConfigDFeV0.binsMl.value.begin(), fConfigDFeV0.binsMl.value.end(), binningFeature) - fConfigDFeV0.binsMl.value.begin() - 1;
Expand Down Expand Up @@ -1823,7 +1823,7 @@
candidate.ptL = trackParCov.getPt();
candidate.massH = o2::constants::physics::MassXiMinus;

std::vector<float> inputFeatures = mlResponseSCTeC.getInputFeatures(candidate);
auto inputFeatures = mlResponseSCTeC.getInputFeatures(candidate);
float binningFeature = mlResponseSCTeC.getBinningFeature(candidate);

int pbin = lower_bound(fConfigDFeC.binsMl.value.begin(), fConfigDFeC.binsMl.value.end(), binningFeature) - fConfigDFeC.binsMl.value.begin() - 1;
Expand Down Expand Up @@ -1874,7 +1874,7 @@
candidate.ptL = trackParCov.getPt();
candidate.massH = o2::constants::physics::MassOmegaMinus;

std::vector<float> inputFeatures = mlResponseSCTeC.getInputFeatures(candidate);
auto inputFeatures = mlResponseSCTeC.getInputFeatures(candidate);
float binningFeature = mlResponseSCTeC.getBinningFeature(candidate);

int pbin = lower_bound(fConfigDFeC.binsMl.value.begin(), fConfigDFeC.binsMl.value.end(), binningFeature) - fConfigDFeC.binsMl.value.begin() - 1;
Expand Down Expand Up @@ -1925,7 +1925,7 @@
candidate.ptL = trackParCov.getPt();
candidate.massH = o2::constants::physics::MassLambda;

std::vector<float> inputFeatures = mlResponseSCTeV0.getInputFeatures(candidate);
auto inputFeatures = mlResponseSCTeV0.getInputFeatures(candidate);
float binningFeature = mlResponseSCTeV0.getBinningFeature(candidate);

int pbin = lower_bound(fConfigDFeV0.binsMl.value.begin(), fConfigDFeV0.binsMl.value.end(), binningFeature) - fConfigDFeV0.binsMl.value.begin() - 1;
Expand Down Expand Up @@ -1976,7 +1976,7 @@
candidate.ptL = trackParCov.getPt();
candidate.massH = o2::constants::physics::MassXiMinus;

std::vector<float> inputFeatures = mlResponseSCTeC.getInputFeatures(candidate);
auto inputFeatures = mlResponseSCTeC.getInputFeatures(candidate);
float binningFeature = mlResponseSCTeC.getBinningFeature(candidate);

int pbin = lower_bound(fConfigDFeC.binsMl.value.begin(), fConfigDFeC.binsMl.value.end(), binningFeature) - fConfigDFeC.binsMl.value.begin() - 1;
Expand Down Expand Up @@ -2027,7 +2027,7 @@
candidate.ptL = trackParCov.getPt();
candidate.massH = o2::constants::physics::MassOmegaMinus;

std::vector<float> inputFeatures = mlResponseSCTeC.getInputFeatures(candidate);
auto inputFeatures = mlResponseSCTeC.getInputFeatures(candidate);
float binningFeature = mlResponseSCTeC.getBinningFeature(candidate);

int pbin = lower_bound(fConfigDFeC.binsMl.value.begin(), fConfigDFeC.binsMl.value.end(), binningFeature) - fConfigDFeC.binsMl.value.begin() - 1;
Expand Down
Loading