Add conversion of MCRecoCaloParticleAssociations

key4hep · Jun 2, 2023 · afb9450 · afb9450
1 parent 19ed20b
commit afb9450
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/k4EDM4hep2LcioConv/src/k4Lcio2EDM4hepConv.cpp b/k4EDM4hep2LcioConv/src/k4Lcio2EDM4hepConv.cpp
@@ -1001,6 +1001,16 @@ namespace LCIO2EDM4hepConv {
           relations, typeMapping.vertices, typeMapping.recoParticles);
         assoCollVec.emplace_back(name, std::move(mc_a));
       }
+      else if (fromType == "CalorimeterHit" && toType == "MCParticle") {
+        auto assoc = createAssociationCollection<edm4hep::MCRecoCaloParticleAssociationCollection, true>(
+          relations, typeMapping.caloHits, typeMapping.mcParticles);
+        assoCollVec.emplace_back(name, std::move(assoc));
+      }
+      else if (fromType == "MCParticle" && toType == "CalorimeterHit") {
+        auto assoc = createAssociationCollection<edm4hep::MCRecoCaloParticleAssociationCollection, false>(
+          relations, typeMapping.mcParticles, typeMapping.caloHits);
+        assoCollVec.emplace_back(name, std::move(assoc));
+      }
       else {
         std::cout << "Relation from: " << fromType << " to: " << toType << " (" << name
                   << ") is not beeing handled during creation of associations" << std::endl;