Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 9 additions & 21 deletions ARCdigi/test/runARCdigitizer.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
import os
from Gaudi.Configuration import *
from Configurables import ApplicationMgr

from Configurables import GeoSvc
from Configurables import GeoSvc, EventDataSvc
from k4FWCore import ApplicationMgr, IOSvc

geoservice = GeoSvc("GeoSvc")
path_to_detector = os.environ.get("LCGEO")
Expand All @@ -11,41 +11,29 @@
]
geoservice.detectors = [os.path.join(path_to_detector, _det) for _det in detectors_to_use]

from Configurables import k4DataSvc

dataservice = k4DataSvc(
"EventDataSvc", input=vars().get("input", "data/arcsim_kaon+_edm4hep.root")
)

from Configurables import PodioInput

podioinput = PodioInput("PodioInput", collections=["ARC_HITS"], OutputLevel=DEBUG)
iosvc = IOSvc()
iosvc.Input = vars().get("input", "data/arcsim_kaon+_edm4hep.root")
iosvc.CollectionNames = ["ARC_HITS"]
iosvc.Output = vars().get("output", "digi.root")
iosvc.outputCommands = ["keep *"]

from Configurables import ARCdigitizer

arc_digitizer = ARCdigitizer(
"ARCdigitizer", inputSimHits="ARC_HITS", outputDigiHits="ARC_DIGI_HITS"
)

from Configurables import PodioOutput

podiooutput = PodioOutput(
"PodioOutput", filename=vars().get("output", "digi.root"), OutputLevel=DEBUG
)
podiooutput.outputCommands = ["keep *"]

# CPU information
from Configurables import AuditorSvc, ChronoAuditor

chra = ChronoAuditor()
audsvc = AuditorSvc()
audsvc.Auditors = [chra]
arc_digitizer.AuditExecute = True
podiooutput.AuditExecute = True

ApplicationMgr(
TopAlg=[podioinput, arc_digitizer, podiooutput],
TopAlg=[arc_digitizer],
EvtSel="NONE",
EvtMax=10,
ExtSvc=[geoservice, dataservice],
ExtSvc=[geoservice, EventDataSvc("EventDataSvc")],
)
21 changes: 7 additions & 14 deletions DCHdigi/test/runDCHsimpleDigitizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,10 @@

from Gaudi.Configuration import *

from Configurables import FCCDataSvc
from Configurables import EventDataSvc
from k4FWCore import ApplicationMgr, IOSvc

podioevent = FCCDataSvc("EventDataSvc")
podioevent = EventDataSvc("EventDataSvc")

from GaudiKernel.SystemOfUnits import MeV, GeV, tesla

Expand Down Expand Up @@ -142,14 +143,8 @@
)

################ Output
from Configurables import PodioOutput

out = PodioOutput("out", OutputLevel=INFO)
out.outputCommands = ["keep *"]

import uuid

