Skip to content

[PWGHF] Add ITS chi2 requirement, TPC and ITS refit conditions #12612

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
jpxrk merged 3 commits into from
Aug 18, 2025
Merged

[PWGHF] Add ITS chi2 requirement, TPC and ITS refit conditions #12612

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
8b56286
Change variable names
jpxrk Aug 13, 2025
5bbb7c5
Change enumeration names
jpxrk Aug 14, 2025
c5a777b
Add ITS chi2 requirement, TPC and ITS refit conditions
jpxrk Aug 18, 2025
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
176 changes: 95 additions & 81 deletions PWGHF/HFL/Tasks/taskSingleElectron.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,34 +38,34 @@ enum PdgCode {
};

enum SourceType {
fNotElec = 0, // not electron
fDirectCharm = 1, // electrons from prompt charm hadrons
fDirectBeauty = 2, // electrons from primary beauty hadrons
fBeautyCharm = 3, // electrons from non-prompt charm hadrons
fDirectGamma = 4, // electrons from direct photon
fGammaPi0 = 5,
fGammaEta = 6,
fGammaOmega = 7,
fGammaPhi = 8,
fGammaEtaPrime = 9,
fGammaRho0 = 10,
fGammaK0s = 11,
fGammaK0l = 12,
fGammaKe3 = 13,
fGammaLambda0 = 14,
fGammaSigma = 15,
fPi0 = 16,
fEta = 17,
fOmega = 18,
fPhi = 19,
fEtaPrime = 20,
fRho0 = 21,
fK0s = 22,
fK0l = 23,
fKe3 = 24,
fLambda0 = 25,
fSigma = 26,
fElse = 27
NotElec = 0, // not electron
DirectCharm = 1, // electrons from prompt charm hadrons
DirectBeauty = 2, // electrons from primary beauty hadrons
BeautyCharm = 3, // electrons from non-prompt charm hadrons
DirectGamma = 4, // electrons from direct photon
GammaPi0 = 5,
GammaEta = 6,
GammaOmega = 7,
GammaPhi = 8,
GammaEtaPrime = 9,
GammaRho0 = 10,
GammaK0s = 11,
GammaK0l = 12,
GammaKe3 = 13,
GammaLambda0 = 14,
GammaSigma = 15,
Pi0 = 16,
Eta = 17,
Omega = 18,
Phi = 19,
EtaPrime = 20,
Rho0 = 21,
K0s = 22,
K0l = 23,
Ke3 = 24,
Lambda0 = 25,
Sigma = 26,
Else = 27
};

