From c4456536ca49753772cb716183c9dc331ea63c04 Mon Sep 17 00:00:00 2001
From: lintao <lintao51@gmail.com>
Date: Wed, 16 Dec 2020 22:08:52 +0800
Subject: [PATCH] WIP: read the DC hits with MC particles.
---
Examples/src/Edm4hepTest/Edm4hepReadDCAlg.cpp | 85 +++++++++++++++++++
Examples/src/Edm4hepTest/Edm4hepReadDCAlg.h | 33 +++++++
2 files changed, 118 insertions(+)
create mode 100644 Examples/src/Edm4hepTest/Edm4hepReadDCAlg.cpp
create mode 100644 Examples/src/Edm4hepTest/Edm4hepReadDCAlg.h
diff --git a/Examples/src/Edm4hepTest/Edm4hepReadDCAlg.cpp b/Examples/src/Edm4hepTest/Edm4hepReadDCAlg.cpp
new file mode 100644
index 00000000..3c0baa99
--- /dev/null
+++ b/Examples/src/Edm4hepTest/Edm4hepReadDCAlg.cpp
@@ -0,0 +1,85 @@
+#include "Edm4hepReadDCAlg.h"
+#include "edm4hep/EventHeaderCollection.h"
+#include "edm4hep/MCParticleCollection.h"
+#include "edm4hep/SimTrackerHitCollection.h"
+
+DECLARE_COMPONENT(Edm4hepReadDCAlg)
+
+Edm4hepReadDCAlg::Edm4hepReadDCAlg(const std::string& name, ISvcLocator* svcLoc)
+ : GaudiAlgorithm(name, svcLoc)
+{
+ declareProperty("MCParticleCol", m_mcParCol, "MCParticle collection (input)");
+ declareProperty("DCHitCol", m_dcCol, "Drift Chamber collections (input)");
+}
+
+StatusCode Edm4hepReadDCAlg::initialize()
+{
+ debug() << "begin initialize Edm4hepReadDCAlg" << endmsg;
+ return GaudiAlgorithm::initialize();
+}
+
+StatusCode Edm4hepReadDCAlg::execute()
+{
+ debug() << "begin execute Edm4hepReadDCAlg" << endmsg;
+
+ auto mcCol = m_mcParCol.get();
+ for ( auto p : *mcCol ) {
+ info() << p.getObjectID().index << " : [";
+ for ( auto it = p.daughters_begin(), end = p.daughters_end(); it != end; ++it ) {
+ info() << " " << it->getObjectID().index;
+ }
+ info() << " ]; ";
+ }
+ info() << "}" << endmsg;
+
+ auto trkCol = m_dcCol.get();
+ for (auto trkhit : *trkCol) {
+ auto position = trkhit.getPosition();
+ auto momentum = trkhit.getMomentum();
+ auto primary_particle = trkhit.getMCParticle();
+
+ info() << " cellID: " << trkhit.getCellID()
+ << " edep: " << trkhit.getEDep()
+ << " time: " << trkhit.getTime()
+ << " length: " << trkhit.getPathLength()
+ << " quality: " << trkhit.getQuality()
+ << " position: ("
+ << position[0] << ", " << position[1] << ", " << position[2]
+ << ")"
+ << " momentum: ("
+ << momentum[0] << ", " << momentum[1] << ", " << momentum[2]
+ << ")"
+ << " primary track: "
+ << primary_particle.getPDG()
+ << endmsg;
+
+
+ // unsigned int contrib_size = calohit.contributions_size();
+ // info() << " contributions_size: "
+ // << contrib_size
+ // << endmsg;
+ // for (unsigned int i = 0; i < contrib_size; ++i) {
+ // auto contrib = calohit.getContributions(i);
+ // auto primary_particle = contrib.getParticle();
+
+ // info() << " - #" << i << ": "
+ // << " track with "
+ // << " PDG: " << contrib.getPDG() // current track
+ // << ". "
+ // << " primary track with "
+ // << " PDG: " << primary_particle.getPDG()
+ // << endmsg;
+
+
+ // }
+
+ }
+
+ return StatusCode::SUCCESS;
+}
+
+StatusCode Edm4hepReadDCAlg::finalize()
+{
+ debug() << "begin finalize Edm4hepReadDCAlg" << endmsg;
+ return GaudiAlgorithm::finalize();
+}
diff --git a/Examples/src/Edm4hepTest/Edm4hepReadDCAlg.h b/Examples/src/Edm4hepTest/Edm4hepReadDCAlg.h
new file mode 100644
index 00000000..7a09eb6d
--- /dev/null
+++ b/Examples/src/Edm4hepTest/Edm4hepReadDCAlg.h
@@ -0,0 +1,33 @@
+#ifndef TEST_EDM4HEP_READ_DC_ALG_H
+#define TEST_EDM4HEP_READ_DC_ALG_H
+
+#include "k4FWCore/DataHandle.h"
+#include "GaudiAlg/GaudiAlgorithm.h"
+
+namespace edm4hep {
+ class EventHeaderCollection;
+ class MCParticleCollection;
+ class SimTrackerHitCollection;
+}
+
+class Edm4hepReadDCAlg : public GaudiAlgorithm
+{
+
+ public :
+
+ Edm4hepReadDCAlg(const std::string& name, ISvcLocator* svcLoc);
+
+ virtual StatusCode initialize();
+ virtual StatusCode execute();
+ virtual StatusCode finalize();
+
+ private :
+
+ // DataHandle<edm4hep::EventHeaderCollection> m_headerCol{"EventHeader", Gaudi::DataHandle::Reader, this};
+ DataHandle<edm4hep::MCParticleCollection> m_mcParCol{"MCParticle", Gaudi::DataHandle::Reader, this};
+ DataHandle<edm4hep::SimTrackerHitCollection> m_dcCol{"DriftChamberHitsCollection",
+ Gaudi::DataHandle::Reader, this};
+
+};
+
+#endif
--
GitLab