diff --git a/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.cpp b/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.cpp index 951ff463723de7ccd5af6004adc45feca0ee41da..2736ac984c15b53f291dfac472a5e3307f823a60 100644 --- a/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.cpp +++ b/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.cpp @@ -66,6 +66,13 @@ Edm4hepWriterAnaElemTool::EndOfEventAction(const G4Event* anEvent) { auto hcalendcapringcol = m_HcalEndcapRingCol.createAndPut(); auto hcalendcapringcontribcol = m_HcalEndcapRingContributionCol.createAndPut(); + auto coilcols = m_COILCol.createAndPut(); + + auto muonbarrelcol = m_MuonBarrelCol.createAndPut(); + auto muonbarrelcontribcols = m_MuonBarrelContributionCol.createAndPut(); + auto muonendcapscol = m_MuonEndcapsCol.createAndPut(); + auto muonendcapscontribcols = m_MuonEndcapsContributionCol.createAndPut(); + auto driftchamberhitscol = m_DriftChamberHitsCol.createAndPut(); // readout defined in DD4hep @@ -137,6 +144,14 @@ Edm4hepWriterAnaElemTool::EndOfEventAction(const G4Event* anEvent) { } else if (collect->GetName() == "HcalEndcapRingCollection") { calo_col_ptr = hcalendcapringcol; calo_contrib_col_ptr = hcalendcapringcontribcol; + } else if (collect->GetName() == "COILCollection") { + tracker_col_ptr = coilcols; + } else if (collect->GetName() == "MuonBarrelCollection") { + calo_col_ptr = muonbarrelcol; + calo_contrib_col_ptr = muonbarrelcontribcols; + } else if (collect->GetName() == "MuonEndcapsCollection") { + calo_col_ptr = muonendcapscol; + calo_contrib_col_ptr = muonendcapscontribcols; } else if (collect->GetName() == "DriftChamberHitsCollection") { tracker_col_ptr = driftchamberhitscol; } else { diff --git a/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.h b/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.h index bd71f932377845ca8d56b1ee84d238a6f7ce3e1d..26a1ba48a1ab00a05fa65a4fa011320d1770d988 100644 --- a/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.h +++ b/Simulation/DetSimAna/src/Edm4hepWriterAnaElemTool.h @@ -98,6 +98,22 @@ private: "HcalEndcapRingContributionCollection", Gaudi::DataHandle::Writer, this}; + // Coil + DataHandle<edm4hep::SimTrackerHitCollection> m_COILCol{"COILCollection", + Gaudi::DataHandle::Writer, this}; + + // Muon + DataHandle<edm4hep::SimCalorimeterHitCollection> m_MuonBarrelCol{"MuonBarrelCollection", + Gaudi::DataHandle::Writer, this}; + DataHandle<edm4hep::CaloHitContributionCollection> m_MuonBarrelContributionCol{ + "MuonBarrelContributionCollection", + Gaudi::DataHandle::Writer, this}; + DataHandle<edm4hep::SimCalorimeterHitCollection> m_MuonEndcapsCol{"MuonEndcapsCollection", + Gaudi::DataHandle::Writer, this}; + DataHandle<edm4hep::CaloHitContributionCollection> m_MuonEndcapsContributionCol{ + "MuonEndcapsContributionCollection", + Gaudi::DataHandle::Writer, this}; + // Drift Chamber // - DriftChamberHitsCollection DataHandle<edm4hep::SimTrackerHitCollection> m_DriftChamberHitsCol{