Add the conversion of subdetector energies

key4hep · Nov 3, 2023 · 30e7afa · 30e7afa
1 parent 37c9a9f
commit 30e7afa
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/k4EDM4hep2LcioConv/include/k4EDM4hep2LcioConv/k4EDM4hep2LcioConv.ipp b/k4EDM4hep2LcioConv/include/k4EDM4hep2LcioConv/k4EDM4hep2LcioConv.ipp
@@ -427,6 +427,12 @@ namespace EDM4hep2LCIOConv {
         }
         lcio_cluster->setShape(shape_vec);
 
+        auto& subdetEnergies = lcio_cluster->subdetectorEnergies();
+        for (const auto edmEnergy : edm_cluster.getSubdetectorEnergies()) {
+          std::cout << "Adding subdetector energy " << edmEnergy << '\n';
+          subdetEnergies.push_back(edmEnergy);
+        }
+
         // Convert ParticleIDs associated to the recoparticle
         for (const auto& edm_pid : edm_cluster.getParticleIDs()) {
           if (edm_pid.isAvailable()) {