From b2e07ab4f6e32427dd78b03c33a90655967386f8 Mon Sep 17 00:00:00 2001
From: FU Chengdong <fucd@ihep.ac.cn>
Date: Mon, 4 Nov 2024 03:23:05 +0000
Subject: [PATCH] SIM: add TPCLowPtCollection and TPCSpacePointCollection
---
Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.cpp | 10 +++++++---
Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.h | 4 ++++
2 files changed, 11 insertions(+), 3 deletions(-)
diff --git a/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.cpp b/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.cpp
index e3728357..4a7f0341 100644
--- a/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.cpp
+++ b/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.cpp
@@ -116,6 +116,8 @@ Edm4hepWriterAnaElemTool::EndOfEventAction(const G4Event* anEvent) {
auto ftdcols = m_FTDCol.createAndPut();
auto sitcols = m_SITCol.createAndPut();
auto tpccols = m_TPCCol.createAndPut();
+ auto tpclowptcols = m_TPCLowPtCol.createAndPut();
+ auto tpcspcols = m_TPCSpacePointCol.createAndPut();
auto setcols = m_SETCol.createAndPut();
auto otkbarrelcols = m_OTKBarrelCol.createAndPut();
auto otkendcapcols = m_OTKEndcapCol.createAndPut();
@@ -190,9 +192,11 @@ Edm4hepWriterAnaElemTool::EndOfEventAction(const G4Event* anEvent) {
tracker_col_ptr = sitcols;
} else if (collect->GetName() == "TPCCollection") {
tracker_col_ptr = tpccols;
- } else if (collect->GetName() == "SETCollection") {
- tracker_col_ptr = setcols;
- } else if (collect->GetName() == "SETCollection") {
+ } else if (collect->GetName() == "TPCLowPtCollection") {
+ tracker_col_ptr = tpclowptcols;
+ } else if (collect->GetName() == "TPCSpacePointCollection") {
+ tracker_col_ptr = tpcspcols;
+ } else if (collect->GetName() == "SETCollection") {
tracker_col_ptr = setcols;
} else if (collect->GetName() == "OTKBarrelCollection") {
tracker_col_ptr = otkbarrelcols;
diff --git a/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.h b/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.h
index dffb1a82..174f64fa 100644
--- a/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.h
+++ b/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.h
@@ -70,6 +70,10 @@ private:
Gaudi::DataHandle::Writer, this};
DataHandle<edm4hep::SimTrackerHitCollection> m_TPCCol{"TPCCollection",
Gaudi::DataHandle::Writer, this};
+ DataHandle<edm4hep::SimTrackerHitCollection> m_TPCLowPtCol{"TPCLowPtCollection",
+ Gaudi::DataHandle::Writer, this};
+ DataHandle<edm4hep::SimTrackerHitCollection> m_TPCSpacePointCol{"TPCSpacePointCollection",
+ Gaudi::DataHandle::Writer, this};
DataHandle<edm4hep::SimTrackerHitCollection> m_SETCol{"SETCollection",
Gaudi::DataHandle::Writer, this};
DataHandle<edm4hep::SimTrackerHitCollection> m_OTKBarrelCol{"OTKBarrelCollection",
--
GitLab