Skip to content

[PWGLF] updated rec loop for eff #14062

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

[PWGLF] updated rec loop for eff #14062

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
7f6e2fd
updated rec loop for eff
Dec 2, 2025
ab4fe4b
removed unused variable
Dec 3, 2025
2bf69a7
removed unused variable2
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
273 changes: 151 additions & 122 deletions PWGLF/Tasks/Strangeness/lambdak0seff.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@
#include <TH1F.h>
#include <TH2F.h>
#include <THn.h>
#include <TLorentzVector.h>

Check failure on line 51 in PWGLF/Tasks/Strangeness/lambdak0seff.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[root/lorentz-vector] 

Do not use the TLorentzVector legacy class. Use std::array with RecoDecay methods or the ROOT::Math::LorentzVector template instead.
#include <TMath.h>
#include <TObjArray.h>
#include <TPDGCode.h>
Expand All @@ -58,7 +58,7 @@
#include <cmath>
#include <cstdlib>
#include <deque>
#include <iostream>

Check failure on line 61 in PWGLF/Tasks/Strangeness/lambdak0seff.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[include-iostream] 

Do not include iostream. Use O2 logging instead.
#include <iterator>
#include <set> // <<< CHANGED: for dedup sets
#include <string>
Expand Down Expand Up @@ -146,6 +146,7 @@
std::vector<AxisSpec> runaxes2 = {thnAxisInvMass, axisGrp.configthnAxispT, axisGrp.configetaAxis, axisGrp.configvzAxis, axisGrp.configcentAxis};

