From 6d4aed7895014da104e99cd3a6b2fb085c890be1 Mon Sep 17 00:00:00 2001
From: Lottie Cavanagh <ccavanagh@hep.ph.liv.ac.uk>
Date: Tue, 20 Dec 2022 16:15:16 +0000
Subject: [PATCH] Fix for entries in truth plots
---
.../NtupleDumper/src/NtupleDumperAlg.cxx | 29 +++++++++++++++++--
1 file changed, 27 insertions(+), 2 deletions(-)
diff --git a/PhysicsAnalysis/NtupleDumper/src/NtupleDumperAlg.cxx b/PhysicsAnalysis/NtupleDumper/src/NtupleDumperAlg.cxx
index 1369684d1..b61be213a 100644
--- a/PhysicsAnalysis/NtupleDumper/src/NtupleDumperAlg.cxx
+++ b/PhysicsAnalysis/NtupleDumper/src/NtupleDumperAlg.cxx
@@ -548,10 +548,10 @@ StatusCode NtupleDumperAlg::execute(const EventContext &ctx) const
{
for (auto particle : *truthParticleContainer)
{
- if ( particle->barcode() <= 3)
+ if ( particle->barcode() == 1 || particle->barcode() == 2 || particle->barcode() == 3 )
{
- if ( particle->barcode() == 1) // mother particle (A')
+ if ( particle->pdgId() == 32) // mother particle (A')
{
m_truthM_P.push_back(particle->p4().P());
@@ -1218,6 +1218,31 @@ NtupleDumperAlg::clearTree() const
m_truthLeptonMomentum = 0;
m_truthBarcode = 0;
m_truthPdg = 0;
+
+ m_truthM_P.clear();
+ m_truthM_px.clear();
+ m_truthM_py.clear();
+ m_truthM_pz.clear();
+ m_truthM_x.clear();
+ m_truthM_y.clear();
+ m_truthM_z.clear();
+
+ m_truthd0_P.clear();
+ m_truthd0_px.clear();
+ m_truthd0_py.clear();
+ m_truthd0_pz.clear();
+ m_truthd0_x.clear();
+ m_truthd0_y.clear();
+ m_truthd0_z.clear();
+
+ m_truthd1_P.clear();
+ m_truthd1_px.clear();
+ m_truthd1_py.clear();
+ m_truthd1_pz.clear();
+ m_truthd1_x.clear();
+ m_truthd1_y.clear();
+ m_truthd1_z.clear();
+
}
void NtupleDumperAlg::setNaN() const {
--
GitLab