Skip to content
Merged
Changes from all commits
Commits
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
47 changes: 22 additions & 25 deletions PWGLF/Tasks/Strangeness/strangeness_in_jets.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -138,6 +138,9 @@ struct strangeness_in_jets {
registryMC.add("K0s_reconstructed", "K0s_reconstructed", HistType::kTH2F, {multBinning, {100, 0.0, 10.0, "#it{p}_{T} (GeV/#it{c})"}});
registryMC.add("Lambda_reconstructed", "Lambda_reconstructed", HistType::kTH2F, {multBinning, {100, 0.0, 10.0, "#it{p}_{T} (GeV/#it{c})"}});
registryMC.add("AntiLambda_reconstructed", "AntiLambda_reconstructed", HistType::kTH2F, {multBinning, {100, 0.0, 10.0, "#it{p}_{T} (GeV/#it{c})"}});
registryMC.add("K0s_reconstructed_incl", "K0s_reconstructed_incl", HistType::kTH2F, {multBinning, {100, 0.0, 10.0, "#it{p}_{T} (GeV/#it{c})"}});
registryMC.add("Lambda_reconstructed_incl", "Lambda_reconstructed_incl", HistType::kTH2F, {multBinning, {100, 0.0, 10.0, "#it{p}_{T} (GeV/#it{c})"}});
registryMC.add("AntiLambda_reconstructed_incl", "AntiLambda_reconstructed_incl", HistType::kTH2F, {multBinning, {100, 0.0, 10.0, "#it{p}_{T} (GeV/#it{c})"}});

// Histograms for reweighting
registryMC.add("K0s_eta_pt_jet", "K0s_eta_pt_jet", HistType::kTH2F, {{100, 0.0, 10.0, "#it{p}_{T} (GeV/#it{c})"}, {18, -0.9, 0.9, "#eta"}});
Expand Down Expand Up @@ -468,11 +471,6 @@ struct strangeness_in_jets {

for (auto track : tracks) {

if (!track.passedITSRefit())
continue;
if (!track.passedTPCRefit())
continue;

int i = track.globalIndex();
if (!passedTrackSelectionForJets(track))
continue;
Expand Down Expand Up @@ -587,10 +585,6 @@ struct strangeness_in_jets {

const auto& pos = v0.posTrack_as<FullTracks>();
const auto& neg = v0.negTrack_as<FullTracks>();
if (!pos.passedTPCRefit())
continue;
if (!neg.passedTPCRefit())
continue;

TVector3 v0dir(pos.px() + neg.px(), pos.py() + neg.py(), pos.pz() + neg.pz());

Expand Down Expand Up @@ -665,10 +659,6 @@ struct strangeness_in_jets {

const auto& pos = v0.posTrack_as<MCTracks>();
const auto& neg = v0.negTrack_as<MCTracks>();
if (!pos.passedTPCRefit())
continue;
if (!neg.passedTPCRefit())
continue;
if (!pos.has_mcParticle())
continue;
if (!neg.has_mcParticle())
Expand All @@ -693,6 +683,17 @@ struct strangeness_in_jets {
}
if (pdg_parent == 0)
continue;

// K0s
if (passedK0ShortSelection(v0, pos, neg, collision) && pdg_parent == 310) {
registryMC.fill(HIST("K0s_reconstructed_incl"), multiplicity, v0.pt());
}
if (passedLambdaSelection(v0, pos, neg, collision) && pdg_parent == 3122) {
registryMC.fill(HIST("Lambda_reconstructed_incl"), multiplicity, v0.pt());
}
if (passedAntiLambdaSelection(v0, pos, neg, collision) && pdg_parent == -3122) {
registryMC.fill(HIST("AntiLambda_reconstructed_incl"), multiplicity, v0.pt());
}
if (!isPhysPrim)
continue;

Expand All @@ -718,14 +719,17 @@ struct strangeness_in_jets {
// K0s
if (mcParticle.pdgCode() == 310) {
registryMC.fill(HIST("K0s_Generated"), multiplicity, mcParticle.pt());
registryMC.fill(HIST("K0s_eta_pt_pythia"), mcParticle.pt(), mcParticle.eta());
}
// Lambda
if (mcParticle.pdgCode() == 3122) {
registryMC.fill(HIST("Lambda_Generated"), multiplicity, mcParticle.pt());
registryMC.fill(HIST("Lambda_eta_pt_pythia"), mcParticle.pt(), mcParticle.eta());
}
// AntiLambda
if (mcParticle.pdgCode() == -3122) {
registryMC.fill(HIST("AntiLambda_Generated"), multiplicity, mcParticle.pt());
registryMC.fill(HIST("AntiLambda_eta_pt_pythia"), mcParticle.pt(), mcParticle.eta());
}
}
}
Expand Down Expand Up @@ -753,18 +757,6 @@ struct strangeness_in_jets {
// Global Index
int i = particle.globalIndex();

if (particle.pdgCode() == 310) {
registryMC.fill(HIST("K0s_eta_pt_pythia"), particle.pt(), particle.eta());
}

if (particle.pdgCode() == 3122) {
registryMC.fill(HIST("Lambda_eta_pt_pythia"), particle.pt(), particle.eta());
}

if (particle.pdgCode() == -3122) {
registryMC.fill(HIST("AntiLambda_eta_pt_pythia"), particle.pt(), particle.eta());
}

// Select Primary Particles
float deltaX = particle.vx() - collision.posX();
float deltaY = particle.vy() - collision.posY();
Expand Down Expand Up @@ -896,6 +888,11 @@ struct strangeness_in_jets {
// PDG Selection
int pdg = particle.pdgCode();

if (!particle.isPhysicalPrimary())
continue;
if (particle.y() < yMin || particle.y() > yMax)
continue;

TVector3 p_particle(particle.px(), particle.py(), particle.pz());
float deltaEta_jet = p_particle.Eta() - jet_axis.Eta();
float deltaPhi_jet = GetDeltaPhi(p_particle.Phi(), jet_axis.Phi());
Expand Down