Keep cell relation in corrected clusters (#40)

HEP-FCC · Mar 28, 2023 · f8d3ad8 · f8d3ad8
1 parent b5b0d2b
commit f8d3ad8
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/RecCalorimeter/src/components/CorrectCaloClusters.cpp b/RecCalorimeter/src/components/CorrectCaloClusters.cpp
@@ -166,14 +166,12 @@ edm4hep::ClusterCollection* CorrectCaloClusters::initializeOutputClusters(
   edm4hep::ClusterCollection* outClusters = m_outClusters.createAndPut();
 
   for (auto const& inCluster: *inClusters) {
-    auto outCluster = outClusters->create();
-
-    outCluster.setPosition(inCluster.getPosition());
+    auto outCluster = inCluster.clone();
     verbose() << "Cluster position:" << endmsg;
     verbose() << "    x: " << outCluster.position().x << endmsg;
     verbose() << "    y: " << outCluster.position().y << endmsg;
     verbose() << "    z: " << outCluster.position().z << endmsg;
-    outCluster.setEnergy(inCluster.getEnergy());
+    outClusters->push_back(outCluster);
   }
 
   return outClusters;