Skip to content

Commit f04708f

Browse files
committed
Fix histo filling for secondaries
1 parent f2b8377 commit f04708f

File tree

1 file changed

+48
-2
lines changed

1 file changed

+48
-2
lines changed

PWGMM/Mult/Tasks/dndetaMFTPbPb.cxx

Lines changed: 48 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -932,6 +932,12 @@ struct DndetaMFTPbPb {
932932
hNevt->GetXaxis()->SetBinLabel(1, "Reconstructed collisions");
933933
hNevt->GetXaxis()->SetBinLabel(2, "Generated collisions");
934934
if (doprocessSecondariesMCInlcusive) {
935+
registry.add({"Events/EvtGenRec", ";status", {HistType::kTH1F, {{3, 0.5, 3.5}}}});
936+
auto heff = registry.get<TH2>(HIST("Events/EvtGenRec"));
937+
auto* h = heff->GetXaxis();
938+
h->SetBinLabel(1, "All generated");
939+
h->SetBinLabel(2, "All reconstructed");
940+
h->SetBinLabel(3, "Selected reconstructed");
935941
registry.add({"Tracks/THnRecAmb", "; p_{T} (GeV/c); #eta; Z_{vtx} (cm)", {HistType::kTHnSparseF, {ptAxis, etaAxis, zAxis}}});
936942
registry.add({"Tracks/THnRec", "; p_{T} (GeV/c); #eta; Z_{vtx} (cm)", {HistType::kTHnSparseF, {ptAxis, etaAxis, zAxis}}});
937943
registry.add({"Tracks/THnRecNonAmb", "; p_{T} (GeV/c); #eta; Z_{vtx} (cm)", {HistType::kTHnSparseF, {ptAxis, etaAxis, zAxis}}});
@@ -944,6 +950,11 @@ struct DndetaMFTPbPb {
944950
registry.add({"Tracks/THnGenSecMatAmb", "; p_{T} (GeV/c); #eta; Z_{vtx} (cm)", {HistType::kTHnSparseF, {ptAxis, etaAxis, zAxis}}});
945951
}
946952
if (doprocessSecondariesMCCentFT0C) {
953+
registry.add({"Events/Centrality/EvtGenRec", ";status;centrality", {HistType::kTH2F, {{3, 0.5, 3.5}, centralityAxis}}});
954+
auto heff = registry.get<THnSparse>(HIST("Events/Centrality/EvtGenRec"));
955+
heff->GetAxis(0)->SetBinLabel(1, "All generated");
956+
heff->GetAxis(0)->SetBinLabel(2, "All reconstructed");
957+
heff->GetAxis(0)->SetBinLabel(3, "Selected reconstructed");
947958
registry.add({"Tracks/Centrality/THnRecAmb", "; p_{T} (GeV/c); #eta; Z_{vtx} (cm); centrality", {HistType::kTHnSparseF, {ptAxis, etaAxis, zAxis, centralityAxis}}});
948959
registry.add({"Tracks/Centrality/THnRec", "; p_{T} (GeV/c); #eta; Z_{vtx} (cm); centrality", {HistType::kTHnSparseF, {ptAxis, etaAxis, zAxis, centralityAxis}}});
949960
registry.add({"Tracks/Centrality/THnRecNonAmb", "; p_{T} (GeV/c); #eta; Z_{vtx} (cm); centrality", {HistType::kTHnSparseF, {ptAxis, etaAxis, zAxis, centralityAxis}}});
@@ -2883,11 +2894,46 @@ struct DndetaMFTPbPb {
28832894
registry.fill(HIST("Events/hNGenRecColls"), 1.f, collisions.size());
28842895
registry.fill(HIST("Events/hNGenRecColls"), 2.f, mcCollisions.size());
28852896

2897+
float cGen = -1;
2898+
if constexpr (has_reco_cent<C>) {
2899+
float crecMin = 105.f;
2900+
for (const auto& collision : collisions) {
2901+
if (isGoodEvent<false>(collision)) {
2902+
float c = getRecoCent(collision);
2903+
if (c < crecMin) {
2904+
crecMin = c;
2905+
}
2906+
}
2907+
}
2908+
if (cGen < 0)
2909+
cGen = crecMin;
2910+
}
2911+
2912+
if constexpr (has_reco_cent<C>) {
2913+
registry.fill(HIST("Events/Centrality/EvtGenRec"), 1., cGen);
2914+
} else {
2915+
registry.fill(HIST("Events/EvtGenRec"), 1.);
2916+
}
2917+
28862918
for (const auto& collision : collisions) {
28872919
float crec = getRecoCent(collision);
2920+
2921+
if constexpr (has_reco_cent<C>) {
2922+
registry.fill(HIST("Events/Centrality/EvtGenRec"), 2., crec);
2923+
} else {
2924+
registry.fill(HIST("Events/EvtGenRec"), 2.);
2925+
}
2926+
28882927
if (!isGoodEvent<false>(collision)) {
28892928
continue;
28902929
}
2930+
2931+
if constexpr (has_reco_cent<C>) {
2932+
registry.fill(HIST("Events/Centrality/EvtGenRec"), 3., crec);
2933+
} else {
2934+
registry.fill(HIST("Events/EvtGenRec"), 3.);
2935+
}
2936+
28912937
if (!collision.has_mcCollision()) {
28922938
continue;
28932939
}
@@ -3111,9 +3157,9 @@ struct DndetaMFTPbPb {
31113157
if (!particle.isPhysicalPrimary()) { // Secondaries (weak decays and material)
31123158
if (particle.getProcess() == kNumDecay) { // Particles from decay
31133159
if constexpr (has_reco_cent<C>) {
3114-
registry.fill(HIST("Tracks/Centrality/THnDCAxyBestGenPrim"), itrack.pt(), itrack.eta(), mcCollision.posZ(), atrack.bestDCAXY(), atrack.bestDCAZ(), crec);
3160+
registry.fill(HIST("Tracks/Centrality/THnDCAxyBestGenSecWeak"), itrack.pt(), itrack.eta(), mcCollision.posZ(), atrack.bestDCAXY(), atrack.bestDCAZ(), crec);
31153161
} else {
3116-
registry.fill(HIST("Tracks/THnDCAxyBestGenPrim"), itrack.pt(), itrack.eta(), mcCollision.posZ(), atrack.bestDCAXY(), atrack.bestDCAZ());
3162+
registry.fill(HIST("Tracks/THnDCAxyBestGenSecWeak"), itrack.pt(), itrack.eta(), mcCollision.posZ(), atrack.bestDCAXY(), atrack.bestDCAZ());
31173163
}
31183164
} else { // Particles from the material
31193165
if constexpr (has_reco_cent<C>) {

0 commit comments

Comments
 (0)