Skip to content

Commit 415d2ab

Browse files
njacazionjacazio
authored
[PWGLF] StrInJet: Fix pion histogram (#9499)
1 parent 7c604bb commit 415d2ab

File tree

1 file changed

+92
-78
lines changed

1 file changed

+92
-78
lines changed

‎PWGLF/Tasks/Strangeness/strangenessInJets.cxx‎

Lines changed: 92 additions & 78 deletions
Original file line numberDiff line numberDiff line change
@@ -269,9 +269,13 @@ struct StrangenessInJets {
269269
registryMC.add("AntiLambda_reconstructed_incl", "AntiLambda_reconstructed_incl", HistType::kTH2F, {multBinning, {100, 0.0, 10.0, "#it{p}_{T} (GeV/#it{c})"}});
270270

271271
// Histograms for 2d reweighting (pion)
272-
registryMC.add("Pion_eta_pt_jet", "Pion_eta_pt_jet", HistType::kTH2F, {ptAxis, etaAxis});
273-
registryMC.add("Pion_eta_pt_ue", "Pion_eta_pt_ue", HistType::kTH2F, {ptAxis, etaAxis});
274-
registryMC.add("Pion_eta_pt_pythia", "Pion_eta_pt_pythia", HistType::kTH2F, {ptAxis, etaAxis});
272+
registryMC.add("pi_plus_eta_pt_jet", "pi_plus_eta_pt_jet", HistType::kTH2F, {ptAxis, etaAxis});
273+
registryMC.add("pi_plus_eta_pt_ue", "pi_plus_eta_pt_ue", HistType::kTH2F, {ptAxis, etaAxis});
274+
registryMC.add("pi_plus_eta_pt_pythia", "pi_plus_eta_pt_pythia", HistType::kTH2F, {ptAxis, etaAxis});
275+
276+
registryMC.add("pi_minus_eta_pt_jet", "pi_minus_eta_pt_jet", HistType::kTH2F, {ptAxis, etaAxis});
277+
registryMC.add("pi_minus_eta_pt_ue", "pi_minus_eta_pt_ue", HistType::kTH2F, {ptAxis, etaAxis});
278+
registryMC.add("pi_minus_eta_pt_pythia", "pi_minus_eta_pt_pythia", HistType::kTH2F, {ptAxis, etaAxis});
275279

276280
// Histograms for 2d reweighting (K0s)
277281
registryMC.add("K0s_eta_pt_jet", "K0s_eta_pt_jet", HistType::kTH2F, {{100, 0.0, 10.0, "#it{p}_{T} (GeV/#it{c})"}, etaAxis});
@@ -1708,10 +1712,12 @@ struct StrangenessInJets {
17081712
case 211: // Pi+
17091713
registryMC.fill(HIST("pi_plus_gen_in_jet"), multiplicity, mcParticle.pt(), wPiplusJet);
17101714
registryMC.fill(HIST("pi_plus_gen_in_ue"), multiplicity, mcParticle.pt(), wPiplusUe);
1715+
registryMC.fill(HIST("pi_plus_eta_pt_pythia"), mcParticle.pt(), mcParticle.eta());
17111716
break;
17121717
case -211: // Pi-
17131718
registryMC.fill(HIST("pi_minus_gen_in_jet"), multiplicity, mcParticle.pt(), wPiminusJet);
17141719
registryMC.fill(HIST("pi_minus_gen_in_ue"), multiplicity, mcParticle.pt(), wPiminusUe);
1720+
registryMC.fill(HIST("pi_minus_eta_pt_pythia"), mcParticle.pt(), mcParticle.eta());
17151721
break;
17161722
case 310: // K0s
17171723
registryMC.fill(HIST("K0s_generated_jet"), multiplicity, mcParticle.pt(), wK0jet);
@@ -1928,81 +1934,89 @@ struct StrangenessInJets {
19281934
continue;
19291935

19301936
TVector3 particleDir(particle.px(), particle.py(), particle.pz());
1931-
double deltaEtaJet = particleDir.Eta() - jet[i].Eta();
1932-
double deltaPhiJet = getDeltaPhi(particleDir.Phi(), jet[i].Phi());
1933-
double deltaRjet = std::sqrt(deltaEtaJet * deltaEtaJet + deltaPhiJet * deltaPhiJet);
1934-
double deltaEtaUe1 = particleDir.Eta() - ue1[i].Eta();
1935-
double deltaPhiUe1 = getDeltaPhi(particleDir.Phi(), ue1[i].Phi());
1936-
double deltaRue1 = std::sqrt(deltaEtaUe1 * deltaEtaUe1 + deltaPhiUe1 * deltaPhiUe1);
1937-
double deltaEtaUe2 = particleDir.Eta() - ue2[i].Eta();
1938-
double deltaPhiUe2 = getDeltaPhi(particleDir.Phi(), ue2[i].Phi());
1939-
double deltaRue2 = std::sqrt(deltaEtaUe2 * deltaEtaUe2 + deltaPhiUe2 * deltaPhiUe2);
1940-
1941-
int pdg = particle.pdgCode();
1942-
1943-
if (pdg == 211) {
1944-
if (deltaRjet < rJet) {
1945-
registryMC.fill(HIST("Pion_eta_pt_jet"), particle.pt(), particle.eta());
1946-
}
1947-
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1948-
registryMC.fill(HIST("Pion_eta_pt_ue"), particle.pt(), particle.eta());
1949-
}
1950-
}
1951-
if (pdg == 310) {
1952-
if (deltaRjet < rJet) {
1953-
registryMC.fill(HIST("K0s_eta_pt_jet"), particle.pt(), particle.eta());
1954-
}
1955-
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1956-
registryMC.fill(HIST("K0s_eta_pt_ue"), particle.pt(), particle.eta());
1957-
}
1958-
}
1959-
if (pdg == 3122) {
1960-
if (deltaRjet < rJet) {
1961-
registryMC.fill(HIST("Lambda_eta_pt_jet"), particle.pt(), particle.eta());
1962-
}
1963-
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1964-
registryMC.fill(HIST("Lambda_eta_pt_ue"), particle.pt(), particle.eta());
1965-
}
1966-
}
1967-
if (pdg == -3122) {
1968-
if (deltaRjet < rJet) {
1969-
registryMC.fill(HIST("AntiLambda_eta_pt_jet"), particle.pt(), particle.eta());
1970-
}
1971-
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1972-
registryMC.fill(HIST("AntiLambda_eta_pt_ue"), particle.pt(), particle.eta());
1973-
}
1974-
}
1975-
if (pdg == 3312) {
1976-
if (deltaRjet < rJet) {
1977-
registryMC.fill(HIST("Xi_eta_pt_jet"), particle.pt(), particle.eta());
1978-
}
1979-
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1980-
registryMC.fill(HIST("Xi_eta_pt_ue"), particle.pt(), particle.eta());
1981-
}
1982-
}
1983-
if (pdg == -3312) {
1984-
if (deltaRjet < rJet) {
1985-
registryMC.fill(HIST("AntiXi_eta_pt_jet"), particle.pt(), particle.eta());
1986-
}
1987-
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1988-
registryMC.fill(HIST("AntiXi_eta_pt_ue"), particle.pt(), particle.eta());
1989-
}
1990-
}
1991-
if (pdg == 3334) {
1992-
if (deltaRjet < rJet) {
1993-
registryMC.fill(HIST("Omega_eta_pt_jet"), particle.pt(), particle.eta());
1994-
}
1995-
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1996-
registryMC.fill(HIST("Omega_eta_pt_ue"), particle.pt(), particle.eta());
1997-
}
1998-
}
1999-
if (pdg == -3334) {
2000-
if (deltaRjet < rJet) {
2001-
registryMC.fill(HIST("AntiOmega_eta_pt_jet"), particle.pt(), particle.eta());
2002-
}
2003-
if (deltaRue1 < rJet || deltaRue2 < rJet) {
2004-
registryMC.fill(HIST("AntiOmega_eta_pt_ue"), particle.pt(), particle.eta());
2005-
}
1937+
const double deltaEtaJet = particleDir.Eta() - jet[i].Eta();
1938+
const double deltaPhiJet = getDeltaPhi(particleDir.Phi(), jet[i].Phi());
1939+
const double deltaRjet = std::sqrt(deltaEtaJet * deltaEtaJet + deltaPhiJet * deltaPhiJet);
1940+
const double deltaEtaUe1 = particleDir.Eta() - ue1[i].Eta();
1941+
const double deltaPhiUe1 = getDeltaPhi(particleDir.Phi(), ue1[i].Phi());
1942+
const double deltaRue1 = std::sqrt(deltaEtaUe1 * deltaEtaUe1 + deltaPhiUe1 * deltaPhiUe1);
1943+
const double deltaEtaUe2 = particleDir.Eta() - ue2[i].Eta();
1944+
const double deltaPhiUe2 = getDeltaPhi(particleDir.Phi(), ue2[i].Phi());
1945+
const double deltaRue2 = std::sqrt(deltaEtaUe2 * deltaEtaUe2 + deltaPhiUe2 * deltaPhiUe2);
1946+
1947+
switch (particle.pdgCode()) {
1948+
case 211:
1949+
if (deltaRjet < rJet) {
1950+
registryMC.fill(HIST("pi_plus_eta_pt_jet"), particle.pt(), particle.eta());
1951+
}
1952+
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1953+
registryMC.fill(HIST("pi_plus_eta_pt_ue"), particle.pt(), particle.eta());
1954+
}
1955+
break;
1956+
case -211:
1957+
if (deltaRjet < rJet) {
1958+
registryMC.fill(HIST("pi_minus_eta_pt_jet"), particle.pt(), particle.eta());
1959+
}
1960+
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1961+
registryMC.fill(HIST("pi_minus_eta_pt_ue"), particle.pt(), particle.eta());
1962+
}
1963+
break;
1964+
case 310:
1965+
if (deltaRjet < rJet) {
1966+
registryMC.fill(HIST("K0s_eta_pt_jet"), particle.pt(), particle.eta());
1967+
}
1968+
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1969+
registryMC.fill(HIST("K0s_eta_pt_ue"), particle.pt(), particle.eta());
1970+
}
1971+
break;
1972+
case 3122:
1973+
if (deltaRjet < rJet) {
1974+
registryMC.fill(HIST("Lambda_eta_pt_jet"), particle.pt(), particle.eta());
1975+
}
1976+
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1977+
registryMC.fill(HIST("Lambda_eta_pt_ue"), particle.pt(), particle.eta());
1978+
}
1979+
break;
1980+
case -3122:
1981+
if (deltaRjet < rJet) {
1982+
registryMC.fill(HIST("AntiLambda_eta_pt_jet"), particle.pt(), particle.eta());
1983+
}
1984+
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1985+
registryMC.fill(HIST("AntiLambda_eta_pt_ue"), particle.pt(), particle.eta());
1986+
}
1987+
break;
1988+
case 3312:
1989+
if (deltaRjet < rJet) {
1990+
registryMC.fill(HIST("Xi_eta_pt_jet"), particle.pt(), particle.eta());
1991+
}
1992+
if (deltaRue1 < rJet || deltaRue2 < rJet) {
1993+
registryMC.fill(HIST("Xi_eta_pt_ue"), particle.pt(), particle.eta());
1994+
}
1995+
break;
1996+
case -3312:
1997+
if (deltaRjet < rJet) {
1998+
registryMC.fill(HIST("AntiXi_eta_pt_jet"), particle.pt(), particle.eta());
1999+
}
2000+
if (deltaRue1 < rJet || deltaRue2 < rJet) {
2001+
registryMC.fill(HIST("AntiXi_eta_pt_ue"), particle.pt(), particle.eta());
2002+
}
2003+
break;
2004+
case 3334:
2005+
if (deltaRjet < rJet) {
2006+
registryMC.fill(HIST("Omega_eta_pt_jet"), particle.pt(), particle.eta());
2007+
}
2008+
if (deltaRue1 < rJet || deltaRue2 < rJet) {
2009+
registryMC.fill(HIST("Omega_eta_pt_ue"), particle.pt(), particle.eta());
2010+
}
2011+
break;
2012+
case -3334:
2013+
if (deltaRjet < rJet) {
2014+
registryMC.fill(HIST("AntiOmega_eta_pt_jet"), particle.pt(), particle.eta());
2015+
}
2016+
if (deltaRue1 < rJet || deltaRue2 < rJet) {
2017+
registryMC.fill(HIST("AntiOmega_eta_pt_ue"), particle.pt(), particle.eta());
2018+
}
2019+
break;
20062020
}
20072021
}
20082022
}

0 commit comments

Comments
 (0)