Myrama

benchmarks/clustering/options/hcal_clustering.py

@@ -45,7 +45,7 @@ from Configurables import Jug__Reco__TopologicalCellCluster as TopoCluster
 from Configurables import Jug__Reco__CalorimeterIslandCluster as IslandCluster
 from Configurables import Jug__Reco__ClusterRecoCoG as RecoCoG
 
-podioinput = PodioInput("PodioReader", collections=["mcparticles","HcalBarrelHits"], OutputLevel=DEBUG)
+podioinput = PodioInput("PodioReader", collections=["mcparticles","HcalBarrelHits","HcalEndcapHits"], OutputLevel=DEBUG)
 
 ## copiers to get around input --> output copy bug. Note the "2" appended to the output collection.
 copier = MCCopier("MCCopier", 
@@ -70,9 +70,24 @@ hcaldigi = HadCalorimeterDigi("hcal_barrel_digi",
          dynamicRangeADC=50.*units.MeV,
          capacityADC=32768,
          pedestalMean=400,
-         pedestalSigma=10,
+         pedestalSigma=5,
          OutputLevel=DEBUG                     
          )
+endcap_hcaldigi = HadCalorimeterDigi("hcal_endcap_digi",
+         inputHitCollection="HcalEndcapHits",
+         outputHitCollection="RawHcalEndcapHits",
+         inputEnergyUnit=units.GeV,
+         inputTimeUnit=units.ns,
+         energyResolutions=[0.0, 0., 0.],
+	 dynamicRangeADC=50.*units.MeV,
+         capacityADC=32768,
+         pedestalMean=400,
+         pedestalSigma=5,
+         OutputLevel=DEBUG
+         )
+
+
+
 
 #reconstructed hits
 
@@ -82,23 +97,50 @@ hcal_reco = HCalReconstruction("hcal_reco",
         dynamicRangeADC=50.*units.MeV,
         capacityADC=32768,
         pedestalMean=400,
-        pedestalSigma=10,
-        thresholdFactor=5.0,
+        pedestalSigma=5,
+        thresholdFactor=2.0,
         OutputLevel=DEBUG)
 
+endcaphcal_reco = HCalReconstruction("endcaphcal_reco",
+        inputHitCollection="RawHcalEndcapHits",
+        outputHitCollection="RecHcalEndcapHits",
+        dynamicRangeADC=50.*units.MeV,
+        capacityADC=32768,
+	pedestalMean=400,
+	pedestalSigma=5,
+	thresholdFactor=2.0,
+	OutputLevel=DEBUG)
+
+
 #clusters
-xymerger = CalorimeterHitsMerger("hcal_xy_merger",                             
+xymerger = CalorimeterHitsMerger("hcal_xy_merger",                            # 
           fields=["layer", "slice"],
           inputHitCollection="RecHcalBarrelHits",
-          outputHitCollection="RecHcalBarrelHitsXY")
+          outputHitCollection="RecHcalBarrelHitsXY",
+          OutputLevel=DEBUG)                       
+
+endcap_xymerger = CalorimeterHitsMerger("endcaphcal_xy_merger",                            #\
+                                                                                
+          fields=["layer", "slice"],
+          inputHitCollection="RecHcalEndcapHits",
+	  outputHitCollection="RecHcalEndcapHitsXY",
+	  OutputLevel=DEBUG)
+
 
 hcal_barrel_cluster = IslandCluster("hcal_barrel_cluster", 
-        inputHitCollection="RecHcalBarrelHitsXY", 
+        inputHitCollection="RecHcalBarrelHits", 
         outputClusterCollection="HcalBarrelClusters",
         minClusterCenterEdep=30*units.MeV, 
         groupRange=2.0,
         OutputLevel=DEBUG)
 
+hcal_endcap_cluster = IslandCluster("hcal_endcap_cluster",
+        inputHitCollection="RecHcalEndcapHits",
+        outputClusterCollection="HcalEndcapClusters",
+        minClusterCenterEdep=30*units.MeV,
+        groupRange=2.0,
+	OutputLevel=DEBUG)
+
 
 # finalizing clustering (center-of-gravity step)
 
@@ -108,6 +150,13 @@ hcal_barrel_clusterreco = RecoCoG("hcal_barrel_clusterreco",
         samplingFraction=sf,
         OutputLevel=DEBUG) 
 
+hcal_endcap_clusterreco = RecoCoG("hcal_endcap_clusterreco",
+        clusterCollection="HcalEndcapClusters",
+	logWeightBase=6.2,
+	samplingFraction=sf,
+        OutputLevel=DEBUG)
+
+
 out = PodioOutput("out", filename=output_rec_file)
 
 out.outputCommands = ["keep *"]
@@ -115,10 +164,15 @@ out.outputCommands = ["keep *"]
 ApplicationMgr(
     TopAlg = [podioinput, copier, calcopier,
               hcaldigi,
+              endcap_hcaldigi,
               hcal_reco,
+              endcaphcal_reco,
               xymerger,
+              endcap_xymerger,
               hcal_barrel_cluster,
+              #hcal_endcap_cluster,
               hcal_barrel_clusterreco,
+              #hcal_endcap_clusterreco,
               out],
     EvtSel = 'NONE',
     EvtMax   = n_events,