fix: black on options files

Any consistent formatting is better than no formatting.

options/reconstruction.ecal.py

@@ -7,28 +7,30 @@ from GaudiKernel.SystemOfUnits import eV, MeV, GeV, mm, cm, mrad
 import json
 
 detector_name = "athena"
-if "JUGGLER_DETECTOR" in os.environ :
+if "JUGGLER_DETECTOR" in os.environ:
     detector_name = str(os.environ["JUGGLER_DETECTOR"])
 
 detector_config = detector_name
-if "JUGGLER_DETECTOR_CONFIG" in os.environ :
+if "JUGGLER_DETECTOR_CONFIG" in os.environ:
     detector_config = str(os.environ["JUGGLER_DETECTOR_CONFIG"])
 
 detector_path = ""
-if "DETECTOR_PATH" in os.environ :
-  detector_path = str(os.environ["DETECTOR_PATH"])
+if "DETECTOR_PATH" in os.environ:
+    detector_path = str(os.environ["DETECTOR_PATH"])
 
-detector_version = 'default'
+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'
+    if "acadia" in env_version:
+        detector_version = "acadia"
 
 # Detector features that affect reconstruction
 has_ecal_barrel_scfi = False
-if 'athena' in detector_name:
+if "athena" in detector_name:
     has_ecal_barrel_scfi = True
-if 'ecce' in detector_name and 'imaging' in detector_config:
+if "ecce" in detector_name and "imaging" in detector_config:
+    has_ecal_barrel_scfi = True
+if "epic" in detector_name and "imaging" in detector_config:
     has_ecal_barrel_scfi = True
 
 # CAL reconstruction
@@ -37,28 +39,38 @@ ci_ecal_sf = float(os.environ.get("CI_ECAL_SAMP_FRAC", 0.253))
 
 # input calorimeter DAQ info
 calo_daq = {}
-with open('{}/calibrations/calo_digi_{}.json'.format(detector_path, detector_version)) as f:
+with open(
+    "{}/calibrations/calo_digi_{}.json".format(detector_path, detector_version)
+) as f:
     calo_config = json.load(f)
     ## add proper ADC capacity based on bit depth
     for sys in calo_config:
         cfg = calo_config[sys]
         calo_daq[sys] = {
-            'dynamicRangeADC': eval(cfg['dynamicRange']),
-            'capacityADC': 2**int(cfg['capacityBitsADC']),
-            'pedestalMean': int(cfg['pedestalMean']),
-            'pedestalSigma': float(cfg['pedestalSigma'])
+            "dynamicRangeADC": eval(cfg["dynamicRange"]),
+            "capacityADC": 2 ** int(cfg["capacityBitsADC"]),
+            "pedestalMean": int(cfg["pedestalMean"]),
+            "pedestalSigma": float(cfg["pedestalSigma"]),
         }
 print(calo_daq)
 
 # input and output
-input_sims = [f.strip() for f in str.split(os.environ["JUGGLER_SIM_FILE"], ",") if f.strip()]
+input_sims = [
+    f.strip() for f in str.split(os.environ["JUGGLER_SIM_FILE"], ",") if f.strip()
+]
 output_rec = str(os.environ["JUGGLER_REC_FILE"])
 n_events = int(os.environ["JUGGLER_N_EVENTS"])
 
 # services
 services = []
 # geometry service
-services.append(GeoSvc("GeoSvc", detectors=["{}/{}.xml".format(detector_path,detector_config)], OutputLevel=WARNING))
+services.append(
+    GeoSvc(
+        "GeoSvc",
+        detectors=["{}/{}.xml".format(detector_path, detector_config)],
+        OutputLevel=WARNING,
+    )
+)
 # data service
 services.append(EICDataSvc("EventDataSvc", inputs=input_sims, OutputLevel=WARNING))
 
@@ -70,15 +82,16 @@ from Configurables import Jug__Reco__CalorimeterHitsMerger as CalHitsMerger
 from Configurables import Jug__Reco__CalorimeterIslandCluster as IslandCluster
 from Configurables import Jug__Reco__ClusterRecoCoG as RecoCoG
 from Configurables import Jug__Fast__TruthClustering as TruthClustering
-#from Configurables import Jug__Fast__ClusterMerger as ClusterMerger
+
+# from Configurables import Jug__Fast__ClusterMerger as ClusterMerger
 
 # branches needed from simulation root file
 sim_coll = [
     "MCParticles",
-    'EcalEndcapNHits',
-    'EcalEndcapNHitsContributions',
-    'EcalEndcapPHits',
-    'EcalEndcapPHitsContributions',
+    "EcalEndcapNHits",
+    "EcalEndcapNHitsContributions",
+    "EcalEndcapPHits",
+    "EcalEndcapPHitsContributions",
 ]
 
 # list of algorithms
@@ -89,29 +102,35 @@ podin = PodioInput("PodioReader", collections=sim_coll)
 algorithms.append(podin)
 
 # Crystal Endcap Ecal