histos.add("hCentrality", "Centrality distribution", kTH1F, {{axisGrp.configcentAxis}});
histos.add("hCentralitymc", "Centrality distribution MC", kTH1F, {{axisGrp.configcentAxis}});
histos.add("hSparseGenLambda", "hSparseGenLambda", HistType::kTHnSparseF, runaxes2, true);
histos.add("hSparseGenAntiLambda", "hSparseGenAntiLambda", HistType::kTHnSparseF, runaxes2, true);
histos.add("hSparseRecLambda", "hSparseRecLambda", HistType::kTHnSparseF, runaxes2, true);
Expand Down Expand Up @@ -239,10 +240,10 @@
return false;
}
if (pid == 0 && (candidate.positivept() < cfgDaughPrPt || candidate.negativept() < cfgDaughPiPt)) {
return false; // doesn´t pass lambda pT sels
return false;
}
if (pid == 1 && (candidate.positivept() < cfgDaughPiPt || candidate.negativept() < cfgDaughPrPt)) {
return false; // doesn´t pass antilambda pT sels
return false;
}
if (std::abs(candidate.positiveeta()) > ConfDaughEta || std::abs(candidate.negativeeta()) > ConfDaughEta) {
return false;
Expand All @@ -257,7 +258,7 @@
if (analyzeK0s && (TMath::Abs(candidate.dcapostopv()) < cMinV0DCAPi || TMath::Abs(candidate.dcanegtopv()) < cMinV0DCAPi)) {
return false;
}
if (analyzeK0s && (candidate.qtarm() / (std::abs(candidate.alpha()))) < 0.2) {

Check failure on line 261 in PWGLF/Tasks/Strangeness/lambdak0seff.cxx

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.
return false;
}

Expand Down Expand Up @@ -285,166 +286,194 @@
Filter centralityFilter = (nabs(aod::cent::centFT0C) < cfgCutCentralityMax && nabs(aod::cent::centFT0C) > cfgCutCentralityMin);
Filter acceptanceFilter = (nabs(aod::track::eta) < cfgCutEta && nabs(aod::track::pt) > cfgCutPT);

using CollisionMCTrueTable = aod::McCollisions;
// using EventCandidatesMC = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Cs, aod::McCollisionLabels>>;
using EventCandidatesMC = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Cs>>;
using AllTrackCandidates = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::TrackSelection, aod::pidTPCFullPi, aod::pidTPCFullPr, aod::pidTPCFullKa>>;
using ResoV0s = aod::V0Datas;

using TrackMCTrueTable = aod::McParticles;
Preslice<ResoV0s> perCollision = aod::v0data::collisionId;

ROOT::Math::PxPyPzMVector lambdadummymc, antiLambdadummymc, kshortdummymc, protonmc, pionmc, antiProtonmc, antiPionmc;

void processMC(EventCandidatesMC::iterator const& collision, AllTrackCandidates const& /*tracks*/, TrackMCTrueTable const& GenParticles, ResoV0s const& V0s)
void processMC(CollisionMCTrueTable::iterator const&, EventCandidatesMC const& RecCollisions, TrackMCTrueTable const& GenParticles, ResoV0s const& V0s, AllTrackCandidates const&)
{
if (!collision.sel8()) {
return;
}
double centrality = -999.;
centrality = collision.centFT0C();
// centrality = collision.multiplicity();
double vz = collision.posZ();

if (evselGrp.additionalEvSel && (!collision.selection_bit(aod::evsel::kNoSameBunchPileup) || !collision.selection_bit(aod::evsel::kIsGoodZvtxFT0vsPV))) {
return;
}
if (evselGrp.additionalEvSel2 && (!collision.selection_bit(aod::evsel::kNoTimeFrameBorder) || !collision.selection_bit(aod::evsel::kNoITSROFrameBorder))) {
return;
}
if (evselGrp.additionalEvSel3 && !collision.selection_bit(o2::aod::evsel::kIsGoodITSLayersAll)) {
return;
}

histos.fill(HIST("hCentrality"), centrality);

for (const auto& v0 : V0s) {
for (auto& collision : RecCollisions) {

Check failure on line 303 in PWGLF/Tasks/Strangeness/lambdak0seff.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[const-ref-in-for-loop] 

Use constant references for non-modified iterators in range-based for loops.

auto postrack = v0.template posTrack_as<AllTrackCandidates>();
auto negtrack = v0.template negTrack_as<AllTrackCandidates>();
if (!collision.sel8()) {
continue;
}
double centrality = -999.;
centrality = collision.centFT0C();
double vz = collision.posZ();

if (analyzeLambda && analyzeK0s)
if (evselGrp.additionalEvSel && (!collision.selection_bit(aod::evsel::kNoSameBunchPileup) || !collision.selection_bit(aod::evsel::kIsGoodZvtxFT0vsPV))) {
continue;
if (!analyzeLambda && !analyzeK0s)
}
if (evselGrp.additionalEvSel2 && (!collision.selection_bit(aod::evsel::kNoTimeFrameBorder) || !collision.selection_bit(aod::evsel::kNoITSROFrameBorder))) {
continue;
}
if (evselGrp.additionalEvSel3 && !collision.selection_bit(o2::aod::evsel::kIsGoodITSLayersAll)) {
continue;
}

int LambdaTag = 0;
int aLambdaTag = 0;
int K0sTag = 0;
histos.fill(HIST("hCentrality"), centrality);

const auto signpos = postrack.sign();
const auto signneg = negtrack.sign();
auto v0sThisColl = V0s.sliceBy(perCollision, collision.globalIndex());

if (signpos < 0 || signneg > 0) {
continue;
}
if (analyzeLambda) {
if (isSelectedV0Daughter(v0, postrack, 0, 0) && isSelectedV0Daughter(v0, negtrack, 1, 0)) {
LambdaTag = 1;
for (const auto& v0 : v0sThisColl) {

auto postrack = v0.template posTrack_as<AllTrackCandidates>();
auto negtrack = v0.template negTrack_as<AllTrackCandidates>();

if (analyzeLambda && analyzeK0s)
continue;
if (!analyzeLambda && !analyzeK0s)
continue;

int LambdaTag = 0;
int aLambdaTag = 0;
int K0sTag = 0;

const auto signpos = postrack.sign();
const auto signneg = negtrack.sign();

if (signpos < 0 || signneg > 0) {
continue;
}
if (isSelectedV0Daughter(v0, negtrack, 0, 1) && isSelectedV0Daughter(v0, postrack, 1, 1)) {
aLambdaTag = 1;
if (analyzeLambda) {
if (isSelectedV0Daughter(v0, postrack, 0, 0) && isSelectedV0Daughter(v0, negtrack, 1, 0)) {
LambdaTag = 1;
}
if (isSelectedV0Daughter(v0, negtrack, 0, 1) && isSelectedV0Daughter(v0, postrack, 1, 1)) {
aLambdaTag = 1;
}
}
}
if (analyzeK0s) {
if (isSelectedV0Daughter(v0, postrack, 0, 0) && isSelectedV0Daughter(v0, negtrack, 1, 0)) {
K0sTag = 1;
if (analyzeK0s) {
if (isSelectedV0Daughter(v0, postrack, 0, 0) && isSelectedV0Daughter(v0, negtrack, 1, 0)) {
K0sTag = 1;
}
}
}

if (analyzeLambda && (!LambdaTag && !aLambdaTag))
continue;
if (analyzeK0s && (!K0sTag))
continue;
if (analyzeLambda && (!LambdaTag && !aLambdaTag))
continue;
if (analyzeK0s && (!K0sTag))
continue;

if (!SelectionV0(collision, v0)) {
continue;
}
if (!SelectionV0(collision, v0)) {
continue;
}

if (analyzeLambda) {
if (LambdaTag) {
Proton = ROOT::Math::PxPyPzMVector(v0.pxpos(), v0.pypos(), v0.pzpos(), massPr);
AntiPion = ROOT::Math::PxPyPzMVector(v0.pxneg(), v0.pyneg(), v0.pzneg(), massPi);
Lambdadummy = Proton + AntiPion;
if (analyzeLambda) {
if (LambdaTag) {
Proton = ROOT::Math::PxPyPzMVector(v0.pxpos(), v0.pypos(), v0.pzpos(), massPr);
AntiPion = ROOT::Math::PxPyPzMVector(v0.pxneg(), v0.pyneg(), v0.pzneg(), massPi);
Lambdadummy = Proton + AntiPion;
}
if (aLambdaTag) {
AntiProton = ROOT::Math::PxPyPzMVector(v0.pxneg(), v0.pyneg(), v0.pzneg(), massPr);
Pion = ROOT::Math::PxPyPzMVector(v0.pxpos(), v0.pypos(), v0.pzpos(), massPi);
AntiLambdadummy = AntiProton + Pion;
}

if (shouldReject(LambdaTag, aLambdaTag, Lambdadummy, AntiLambdadummy)) {
continue;
}
}
if (aLambdaTag) {
AntiProton = ROOT::Math::PxPyPzMVector(v0.pxneg(), v0.pyneg(), v0.pzneg(), massPr);
Pion = ROOT::Math::PxPyPzMVector(v0.pxpos(), v0.pypos(), v0.pzpos(), massPi);
AntiLambdadummy = AntiProton + Pion;

if (analyzeK0s) {
if (K0sTag) {
Pion = ROOT::Math::PxPyPzMVector(v0.pxpos(), v0.pypos(), v0.pzpos(), massPi);
AntiPion = ROOT::Math::PxPyPzMVector(v0.pxneg(), v0.pyneg(), v0.pzneg(), massPi);
K0sdummy = Pion + AntiPion;
}
}

if (shouldReject(LambdaTag, aLambdaTag, Lambdadummy, AntiLambdadummy)) {
if (TMath::Abs(v0.eta()) > 0.8)

Check failure on line 394 in PWGLF/Tasks/Strangeness/lambdak0seff.cxx

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.
continue;
}
}

if (analyzeK0s) {
if (LambdaTag) {
Lambda = Proton + AntiPion;
histos.fill(HIST("hSparseRecLambda"), v0.mLambda(), v0.pt(), v0.eta(), vz, centrality);
}
if (aLambdaTag) {
AntiLambda = AntiProton + Pion;
histos.fill(HIST("hSparseRecAntiLambda"), v0.mAntiLambda(), v0.pt(), v0.eta(), vz, centrality);
}
if (K0sTag) {
Pion = ROOT::Math::PxPyPzMVector(v0.pxpos(), v0.pypos(), v0.pzpos(), massPi);
AntiPion = ROOT::Math::PxPyPzMVector(v0.pxneg(), v0.pyneg(), v0.pzneg(), massPi);
K0sdummy = Pion + AntiPion;
histos.fill(HIST("hSparseRecK0s"), v0.mK0Short(), v0.pt(), v0.eta(), vz, centrality);
}
}

if (TMath::Abs(v0.eta()) > 0.8)
continue;

if (LambdaTag) {
Lambda = Proton + AntiPion;
histos.fill(HIST("hSparseRecLambda"), v0.mLambda(), v0.pt(), v0.eta(), vz, centrality);
}
if (aLambdaTag) {
AntiLambda = AntiProton + Pion;
histos.fill(HIST("hSparseRecAntiLambda"), v0.mAntiLambda(), v0.pt(), v0.eta(), vz, centrality);
}
if (K0sTag) {
histos.fill(HIST("hSparseRecK0s"), v0.mK0Short(), v0.pt(), v0.eta(), vz, centrality);
}
}

for (const auto& mcParticle : GenParticles) {
if (analyzeLambda && std::abs(mcParticle.pdgCode()) != PDG_t::kLambda0) {
continue;
}
if (analyzeK0s && std::abs(mcParticle.pdgCode()) != PDG_t::kK0Short) {
continue;
}
if (std::abs(mcParticle.y()) > ConfV0Rap) {
continue;
}
auto pdg1 = mcParticle.pdgCode();
auto kDaughters = mcParticle.daughters_as<aod::McParticles>();
int daughsize = 2;
if (kDaughters.size() != daughsize) {
continue;
}
for (const auto& kCurrentDaughter : kDaughters) {
for (const auto& mcParticle : GenParticles) {

if (std::abs(kCurrentDaughter.pdgCode()) != PDG_t::kProton && std::abs(kCurrentDaughter.pdgCode()) != PDG_t::kPiPlus) {
if (analyzeLambda && std::abs(mcParticle.pdgCode()) != PDG_t::kLambda0) {
continue;
}
if (analyzeK0s && std::abs(mcParticle.pdgCode()) != PDG_t::kK0Short) {
continue;
}
if (std::abs(mcParticle.y()) > ConfV0Rap) {
continue;
}
if (kCurrentDaughter.pdgCode() == PDG_t::kProton) {
protonmc = ROOT::Math::PxPyPzMVector(kCurrentDaughter.px(), kCurrentDaughter.py(), kCurrentDaughter.pz(), o2::constants::physics::MassProton);
if (!mcParticle.isPhysicalPrimary() || !mcParticle.producedByGenerator()) {
continue;
}
if (kCurrentDaughter.pdgCode() == PDG_t::kPiMinus) {
antiPionmc = ROOT::Math::PxPyPzMVector(kCurrentDaughter.px(), kCurrentDaughter.py(), kCurrentDaughter.pz(), o2::constants::physics::MassPionCharged);

auto pdg1 = mcParticle.pdgCode();
auto kDaughters = mcParticle.daughters_as<aod::McParticles>();
int daughsize = 2;
if (kDaughters.size() != daughsize) {
continue;
}

if (kCurrentDaughter.pdgCode() == PDG_t::kProtonBar) {
antiProtonmc = ROOT::Math::PxPyPzMVector(kCurrentDaughter.px(), kCurrentDaughter.py(), kCurrentDaughter.pz(), o2::constants::physics::MassProton);
int lambdacounter = 0;
int antilambdacounter = 0;
int k0scounter = 0;

for (const auto& kCurrentDaughter : kDaughters) {
if (std::abs(kCurrentDaughter.pdgCode()) != PDG_t::kProton && std::abs(kCurrentDaughter.pdgCode()) != PDG_t::kPiPlus) {
continue;
}
if (kCurrentDaughter.pt() < 0.2 || TMath::Abs(kCurrentDaughter.eta()) > 0.8)

Check failure on line 440 in PWGLF/Tasks/Strangeness/lambdak0seff.cxx

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.
continue;

if (kCurrentDaughter.pdgCode() == PDG_t::kProton) {
lambdacounter += 1;
protonmc = ROOT::Math::PxPyPzMVector(kCurrentDaughter.px(), kCurrentDaughter.py(), kCurrentDaughter.pz(), o2::constants::physics::MassProton);
}
if (kCurrentDaughter.pdgCode() == PDG_t::kPiMinus) {
lambdacounter += 1;
k0scounter += 1;
antiPionmc = ROOT::Math::PxPyPzMVector(kCurrentDaughter.px(), kCurrentDaughter.py(), kCurrentDaughter.pz(), o2::constants::physics::MassPionCharged);
}

if (kCurrentDaughter.pdgCode() == PDG_t::kProtonBar) {
antilambdacounter += 1;
antiProtonmc = ROOT::Math::PxPyPzMVector(kCurrentDaughter.px(), kCurrentDaughter.py(), kCurrentDaughter.pz(), o2::constants::physics::MassProton);
}
if (kCurrentDaughter.pdgCode() == PDG_t::kPiPlus) {
antilambdacounter += 1;
k0scounter += 1;
pionmc = ROOT::Math::PxPyPzMVector(kCurrentDaughter.px(), kCurrentDaughter.py(), kCurrentDaughter.pz(), o2::constants::physics::MassPionCharged);
}
}
if (kCurrentDaughter.pdgCode() == PDG_t::kPiPlus) {
pionmc = ROOT::Math::PxPyPzMVector(kCurrentDaughter.px(), kCurrentDaughter.py(), kCurrentDaughter.pz(), o2::constants::physics::MassPionCharged);

if (analyzeLambda && pdg1 == PDG_t::kLambda0 && lambdacounter == 2 && antilambdacounter != 2 && k0scounter != 2) {

Check failure on line 464 in PWGLF/Tasks/Strangeness/lambdak0seff.cxx

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 464 in PWGLF/Tasks/Strangeness/lambdak0seff.cxx

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.
lambdadummymc = protonmc + antiPionmc;
histos.fill(HIST("hSparseGenLambda"), lambdadummymc.M(), lambdadummymc.Pt(), lambdadummymc.Eta(), vz, centrality);
}
}
if (pdg1 == PDG_t::kLambda0) {
lambdadummymc = protonmc + antiPionmc;
histos.fill(HIST("hSparseGenLambda"), lambdadummymc.M(), lambdadummymc.Pt(), lambdadummymc.Eta(), vz, centrality);
}

if (pdg1 == PDG_t::kLambda0Bar) {
antiLambdadummymc = antiProtonmc + pionmc;
histos.fill(HIST("hSparseGenAntiLambda"), antiLambdadummymc.M(), antiLambdadummymc.Pt(), lambdadummymc.Eta(), vz, centrality);
}
if (pdg1 == PDG_t::kK0Short) {
kshortdummymc = antiPionmc + pionmc;
histos.fill(HIST("hSparseGenK0s"), kshortdummymc.M(), kshortdummymc.Pt(), kshortdummymc.Eta(), vz, centrality);
if (analyzeLambda && pdg1 == PDG_t::kLambda0Bar && antilambdacounter == 2 && lambdacounter != 2 && k0scounter != 2) {

Check failure on line 469 in PWGLF/Tasks/Strangeness/lambdak0seff.cxx

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.
antiLambdadummymc = antiProtonmc + pionmc;
histos.fill(HIST("hSparseGenAntiLambda"), antiLambdadummymc.M(), antiLambdadummymc.Pt(), antiLambdadummymc.Eta(), vz, centrality);
}
if (analyzeK0s && pdg1 == PDG_t::kK0Short && k0scounter == 2 && lambdacounter != 2 && antilambdacounter != 2) {

Check failure on line 473 in PWGLF/Tasks/Strangeness/lambdak0seff.cxx

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.
kshortdummymc = antiPionmc + pionmc;
histos.fill(HIST("hSparseGenK0s"), kshortdummymc.M(), kshortdummymc.Pt(), kshortdummymc.Eta(), vz, centrality);
}
}
}
}
Expand Down
Loading