Skip to content

LF: spectra add info for non recoed events #2755

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
njacazio merged 2 commits into from
Jun 5, 2023
Merged

LF: spectra add info for non recoed events #2755

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
5f434d1
LF: spectra add info for non recoed events
njacazio Jun 5, 2023
079da87
Fix
njacazio Jun 5, 2023
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
195 changes: 190 additions & 5 deletions PWGLF/Tasks/spectraTOF.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -425,6 +425,11 @@ struct tofSpectra {
histos.add(hpt_den_prm[i].data(), pTCharge[i], kTH1D, {ptAxis});
histos.add(hpt_den_str[i].data(), pTCharge[i], kTH1D, {ptAxis});
histos.add(hpt_den_mat[i].data(), pTCharge[i], kTH1D, {ptAxis});
histos.add(hpt_den_prm_recoev[i].data(), pTCharge[i], kTH1D, {ptAxis});
histos.add(hpt_den_prm_evsel[i].data(), pTCharge[i], kTH1D, {ptAxis});
histos.add(hpt_den_prm_goodev[i].data(), pTCharge[i], kTH1D, {ptAxis});
histos.add(hpt_den_prm_mcgoodev[i].data(), pTCharge[i], kTH1D, {ptAxis});
histos.add(hpt_den_prm_mcbadev[i].data(), pTCharge[i], kTH1D, {ptAxis});

histos.add(hdcaxyprm[i].data(), pTCharge[i], kTH2D, {ptAxis, dcaXyAxis});
histos.add(hdcazprm[i].data(), pTCharge[i], kTH2D, {ptAxis, dcaZAxis});
Expand Down Expand Up @@ -1135,6 +1140,160 @@ struct tofSpectra {
}
}

template <std::size_t i, typename ParticleType>
void fillParticleHistograms_MCRecoEvs(ParticleType const& mcParticle, CollisionCandidateMC::iterator const& collision)
{

switch (i) {
case 0:
case Np:
if (doprocessFullEl == false && doprocessLfFullEl == false) {
return;
}
break;
case 1:
case Np + 1:
if (doprocessFullMu == false && doprocessLfFullMu == false) {
return;
}
break;
case 2:
case Np + 2:
if (doprocessFullPi == false && doprocessLfFullPi == false) {
return;
}
break;
case 3:
case Np + 3:
if (doprocessFullKa == false && doprocessLfFullKa == false) {
return;
}
break;
case 4:
case Np + 4:
if (doprocessFullPr == false && doprocessLfFullPr == false) {
return;
}
break;
case 5:
case Np + 5:
if (doprocessFullDe == false && doprocessLfFullDe == false) {
return;
}
break;
case 6:
case Np + 6:
if (doprocessFullTr == false && doprocessLfFullTr == false) {
return;
}
break;
case 7:
case Np + 7:
if (doprocessFullHe == false && doprocessLfFullHe == false) {
return;
}
break;
case 8:
case Np + 8:
if (doprocessFullAl == false && doprocessLfFullAl == false) {
return;
}
break;
}

if (mcParticle.pdgCode() != PDGs[i]) {
return;
}

if (mcParticle.isPhysicalPrimary()) {
if (collision.sel8()) {
if (abs(collision.posZ()) < cfgCutVertex) {
histos.fill(HIST(hpt_den_prm_goodev[i]), mcParticle.pt());
} else {
histos.fill(HIST(hpt_den_prm_evsel[i]), mcParticle.pt());
}
} else {
histos.fill(HIST(hpt_den_prm_recoev[i]), mcParticle.pt());
}
}
}

template <std::size_t i, typename ParticleType>
void fillParticleHistograms_MCGenEvs(ParticleType const& mcParticle, aod::McCollision const& mcCollision)
{

switch (i) {
case 0:
case Np:
if (doprocessFullEl == false && doprocessLfFullEl == false) {
return;
}
break;
case 1:
case Np + 1:
if (doprocessFullMu == false && doprocessLfFullMu == false) {
return;
}
break;
case 2:
case Np + 2:
if (doprocessFullPi == false && doprocessLfFullPi == false) {
return;
}
break;
case 3:
case Np + 3:
if (doprocessFullKa == false && doprocessLfFullKa == false) {
return;
}
break;
case 4:
case Np + 4:
if (doprocessFullPr == false && doprocessLfFullPr == false) {
return;
}
break;
case 5:
case Np + 5:
if (doprocessFullDe == false && doprocessLfFullDe == false) {
return;
}
break;
case 6:
case Np + 6:
if (doprocessFullTr == false && doprocessLfFullTr == false) {
return;
}
break;
case 7:
case Np + 7:
if (doprocessFullHe == false && doprocessLfFullHe == false) {
return;
}
break;
case 8:
case Np + 8:
if (doprocessFullAl == false && doprocessLfFullAl == false) {
return;
}
break;
}

if (mcParticle.pdgCode() != PDGs[i]) {
return;
}

if (mcParticle.isPhysicalPrimary()) {
if (abs(mcCollision.posZ()) < cfgCutVertex) {
histos.fill(HIST(hpt_den_prm_mcgoodev[i]), mcParticle.pt());
} else {
histos.fill(HIST(hpt_den_prm_mcbadev[i]), mcParticle.pt());
}
}
}

