diff --git a/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.cpp b/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.cpp index e3728357740c5001984f99b58fc1a12a8b090842..4a7f03413735a5b4b1aade3ef7eed49168c7ee26 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 dffb1a8232ef29ea8e20d1fa606a092d0bd522d9..174f64fa5286fd896432b2f0225c2075690a68ee 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",