-ce_ecal_daq = calo_daq['ecal_neg_endcap']
-ce_ecal_digi = CalHitDigi("ce_ecal_digi",
-        inputHitCollection="EcalEndcapNHits",
-        outputHitCollection="EcalEndcapNRawHits",
-        energyResolutions=[0., 0.02, 0.],
-        **ce_ecal_daq)
+ce_ecal_daq = calo_daq["ecal_neg_endcap"]
+ce_ecal_digi = CalHitDigi(
+    "ce_ecal_digi",
+    inputHitCollection="EcalEndcapNHits",
+    outputHitCollection="EcalEndcapNRawHits",
+    energyResolutions=[0.0, 0.02, 0.0],
+    **ce_ecal_daq
+)
 algorithms.append(ce_ecal_digi)
 
-ce_ecal_reco = CalHitReco("ce_ecal_reco",
-        inputHitCollection=ce_ecal_digi.outputHitCollection,
-        outputHitCollection="EcalEndcapNRecHits",
-        thresholdFactor=4,          # 4 sigma cut on pedestal sigma
-        samplingFraction=0.998,      # this accounts for a small fraction of leakage
-        readoutClass="EcalEndcapNHits",
-        sectorField="sector",
-        **ce_ecal_daq)
+ce_ecal_reco = CalHitReco(
+    "ce_ecal_reco",
+    inputHitCollection=ce_ecal_digi.outputHitCollection,
+    outputHitCollection="EcalEndcapNRecHits",
+    thresholdFactor=4,  # 4 sigma cut on pedestal sigma
+    samplingFraction=0.998,  # this accounts for a small fraction of leakage
+    readoutClass="EcalEndcapNHits",
+    sectorField="sector",
+    **ce_ecal_daq
+)
 algorithms.append(ce_ecal_reco)
 
-ce_ecal_cl = TruthClustering("ce_ecal_cl",
-        inputHits=ce_ecal_reco.outputHitCollection,
-        mcHits="EcalEndcapNHits",
-        outputProtoClusters="EcalEndcapNProtoClusters")
-#ce_ecal_cl = IslandCluster("ce_ecal_cl",
+ce_ecal_cl = TruthClustering(
+    "ce_ecal_cl",
+    inputHits=ce_ecal_reco.outputHitCollection,
+    mcHits="EcalEndcapNHits",
+    outputProtoClusters="EcalEndcapNProtoClusters",
+)
+# ce_ecal_cl = IslandCluster("ce_ecal_cl",
 #        inputHitCollection=ce_ecal_reco.outputHitCollection,
 #        outputProtoClusterCollection="EcalEndcapNProtoClusters",
 #        splitCluster=False,