Preslice<aod::McParticles> perMCCol = aod::mcparticle::mcCollisionId;
SliceCache cache;
void processMC(soa::Join<aod::Tracks, aod::TracksExtra,
aod::TracksDCA, aod::McTrackLabels,
aod::pidTOFFullPi, aod::pidTOFFullKa, aod::pidTOFFullPr,
Expand All @@ -1147,7 +1306,7 @@ struct tofSpectra {
histos.fill(HIST("MC/GenRecoCollisions"), 0.5, mcCollisions.size());
histos.fill(HIST("MC/GenRecoCollisions"), 1.5, collisions.size());
// LOGF(info, "Enter processMC!");
for (auto& track : tracks) {
for (const auto& track : tracks) {
if (!track.has_collision()) {
if (track.sign() > 0) {
histos.fill(HIST("MC/no_collision/pos"), track.pt());
Expand All @@ -1174,20 +1333,46 @@ struct tofSpectra {
});
}

for (auto& mcParticle : mcParticles) {
for (const auto& mcParticle : mcParticles) {
// if (std::abs(mcParticle.eta()) > cfgCutEta) {
// continue;
// }
if (std::abs(mcParticle.y()) > cfgCutY) {
continue;
}
if (!mcParticle.isPhysicalPrimary()) {
continue;
}
static_for<0, 17>([&](auto i) {
fillParticleHistograms_MC<i>(mcParticle);
});
}

// Loop on reconstructed collisions
for (const auto& collision : collisions) {
if (!collision.has_mcCollision()) {
continue;
}
const auto& particlesInCollision = mcParticles.sliceByCached(aod::mcparticle::mcCollisionId, collision.mcCollision().globalIndex(), cache);
for (const auto& mcParticle : particlesInCollision) {
if (std::abs(mcParticle.y()) > cfgCutY) {
continue;
}
static_for<0, 17>([&](auto i) {
fillParticleHistograms_MCRecoEvs<i>(mcParticle, collision);
});
}
}

// Loop on generated collisions
for (const auto& mcCollision : mcCollisions) {
const auto& particlesInCollision = mcParticles.sliceByCached(aod::mcparticle::mcCollisionId, mcCollision.globalIndex(), cache);
for (const auto& mcParticle : particlesInCollision) {
if (std::abs(mcParticle.y()) > cfgCutY) {
continue;
}
static_for<0, 17>([&](auto i) {
fillParticleHistograms_MCGenEvs<i>(mcParticle, mcCollision);
});
}
}
}
PROCESS_SWITCH(tofSpectra, processMC, "Process MC", false);

Expand Down
30 changes: 30 additions & 0 deletions PWGLF/Tasks/spectraTOF.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -116,6 +116,36 @@ static constexpr std::string_view hpt_den_prm[NpCharge] = {"MC/el/pos/prm/pt/den
"MC/el/neg/prm/pt/den", "MC/mu/neg/prm/pt/den", "MC/pi/neg/prm/pt/den",
"MC/ka/neg/prm/pt/den", "MC/pr/neg/prm/pt/den", "MC/de/neg/prm/pt/den",
"MC/tr/neg/prm/pt/den", "MC/he/neg/prm/pt/den", "MC/al/neg/prm/pt/den"};
static constexpr std::string_view hpt_den_prm_recoev[NpCharge] = {"MC/el/pos/prm/pt/denrecoev", "MC/mu/pos/prm/pt/denrecoev", "MC/pi/pos/prm/pt/denrecoev",
"MC/ka/pos/prm/pt/denrecoev", "MC/pr/pos/prm/pt/denrecoev", "MC/de/pos/prm/pt/denrecoev",
"MC/tr/pos/prm/pt/denrecoev", "MC/he/pos/prm/pt/denrecoev", "MC/al/pos/prm/pt/denrecoev",
"MC/el/neg/prm/pt/denrecoev", "MC/mu/neg/prm/pt/denrecoev", "MC/pi/neg/prm/pt/denrecoev",
"MC/ka/neg/prm/pt/denrecoev", "MC/pr/neg/prm/pt/denrecoev", "MC/de/neg/prm/pt/denrecoev",
"MC/tr/neg/prm/pt/denrecoev", "MC/he/neg/prm/pt/denrecoev", "MC/al/neg/prm/pt/denrecoev"};
static constexpr std::string_view hpt_den_prm_evsel[NpCharge] = {"MC/el/pos/prm/pt/denevsel", "MC/mu/pos/prm/pt/denevsel", "MC/pi/pos/prm/pt/denevsel",
"MC/ka/pos/prm/pt/denevsel", "MC/pr/pos/prm/pt/denevsel", "MC/de/pos/prm/pt/denevsel",
"MC/tr/pos/prm/pt/denevsel", "MC/he/pos/prm/pt/denevsel", "MC/al/pos/prm/pt/denevsel",
"MC/el/neg/prm/pt/denevsel", "MC/mu/neg/prm/pt/denevsel", "MC/pi/neg/prm/pt/denevsel",
"MC/ka/neg/prm/pt/denevsel", "MC/pr/neg/prm/pt/denevsel", "MC/de/neg/prm/pt/denevsel",
"MC/tr/neg/prm/pt/denevsel", "MC/he/neg/prm/pt/denevsel", "MC/al/neg/prm/pt/denevsel"};
static constexpr std::string_view hpt_den_prm_goodev[NpCharge] = {"MC/el/pos/prm/pt/dengoodev", "MC/mu/pos/prm/pt/dengoodev", "MC/pi/pos/prm/pt/dengoodev",
"MC/ka/pos/prm/pt/dengoodev", "MC/pr/pos/prm/pt/dengoodev", "MC/de/pos/prm/pt/dengoodev",
"MC/tr/pos/prm/pt/dengoodev", "MC/he/pos/prm/pt/dengoodev", "MC/al/pos/prm/pt/dengoodev",
"MC/el/neg/prm/pt/dengoodev", "MC/mu/neg/prm/pt/dengoodev", "MC/pi/neg/prm/pt/dengoodev",
"MC/ka/neg/prm/pt/dengoodev", "MC/pr/neg/prm/pt/dengoodev", "MC/de/neg/prm/pt/dengoodev",
"MC/tr/neg/prm/pt/dengoodev", "MC/he/neg/prm/pt/dengoodev", "MC/al/neg/prm/pt/dengoodev"};
static constexpr std::string_view hpt_den_prm_mcgoodev[NpCharge] = {"MC/el/pos/prm/pt/denmcgoodev", "MC/mu/pos/prm/pt/denmcgoodev", "MC/pi/pos/prm/pt/denmcgoodev",
"MC/ka/pos/prm/pt/denmcgoodev", "MC/pr/pos/prm/pt/denmcgoodev", "MC/de/pos/prm/pt/denmcgoodev",
"MC/tr/pos/prm/pt/denmcgoodev", "MC/he/pos/prm/pt/denmcgoodev", "MC/al/pos/prm/pt/denmcgoodev",
"MC/el/neg/prm/pt/denmcgoodev", "MC/mu/neg/prm/pt/denmcgoodev", "MC/pi/neg/prm/pt/denmcgoodev",
"MC/ka/neg/prm/pt/denmcgoodev", "MC/pr/neg/prm/pt/denmcgoodev", "MC/de/neg/prm/pt/denmcgoodev",
"MC/tr/neg/prm/pt/denmcgoodev", "MC/he/neg/prm/pt/denmcgoodev", "MC/al/neg/prm/pt/denmcgoodev"};
static constexpr std::string_view hpt_den_prm_mcbadev[NpCharge] = {"MC/el/pos/prm/pt/denmcbadev", "MC/mu/pos/prm/pt/denmcbadev", "MC/pi/pos/prm/pt/denmcbadev",
"MC/ka/pos/prm/pt/denmcbadev", "MC/pr/pos/prm/pt/denmcbadev", "MC/de/pos/prm/pt/denmcbadev",
"MC/tr/pos/prm/pt/denmcbadev", "MC/he/pos/prm/pt/denmcbadev", "MC/al/pos/prm/pt/denmcbadev",
"MC/el/neg/prm/pt/denmcbadev", "MC/mu/neg/prm/pt/denmcbadev", "MC/pi/neg/prm/pt/denmcbadev",
"MC/ka/neg/prm/pt/denmcbadev", "MC/pr/neg/prm/pt/denmcbadev", "MC/de/neg/prm/pt/denmcbadev",
"MC/tr/neg/prm/pt/denmcbadev", "MC/he/neg/prm/pt/denmcbadev", "MC/al/neg/prm/pt/denmcbadev"};
static constexpr std::string_view hpt_num_str[NpCharge] = {"MC/el/pos/str/pt/num", "MC/mu/pos/str/pt/num", "MC/pi/pos/str/pt/num",
"MC/ka/pos/str/pt/num", "MC/pr/pos/str/pt/num", "MC/de/pos/str/pt/num",
"MC/tr/pos/str/pt/num", "MC/he/pos/str/pt/num", "MC/al/pos/str/pt/num",
Expand Down