From e506497aa2b1ecbab818d2e851ba0b45734ce4f8 Mon Sep 17 00:00:00 2001
From: tmadlener <thomas.madlener@desy.de>
Date: Mon, 29 Jul 2024 17:48:21 +0200
Subject: [PATCH] Fix names in python script creating all types

---
 scripts/createEDM4hepFile.py | 96 ++++++++++++++++++------------------
 1 file changed, 48 insertions(+), 48 deletions(-)

diff --git a/scripts/createEDM4hepFile.py b/scripts/createEDM4hepFile.py
index f838adeed..9ade63cf0 100644
--- 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()