@@ -121,92 +140,104 @@ ce_ecal_cl = TruthClustering("ce_ecal_cl",
 #        dimScaledLocalDistXY=[1.8, 1.8]) # dimension scaled dist is good for hybrid sectors with different module size
 algorithms.append(ce_ecal_cl)
 
-ce_ecal_clreco = RecoCoG("ce_ecal_clreco",
-        inputProtoClusterCollection=ce_ecal_cl.outputProtoClusters,
-        outputClusterCollection="EcalEndcapNClusters",
-        logWeightBase=4.6)
+ce_ecal_clreco = RecoCoG(
+    "ce_ecal_clreco",
+    inputProtoClusterCollection=ce_ecal_cl.outputProtoClusters,
+    outputClusterCollection="EcalEndcapNClusters",
+    logWeightBase=4.6,
+)
 algorithms.append(ce_ecal_clreco)
 
-#ce_ecal_clmerger = ClusterMerger("ce_ecal_clmerger",
+# ce_ecal_clmerger = ClusterMerger("ce_ecal_clmerger",
 #        inputClusters = ce_ecal_clreco.outputClusterCollection,
 #        outputClusters = "EcalEndcapNMergedClusters",
 #        outputRelations = "EcalEndcapNMergedClusterRelations")
-#algorithms.append(ce_ecal_clmerger)
+# algorithms.append(ce_ecal_clmerger)
 
 # Endcap ScFi Ecal
-ci_ecal_daq = calo_daq['ecal_pos_endcap']
-
-ci_ecal_digi = CalHitDigi("ci_ecal_digi",
-        inputHitCollection="EcalEndcapPHits",
-        outputHitCollection="EcalEndcapPRawHits",
-        scaleResponse=ci_ecal_sf,
-        energyResolutions=[.1, .0015, 0.],
-        **ci_ecal_daq)
+ci_ecal_daq = calo_daq["ecal_pos_endcap"]
+
+ci_ecal_digi = CalHitDigi(
+    "ci_ecal_digi",
+    inputHitCollection="EcalEndcapPHits",
+    outputHitCollection="EcalEndcapPRawHits",
+    scaleResponse=ci_ecal_sf,
+    energyResolutions=[0.1, 0.0015, 0.0],
+    **ci_ecal_daq
+)
 algorithms.append(ci_ecal_digi)
 
-ci_ecal_reco = CalHitReco("ci_ecal_reco",
-        inputHitCollection=ci_ecal_digi.outputHitCollection,
-        outputHitCollection="EcalEndcapPRecHits",
-        thresholdFactor=5.0,
-        samplingFraction=ci_ecal_sf,
-        **ci_ecal_daq)
+ci_ecal_reco = CalHitReco(
+    "ci_ecal_reco",
+    inputHitCollection=ci_ecal_digi.outputHitCollection,
+    outputHitCollection="EcalEndcapPRecHits",
+    thresholdFactor=5.0,
+    samplingFraction=ci_ecal_sf,
+    **ci_ecal_daq
+)
 algorithms.append(ci_ecal_reco)
 
 # merge hits in different layer (projection to local x-y plane)
-ci_ecal_merger = CalHitsMerger("ci_ecal_merger",
-        inputHitCollection=ci_ecal_reco.outputHitCollection,
-        outputHitCollection="EcalEndcapPRecMergedHits",
-        fields=["fiber_x", "fiber_y"],
-        fieldRefNumbers=[1, 1],
-        # fields=["layer", "slice"],
-        # fieldRefNumbers=[1, 0],
-        readoutClass="EcalEndcapPHits")
+ci_ecal_merger = CalHitsMerger(
+    "ci_ecal_merger",
+    inputHitCollection=ci_ecal_reco.outputHitCollection,
+    outputHitCollection="EcalEndcapPRecMergedHits",
+    fields=["fiber_x", "fiber_y"],
+    fieldRefNumbers=[1, 1],
+    # fields=["layer", "slice"],
+    # fieldRefNumbers=[1, 0],
+    readoutClass="EcalEndcapPHits",
+)
 algorithms.append(ci_ecal_merger)
 
-ci_ecal_cl = TruthClustering("ci_ecal_cl",
-        inputHits=ci_ecal_reco.outputHitCollection,
-        mcHits="EcalEndcapPHits",
-        outputProtoClusters="EcalEndcapPProtoClusters")
-#ci_ecal_cl = IslandCluster("ci_ecal_cl",
-        #inputHitCollection=ci_ecal_merger.outputHitCollection,
-        #outputProtoClusterCollection="EcalEndcapPProtoClusters",
-        #splitCluster=False,
-        #minClusterCenterEdep=10.*MeV,
-        #localDistXY=[10*mm, 10*mm])
+ci_ecal_cl = TruthClustering(
+    "ci_ecal_cl",
+    inputHits=ci_ecal_reco.outputHitCollection,
+    mcHits="EcalEndcapPHits",
+    outputProtoClusters="EcalEndcapPProtoClusters",
+)
+# ci_ecal_cl = IslandCluster("ci_ecal_cl",
+# inputHitCollection=ci_ecal_merger.outputHitCollection,
+# outputProtoClusterCollection="EcalEndcapPProtoClusters",
+# splitCluster=False,
+# minClusterCenterEdep=10.*MeV,
+# localDistXY=[10*mm, 10*mm])
 algorithms.append(ci_ecal_cl)
 
-ci_ecal_clreco = RecoCoG("ci_ecal_clreco",
-        inputProtoClusterCollection=ci_ecal_cl.outputProtoClusters,
-        outputClusterCollection="EcalEndcapPClusters",
-        enableEtaBounds=True,
-        logWeightBase=6.2)
+ci_ecal_clreco = RecoCoG(
+    "ci_ecal_clreco",
+    inputProtoClusterCollection=ci_ecal_cl.outputProtoClusters,
+    outputClusterCollection="EcalEndcapPClusters",
+    enableEtaBounds=True,
+    logWeightBase=6.2,
+)
 algorithms.append(ci_ecal_clreco)
 
-#ci_ecal_clmerger = ClusterMerger("ci_ecal_clmerger",
+# ci_ecal_clmerger = ClusterMerger("ci_ecal_clmerger",
 #        inputClusters = ci_ecal_clreco.outputClusterCollection,
 #        outputClusters = "EcalEndcapPMergedClusters",
 #        outputRelations = "EcalEndcapPMergedClusterRelations")
-#algorithms.append(ci_ecal_clmerger)
+# algorithms.append(ci_ecal_clmerger)
 
 # Output
 podout = PodioOutput("out", filename=output_rec)
 podout.outputCommands = [
-        "keep *",
-        "drop *Hits",
-        "keep *RecHits",
-        "keep *Layers",
-        "keep *Clusters",
-        "drop *ProtoClusters",
-        "drop outputParticles",
-        "drop InitTrackParams",
-        ] + [ "drop " + c for c in sim_coll]
+    "keep *",
+    "drop *Hits",
+    "keep *RecHits",
+    "keep *Layers",
+    "keep *Clusters",
+    "drop *ProtoClusters",
+    "drop outputParticles",
+    "drop InitTrackParams",
+] + ["drop " + c for c in sim_coll]
 algorithms.append(podout)
 
 ApplicationMgr(
-    TopAlg = algorithms,
-    EvtSel = 'NONE',
-    EvtMax = n_events,
-    ExtSvc = services,
-    OutputLevel = WARNING,
-    AuditAlgorithms = True
- )
+    TopAlg=algorithms,
+    EvtSel="NONE",
+    EvtMax=n_events,
+    ExtSvc=services,
+    OutputLevel=WARNING,
+    AuditAlgorithms=True,
+)