Skip to content

[PWGLF] Add histograms for efficiency factorization #14068

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
alcaliva merged 3 commits into from
Dec 4, 2025
Merged

[PWGLF] Add histograms for efficiency factorization #14068

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
4ec9cc8
[PWGLF] Add histograms for efficiency factorization
alcaliva Dec 3, 2025
2a4f6cf
removed trailing spaces
alcaliva Dec 3, 2025
90feb3c
fixed typos
alcaliva Dec 3, 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
269 changes: 178 additions & 91 deletions PWGLF/Tasks/Strangeness/strangenessInJets.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -302,6 +302,19 @@ struct StrangenessInJets {
registryMC.add("Proton_generated_in_jet", "Proton_generated_in_jet", HistType::kTH2F, {multBinning, ptAxisLongLived});
registryMC.add("Proton_generated_in_ue", "Proton_generated_in_ue", HistType::kTH2F, {multBinning, ptAxisLongLived});
}

// Histograms to calculate probability of hyperons to be found within jets
if (enabledSignals.value[ParticleOfInterest::kV0Particles]) {
registryMC.add("K0s_generated_fullevent", "K0s_generated_fullevent", HistType::kTH2F, {multBinning, ptAxis});
registryMC.add("Lambda_generated_fullevent", "Lambda_generated_fullevent", HistType::kTH2F, {multBinning, ptAxis});
registryMC.add("AntiLambda_generated_fullevent", "AntiLambda_generated_fullevent", HistType::kTH2F, {multBinning, ptAxis});
}
if (enabledSignals.value[ParticleOfInterest::kCascades]) {
registryMC.add("XiPos_generated_fullevent", "XiPos_generated_fullevent", HistType::kTH2F, {multBinning, ptAxis});
registryMC.add("XiNeg_generated_fullevent", "XiNeg_generated_fullevent", HistType::kTH2F, {multBinning, ptAxis});
registryMC.add("OmegaPos_generated_fullevent", "OmegaPos_generated_fullevent", HistType::kTH2F, {multBinning, ptAxis});
registryMC.add("OmegaNeg_generated_fullevent", "OmegaNeg_generated_fullevent", HistType::kTH2F, {multBinning, ptAxis});
}
}

// Histograms for mc reconstructed
Expand Down Expand Up @@ -350,58 +363,21 @@ struct StrangenessInJets {
registryMC.add("Proton_reconstructed_in_jet", "Proton_reconstructed_in_jet", HistType::kTH2F, {multBinning, ptAxisLongLived});
registryMC.add("Proton_reconstructed_in_ue", "Proton_reconstructed_in_ue", HistType::kTH2F, {multBinning, ptAxisLongLived});
}
}
}

/*
// Calculation of perpendicular axes
void getPerpendicularAxis(TVector3 p, TVector3& u, double sign)
{
// initialization
double ux(0), uy(0), uz(0);

// components of vector p
const double px = p.X();
const double py = p.Y();
const double pz = p.Z();

// protection 1
if (px == 0 && py != 0) {
uy = -(pz * pz) / py;
ux = sign * std::sqrt(py * py - (pz * pz * pz * pz) / (py * py));
uz = pz;
u.SetXYZ(ux, uy, uz);
return;
}

// protection 2
if (py == 0 && px != 0) {
ux = -(pz * pz) / px;
uy = sign * std::sqrt(px * px - (pz * pz * pz * pz) / (px * px));
uz = pz;
u.SetXYZ(ux, uy, uz);
return;
}

// equation parameters
const double a = px * px + py * py;
const double b = 2.0 * px * pz * pz;
const double c = pz * pz * pz * pz - py * py * py * py - px * px * py * py;
const double delta = b * b - 4.0 * a * c;

// protection agains delta<0
if (delta < 0) {
return;
// Histograms to calculate probability of hyperons to be found within jets
if (enabledSignals.value[ParticleOfInterest::kV0Particles]) {
registryMC.add("K0s_reconstructed_fullevent", "K0s_reconstructed_fullevent", HistType::kTH2F, {multBinning, ptAxis});
registryMC.add("Lambda_reconstructed_fullevent", "Lambda_reconstructed_fullevent", HistType::kTH2F, {multBinning, ptAxis});
registryMC.add("AntiLambda_reconstructed_fullevent", "AntiLambda_reconstructed_fullevent", HistType::kTH2F, {multBinning, ptAxis});
}
if (enabledSignals.value[ParticleOfInterest::kCascades]) {
registryMC.add("XiPos_reconstructed_fullevent", "XiPos_reconstructed_fullevent", HistType::kTH2F, {multBinning, ptAxis});
registryMC.add("XiNeg_reconstructed_fullevent", "XiNeg_reconstructed_fullevent", HistType::kTH2F, {multBinning, ptAxis});
registryMC.add("OmegaPos_reconstructed_fullevent", "OmegaPos_reconstructed_fullevent", HistType::kTH2F, {multBinning, ptAxis});
registryMC.add("OmegaNeg_reconstructed_fullevent", "OmegaNeg_reconstructed_fullevent", HistType::kTH2F, {multBinning, ptAxis});
}
}

// solutions
ux = (-b + sign * std::sqrt(delta)) / (2.0 * a);
uy = (-pz * pz - px * ux) / py;
uz = pz;
u.SetXYZ(ux, uy, uz);
return;
}
*/

