Skip to content
Snippets Groups Projects

Prepare detector benchmarks to be used with truly different detector versions

Merged Sylvester Joosten requested to merge prepare_canyonlands into master
Files
4
@@ -5,11 +5,17 @@ from GaudiKernel import SystemOfUnits as units
@@ -5,11 +5,17 @@ from GaudiKernel import SystemOfUnits as units
detector_name = "topside"
detector_name = "topside"
if "JUGGLER_DETECTOR" in os.environ :
if "JUGGLER_DETECTOR" in os.environ :
detector_name = str(os.environ["JUGGLER_DETECTOR"])
detector_name = str(os.environ["JUGGLER_DETECTOR"])
detector_path = ""
detector_path = ""
if "DETECTOR_PATH" in os.environ :
if "DETECTOR_PATH" in os.environ :
detector_path = str(os.environ["DETECTOR_PATH"])
detector_path = str(os.environ["DETECTOR_PATH"])
 
 
detector_version = 'default'
 
if "JUGGLER_DETECTOR_VERSION" in os.environ:
 
env_version = str(os.environ["JUGGLER_DETECTOR_VERSION"])
 
if 'acadia' in env_version:
 
detector_version = 'acadia'
# todo add checks
# todo add checks
input_sim_file = str(os.environ["JUGGLER_SIM_FILE"])
input_sim_file = str(os.environ["JUGGLER_SIM_FILE"])
@@ -40,8 +46,13 @@ from Configurables import Jug__Reco__SimpleClustering as SimpleClustering
@@ -40,8 +46,13 @@ from Configurables import Jug__Reco__SimpleClustering as SimpleClustering
algorithms = [ ]
algorithms = [ ]
 
input_collections = ['mcparticles','TrackerEndcapHits','TrackerBarrelHits','VertexBarrelHits','GEMTrackerEndcapHits']
 
if 'acadia' in detector_version:
 
input_collections.append('VertexEndcapHits')
 
else:
 
input_collections.append('MPGDTrackerBarrelHits')
podioinput = PodioInput("PodioReader",
podioinput = PodioInput("PodioReader",
collections=["mcparticles","TrackerEndcapHits","TrackerBarrelHits","VertexBarrelHits","VertexEndcapHits","GEMTrackerEndcapHits"])
collections=input_collections)#, OutputLevel=DEBUG)
algorithms.append( podioinput )
algorithms.append( podioinput )
trk_b_digi = TrackerDigi("trk_b_digi",
trk_b_digi = TrackerDigi("trk_b_digi",
@@ -61,11 +72,18 @@ vtx_b_digi = TrackerDigi("vtx_b_digi",
@@ -61,11 +72,18 @@ vtx_b_digi = TrackerDigi("vtx_b_digi",
timeResolution=8)
timeResolution=8)
algorithms.append( vtx_b_digi )
algorithms.append( vtx_b_digi )
vtx_ec_digi = TrackerDigi("vtx_ec_digi",
if 'acadia' in detector_version:
inputHitCollection="VertexEndcapHits",
vtx_ec_digi = TrackerDigi("vtx_ec_digi",
outputHitCollection="VertexEndcapRawHits",
inputHitCollection="VertexEndcapHits",
timeResolution=8)
outputHitCollection="VertexEndcapRawHits",
algorithms.append( vtx_ec_digi )
timeResolution=8)
 
algorithms.append( vtx_ec_digi )
 
else:
 
mm_b_digi = TrackerDigi("mm_b_digi",
 
inputHitCollection="MPGDTrackerBarrelHits",
 
outputHitCollection="MPGDTrackerBarrelRawHits",
 
timeResolution=8)
 
algorithms.append( mm_b_digi )
gem_ec_digi = TrackerDigi("gem_ec_digi",
gem_ec_digi = TrackerDigi("gem_ec_digi",
inputHitCollection="GEMTrackerEndcapHits",
inputHitCollection="GEMTrackerEndcapHits",
@@ -89,23 +107,35 @@ vtx_b_reco = TrackerHitReconstruction("vtx_b_reco",
@@ -89,23 +107,35 @@ vtx_b_reco = TrackerHitReconstruction("vtx_b_reco",
outputHitCollection="VertexBarrelRecHits")
outputHitCollection="VertexBarrelRecHits")
algorithms.append( vtx_b_reco )
algorithms.append( vtx_b_reco )
vtx_ec_reco = TrackerHitReconstruction("vtx_ec_reco",
if 'acadia' in detector_version:
inputHitCollection = vtx_ec_digi.outputHitCollection,
vtx_ec_reco = TrackerHitReconstruction("vtx_ec_reco",
outputHitCollection="VertexEndcapRecHits")
inputHitCollection = vtx_ec_digi.outputHitCollection,
algorithms.append( vtx_ec_reco )
outputHitCollection="VertexEndcapRecHits")
 
algorithms.append( vtx_ec_reco )
 
else:
 
mm_b_reco = TrackerHitReconstruction("mm_b_reco",
 
inputHitCollection = mm_b_digi.outputHitCollection,
 
outputHitCollection="MPGDTrackerBarrelRecHits")
 
algorithms.append( mm_b_reco )
 
gem_ec_reco = TrackerHitReconstruction("gem_ec_reco",
gem_ec_reco = TrackerHitReconstruction("gem_ec_reco",
inputHitCollection=gem_ec_digi.outputHitCollection,
inputHitCollection=gem_ec_digi.outputHitCollection,
outputHitCollection="GEMTrackerEndcapRecHits")
outputHitCollection="GEMTrackerEndcapRecHits")
algorithms.append(gem_ec_reco)
algorithms.append(gem_ec_reco)
 
input_tracking_hits = [
 
str(trk_b_reco.outputHitCollection),
 
str(trk_ec_reco.outputHitCollection),
 
str(vtx_b_reco.outputHitCollection),
 
str(gem_ec_reco.outputHitCollection) ]
 
if 'acadia' in detector_version:
 
input_tracking_hits.append(str(vtx_ec_reco.outputHitCollection))
 
else:
 
input_tracking_hits.append(str(mm_b_reco.outputHitCollection))
 
trk_hit_col = TrackingHitsCollector("trk_hit_col",
trk_hit_col = TrackingHitsCollector("trk_hit_col",
inputTrackingHits=[
inputTrackingHits=input_tracking_hits,
str(trk_b_reco.outputHitCollection),
str(trk_ec_reco.outputHitCollection),
str(vtx_b_reco.outputHitCollection),
str(vtx_ec_reco.outputHitCollection),
str(gem_ec_reco.outputHitCollection) ],
trackingHits="trackingHits",
trackingHits="trackingHits",
OutputLevel=DEBUG)
OutputLevel=DEBUG)
algorithms.append( trk_hit_col )
algorithms.append( trk_hit_col )
Loading