out.filename = (
iosvc = IOSvc()
iosvc.Output = (
"output_simplifiedDriftChamber_MagneticField_"
+ str(magneticField)
+ "_pMin_"
Expand All @@ -163,6 +158,7 @@
+ str(pdgCode)
+ ".root"
)
iosvc.outputCommands = ["keep *"]

# CPU information
from Configurables import AuditorSvc, ChronoAuditor
Expand All @@ -173,12 +169,9 @@
genAlg.AuditExecute = True
hepmc_converter.AuditExecute = True
geantsim.AuditExecute = True
out.AuditExecute = True

from Configurables import ApplicationMgr

ApplicationMgr(
TopAlg=[genAlg, hepmc_converter, geantsim, dch_digitizer, out],
TopAlg=[genAlg, hepmc_converter, geantsim, dch_digitizer],
EvtSel="NONE",
EvtMax=10,
ExtSvc=[geoservice, podioevent, geantservice, audsvc],
Expand Down
26 changes: 9 additions & 17 deletions DCHdigi/test/runDCHsimpleDigitizerExtendedEdm.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,10 @@

from Gaudi.Configuration import *

from Configurables import FCCDataSvc
from Configurables import EventDataSvc
from k4FWCore import ApplicationMgr, IOSvc

podioevent = FCCDataSvc("EventDataSvc")
podioevent = EventDataSvc("EventDataSvc")

from GaudiKernel.SystemOfUnits import MeV, GeV, tesla

Expand Down Expand Up @@ -174,17 +175,8 @@


################ Output
from Configurables import PodioOutput

out = PodioOutput("out", OutputLevel=INFO)
out.outputCommands = ["keep *"]
if not dch_digitizer.debugMode:
out.outputCommands.append("drop *HitSimHitDelta*")
out.outputCommands.append("drop outputDigiLocalHits")

import uuid

out.filename = (
iosvc = IOSvc()
iosvc.Output = (
"output_simplifiedDriftChamber_MagneticField_"
+ str(magneticField)
+ "_pMin_"
Expand All @@ -198,6 +190,10 @@
+ str(pdgCode)
+ "_stepLength_default.root"
)
iosvc.outputCommands = ["keep *"]
if not dch_digitizer.debugMode:
iosvc.outputCommands.append("drop *HitSimHitDelta*")
iosvc.outputCommands.append("drop outputDigiLocalHits")

# CPU information
from Configurables import AuditorSvc, ChronoAuditor
Expand All @@ -208,9 +204,6 @@
genAlg.AuditExecute = True
hepmc_converter.AuditExecute = True
geantsim.AuditExecute = True
out.AuditExecute = True

from Configurables import ApplicationMgr

ApplicationMgr(
TopAlg=[
Expand All @@ -219,7 +212,6 @@
geantsim,
dch_digitizer,
# dch_perf,
out,
],
EvtSel="NONE",
EvtMax=100,
Expand Down
20 changes: 6 additions & 14 deletions VTXdigi/test/runVTXdigitizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,10 @@

from Gaudi.Configuration import *

from Configurables import FCCDataSvc
from Configurables import EventDataSvc
from k4FWCore import ApplicationMgr, IOSvc

podioevent = FCCDataSvc("EventDataSvc")
podioevent = EventDataSvc("EventDataSvc")

from GaudiKernel.SystemOfUnits import MeV, GeV, tesla

Expand Down Expand Up @@ -314,14 +315,8 @@
# genfitter = GenFitter("GenFitter", inputHits = savetrackertool.SimTrackHits.Path.replace("sim", "digi"), outputTracks = "genfit_tracks")

################ Output
from Configurables import PodioOutput

out = PodioOutput("out", OutputLevel=INFO)
out.outputCommands = ["keep *"]

import uuid

out.filename = (
iosvc = IOSvc()
iosvc.Output = (
"output_vertex_"
+ str(magneticField)
+ "_pMin_"
Expand All @@ -335,6 +330,7 @@
+ str(pdgCode)
+ ".root"
)
iosvc.outputCommands = ["keep *"]

# CPU information
from Configurables import AuditorSvc, ChronoAuditor
Expand All @@ -345,9 +341,6 @@
genAlg.AuditExecute = True
hepmc_converter.AuditExecute = True
geantsim.AuditExecute = True
out.AuditExecute = True

from Configurables import ApplicationMgr

# # CLD
# ApplicationMgr(
Expand All @@ -374,7 +367,6 @@
idea_vtxd_digitizer,
idea_siwrb_digitizer,
idea_siwrd_digitizer,
out,
],
EvtSel="NONE",
EvtMax=10,
Expand Down
2 changes: 1 addition & 1 deletion VTXdigiDetailed/test/py_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
import importlib.util
import importlib.abc
from importlib.machinery import SourceFileLoader
from Configurables import PodioOutput, MarlinProcessorWrapper
from Configurables import MarlinProcessorWrapper
from typing import Iterable
from Gaudi.Configuration import WARNING

Expand Down
30 changes: 9 additions & 21 deletions share/runDigi.py
Original file line number Diff line number Diff line change
@@ -1,51 +1,39 @@
from Gaudi.Configuration import *
from Configurables import ApplicationMgr

from Configurables import GeoSvc
from Configurables import GeoSvc, EventDataSvc
from k4FWCore import ApplicationMgr, IOSvc

geoservice = GeoSvc("GeoSvc")
geoservice.detectors.append("/path/to/k4geo/FCCee/compact/CLD_ARC/arc_full_v0.xml")

from Configurables import k4DataSvc

dataservice = k4DataSvc(
"EventDataSvc", input=vars().get("input", "data/arcsim_kaon+_edm4hep.root")
)

from Configurables import PodioInput

podioinput = PodioInput("PodioInput", collections=["ARC_HITS"], OutputLevel=DEBUG)
iosvc = IOSvc()
iosvc.Input = vars().get("input", "data/arcsim_kaon+_edm4hep.root")
iosvc.CollectionNames = ["ARC_HITS"]
iosvc.Output = vars().get("output", "digi.root")
iosvc.outputCommands = ["keep *"]

from Configurables import ARCdigitizer

arc_digitizer = ARCdigitizer(
"ARCdigitizer", inputSimHits="ARC_HITS", outputDigiHits="ARC_DIGI_HITS"
)

from Configurables import PodioOutput

podiooutput = PodioOutput(
"PodioOutput", filename=vars().get("output", "digi.root"), OutputLevel=DEBUG
)
podiooutput.outputCommands = ["keep *"]

# CPU information
from Configurables import AuditorSvc, ChronoAuditor

chra = ChronoAuditor()
audsvc = AuditorSvc()
audsvc.Auditors = [chra]
arc_digitizer.AuditExecute = True
podiooutput.AuditExecute = True

from Configurables import EventCounter

event_counter = EventCounter("event_counter")
event_counter.Frequency = 1

ApplicationMgr(
TopAlg=[event_counter, podioinput, arc_digitizer, podiooutput],
TopAlg=[event_counter, arc_digitizer],
EvtSel="NONE",
EvtMax=10,
ExtSvc=[geoservice, dataservice],
ExtSvc=[geoservice, EventDataSvc("EventDataSvc")],
)
Loading