// Delta phi calculation
double getDeltaPhi(double a1, double a2)
Expand Down Expand Up @@ -1246,14 +1222,58 @@ struct StrangenessInJets {
strHadronMomentum.emplace_back(particle.px(), particle.py(), particle.pz());
}

// Select physical primary particles or HF decay products
if (!isPhysicalPrimaryOrFromHF(particle, mcParticles))
continue;

// Eta and pt selection
double minPtParticle = 0.1;
if (particle.eta() < etaMin || particle.eta() > etaMax || particle.pt() < minPtParticle)
continue;

// Fill generated histograms for full event
if (particle.isPhysicalPrimary()) {
switch (particle.pdgCode()) {
case kK0Short:
if (enabledSignals.value[ParticleOfInterest::kV0Particles]) {
registryMC.fill(HIST("K0s_generated_fullevent"), genMultiplicity, particle.pt());
}
break;
case kLambda0:
if (enabledSignals.value[ParticleOfInterest::kV0Particles]) {
registryMC.fill(HIST("Lambda_generated_fullevent"), genMultiplicity, particle.pt());
}
break;
case kLambda0Bar:
if (enabledSignals.value[ParticleOfInterest::kV0Particles]) {
registryMC.fill(HIST("AntiLambda_generated_fullevent"), genMultiplicity, particle.pt());
}
break;
case kXiMinus:
if (enabledSignals.value[ParticleOfInterest::kCascades]) {
registryMC.fill(HIST("XiNeg_generated_fullevent"), genMultiplicity, particle.pt());
}
break;
case kXiPlusBar:
if (enabledSignals.value[ParticleOfInterest::kCascades]) {
registryMC.fill(HIST("XiPos_generated_fullevent"), genMultiplicity, particle.pt());
}
break;
case kOmegaMinus:
if (enabledSignals.value[ParticleOfInterest::kCascades]) {
registryMC.fill(HIST("OmegaNeg_generated_fullevent"), genMultiplicity, particle.pt());
}
break;
case kOmegaPlusBar:
if (enabledSignals.value[ParticleOfInterest::kCascades]) {
registryMC.fill(HIST("OmegaPos_generated_fullevent"), genMultiplicity, particle.pt());
}
break;
default:
break;
}
}

// Select physical primary particles or HF decay products
if (!isPhysicalPrimaryOrFromHF(particle, mcParticles))
continue;

