add collections for COIL and Muon

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 {

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{