Migrate to IOSvc

key4hep · Aug 6, 2024 · a9f592f · a9f592f
1 parent c3a17f5
commit a9f592f
Showing 1 changed file with 16 additions and 18 deletions.
diff --git a/Tracking/test/runTracksFromGenParticles.py b/Tracking/test/runTracksFromGenParticles.py
@@ -1,18 +1,15 @@
-from Configurables import ApplicationMgr
-from Configurables import EventCounter
+from k4FWCore import ApplicationMgr
 from Gaudi.Configuration import INFO, WARNING, DEBUG
 import os
 
 if not os.path.isfile("ddsim_output_edm4hep.root"):
     os.system("ddsim --enableGun --gun.distribution uniform --gun.energy '10*GeV' --gun.particle e- --numberOfEvents 100 --outputFile ddsim_output_edm4hep.root --random.enableEventSeed --random.seed 42 --compactFile $K4GEO/FCCee/IDEA/compact/IDEA_o1_v02/IDEA_o1_v02.xml")
 
 # Loading the output of the SIM step
-from Configurables import k4DataSvc, PodioInput, PodioOutput
-evtsvc = k4DataSvc('EventDataSvc')
-evtsvc.input = "ddsim_output_edm4hep.root"
-#evtsvc.input = "/afs/cern.ch/user/b/brfranco/work/public/MIT_tutorial/CLDConfig/CLDConfig/wzp6_ee_mumuH_ecm240_CLD_RECO_edm4hep.root"
-Nevts = -1
-input_reader = PodioInput('InputReader')
+from k4FWCore import IOSvc
+io_svc = IOSvc("IOSvc")
+io_svc.input = "ddsim_output_edm4hep.root"
+io_svc.output = "tracks_from_genParticle_output.root"
 
 # Calling TracksFromGenParticles
 from Configurables import TracksFromGenParticles
@@ -30,23 +27,24 @@
                                              InputTracksFromGenParticlesAssociation = tracksFromGenParticles.OutputMCRecoTrackParticleAssociation, 
                                              Bz = 2.0)
 
-# Output
-from Configurables import PodioOutput
-out = PodioOutput("out",
-                  OutputLevel=INFO)
-out.outputCommands = ["keep *"]
-out.filename = "tracks_from_genParticle_output.root"
-
 # Set auditor service
 from Configurables import AuditorSvc, ChronoAuditor
 chra = ChronoAuditor()
 audsvc = AuditorSvc()
 audsvc.Auditors = [chra]
+tracksFromGenParticles.AuditExecute = True
+plotTrackHitDistances.AuditExecute = True
+
+# event counter
+from Configurables import EventCounter
+event_counter = EventCounter('event_counter')
+event_counter.Frequency = 10
 
+from Configurables import EventDataSvc
 ApplicationMgr(
-    TopAlg= [input_reader, tracksFromGenParticles, plotTrackHitDistances, out],
+    TopAlg= [event_counter, tracksFromGenParticles, plotTrackHitDistances],
     EvtSel='NONE',
-    EvtMax=Nevts,
-    ExtSvc=[evtsvc, audsvc],
+    EvtMax=-1,
+    ExtSvc=[EventDataSvc("EventDataSvc"), audsvc],
     StopOnSignal=True,
 )