From 7a8a4cc4e1c538432902aa3dc908c51d19705160 Mon Sep 17 00:00:00 2001
From: Wouter Deconinck <wdconinc@gmail.com>
Date: Thu, 30 Jun 2022 15:08:18 -0500
Subject: [PATCH] Use has_ecal_barrel_scfi
---
.../imaging_ecal/options/scfi_cluster.py | 50 +++++++++++++++----
1 file changed, 40 insertions(+), 10 deletions(-)
diff --git a/benchmarks/imaging_ecal/options/scfi_cluster.py b/benchmarks/imaging_ecal/options/scfi_cluster.py
index bb55aba3..802cf6b4 100644
--- a/benchmarks/imaging_ecal/options/scfi_cluster.py
+++ b/benchmarks/imaging_ecal/options/scfi_cluster.py
@@ -12,8 +12,19 @@ from Configurables import Jug__Reco__CalorimeterHitsMerger as CalHitsMerger
from Configurables import Jug__Reco__CalorimeterIslandCluster as IslandCluster
from Configurables import Jug__Reco__ClusterRecoCoG as RecoCoG
+detector_name = str(os.environ.get("JUGGLER_DETECTOR", "athena"))
+detector_config = str(os.environ.get("JUGGLER_DETECTOR_CONFIG", detector_name))
+detector_version = str(os.environ.get("JUGGLER_DETECTOR_VERSION", "master"))
+
detector_path = str(os.environ.get("DETECTOR_PATH", "."))
+# Detector features that affect reconstruction
+has_ecal_barrel_scfi = False
+if 'athena' in detector_name:
+ has_ecal_barrel_scfi = True
+if 'ecce' in detector_name and 'imaging' in detector_config:
+ has_ecal_barrel_scfi = True
+
# input arguments from calibration file
with open(f'{detector_path}/calibrations/emcal_barrel_calibration.json') as f:
calib_data = json.load(f)['electron']
@@ -38,22 +49,38 @@ sf = float(os.environ.get('CB_EMCAL_SAMP_FRAC', '1.0'))
geo_service = GeoSvc("GeoSvc", detectors=kwargs['compact'].split(','), OutputLevel=INFO)
podioevent = EICDataSvc("EventDataSvc", inputs=kwargs['input'].split(','), OutputLevel=DEBUG)
-podin = PodioInput("PodioReader", collections=["MCParticles", "EcalBarrelScFiHits", "EcalBarrelScFiHitsContributions"], OutputLevel=DEBUG)
+sim_coll = [
+ "MCParticles",
+]
+if has_ecal_barrel_scfi:
+ sim_coll += [
+ "EcalBarrelHits",
+ "EcalBarrelHitsContributions",
+ ]
+
+algorithms = []
+
+podin = PodioInput("PodioReader", collections=sim_coll, OutputLevel=DEBUG)
+algorithms.append(podin)
+
podout = PodioOutput("out", filename=kwargs['output'])
+algorithms.append(podout)
-# use the same daq_setting for digi/reco pair
-scfi_barrel_daq = dict(
+if has_ecal_barrel_scfi:
+ # use the same daq_setting for digi/reco pair
+ scfi_barrel_daq = dict(
dynamicRangeADC=50.*MeV,
capacityADC=32768,
pedestalMean=400,
pedestalSigma=10)
-scfi_barrel_digi = CalHitDigi("scfi_barrel_digi",
+ scfi_barrel_digi = CalHitDigi("scfi_barrel_digi",
inputHitCollection="EcalBarrelScFiHits",
outputHitCollection="EcalBarrelScFiHitsDigi",
**scfi_barrel_daq)
+ algorithms.append(scfi_barrel_digi)
-scfi_barrel_reco = CalHitReco("scfi_barrel_reco",
+ scfi_barrel_reco = CalHitReco("scfi_barrel_reco",
inputHitCollection=scfi_barrel_digi.outputHitCollection,
outputHitCollection="EcalBarrelScFiHitsReco",
thresholdFactor=5.0,
@@ -62,31 +89,34 @@ scfi_barrel_reco = CalHitReco("scfi_barrel_reco",
sectorField="module",
localDetFields=["system", "module"], # use local coordinates in each module (stave)
**scfi_barrel_daq)
+ algorithms.append(scfi_barrel_reco)
-# merge hits in different layer (projection to local x-y plane)
-scfi_barrel_merger = CalHitsMerger("scfi_barrel_merger",
+ # merge hits in different layer (projection to local x-y plane)
+ scfi_barrel_merger = CalHitsMerger("scfi_barrel_merger",
# OutputLevel=DEBUG,
inputHitCollection=scfi_barrel_reco.outputHitCollection,
outputHitCollection="EcalBarrelScFiGridReco",
fields=["fiber"],
fieldRefNumbers=[1],
readoutClass="EcalBarrelScFiHits")
+ algorithms.append(scfi_barrel_merger)
-scfi_barrel_cl = IslandCluster("scfi_barrel_cl",
+ scfi_barrel_cl = IslandCluster("scfi_barrel_cl",
# OutputLevel=DEBUG,
inputHitCollection=scfi_barrel_reco.outputHitCollection,
outputProtoClusterCollection="EcalBarrelScFiProtoClusters",
splitCluster=False,
minClusterCenterEdep=10.*MeV,
localDistXZ=[30*mm, 30*mm])
+ algorithms.append(scfi_barrel_cl)
-scfi_barrel_clreco = RecoCoG("scfi_barrel_clreco",
+ scfi_barrel_clreco = RecoCoG("scfi_barrel_clreco",
inputProtoClusterCollection=scfi_barrel.outputProtoClusterCollection,
outputClusterCollection="EcalBarrelScFiClusters",
mcHits="EcalBarrelScFiHits",
logWeightBase=6.2,
samplingFraction=kwargs['sf'])
-
+ algorithms.append(scfi_barrel_clreco)
podout.outputCommands = ["keep *"]
--
GitLab