struct HfTaskSingleElectron {
Expand All @@ -82,6 +82,7 @@ struct HfTaskSingleElectron {
Configurable<float> tpcNClsFoundOverFindableMin{"tpcNClsFoundOverFindableMin", 0.8, "min # of TPC found/findable clusters"};
Configurable<float> tpcChi2perNClMax{"tpcChi2perNClMax", 4., "min # of tpc chi2 per clusters"};
Configurable<int> itsIBClsMin{"itsIBClsMin", 3, "min # of its clusters in IB"};
Configurable<float> itsChi2perNClMax{"itsChi2perNClMax", 6., "min # of tpc chi2 per clusters"};
Configurable<float> dcaxyMax{"dcaxyMax", 1., "max of track dca in xy"};
Configurable<float> dcazMax{"dcazMax", 2., "max of track dca in z"};
Configurable<float> tofNSigmaMax{"tofNSigmaMax", 3., "max of tof nsigma"};
Expand All @@ -95,7 +96,7 @@ struct HfTaskSingleElectron {

// using declarations
using MyCollisions = soa::Join<aod::Collisions, aod::EvSels>;
using TracksEl = soa::Join<aod::Tracks, aod::TrackExtra, aod::TracksDCA, aod::pidTOFFullEl, aod::pidTPCFullEl>;
using TracksEl = soa::Join<aod::Tracks, aod::TrackSelection, aod::TrackSelectionExtension, aod::TracksExtra, aod::TracksDCA, aod::pidTOFFullEl, aod::pidTPCFullEl>;
using McTracksEl = soa::Join<aod::Tracks, aod::TrackExtra, aod::TracksDCA, aod::pidTOFFullEl, aod::pidTPCFullEl, aod::McTrackLabels>;

// Filter
Expand Down Expand Up @@ -131,6 +132,7 @@ struct HfTaskSingleElectron {
histos.add("tpcFoundFindableTrack", "", kTH1D, {{10, 0, 1}});
histos.add("tpcChi2Track", "", kTH1D, {{100, 0, 10}});
histos.add("itsIBClsTrack", "", kTH1D, {{10, 0, 10}});
histos.add("itsChi2Track", "", kTH1D, {{50, 0, 50}});
histos.add("dcaXYTrack", "", kTH1D, {{600, -3, 3}});
histos.add("dcaZTrack", "", kTH1D, {{600, -3, 3}});

Expand Down Expand Up @@ -168,9 +170,11 @@ struct HfTaskSingleElectron {
if (track.tpcNClsCrossedRows() < tpcNCrossedRowMin) {
return false;
}

if (track.tpcCrossedRowsOverFindableCls() < tpcNClsFoundOverFindableMin) {
return false;
}

if (track.tpcChi2NCl() > tpcChi2perNClMax) {
return false;
}
Expand All @@ -179,9 +183,14 @@ struct HfTaskSingleElectron {
return false;
}

if (track.itsChi2NCl() > itsChi2perNClMax) {
return false;
}

if (std::abs(track.dcaXY()) > dcaxyMax) {
return false;
}

if (std::abs(track.dcaZ()) > dcazMax) {
return false;
}
Expand All @@ -194,7 +203,7 @@ struct HfTaskSingleElectron {
{
auto mcpart = track.mcParticle();
if (std::abs(mcpart.pdgCode()) != kElectron) {
return fNotElec;
return NotElec;
}

int motherPdg = -999;
Expand Down Expand Up @@ -223,27 +232,27 @@ struct HfTaskSingleElectron {
auto const& grmothersIdsVec = mctrack.front().mothersIds();

if (grmothersIdsVec.empty()) {
return fDirectCharm;
return DirectCharm;
} else {
grmotherPt = mctrack.front().pt();
grmotherPdg = std::abs(mctrack.front().pdgCode());
if ((static_cast<int>(grmotherPdg / 100.) % 10) == kBottom || (static_cast<int>(grmotherPdg / 1000.) % 10) == kBottom) {
mpt = grmotherPt;
mpdg = grmotherPdg;
return fBeautyCharm;
return BeautyCharm;
}
}
}
partMother = mctrack;
}
} else if ((static_cast<int>(motherPdg / 100.) % 10) == kBottom || (static_cast<int>(motherPdg / 1000.) % 10) == kBottom) { // check if electron from beauty hadrons
return fDirectBeauty;
return DirectBeauty;
} else if (motherPdg == kGamma) { // check if electron from photon conversion
mctrack = partMother.front().template mothers_as<aod::McParticles>();
if (mctrack.size()) {
auto const& grmothersIdsVec = mctrack.front().mothersIds();
if (grmothersIdsVec.empty()) {
return fDirectGamma;
return DirectGamma;
} else {
grmotherPdg = std::abs(mctrack.front().pdgCode());
mpdg = grmotherPdg;
Expand All @@ -255,19 +264,19 @@ struct HfTaskSingleElectron {
auto const& ggrmothersIdsVec = mctrack.front().mothersIds();
if (ggrmothersIdsVec.empty()) {
if (grmotherPdg == kPi0) {
return fGammaPi0;
return GammaPi0;
} else if (grmotherPdg == kEta) {
return fGammaEta;
return GammaEta;
} else if (grmotherPdg == kOmega) {
return fGammaOmega;
return GammaOmega;
} else if (grmotherPdg == kPhi) {
return fGammaPhi;
return GammaPhi;
} else if (grmotherPdg == kEtaPrime) {
return fGammaEtaPrime;
return GammaEtaPrime;
} else if (grmotherPdg == kRho770_0) {
return fGammaRho0;
return GammaRho0;
} else {
return fElse;
return Else;
}
} else {
ggrmotherPdg = mctrack.front().pdgCode();
Expand All @@ -276,42 +285,42 @@ struct HfTaskSingleElectron {
mpt = ggrmotherPt;
if (grmotherPdg == kPi0) {
if (ggrmotherPdg == kK0Short) {
return fGammaK0s;
return GammaK0s;
} else if (ggrmotherPdg == kK0Long) {
return fGammaK0l;
return GammaK0l;
} else if (ggrmotherPdg == kKPlus) {
return fGammaKe3;
return GammaKe3;
} else if (ggrmotherPdg == kLambda0) {
return fGammaLambda0;
return GammaLambda0;
} else if (ggrmotherPdg == kSigmaPlus) {
return fGammaSigma;
return GammaSigma;
} else {
mpdg = grmotherPdg;
mpt = grmotherPt;
return fGammaPi0;
return GammaPi0;
}
} else if (grmotherPdg == kEta) {
mpdg = grmotherPdg;
mpt = grmotherPt;
return fGammaEta;
return GammaEta;
} else if (grmotherPdg == kOmega) {
mpdg = grmotherPdg;
mpt = grmotherPt;
return fGammaOmega;
return GammaOmega;
} else if (grmotherPdg == kPhi) {
mpdg = grmotherPdg;
mpt = grmotherPt;
return fGammaPhi;
return GammaPhi;
} else if (grmotherPdg == kEtaPrime) {
mpdg = grmotherPdg;
mpt = grmotherPt;
return fGammaEtaPrime;
return GammaEtaPrime;
} else if (grmotherPdg == kRho770_0) {
mpdg = grmotherPdg;
mpt = grmotherPt;
return fGammaRho0;
return GammaRho0;
} else {
return fElse;
return Else;
}
}
}
Expand All @@ -323,20 +332,20 @@ struct HfTaskSingleElectron {
auto const& grmothersIdsVec = mctrack.front().mothersIds();
if (grmothersIdsVec.empty()) {
static const std::map<int, SourceType> pdgToSource = {
{kPi0, fPi0},
{kEta, fEta},
{kOmega, fOmega},
{kPhi, fPhi},
{kEtaPrime, fEtaPrime},
{kRho770_0, fRho0},
{kKPlus, fKe3},
{kK0Long, fK0l}};
{kPi0, Pi0},
{kEta, Eta},
{kOmega, Omega},
{kPhi, Phi},
{kEtaPrime, EtaPrime},
{kRho770_0, Rho0},
{kKPlus, Ke3},
{kK0Long, K0l}};

auto it = pdgToSource.find(motherPdg);
if (it != pdgToSource.end()) {
return it->second;
}
return fElse;
return Else;

} else {
if (motherPdg == kPi0) {
Expand All @@ -345,42 +354,42 @@ struct HfTaskSingleElectron {
mpt = grmotherPt;
mpdg = grmotherPdg;
if (grmotherPdg == kK0Short) {
return fK0s;
return K0s;
} else if (grmotherPdg == kK0Long) {
return fK0l;
return K0l;
} else if (grmotherPdg == kKPlus) {
return fKe3;
return Ke3;
} else if (grmotherPdg == kLambda0) {
return fLambda0;
return Lambda0;
} else if (grmotherPdg == kSigmaPlus) {
return fSigma;
return Sigma;
} else {
mpt = motherPt;
mpdg = motherPdg;
return fPi0;
return Pi0;
}
} else if (motherPdg == kEta) {
return fEta;
return Eta;
} else if (motherPdg == kOmega) {
return fOmega;
return Omega;
} else if (motherPdg == kPhi) {
return fPhi;
return Phi;
} else if (motherPdg == kEtaPrime) {
return fEtaPrime;
return EtaPrime;
} else if (motherPdg == kRho770_0) {
return fRho0;
return Rho0;
} else if (motherPdg == kKPlus) {
return fKe3;
return Ke3;
} else if (motherPdg == kK0Long) {
return fK0l;
return K0l;
} else {
return fElse;
return Else;
}
}
}
}

return fElse;
return Else;
}

void processData(soa::Filtered<MyCollisions>::iterator const& collision,
Expand All @@ -405,13 +414,18 @@ struct HfTaskSingleElectron {
continue;
}

if (!(track.passedITSRefit() && track.passedTPCRefit())) {
continue;
}

histos.fill(HIST("etaTrack"), track.eta());
histos.fill(HIST("ptTrack"), track.pt());

histos.fill(HIST("tpcNClsTrack"), track.tpcNClsCrossedRows());
histos.fill(HIST("tpcFoundFindableTrack"), track.tpcCrossedRowsOverFindableCls());
histos.fill(HIST("tpcChi2Track"), track.tpcChi2NCl());
histos.fill(HIST("itsIBClsTrack"), track.itsNClsInnerBarrel());
histos.fill(HIST("itsChi2Track"), track.itsChi2NCl());
histos.fill(HIST("dcaXYTrack"), track.dcaXY());
histos.fill(HIST("dcaZTrack"), track.dcaZ());

Expand Down Expand Up @@ -474,22 +488,22 @@ struct HfTaskSingleElectron {
double mpt; // electron source pt
int source = getElecSource(track, mpt, mpdg);

if (source == fDirectBeauty || source == fBeautyCharm) {
if (source == DirectBeauty || source == BeautyCharm) {
histos.fill(HIST("hPdgB"), mpdg);
histos.fill(HIST("dcaBeauty"), track.pt(), track.dcaXY());
}

if (source == fDirectCharm) {
if (source == DirectCharm) {
histos.fill(HIST("hPdgC"), mpdg);
histos.fill(HIST("dcaCharm"), track.pt(), track.dcaXY());
}

if (source >= fGammaPi0 && source <= fGammaSigma) {
if (source >= GammaPi0 && source <= GammaSigma) {
histos.fill(HIST("hPdgCo"), mpdg);
histos.fill(HIST("dcaConv"), track.pt(), track.dcaXY());
}

if (source >= fPi0 && source <= fSigma) {
if (source >= Pi0 && source <= Sigma) {
histos.fill(HIST("hPdgDa"), mpdg);
histos.fill(HIST("dcaDalitz"), track.pt(), track.dcaXY());
}
Expand Down
Loading