Switch from Associations to Links (#115)

* Switch from Associations to Links * Rename variables to match types better
HEP-FCC · Oct 2, 2024 · 7ddab6a · 7ddab6a
1 parent 8e29e3d
commit 7ddab6a
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/RecCalorimeter/src/components/CreateTruthJet.cpp b/RecCalorimeter/src/components/CreateTruthJet.cpp
@@ -7,7 +7,7 @@
 
 // EDM4hep
 #include "edm4hep/MCParticleCollection.h"
-#include "edm4hep/MCRecoParticleAssociationCollection.h"
+#include "edm4hep/RecoMCParticleLinkCollection.h"
 #include "edm4hep/ReconstructedParticleCollection.h"
 
 // k4RecCalorimeter
@@ -39,7 +39,7 @@
 struct CreateTruthJet final
     : k4FWCore::MultiTransformer<
           std::tuple<edm4hep::ReconstructedParticleCollection,
-                     edm4hep::MCRecoParticleAssociationCollection>(
+                     edm4hep::RecoMCParticleLinkCollection>(
               const edm4hep::MCParticleCollection &)> {
 
   CreateTruthJet(const std::string &name, ISvcLocator *svcLoc)
@@ -61,7 +61,7 @@ struct CreateTruthJet final
   }
 
   std::tuple<edm4hep::ReconstructedParticleCollection,
-             edm4hep::MCRecoParticleAssociationCollection>
+             edm4hep::RecoMCParticleLinkCollection>
   operator()(const edm4hep::MCParticleCollection &input) const override {
 
     std::vector<fastjet::PseudoJet> clustersPJ;
@@ -79,7 +79,7 @@ struct CreateTruthJet final
         m_clusterer->cluster(clustersPJ);
 
     auto edmJets = edm4hep::ReconstructedParticleCollection();
-    auto assoc = edm4hep::MCRecoParticleAssociationCollection();
+    auto links = edm4hep::RecoMCParticleLinkCollection();
 
     for (auto cjet : inclusiveJets) {
       edm4hep::MutableReconstructedParticle jet;
@@ -92,15 +92,15 @@ struct CreateTruthJet final
       for (auto constit : constits) {
         int index = constit.user_info<k4::recCalo::ClusterInfo>().index();
 
-        auto association = assoc.create();
-        association.setRec(jet);
-        association.setSim((input)[index]);
+        auto link = links.create();
+        link.setFrom(jet);
+        link.setTo((input)[index]);
       }
 
       edmJets.push_back(jet);
     }
 
-    return std::make_tuple(std::move(edmJets), std::move(assoc));
+    return std::make_tuple(std::move(edmJets), std::move(links));
   }
 
   StatusCode finalize() override {