// Build 4-momentum assuming charged pion mass
static constexpr float MassPionChargedSquared = o2::constants::physics::MassPionCharged * o2::constants::physics::MassPionCharged;
const double energy = std::sqrt(particle.p() * particle.p() + MassPionChargedSquared);
Expand Down Expand Up @@ -1472,7 +1492,7 @@ struct StrangenessInJets {
// Reconstructed MC events
void processMCreconstructed(SimCollisions const& collisions, soa::Join<aod::McCollisions, aod::McCentFT0Ms> const&,
DaughterTracksMC const& mcTracks, aod::V0Datas const& fullV0s,
aod::CascDataExt const& Cascades, const aod::McParticles&)
aod::CascDataExt const& Cascades, aod::McParticles const& mcParticles)
{
// Define per-event containers
std::vector<fastjet::PseudoJet> fjParticles;
Expand Down Expand Up @@ -1520,6 +1540,89 @@ struct StrangenessInJets {
auto cascPerColl = Cascades.sliceBy(perCollisionCasc, collision.globalIndex());
auto tracksPerColl = mcTracks.sliceBy(perCollisionTrk, collision.globalIndex());

// V0 particles
if (enabledSignals.value[ParticleOfInterest::kV0Particles]) {
for (const auto& v0 : v0sPerColl) {
const auto& pos = v0.posTrack_as<DaughterTracksMC>();
const auto& neg = v0.negTrack_as<DaughterTracksMC>();

// Get MC particles
if (!pos.has_mcParticle() || !neg.has_mcParticle())
continue;
auto posParticle = pos.mcParticle_as<aod::McParticles>();
auto negParticle = neg.mcParticle_as<aod::McParticles>();
if (!posParticle.has_mothers() || !negParticle.has_mothers())
continue;

// Get Mothers
auto motherPos = mcParticles.iteratorAt(posParticle.mothersIds()[0]);
auto motherNeg = mcParticles.iteratorAt(negParticle.mothersIds()[0]);
if (motherPos != motherNeg)
continue;
if (!motherPos.isPhysicalPrimary())
continue;

// K0s
if (passedK0ShortSelection(v0, pos, neg) && motherPos.pdgCode() == kK0Short) {
registryMC.fill(HIST("K0s_reconstructed_fullevent"), multiplicity, v0.pt());
}
// Lambda
if (passedLambdaSelection(v0, pos, neg) && motherPos.pdgCode() == kLambda0) {
registryMC.fill(HIST("Lambda_reconstructed_fullevent"), multiplicity, v0.pt());
}
// AntiLambda
if (passedAntiLambdaSelection(v0, pos, neg) && motherPos.pdgCode() == kLambda0Bar) {
registryMC.fill(HIST("AntiLambda_reconstructed_fullevent"), multiplicity, v0.pt());
}
}
}

// Cascades
if (enabledSignals.value[ParticleOfInterest::kCascades]) {
for (const auto& casc : cascPerColl) {
auto bach = casc.bachelor_as<DaughterTracksMC>();
auto pos = casc.posTrack_as<DaughterTracksMC>();
auto neg = casc.negTrack_as<DaughterTracksMC>();

// Get MC particles
if (!bach.has_mcParticle() || !pos.has_mcParticle() || !neg.has_mcParticle())
continue;
auto posParticle = pos.mcParticle_as<aod::McParticles>();
auto negParticle = neg.mcParticle_as<aod::McParticles>();
auto bachParticle = bach.mcParticle_as<aod::McParticles>();
if (!posParticle.has_mothers() || !negParticle.has_mothers() || !bachParticle.has_mothers())
continue;

// Select particles originating from the same parent
auto motherPos = mcParticles.iteratorAt(posParticle.mothersIds()[0]);
auto motherNeg = mcParticles.iteratorAt(negParticle.mothersIds()[0]);
auto motherBach = mcParticles.iteratorAt(bachParticle.mothersIds()[0]);
if (motherPos != motherNeg)
continue;
if (std::abs(motherPos.pdgCode()) != kLambda0)
continue;
if (!motherBach.isPhysicalPrimary())
continue;

// Xi+
if (passedXiSelection(casc, pos, neg, bach, collision) && bach.sign() > 0 && motherBach.pdgCode() == kXiPlusBar) {
registryMC.fill(HIST("XiPos_reconstructed_fullevent"), multiplicity, casc.pt());
}
// Xi-
if (passedXiSelection(casc, pos, neg, bach, collision) && bach.sign() < 0 && motherBach.pdgCode() == kXiMinus) {
registryMC.fill(HIST("XiNeg_reconstructed_fullevent"), multiplicity, casc.pt());
}
// Omega+
if (passedOmegaSelection(casc, pos, neg, bach, collision) && bach.sign() > 0 && motherBach.pdgCode() == kOmegaPlusBar) {
registryMC.fill(HIST("OmegaPos_reconstructed_fullevent"), multiplicity, casc.pt());
}
// Omega-
if (passedOmegaSelection(casc, pos, neg, bach, collision) && bach.sign() < 0 && motherBach.pdgCode() == kOmegaMinus) {
registryMC.fill(HIST("OmegaNeg_reconstructed_fullevent"), multiplicity, casc.pt());
}
}
}

// Loop over reconstructed tracks
for (auto const& track : tracksPerColl) {
if (!passedTrackSelectionForJetReconstruction(track))
Expand Down Expand Up @@ -1596,18 +1699,11 @@ struct StrangenessInJets {
continue;

// Select particles originating from the same parent
int pdgParent(0);
bool isPhysPrim = false;
for (const auto& particleMotherOfNeg : negParticle.mothers_as<aod::McParticles>()) {
for (const auto& particleMotherOfPos : posParticle.mothers_as<aod::McParticles>()) {
if (particleMotherOfNeg == particleMotherOfPos) {
pdgParent = particleMotherOfNeg.pdgCode();
isPhysPrim = particleMotherOfNeg.isPhysicalPrimary();
}
}
}
if (pdgParent == 0)
auto motherPos = mcParticles.iteratorAt(posParticle.mothersIds()[0]);
auto motherNeg = mcParticles.iteratorAt(negParticle.mothersIds()[0]);
if (motherPos != motherNeg)
continue;
bool isPhysPrim = motherPos.isPhysicalPrimary();

// Compute distance from jet and UE axes
double deltaEtaJet = v0dir.Eta() - selectedJet[i].Eta();
Expand All @@ -1621,7 +1717,7 @@ struct StrangenessInJets {
double deltaRue2 = std::sqrt(deltaEtaUe2 * deltaEtaUe2 + deltaPhiUe2 * deltaPhiUe2);

// K0s
if (passedK0ShortSelection(v0, pos, neg) && pdgParent == kK0Short && isPhysPrim) {
if (passedK0ShortSelection(v0, pos, neg) && motherPos.pdgCode() == kK0Short && isPhysPrim) {
if (deltaRjet < rJet) {
registryMC.fill(HIST("K0s_reconstructed_jet"), multiplicity, v0.pt());
}
Expand All @@ -1630,7 +1726,7 @@ struct StrangenessInJets {
}
}
// Lambda
if (passedLambdaSelection(v0, pos, neg) && pdgParent == kLambda0 && isPhysPrim) {
if (passedLambdaSelection(v0, pos, neg) && motherPos.pdgCode() == kLambda0 && isPhysPrim) {
if (deltaRjet < rJet) {
registryMC.fill(HIST("Lambda_reconstructed_jet"), multiplicity, v0.pt());
}
Expand All @@ -1639,7 +1735,7 @@ struct StrangenessInJets {
}
}
// AntiLambda
if (passedAntiLambdaSelection(v0, pos, neg) && pdgParent == kLambda0Bar && isPhysPrim) {
if (passedAntiLambdaSelection(v0, pos, neg) && motherPos.pdgCode() == kLambda0Bar && isPhysPrim) {
if (deltaRjet < rJet) {
registryMC.fill(HIST("AntiLambda_reconstructed_jet"), multiplicity, v0.pt());
}
Expand All @@ -1650,7 +1746,7 @@ struct StrangenessInJets {

// Fill inclusive spectra
// K0s
if (passedK0ShortSelection(v0, pos, neg) && pdgParent == kK0Short) {
if (passedK0ShortSelection(v0, pos, neg) && motherPos.pdgCode() == kK0Short) {
if (deltaRjet < rJet) {
registryMC.fill(HIST("K0s_reconstructed_jet_incl"), multiplicity, v0.pt());
}
Expand All @@ -1659,7 +1755,7 @@ struct StrangenessInJets {
}
}
// Lambda
if (passedLambdaSelection(v0, pos, neg) && pdgParent == kLambda0) {
if (passedLambdaSelection(v0, pos, neg) && motherPos.pdgCode() == kLambda0) {
if (deltaRjet < rJet) {
registryMC.fill(HIST("Lambda_reconstructed_jet_incl"), multiplicity, v0.pt());
}
Expand All @@ -1668,7 +1764,7 @@ struct StrangenessInJets {
}
}
// AntiLambda
if (passedAntiLambdaSelection(v0, pos, neg) && pdgParent == kLambda0Bar) {
if (passedAntiLambdaSelection(v0, pos, neg) && motherPos.pdgCode() == kLambda0Bar) {
if (deltaRjet < rJet) {
registryMC.fill(HIST("AntiLambda_reconstructed_jet_incl"), multiplicity, v0.pt());
}
Expand Down Expand Up @@ -1696,23 +1792,14 @@ struct StrangenessInJets {
continue;

// Select particles originating from the same parent
int pdgParent(0);
bool isPhysPrim = false;
for (const auto& particleMotherOfNeg : negParticle.mothers_as<aod::McParticles>()) {
for (const auto& particleMotherOfPos : posParticle.mothers_as<aod::McParticles>()) {
for (const auto& particleMotherOfBach : bachParticle.mothers_as<aod::McParticles>()) {
if (particleMotherOfNeg != particleMotherOfPos)
continue;
if (std::abs(particleMotherOfNeg.pdgCode()) != kLambda0)
continue;
isPhysPrim = particleMotherOfBach.isPhysicalPrimary();
pdgParent = particleMotherOfBach.pdgCode();
}
}
}
if (pdgParent == 0)
auto motherPos = mcParticles.iteratorAt(posParticle.mothersIds()[0]);
auto motherNeg = mcParticles.iteratorAt(negParticle.mothersIds()[0]);
auto motherBach = mcParticles.iteratorAt(bachParticle.mothersIds()[0]);
if (motherPos != motherNeg)
continue;
if (std::abs(motherPos.pdgCode()) != kLambda0)
continue;
if (!isPhysPrim)
if (!motherBach.isPhysicalPrimary())
continue;

// Compute distances from jet and UE axes
Expand All @@ -1728,7 +1815,7 @@ struct StrangenessInJets {
double deltaRue2 = std::sqrt(deltaEtaUe2 * deltaEtaUe2 + deltaPhiUe2 * deltaPhiUe2);

// Xi+
if (passedXiSelection(casc, pos, neg, bach, collision) && bach.sign() > 0 && pdgParent == kXiPlusBar) {
if (passedXiSelection(casc, pos, neg, bach, collision) && bach.sign() > 0 && motherBach.pdgCode() == kXiPlusBar) {
if (deltaRjet < rJet) {
registryMC.fill(HIST("XiPos_reconstructed_jet"), multiplicity, casc.pt());
}
Expand All @@ -1737,7 +1824,7 @@ struct StrangenessInJets {
}
}
// Xi-
if (passedXiSelection(casc, pos, neg, bach, collision) && bach.sign() < 0 && pdgParent == kXiMinus) {
if (passedXiSelection(casc, pos, neg, bach, collision) && bach.sign() < 0 && motherBach.pdgCode() == kXiMinus) {
if (deltaRjet < rJet) {
registryMC.fill(HIST("XiNeg_reconstructed_jet"), multiplicity, casc.pt());
}
Expand All @@ -1746,7 +1833,7 @@ struct StrangenessInJets {
}
}
// Omega+
if (passedOmegaSelection(casc, pos, neg, bach, collision) && bach.sign() > 0 && pdgParent == kOmegaPlusBar) {
if (passedOmegaSelection(casc, pos, neg, bach, collision) && bach.sign() > 0 && motherBach.pdgCode() == kOmegaPlusBar) {
if (deltaRjet < rJet) {
registryMC.fill(HIST("OmegaPos_reconstructed_jet"), multiplicity, casc.pt());
}
Expand All @@ -1755,7 +1842,7 @@ struct StrangenessInJets {
}
}
// Omega-
if (passedOmegaSelection(casc, pos, neg, bach, collision) && bach.sign() < 0 && pdgParent == kOmegaMinus) {
if (passedOmegaSelection(casc, pos, neg, bach, collision) && bach.sign() < 0 && motherBach.pdgCode() == kOmegaMinus) {
if (deltaRjet < rJet) {
registryMC.fill(HIST("OmegaNeg_reconstructed_jet"), multiplicity, casc.pt());
}
Expand Down
Loading