Fix names in python script creating all types

key4hep · Jul 30, 2024 · e506497 · e506497
1 parent 0a537f8
commit e506497
Showing 1 changed file with 48 additions and 48 deletions.
diff --git a/scripts/createEDM4hepFile.py b/scripts/createEDM4hepFile.py
@@ -269,54 +269,54 @@
     # if not args.use_pre1:
     #     pid.setParticle(reco_particle)
 
-    assocs = edm4hep.MCRecoParticleAssociationCollection()
-    assoc = assocs.create()
-    assoc.setWeight(next(counter))
-    assoc.setRec(reco_particle)
-    assoc.setSim(particle)
-    frame.put(assocs, "MCRecoParticleAssociationCollection")
-
-    assocs = edm4hep.MCRecoCaloAssociationCollection()
-    assoc = assocs.create()
-    assoc.setWeight(next(counter))
-    assoc.setRec(calo_hit)
-    assoc.setSim(simcalo_hit)
-    frame.put(assocs, "MCRecoCaloAssociationCollection")
-
-    assocs = edm4hep.MCRecoTrackerAssociationCollection()
-    assoc = assocs.create()
-    assoc.setWeight(next(counter))
-    assoc.setRec(tracker_hit)
-    assoc.setSim(simtracker_hit)
-    frame.put(assocs, "MCRecoTrackerAssociationCollection")
-
-    assocs = edm4hep.MCRecoCaloParticleAssociationCollection()
-    assoc = assocs.create()
-    assoc.setWeight(next(counter))
-    assoc.setRec(calo_hit)
-    assoc.setSim(particle)
-    frame.put(assocs, "MCRecoCaloParticleAssociationCollection")
-
-    assocs = edm4hep.MCRecoClusterParticleAssociationCollection()
-    assoc = assocs.create()
-    assoc.setWeight(next(counter))
-    assoc.setRec(cluster)
-    assoc.setSim(particle)
-    frame.put(assocs, "MCRecoClusterParticleAssociationCollection")
-
-    assocs = edm4hep.MCRecoTrackParticleAssociationCollection()
-    assoc = assocs.create()
-    assoc.setWeight(next(counter))
-    assoc.setRec(track)
-    assoc.setSim(particle)
-    frame.put(assocs, "MCRecoTrackParticleAssociationCollection")
-
-    assocs = edm4hep.RecoParticleVertexAssociationCollection()
-    assoc = assocs.create()
-    assoc.setWeight(next(counter))
-    assoc.setRec(reco_particle)
-    assoc.setVertex(v)
-    frame.put(assocs, "MCRecoParticleVertexAssociationCollection")
+    links = edm4hep.RecoMCParticleLinkCollection()
+    link = links.create()
+    link.setWeight(next(counter))
+    link.setRec(reco_particle)
+    link.setSim(particle)
+    frame.put(links, "RecoMCParticleLinkCollection")
+
+    links = edm4hep.CaloHitSimCaloHitLinkCollection()
+    link = links.create()
+    link.setWeight(next(counter))
+    link.setRec(calo_hit)
+    link.setSim(simcalo_hit)
+    frame.put(links, "CaloHitSimCaloHitLinkCollection")
+
+    links = edm4hep.TrackerHitSimTrackerHitLinkCollection()
+    link = links.create()
+    link.setWeight(next(counter))
+    link.setRec(tracker_hit)
+    link.setSim(simtracker_hit)
+    frame.put(links, "TrackerHitSimTrackerHitLinkCollection")
+
+    links = edm4hep.CaloHitMCParticleLinkCollection()
+    link = links.create()
+    link.setWeight(next(counter))
+    link.setRec(calo_hit)
+    link.setSim(particle)
+    frame.put(links, "CaloHitMCParticleLinkCollection")
+
+    links = edm4hep.ClusterMCParticleLinkCollection()
+    link = links.create()
+    link.setWeight(next(counter))
+    link.setRec(cluster)
+    link.setSim(particle)
+    frame.put(links, "ClusterMCParticleLinkCollection")
+
+    links = edm4hep.TrackMCParticleLinkCollection()
+    link = links.create()
+    link.setWeight(next(counter))
+    link.setRec(track)
+    link.setSim(particle)
+    frame.put(links, "TrackMCParticleLinkCollection")
+
+    links = edm4hep.VertexRecoParticleLinkCollection()
+    link = links.create()
+    link.setWeight(next(counter))
+    link.setRec(reco_particle)
+    link.setVertex(v)
+    frame.put(links, "MCVertexRecoParticleLinkCollection")
 
     timeseries = edm4hep.TimeSeriesCollection()
     serie = timeseries.create()