Skip to content

Commit

Permalink
Merge pull request #468 from kjvbrt/tests
Browse files Browse the repository at this point in the history
Adjusting tests
  • Loading branch information
kjvbrt authored Feb 22, 2024
2 parents 5607949 + 4e480ab commit ef79061
Show file tree
Hide file tree
Showing 14 changed files with 76 additions and 44 deletions.
2 changes: 1 addition & 1 deletion Examples/CMakeLists.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@


install(DIRECTORY options
install(DIRECTORY options
DESTINATION ${CMAKE_INSTALL_DATAROOTDIR}/${PROJECT_NAME}/Examples
PATTERN k4_workflow_blocks EXCLUDE
)
Expand Down
1 change: 1 addition & 0 deletions Examples/options/geant_fastsim.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
import os

from Gaudi.Configuration import *
from GaudiKernel import SystemOfUnits as units
Expand Down
2 changes: 2 additions & 0 deletions Examples/options/geant_fastsim_tklayout.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
import os

from Gaudi.Configuration import *

from Configurables import ApplicationMgr
Expand Down
13 changes: 9 additions & 4 deletions Examples/options/geant_fullsim.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
### | read events from a HepMC file | convert `HepMC::GenEvent` to EDM | geometry taken from GDML file (no sensitive detectors!) | FTFP_BERT physics list | empty action initialisation list | write the EDM output to ROOT file using PODIO |


import os

from Gaudi.Configuration import *
from GaudiKernel import SystemOfUnits as units
Expand Down Expand Up @@ -64,9 +64,14 @@
from Configurables import SimG4Alg
geantsim = SimG4Alg("SimG4Alg")
from Configurables import SimG4SaveTrackerHits
savetrackertool = SimG4SaveTrackerHits("saveTrackerHits")
savetrackertool.readoutNames = ["TrackerBarrelReadout", "TrackerEndcapReadout"]
geantsim.outputs = [savetrackertool]
savetrackerbarreltool = SimG4SaveTrackerHits("saveTrackerHitsBarrel")
savetrackerbarreltool.readoutName = "TrackerBarrelReadout"
savetrackerbarreltool.SimTrackHits.Path = "TrackerBarrelReadout"
geantsim.outputs = [savetrackerbarreltool]
savetrackerendcaptool = SimG4SaveTrackerHits("saveTrackerHitsEndcap")
savetrackerendcaptool.readoutName = "TrackerEndcapReadout"
savetrackerendcaptool.SimTrackHits.Path = "TrackerEndcapReadout"
geantsim.outputs += [savetrackerendcaptool]
from Configurables import SimG4PrimariesFromEdmTool
geantsim.eventProvider = SimG4PrimariesFromEdmTool("EdmConverter")
geantsim.eventProvider.GenParticles.Path = "allGenParticles"
Expand Down
21 changes: 10 additions & 11 deletions Examples/options/geant_fullsim_fccee_idea_hepevt.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
import os

from Gaudi.Configuration import *
import GaudiKernel.SystemOfUnits as units

Expand Down Expand Up @@ -29,26 +31,27 @@
ApplicationMgr().ExtSvc += [geoservice]

from Configurables import SimG4Alg
SimG4Alg("SimG4Alg").outputs += []
geantsim = SimG4Alg("SimG4Alg")
geantsim.outputs = []

from Configurables import SimG4SaveTrackerHits
savetrackertool = SimG4SaveTrackerHits("saveTrackerHits_Barrel")
savetrackertool.readoutNames = ["VertexBarrelCollection"]
savetrackertool.SimTrackHits.Path = "TrackerHits_barrel"
SimG4Alg("SimG4Alg").outputs += [savetrackertool]
geantsim.outputs += [savetrackertool]

from Configurables import SimG4SaveTrackerHits
savetrackertool_endcap = SimG4SaveTrackerHits("saveTrackerHits_Endcap")
savetrackertool_endcap.readoutNames = ["VertexEndcapCollection"]
savetrackertool_endcap.SimTrackHits.Path = "positionedHits_endcap"
SimG4Alg("SimG4Alg").outputs += [savetrackertool_endcap]
geantsim.outputs += [savetrackertool_endcap]


from Configurables import SimG4SaveTrackerHits
savetrackertool_DCH = SimG4SaveTrackerHits("saveTrackerHits_DCH")
savetrackertool_DCH.readoutNames = ["SimplifiedDriftChamberCollection"]
savetrackertool_DCH.SimTrackHits.Path = "positionedHits_DCH"
SimG4Alg("SimG4Alg").outputs += [savetrackertool_DCH]
geantsim.outputs += [savetrackertool_DCH]

from Configurables import SimG4ConstantMagneticFieldTool
field = SimG4ConstantMagneticFieldTool()
Expand Down Expand Up @@ -84,18 +87,14 @@
from Configurables import SimG4SaveTrajectory
savetrajectorytool = SimG4SaveTrajectory("saveTrajectory")
savetrajectorytool.TrajectoryPoints.Path = "trajectoryPoints"
# todo: investigate error
# geantsim.outputs += [savetrajectorytool]

from Configurables import SimG4SaveParticleHistory
savehisttool = SimG4SaveParticleHistory("saveHistory")
savehisttool.GenParticles.Path = "SimParticles"
geantsim.outputs += [savehisttool]

from Configurables import SimG4Alg
geantsim = SimG4Alg("SimG4Alg")
geantsim.outputs += [
savehisttool,
#todo: investigate error
#savetrajectorytool,
]
from Configurables import SimG4PrimariesFromEdmTool
geantsim.eventProvider = SimG4PrimariesFromEdmTool("EdmConverter")
geantsim.eventProvider.GenParticles.Path = "GenParticles"
Expand Down
2 changes: 2 additions & 0 deletions Examples/options/geant_fullsim_fccee_idea_pgun.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
import os

from Gaudi.Configuration import *
import GaudiKernel.SystemOfUnits as units

Expand Down
6 changes: 3 additions & 3 deletions Examples/options/geant_fullsim_fccee_lar_pgun.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,9 +46,9 @@
from Configurables import GeoSvc
geoservice = GeoSvc("GeoSvc")
# if FCC_DETECTORS is empty, this should use relative path to working directory
path_to_detector = os.environ.get("FCCDETECTORS", "")
path_to_detector = os.environ.get("K4GEO", "")
detectors_to_use=[
'Detector/DetFCCeeIDEA-LAr/compact/FCCee_DectMaster.xml',
'FCCee/ALLEGRO/compact/ALLEGRO_o1_v01/ALLEGRO_o1_v01.xml',
]
# prefix all xmls with path_to_detector
geoservice.detectors = [os.path.join(path_to_detector, _det) for _det in detectors_to_use]
Expand Down Expand Up @@ -187,7 +187,7 @@

from Configurables import CreateCaloCellPositionsFCCee
createEcalBarrelPositionedCells = CreateCaloCellPositionsFCCee("ECalBarrelPositionedCells")
createEcalBarrelPositionedCells.positionsECalBarrelTool = cellPositionEcalBarrelTool
createEcalBarrelPositionedCells.positionsTool = cellPositionEcalBarrelTool
createEcalBarrelPositionedCells.hits.Path = "ECalBarrelCells"
createEcalBarrelPositionedCells.positionedHits.Path = "ECalBarrelPositionedCells"
ApplicationMgr().TopAlg += [createEcalBarrelPositionedCells]
Expand Down
5 changes: 3 additions & 2 deletions Examples/options/geant_fullsim_fcchh_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,9 @@

# Geant4 algorithm
# Translates EDM to G4Event, passes the event to G4, writes out outputs via tools
from Configurables import SimG4Alg
geantsim = SimG4Alg("SimG4Alg")
# !!! `geantsim` already imported from k4_workflow_blocks.fccsw.detector_fcc_hh_main !!!
# from Configurables import SimG4Alg
# geantsim = SimG4Alg("SimG4Alg")
from Configurables import SimG4PrimariesFromEdmTool
geantsim.eventProvider = SimG4PrimariesFromEdmTool("EdmConverter")
geantsim.eventProvider.GenParticles.Path = "GenParticles"
Expand Down
9 changes: 5 additions & 4 deletions Examples/options/geant_fullsim_fcchh_tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@


from Configurables import ApplicationMgr
ApplicationMgr().EvtSel = 'NONE'
ApplicationMgr().EvtSel = 'NONE'
ApplicationMgr().EvtMax = 2
ApplicationMgr().OutputLevel = INFO
ApplicationMgr().ExtSvc += ['RndmGenSvc']
Expand All @@ -24,7 +24,7 @@
from Configurables import MomentumRangeParticleGun
guntool = MomentumRangeParticleGun()
guntool.ThetaMin = 0
guntool.ThetaMax = 2 * constants.pi
guntool.ThetaMax = 2 * constants.pi
guntool.PdgCodes = [11]

from Configurables import GenAlg
Expand Down Expand Up @@ -53,8 +53,9 @@

# Geant4 algorithm
# Translates EDM to G4Event, passes the event to G4, writes out outputs via tools
from Configurables import SimG4Alg
geantsim = SimG4Alg("SimG4Alg")
# !!! `geantsim` already imported from k4_workflow_blocks.fccsw.detector_fcc_hh_tracker !!!
# from Configurables import SimG4Alg
# geantsim = SimG4Alg("SimG4Alg")
from Configurables import SimG4PrimariesFromEdmTool
geantsim.eventProvider = SimG4PrimariesFromEdmTool("EdmConverter")
geantsim.eventProvider.GenParticles.Path = "GenParticles"
Expand Down
13 changes: 9 additions & 4 deletions Examples/options/geant_fullsim_field.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
### | read events from a HepMC file | convert `HepMC::GenEvent` to EDM | geometry taken from GDML file (no sensitive detectors!) | FTFP_BERT physics list | empty action initialisation list | write the EDM output to ROOT file using PODIO |


import os

from Gaudi.Configuration import *
from GaudiKernel import SystemOfUnits as units
Expand Down Expand Up @@ -76,9 +76,14 @@
savehisttool.GenParticles.Path = "SimParticles"
geantsim.outputs += [savehisttool]
from Configurables import SimG4SaveTrackerHits
savetrackertool = SimG4SaveTrackerHits("saveTrackerHits")
savetrackertool.readoutNames = ["TrackerBarrelReadout", "TrackerEndcapReadout"]
geantsim.outputs += [savetrackertool]
savetrackerbarreltool = SimG4SaveTrackerHits("saveTrackerHits")
savetrackerbarreltool.readoutName = "TrackerBarrelReadout"
savetrackerbarreltool.SimTrackHits.Path = "TrackerBarrelReadout"
geantsim.outputs += [savetrackerbarreltool]
savetrackerendcaptool = SimG4SaveTrackerHits("saveTrackerHits")
savetrackerendcaptool.readoutName = "TrackerEndcapReadout"
savetrackerendcaptool.SimTrackHits.Path += "TrackerEndcapReadout"
geantsim.outputs += [savetrackerendcaptool]
from Configurables import SimG4PrimariesFromEdmTool
geantsim.eventProvider = SimG4PrimariesFromEdmTool("EdmConverter")
geantsim.eventProvider.GenParticles.Path = "GenParticles"
Expand Down
2 changes: 2 additions & 0 deletions Examples/options/geant_fullsim_userlimits.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
import os

from Gaudi.Configuration import *
from GaudiKernel.SystemOfUnits import mm

Expand Down
2 changes: 1 addition & 1 deletion Examples/options/read_podio_input.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

from Configurables import ApplicationMgr
ApplicationMgr().EvtSel = 'NONE'
ApplicationMgr().EvtMax = 3
ApplicationMgr().EvtMax = 2

from Configurables import FCCDataSvc
podioevent = FCCDataSvc("EventDataSvc")
Expand Down
25 changes: 16 additions & 9 deletions k4_workflow_blocks/fccsw/detector_fcc_hh_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,37 +13,44 @@
geoservice.OutputLevel = INFO
ApplicationMgr().ExtSvc += [geoservice]

from Configurables import SimG4Alg
geantsim = SimG4Alg("SimG4Alg")

# Set up SaveTools that write

SimG4Alg("SimG4Alg").outputs = []
geantsim.outputs = []

from Configurables import SimG4SaveTrackerHits
savetrackertool = SimG4SaveTrackerHits("saveTrackerHits")
savetrackertool.readoutNames = ["TrackerBarrelReadout", "TrackerEndcapReadout"]
from Configurables import SimG4Alg
SimG4Alg("SimG4Alg").outputs += [savetrackertool]
savetrackerbarreltool = SimG4SaveTrackerHits("saveTrackerHits")
savetrackerbarreltool.readoutName = "TrackerBarrelReadout"
savetrackerbarreltool.SimTrackHits.Path = "TrackerBarrelReadout"
geantsim.outputs += [savetrackerbarreltool]
savetrackerendcaptool = SimG4SaveTrackerHits("saveTrackerHits")
savetrackerendcaptool.readoutName = "TrackerEndcapReadout"
savetrackerendcaptool.SimTrackHits.Path = "TrackerEndcapReadout"
geantsim.outputs += [savetrackerendcaptool]

from Configurables import SimG4SaveCalHits
saveecaltool = SimG4SaveCalHits("saveECalBarrelHits")
saveecaltool.readoutNames = ["ECalBarrelEta"]
saveecaltool.CaloHits.Path = "ECalBarrelHits"
SimG4Alg("SimG4Alg").outputs += [saveecaltool]
geantsim.outputs += [saveecaltool]

from Configurables import SimG4SaveCalHits
saveendcaptool = SimG4SaveCalHits("saveECalEndcapHits")
saveendcaptool.readoutNames = ["EMECPhiEta"]
saveendcaptool.CaloHits.Path = "ECalEndcapHits"
SimG4Alg("SimG4Alg").outputs += [saveendcaptool]
geantsim.outputs += [saveendcaptool]

from Configurables import SimG4SaveCalHits
savefwdtool = SimG4SaveCalHits("saveECalFwdHits")
savefwdtool.readoutNames = ["EMFwdPhiEta"]
savefwdtool.CaloHits.Path = "ECalFwdHits"
SimG4Alg("SimG4Alg").outputs += [savefwdtool]
geantsim.outputs += [savefwdtool]

from Configurables import SimG4SaveCalHits
savehcaltool = SimG4SaveCalHits("saveHCalHits")
savehcaltool.readoutNames = ["HCalBarrelReadout"]
savehcaltool.CaloHits.Path = "HCalBarrelHits"
SimG4Alg("SimG4Alg").outputs += [savehcaltool]
geantsim.outputs += [savehcaltool]

17 changes: 12 additions & 5 deletions k4_workflow_blocks/fccsw/detector_fcc_hh_tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,18 @@
'Detector/DetFCChhTrackerTkLayout/compact/Tracker.xml',
]
geoservice.detectors = [ os.path.join(path_to_detectors, d) for d in detectors ]
geoservice.OutputLevel = INFO
geoservice.OutputLevel = INFO
ApplicationMgr().ExtSvc += [geoservice]

from Configurables import SimG4SaveTrackerHits
savetrackertool = SimG4SaveTrackerHits("saveTrackerHits")
savetrackertool.readoutNames = ["TrackerBarrelReadout", "TrackerEndcapReadout"]
from Configurables import SimG4Alg
SimG4Alg("SimG4Alg").outputs = [savetrackertool]
geantsim = SimG4Alg("SimG4Alg")

from Configurables import SimG4SaveTrackerHits
savetrackerbarreltool = SimG4SaveTrackerHits("saveTrackerHits")
savetrackerbarreltool.readoutName = "TrackerBarrelReadout"
savetrackerbarreltool.SimTrackHits.Path = "TrackerBarrelReadout"
geantsim.outputs = [savetrackerbarreltool]
savetrackerendcaptool = SimG4SaveTrackerHits("saveTrackerHits")
savetrackerendcaptool.readoutName = "TrackerEndcapReadout"
savetrackerendcaptool.SimTrackHits.Path = "TrackerEndcapReadout"
geantsim.outputs += [savetrackerendcaptool]

0 comments on commit ef79061

Please sign in to comment.