Skip to content
Snippets Groups Projects
Commit fae76bb0 authored by Wouter Deconinck's avatar Wouter Deconinck
Browse files

Reenable algorithms again

parent 98a28813
No related branches found
No related tags found
1 merge request!139edm4hep updates
This commit is part of merge request !139. Comments created here will be created in the context of that merge request.
...@@ -167,29 +167,29 @@ from Configurables import Jug__Reco__ParticleCollector as ParticleCollector ...@@ -167,29 +167,29 @@ from Configurables import Jug__Reco__ParticleCollector as ParticleCollector
# branches needed from simulation root file # branches needed from simulation root file
sim_coll = [ sim_coll = [
'MCParticles', 'MCParticles',
# 'B0TrackerHits', 'B0TrackerHits',
# 'ForwardRomanPotHits', 'ForwardRomanPotHits',
# 'ForwardOffMTrackerHits', 'ForwardOffMTrackerHits',
# 'EcalEndcapNHits', 'EcalEndcapNHits',
# 'EcalEndcapPHits', 'EcalEndcapPHits',
# 'EcalBarrelHits', 'EcalBarrelHits',
# 'EcalBarrelScFiHits', 'EcalBarrelScFiHits',
# 'HcalBarrelHits', 'HcalBarrelHits',
# 'HcalEndcapPHits', 'HcalEndcapPHits',
# 'HcalEndcapNHits', 'HcalEndcapNHits',
# 'TrackerEndcapHits', 'TrackerEndcapHits',
# 'TrackerBarrelHits', 'TrackerBarrelHits',
# 'GEMTrackerEndcapHits', 'GEMTrackerEndcapHits',
# 'VertexBarrelHits', 'VertexBarrelHits',
# 'DRICHHits', 'DRICHHits',
# 'ZDCEcalHits', 'ZDCEcalHits',
# 'ZDCHcalHits', 'ZDCHcalHits',
] ]
#if 'acadia' in detector_version: if 'acadia' in detector_version:
# sim_coll.append('VertexEndcapHits') sim_coll.append('VertexEndcapHits')
# sim_coll.append('MRICHHits') sim_coll.append('MRICHHits')
#else: else:
# sim_coll.append('MPGDTrackerBarrelHits') sim_coll.append('MPGDTrackerBarrelHits')
# list of algorithms # list of algorithms
algorithms = [] algorithms = []
...@@ -217,65 +217,65 @@ ffi_romanpot_digi = TrackerDigi("ffi_romanpot_digi", ...@@ -217,65 +217,65 @@ ffi_romanpot_digi = TrackerDigi("ffi_romanpot_digi",
inputHitCollection = "ForwardRomanPotHits", inputHitCollection = "ForwardRomanPotHits",
outputHitCollection = "ForwardRomanPotRawHits", outputHitCollection = "ForwardRomanPotRawHits",
timeResolution = 8) timeResolution = 8)
#algorithms.append(ffi_romanpot_digi) algorithms.append(ffi_romanpot_digi)
ffi_romanpot_reco = TrackerHitReconstruction("ffi_romanpot_reco", ffi_romanpot_reco = TrackerHitReconstruction("ffi_romanpot_reco",
inputHitCollection = ffi_romanpot_digi.outputHitCollection, inputHitCollection = ffi_romanpot_digi.outputHitCollection,
outputHitCollection = "ForwardRomanPotRecHits") outputHitCollection = "ForwardRomanPotRecHits")
#algorithms.append(ffi_romanpot_reco) algorithms.append(ffi_romanpot_reco)
ffi_romanpot_parts = FarForwardParticles("ffi_romanpot_parts", ffi_romanpot_parts = FarForwardParticles("ffi_romanpot_parts",
inputCollection = ffi_romanpot_reco.outputHitCollection, inputCollection = ffi_romanpot_reco.outputHitCollection,
outputCollection = "ForwardRomanPotParticles") outputCollection = "ForwardRomanPotParticles")
#algorithms.append(ffi_romanpot_parts) algorithms.append(ffi_romanpot_parts)
## Off momentum tracker ## Off momentum tracker
ffi_offmtracker_digi = TrackerDigi("ffi_offmtracker_digi", ffi_offmtracker_digi = TrackerDigi("ffi_offmtracker_digi",
inputHitCollection = "ForwardOffMTrackerHits", inputHitCollection = "ForwardOffMTrackerHits",
outputHitCollection = "ForwardOffMTrackerRawHits", outputHitCollection = "ForwardOffMTrackerRawHits",
timeResolution = 8) timeResolution = 8)
#algorithms.append(ffi_offmtracker_digi) algorithms.append(ffi_offmtracker_digi)
ffi_offmtracker_reco = TrackerHitReconstruction("ffi_offmtracker_reco", ffi_offmtracker_reco = TrackerHitReconstruction("ffi_offmtracker_reco",
inputHitCollection = ffi_offmtracker_digi.outputHitCollection, inputHitCollection = ffi_offmtracker_digi.outputHitCollection,
outputHitCollection = "ForwardOffMTrackerRecHits") outputHitCollection = "ForwardOffMTrackerRecHits")
#algorithms.append(ffi_offmtracker_reco) algorithms.append(ffi_offmtracker_reco)
ffi_offmtracker_parts = FarForwardParticles("ffi_offmtracker_parts", ffi_offmtracker_parts = FarForwardParticles("ffi_offmtracker_parts",
inputCollection = ffi_offmtracker_reco.outputHitCollection, inputCollection = ffi_offmtracker_reco.outputHitCollection,
outputCollection = "ForwardOffMTrackerParticles") outputCollection = "ForwardOffMTrackerParticles")
#algorithms.append(ffi_offmtracker_parts) algorithms.append(ffi_offmtracker_parts)
## B0 tracker ## B0 tracker
trk_b0_digi = TrackerDigi("trk_b0_digi", trk_b0_digi = TrackerDigi("trk_b0_digi",
inputHitCollection="B0TrackerHits", inputHitCollection="B0TrackerHits",
outputHitCollection="B0TrackerRawHits", outputHitCollection="B0TrackerRawHits",
timeResolution=8) timeResolution=8)
#algorithms.append(trk_b0_digi) algorithms.append(trk_b0_digi)
trk_b0_reco = TrackerHitReconstruction("trk_b0_reco", trk_b0_reco = TrackerHitReconstruction("trk_b0_reco",
inputHitCollection = trk_b0_digi.outputHitCollection, inputHitCollection = trk_b0_digi.outputHitCollection,
outputHitCollection="B0TrackerRecHits") outputHitCollection="B0TrackerRecHits")
#algorithms.append(trk_b0_reco) algorithms.append(trk_b0_reco)
# ZDC ECAL WSciFi # ZDC ECAL WSciFi
ffi_zdc_ecal_digi = CalHitDigi('ffi_zdc_ecal_digi', ffi_zdc_ecal_digi = CalHitDigi('ffi_zdc_ecal_digi',
inputHitCollection = 'ZDCEcalHits', inputHitCollection = 'ZDCEcalHits',
outputHitCollection = 'ZDCEcalRawHits') outputHitCollection = 'ZDCEcalRawHits')
#algorithms.append(ffi_zdc_ecal_digi) algorithms.append(ffi_zdc_ecal_digi)
ffi_zdc_ecal_reco = CalHitReco('ffi_zdc_ecal_reco', ffi_zdc_ecal_reco = CalHitReco('ffi_zdc_ecal_reco',
inputHitCollection = ffi_zdc_ecal_digi.outputHitCollection, inputHitCollection = ffi_zdc_ecal_digi.outputHitCollection,
outputHitCollection = 'ZDCEcalRecHits', outputHitCollection = 'ZDCEcalRecHits',
readoutClass = 'ZDCEcalHits', readoutClass = 'ZDCEcalHits',
localDetFields = ['system']) localDetFields = ['system'])
#algorithms.append(ffi_zdc_ecal_reco) algorithms.append(ffi_zdc_ecal_reco)
ffi_zdc_ecal_cl = IslandCluster('ffi_zdc_ecal_cl', ffi_zdc_ecal_cl = IslandCluster('ffi_zdc_ecal_cl',
inputHitCollection = ffi_zdc_ecal_reco.outputHitCollection, inputHitCollection = ffi_zdc_ecal_reco.outputHitCollection,
outputProtoClusterCollection = 'ZDCEcalProtoClusters', outputProtoClusterCollection = 'ZDCEcalProtoClusters',
**ffi_zdc_ecal_cl_kwargs) **ffi_zdc_ecal_cl_kwargs)
#algorithms.append(ffi_zdc_ecal_cl) algorithms.append(ffi_zdc_ecal_cl)
ffi_zdc_ecal_clreco = RecoCoG('ffi_zdc_ecal_clreco', ffi_zdc_ecal_clreco = RecoCoG('ffi_zdc_ecal_clreco',
inputHitCollection = ffi_zdc_ecal_cl.inputHitCollection, inputHitCollection = ffi_zdc_ecal_cl.inputHitCollection,
...@@ -284,26 +284,26 @@ ffi_zdc_ecal_clreco = RecoCoG('ffi_zdc_ecal_clreco', ...@@ -284,26 +284,26 @@ ffi_zdc_ecal_clreco = RecoCoG('ffi_zdc_ecal_clreco',
mcHits = "ZDCEcalHits", mcHits = "ZDCEcalHits",
logWeightBase = 3.6, logWeightBase = 3.6,
samplingFraction = ffi_zdc_ecal_sf) samplingFraction = ffi_zdc_ecal_sf)
#algorithms.append(ffi_zdc_ecal_clreco) algorithms.append(ffi_zdc_ecal_clreco)
# ZDC HCAL PbSciFi # ZDC HCAL PbSciFi
ffi_zdc_hcal_digi = CalHitDigi('ffi_zdc_hcal_digi', ffi_zdc_hcal_digi = CalHitDigi('ffi_zdc_hcal_digi',
inputHitCollection = 'ZDCHcalHits', inputHitCollection = 'ZDCHcalHits',
outputHitCollection = 'ZDCHcalRawHits') outputHitCollection = 'ZDCHcalRawHits')
#algorithms.append(ffi_zdc_hcal_digi) algorithms.append(ffi_zdc_hcal_digi)
ffi_zdc_hcal_reco = CalHitReco('ffi_zdc_hcal_reco', ffi_zdc_hcal_reco = CalHitReco('ffi_zdc_hcal_reco',
inputHitCollection = ffi_zdc_hcal_digi.outputHitCollection, inputHitCollection = ffi_zdc_hcal_digi.outputHitCollection,
outputHitCollection = 'ZDCHcalRecHits', outputHitCollection = 'ZDCHcalRecHits',
readoutClass = 'ZDCHcalHits', readoutClass = 'ZDCHcalHits',
localDetFields = ['system']) localDetFields = ['system'])
#algorithms.append(ffi_zdc_hcal_reco) algorithms.append(ffi_zdc_hcal_reco)
ffi_zdc_hcal_cl = IslandCluster('ffi_zdc_hcal_cl', ffi_zdc_hcal_cl = IslandCluster('ffi_zdc_hcal_cl',
inputHitCollection = ffi_zdc_hcal_reco.outputHitCollection, inputHitCollection = ffi_zdc_hcal_reco.outputHitCollection,
outputProtoClusterCollection = 'ZDCHcalProtoClusters', outputProtoClusterCollection = 'ZDCHcalProtoClusters',
**ffi_zdc_hcal_cl_kwargs) **ffi_zdc_hcal_cl_kwargs)
#algorithms.append(ffi_zdc_hcal_cl) algorithms.append(ffi_zdc_hcal_cl)
ffi_zdc_hcal_clreco = RecoCoG('ffi_zdc_hcal_clreco', ffi_zdc_hcal_clreco = RecoCoG('ffi_zdc_hcal_clreco',
inputHitCollection = ffi_zdc_hcal_cl.inputHitCollection, inputHitCollection = ffi_zdc_hcal_cl.inputHitCollection,
...@@ -312,7 +312,7 @@ ffi_zdc_hcal_clreco = RecoCoG('ffi_zdc_hcal_clreco', ...@@ -312,7 +312,7 @@ ffi_zdc_hcal_clreco = RecoCoG('ffi_zdc_hcal_clreco',
mcHits = "ZDCHcalHits", mcHits = "ZDCHcalHits",
logWeightBase = 3.6, logWeightBase = 3.6,
samplingFraction = ffi_zdc_hcal_sf) samplingFraction = ffi_zdc_hcal_sf)
#algorithms.append(ffi_zdc_hcal_clreco) algorithms.append(ffi_zdc_hcal_clreco)
# Crystal Endcap Ecal # Crystal Endcap Ecal
ce_ecal_daq = calo_daq['ecal_neg_endcap'] ce_ecal_daq = calo_daq['ecal_neg_endcap']
...@@ -321,7 +321,7 @@ ce_ecal_digi = CalHitDigi("ce_ecal_digi", ...@@ -321,7 +321,7 @@ ce_ecal_digi = CalHitDigi("ce_ecal_digi",
outputHitCollection="EcalEndcapNRawHits", outputHitCollection="EcalEndcapNRawHits",
energyResolutions=[0., 0.02, 0.], energyResolutions=[0., 0.02, 0.],
**ce_ecal_daq) **ce_ecal_daq)
#algorithms.append(ce_ecal_digi) algorithms.append(ce_ecal_digi)
ce_ecal_reco = CalHitReco("ce_ecal_reco", ce_ecal_reco = CalHitReco("ce_ecal_reco",
inputHitCollection=ce_ecal_digi.outputHitCollection, inputHitCollection=ce_ecal_digi.outputHitCollection,
...@@ -331,7 +331,7 @@ ce_ecal_reco = CalHitReco("ce_ecal_reco", ...@@ -331,7 +331,7 @@ ce_ecal_reco = CalHitReco("ce_ecal_reco",
readoutClass="EcalEndcapNHits", readoutClass="EcalEndcapNHits",
sectorField="sector", sectorField="sector",
**ce_ecal_daq) **ce_ecal_daq)
#algorithms.append(ce_ecal_reco) algorithms.append(ce_ecal_reco)
ce_ecal_cl = TruthClustering("ce_ecal_cl", ce_ecal_cl = TruthClustering("ce_ecal_cl",
inputHits=ce_ecal_reco.outputHitCollection, inputHits=ce_ecal_reco.outputHitCollection,
...@@ -345,7 +345,7 @@ ce_ecal_cl = TruthClustering("ce_ecal_cl", ...@@ -345,7 +345,7 @@ ce_ecal_cl = TruthClustering("ce_ecal_cl",
# minClusterCenterEdep=30*units.MeV, # minClusterCenterEdep=30*units.MeV,
# sectorDist=5.0*units.cm, # sectorDist=5.0*units.cm,
# dimScaledLocalDistXY=[1.8, 1.8]) # dimension scaled dist is good for hybrid sectors with different module size # dimScaledLocalDistXY=[1.8, 1.8]) # dimension scaled dist is good for hybrid sectors with different module size
#algorithms.append(ce_ecal_cl) algorithms.append(ce_ecal_cl)
ce_ecal_clreco = RecoCoG("ce_ecal_clreco", ce_ecal_clreco = RecoCoG("ce_ecal_clreco",
#inputHitCollection=ce_ecal_cl.inputHitCollection, #inputHitCollection=ce_ecal_cl.inputHitCollection,
...@@ -355,13 +355,13 @@ ce_ecal_clreco = RecoCoG("ce_ecal_clreco", ...@@ -355,13 +355,13 @@ ce_ecal_clreco = RecoCoG("ce_ecal_clreco",
outputClusterCollection="EcalEndcapNClusters", outputClusterCollection="EcalEndcapNClusters",
mcHits="EcalEndcapNHits", mcHits="EcalEndcapNHits",
logWeightBase=4.6) logWeightBase=4.6)
#algorithms.append(ce_ecal_clreco) algorithms.append(ce_ecal_clreco)
ce_ecal_clmerger = ClusterMerger("ce_ecal_clmerger", ce_ecal_clmerger = ClusterMerger("ce_ecal_clmerger",
inputClusters = ce_ecal_clreco.outputClusterCollection, inputClusters = ce_ecal_clreco.outputClusterCollection,
outputClusters = "EcalEndcapNMergedClusters", outputClusters = "EcalEndcapNMergedClusters",
outputRelations = "EcalEndcapNMergedClusterRelations") outputRelations = "EcalEndcapNMergedClusterRelations")
#algorithms.append(ce_ecal_clmerger) algorithms.append(ce_ecal_clmerger)
# Endcap ScFi Ecal # Endcap ScFi Ecal
ci_ecal_daq = calo_daq['ecal_pos_endcap'] ci_ecal_daq = calo_daq['ecal_pos_endcap']
...@@ -372,7 +372,7 @@ ci_ecal_digi = CalHitDigi("ci_ecal_digi", ...@@ -372,7 +372,7 @@ ci_ecal_digi = CalHitDigi("ci_ecal_digi",
scaleResponse=ci_ecal_sf, scaleResponse=ci_ecal_sf,
energyResolutions=[.1, .0015, 0.], energyResolutions=[.1, .0015, 0.],
**ci_ecal_daq) **ci_ecal_daq)
#algorithms.append(ci_ecal_digi) algorithms.append(ci_ecal_digi)
ci_ecal_reco = CalHitReco("ci_ecal_reco", ci_ecal_reco = CalHitReco("ci_ecal_reco",
inputHitCollection=ci_ecal_digi.outputHitCollection, inputHitCollection=ci_ecal_digi.outputHitCollection,
...@@ -380,7 +380,7 @@ ci_ecal_reco = CalHitReco("ci_ecal_reco", ...@@ -380,7 +380,7 @@ ci_ecal_reco = CalHitReco("ci_ecal_reco",
thresholdFactor=5.0, thresholdFactor=5.0,
samplingFraction=ci_ecal_sf, samplingFraction=ci_ecal_sf,
**ci_ecal_daq) **ci_ecal_daq)
#algorithms.append(ci_ecal_reco) algorithms.append(ci_ecal_reco)
# merge hits in different layer (projection to local x-y plane) # merge hits in different layer (projection to local x-y plane)
ci_ecal_merger = CalHitsMerger("ci_ecal_merger", ci_ecal_merger = CalHitsMerger("ci_ecal_merger",
...@@ -391,7 +391,7 @@ ci_ecal_merger = CalHitsMerger("ci_ecal_merger", ...@@ -391,7 +391,7 @@ ci_ecal_merger = CalHitsMerger("ci_ecal_merger",
# fields=["layer", "slice"], # fields=["layer", "slice"],
# fieldRefNumbers=[1, 0], # fieldRefNumbers=[1, 0],
readoutClass="EcalEndcapPHits") readoutClass="EcalEndcapPHits")
#algorithms.append(ci_ecal_merger) algorithms.append(ci_ecal_merger)
ci_ecal_cl = TruthClustering("ci_ecal_cl", ci_ecal_cl = TruthClustering("ci_ecal_cl",
inputHits=ci_ecal_reco.outputHitCollection, inputHits=ci_ecal_reco.outputHitCollection,
...@@ -403,7 +403,7 @@ ci_ecal_cl = TruthClustering("ci_ecal_cl", ...@@ -403,7 +403,7 @@ ci_ecal_cl = TruthClustering("ci_ecal_cl",
#splitCluster=False, #splitCluster=False,
#minClusterCenterEdep=10.*units.MeV, #minClusterCenterEdep=10.*units.MeV,
#localDistXY=[10*units.mm, 10*units.mm]) #localDistXY=[10*units.mm, 10*units.mm])
#algorithms.append(ci_ecal_cl) algorithms.append(ci_ecal_cl)
ci_ecal_clreco = RecoCoG("ci_ecal_clreco", ci_ecal_clreco = RecoCoG("ci_ecal_clreco",
#inputHitCollection=ci_ecal_cl.inputHitCollection, #inputHitCollection=ci_ecal_cl.inputHitCollection,
...@@ -414,13 +414,13 @@ ci_ecal_clreco = RecoCoG("ci_ecal_clreco", ...@@ -414,13 +414,13 @@ ci_ecal_clreco = RecoCoG("ci_ecal_clreco",
enableEtaBounds=True, enableEtaBounds=True,
mcHits="EcalEndcapPHits", mcHits="EcalEndcapPHits",
logWeightBase=6.2) logWeightBase=6.2)
#algorithms.append(ci_ecal_clreco) algorithms.append(ci_ecal_clreco)
ci_ecal_clmerger = ClusterMerger("ci_ecal_clmerger", ci_ecal_clmerger = ClusterMerger("ci_ecal_clmerger",
inputClusters = ci_ecal_clreco.outputClusterCollection, inputClusters = ci_ecal_clreco.outputClusterCollection,
outputClusters = "EcalEndcapPMergedClusters", outputClusters = "EcalEndcapPMergedClusters",
outputRelations = "EcalEndcapPMergedClusterRelations") outputRelations = "EcalEndcapPMergedClusterRelations")
#algorithms.append(ci_ecal_clmerger) algorithms.append(ci_ecal_clmerger)
# Central Barrel Ecal (Imaging Cal.) # Central Barrel Ecal (Imaging Cal.)
img_barrel_daq = calo_daq['ecal_barrel_imaging'] img_barrel_daq = calo_daq['ecal_barrel_imaging']
...@@ -430,7 +430,7 @@ img_barrel_digi = CalHitDigi("img_barrel_digi", ...@@ -430,7 +430,7 @@ img_barrel_digi = CalHitDigi("img_barrel_digi",
outputHitCollection="EcalBarrelImagingRawHits", outputHitCollection="EcalBarrelImagingRawHits",
energyResolutions=[0., 0.02, 0.], # 2% flat resolution energyResolutions=[0., 0.02, 0.], # 2% flat resolution
**img_barrel_daq) **img_barrel_daq)
#algorithms.append(img_barrel_digi) algorithms.append(img_barrel_digi)
img_barrel_reco = ImCalPixelReco("img_barrel_reco", img_barrel_reco = ImCalPixelReco("img_barrel_reco",
inputHitCollection=img_barrel_digi.outputHitCollection, inputHitCollection=img_barrel_digi.outputHitCollection,
...@@ -441,7 +441,7 @@ img_barrel_reco = ImCalPixelReco("img_barrel_reco", ...@@ -441,7 +441,7 @@ img_barrel_reco = ImCalPixelReco("img_barrel_reco",
layerField="layer", # field to get layer id layerField="layer", # field to get layer id
sectorField="module", # field to get sector id sectorField="module", # field to get sector id
**img_barrel_daq) **img_barrel_daq)
#algorithms.append(img_barrel_reco) algorithms.append(img_barrel_reco)
img_barrel_cl = ImagingCluster("img_barrel_cl", img_barrel_cl = ImagingCluster("img_barrel_cl",
inputHitCollection=img_barrel_reco.outputHitCollection, inputHitCollection=img_barrel_reco.outputHitCollection,
...@@ -450,7 +450,7 @@ img_barrel_cl = ImagingCluster("img_barrel_cl", ...@@ -450,7 +450,7 @@ img_barrel_cl = ImagingCluster("img_barrel_cl",
layerDistEtaPhi=[10*units.mrad, 10*units.mrad], # adjacent layer layerDistEtaPhi=[10*units.mrad, 10*units.mrad], # adjacent layer
neighbourLayersRange=2, # id diff for adjacent layer neighbourLayersRange=2, # id diff for adjacent layer
sectorDist=3.*units.cm) # different sector sectorDist=3.*units.cm) # different sector
#algorithms.append(img_barrel_cl) algorithms.append(img_barrel_cl)
img_barrel_clreco = ImagingClusterReco("img_barrel_clreco", img_barrel_clreco = ImagingClusterReco("img_barrel_clreco",
inputHitCollection=img_barrel_cl.inputHitCollection, inputHitCollection=img_barrel_cl.inputHitCollection,
...@@ -458,7 +458,7 @@ img_barrel_clreco = ImagingClusterReco("img_barrel_clreco", ...@@ -458,7 +458,7 @@ img_barrel_clreco = ImagingClusterReco("img_barrel_clreco",
outputClusterCollection="EcalBarrelImagingClusters", outputClusterCollection="EcalBarrelImagingClusters",
mcHits="EcalBarrelHits", mcHits="EcalBarrelHits",
outputLayerCollection="EcalBarrelImagingLayers") outputLayerCollection="EcalBarrelImagingLayers")
#algorithms.append(img_barrel_clreco) algorithms.append(img_barrel_clreco)
# Central ECAL SciFi # Central ECAL SciFi
scfi_barrel_daq = calo_daq['ecal_barrel_scfi'] scfi_barrel_daq = calo_daq['ecal_barrel_scfi']
...@@ -467,7 +467,7 @@ scfi_barrel_digi = CalHitDigi("scfi_barrel_digi", ...@@ -467,7 +467,7 @@ scfi_barrel_digi = CalHitDigi("scfi_barrel_digi",
inputHitCollection="EcalBarrelScFiHits", inputHitCollection="EcalBarrelScFiHits",
outputHitCollection="EcalBarrelScFiRawHits", outputHitCollection="EcalBarrelScFiRawHits",
**scfi_barrel_daq) **scfi_barrel_daq)
#algorithms.append(scfi_barrel_digi) algorithms.append(scfi_barrel_digi)
scfi_barrel_reco = CalHitReco("scfi_barrel_reco", scfi_barrel_reco = CalHitReco("scfi_barrel_reco",
inputHitCollection=scfi_barrel_digi.outputHitCollection, inputHitCollection=scfi_barrel_digi.outputHitCollection,
...@@ -479,7 +479,7 @@ scfi_barrel_reco = CalHitReco("scfi_barrel_reco", ...@@ -479,7 +479,7 @@ scfi_barrel_reco = CalHitReco("scfi_barrel_reco",
sectorField="module", sectorField="module",
localDetFields=["system", "module"], # use local coordinates in each module (stave) localDetFields=["system", "module"], # use local coordinates in each module (stave)
**scfi_barrel_daq) **scfi_barrel_daq)
#algorithms.append(scfi_barrel_reco) algorithms.append(scfi_barrel_reco)
# merge hits in different layer (projection to local x-y plane) # merge hits in different layer (projection to local x-y plane)
scfi_barrel_merger = CalHitsMerger("scfi_barrel_merger", scfi_barrel_merger = CalHitsMerger("scfi_barrel_merger",
...@@ -488,7 +488,7 @@ scfi_barrel_merger = CalHitsMerger("scfi_barrel_merger", ...@@ -488,7 +488,7 @@ scfi_barrel_merger = CalHitsMerger("scfi_barrel_merger",
fields=["fiber"], fields=["fiber"],
fieldRefNumbers=[1], fieldRefNumbers=[1],
readoutClass="EcalBarrelScFiHits") readoutClass="EcalBarrelScFiHits")
#algorithms.append(scfi_barrel_merger) algorithms.append(scfi_barrel_merger)
scfi_barrel_cl = IslandCluster("scfi_barrel_cl", scfi_barrel_cl = IslandCluster("scfi_barrel_cl",
inputHitCollection=scfi_barrel_merger.outputHitCollection, inputHitCollection=scfi_barrel_merger.outputHitCollection,
...@@ -496,7 +496,7 @@ scfi_barrel_cl = IslandCluster("scfi_barrel_cl", ...@@ -496,7 +496,7 @@ scfi_barrel_cl = IslandCluster("scfi_barrel_cl",
splitCluster=False, splitCluster=False,
minClusterCenterEdep=10.*MeV, minClusterCenterEdep=10.*MeV,
localDistXZ=[30*mm, 30*mm]) localDistXZ=[30*mm, 30*mm])
#algorithms.append(scfi_barrel_cl) algorithms.append(scfi_barrel_cl)
scfi_barrel_clreco = RecoCoG("scfi_barrel_clreco", scfi_barrel_clreco = RecoCoG("scfi_barrel_clreco",
inputHitCollection=scfi_barrel_cl.inputHitCollection, inputHitCollection=scfi_barrel_cl.inputHitCollection,
...@@ -504,7 +504,7 @@ scfi_barrel_clreco = RecoCoG("scfi_barrel_clreco", ...@@ -504,7 +504,7 @@ scfi_barrel_clreco = RecoCoG("scfi_barrel_clreco",
outputClusterCollection="EcalBarrelScFiClusters", outputClusterCollection="EcalBarrelScFiClusters",
mcHits="EcalBarrelScFiHits", mcHits="EcalBarrelScFiHits",
logWeightBase=6.2) logWeightBase=6.2)
#algorithms.append(scfi_barrel_clreco) algorithms.append(scfi_barrel_clreco)
## barrel cluster merger ## barrel cluster merger
barrel_clus_merger = EnergyPositionClusterMerger("barrel_clus_merger", barrel_clus_merger = EnergyPositionClusterMerger("barrel_clus_merger",
...@@ -513,7 +513,7 @@ barrel_clus_merger = EnergyPositionClusterMerger("barrel_clus_merger", ...@@ -513,7 +513,7 @@ barrel_clus_merger = EnergyPositionClusterMerger("barrel_clus_merger",
inputPositionClusters = img_barrel_clreco.outputClusterCollection, inputPositionClusters = img_barrel_clreco.outputClusterCollection,
outputClusters = "EcalBarrelMergedClusters", outputClusters = "EcalBarrelMergedClusters",
outputRelations = "EcalBarrelMergedClusterRelations") outputRelations = "EcalBarrelMergedClusterRelations")
#algorithms.append(barrel_clus_merger) algorithms.append(barrel_clus_merger)
# Central Barrel Hcal # Central Barrel Hcal
...@@ -523,7 +523,7 @@ cb_hcal_digi = CalHitDigi("cb_hcal_digi", ...@@ -523,7 +523,7 @@ cb_hcal_digi = CalHitDigi("cb_hcal_digi",
inputHitCollection="HcalBarrelHits", inputHitCollection="HcalBarrelHits",
outputHitCollection="HcalBarrelRawHits", outputHitCollection="HcalBarrelRawHits",
**cb_hcal_daq) **cb_hcal_daq)
#algorithms.append(cb_hcal_digi) algorithms.append(cb_hcal_digi)
cb_hcal_reco = CalHitReco("cb_hcal_reco", cb_hcal_reco = CalHitReco("cb_hcal_reco",
inputHitCollection=cb_hcal_digi.outputHitCollection, inputHitCollection=cb_hcal_digi.outputHitCollection,
...@@ -534,7 +534,7 @@ cb_hcal_reco = CalHitReco("cb_hcal_reco", ...@@ -534,7 +534,7 @@ cb_hcal_reco = CalHitReco("cb_hcal_reco",
layerField="layer", layerField="layer",
sectorField="module", sectorField="module",
**cb_hcal_daq) **cb_hcal_daq)
#algorithms.append(cb_hcal_reco) algorithms.append(cb_hcal_reco)
cb_hcal_merger = CalHitsMerger("cb_hcal_merger", cb_hcal_merger = CalHitsMerger("cb_hcal_merger",
inputHitCollection=cb_hcal_reco.outputHitCollection, inputHitCollection=cb_hcal_reco.outputHitCollection,
...@@ -542,7 +542,7 @@ cb_hcal_merger = CalHitsMerger("cb_hcal_merger", ...@@ -542,7 +542,7 @@ cb_hcal_merger = CalHitsMerger("cb_hcal_merger",
readoutClass="HcalBarrelHits", readoutClass="HcalBarrelHits",
fields=["layer", "slice"], fields=["layer", "slice"],
fieldRefNumbers=[1, 0]) fieldRefNumbers=[1, 0])
#algorithms.append(cb_hcal_merger) algorithms.append(cb_hcal_merger)
cb_hcal_cl = IslandCluster("cb_hcal_cl", cb_hcal_cl = IslandCluster("cb_hcal_cl",
inputHitCollection=cb_hcal_merger.outputHitCollection, inputHitCollection=cb_hcal_merger.outputHitCollection,
...@@ -550,7 +550,7 @@ cb_hcal_cl = IslandCluster("cb_hcal_cl", ...@@ -550,7 +550,7 @@ cb_hcal_cl = IslandCluster("cb_hcal_cl",
splitCluster=False, splitCluster=False,
minClusterCenterEdep=30.*units.MeV, minClusterCenterEdep=30.*units.MeV,
localDistXY=[15.*units.cm, 15.*units.cm]) localDistXY=[15.*units.cm, 15.*units.cm])
#algorithms.append(cb_hcal_cl) algorithms.append(cb_hcal_cl)
cb_hcal_clreco = RecoCoG("cb_hcal_clreco", cb_hcal_clreco = RecoCoG("cb_hcal_clreco",
inputHitCollection=cb_hcal_cl.inputHitCollection, inputHitCollection=cb_hcal_cl.inputHitCollection,
...@@ -558,7 +558,7 @@ cb_hcal_clreco = RecoCoG("cb_hcal_clreco", ...@@ -558,7 +558,7 @@ cb_hcal_clreco = RecoCoG("cb_hcal_clreco",
outputClusterCollection="HcalBarrelClusters", outputClusterCollection="HcalBarrelClusters",
mcHits="HcalBarrelHits", mcHits="HcalBarrelHits",
logWeightBase=6.2) logWeightBase=6.2)
#algorithms.append(cb_hcal_clreco) algorithms.append(cb_hcal_clreco)
# Hcal Hadron Endcap # Hcal Hadron Endcap
ci_hcal_daq = calo_daq['hcal_pos_endcap'] ci_hcal_daq = calo_daq['hcal_pos_endcap']
...@@ -567,7 +567,7 @@ ci_hcal_digi = CalHitDigi("ci_hcal_digi", ...@@ -567,7 +567,7 @@ ci_hcal_digi = CalHitDigi("ci_hcal_digi",
inputHitCollection="HcalEndcapPHits", inputHitCollection="HcalEndcapPHits",
outputHitCollection="HcalEndcapPRawHits", outputHitCollection="HcalEndcapPRawHits",
**ci_hcal_daq) **ci_hcal_daq)
#algorithms.append(ci_hcal_digi) algorithms.append(ci_hcal_digi)
ci_hcal_reco = CalHitReco("ci_hcal_reco", ci_hcal_reco = CalHitReco("ci_hcal_reco",
inputHitCollection=ci_hcal_digi.outputHitCollection, inputHitCollection=ci_hcal_digi.outputHitCollection,
...@@ -575,7 +575,7 @@ ci_hcal_reco = CalHitReco("ci_hcal_reco", ...@@ -575,7 +575,7 @@ ci_hcal_reco = CalHitReco("ci_hcal_reco",
thresholdFactor=5.0, thresholdFactor=5.0,
samplingFraction=ci_hcal_sf, samplingFraction=ci_hcal_sf,
**ci_hcal_daq) **ci_hcal_daq)
#algorithms.append(ci_hcal_reco) algorithms.append(ci_hcal_reco)
ci_hcal_merger = CalHitsMerger("ci_hcal_merger", ci_hcal_merger = CalHitsMerger("ci_hcal_merger",
inputHitCollection=ci_hcal_reco.outputHitCollection, inputHitCollection=ci_hcal_reco.outputHitCollection,
...@@ -583,7 +583,7 @@ ci_hcal_merger = CalHitsMerger("ci_hcal_merger", ...@@ -583,7 +583,7 @@ ci_hcal_merger = CalHitsMerger("ci_hcal_merger",
readoutClass="HcalEndcapPHits", readoutClass="HcalEndcapPHits",
fields=["layer", "slice"], fields=["layer", "slice"],
fieldRefNumbers=[1, 0]) fieldRefNumbers=[1, 0])
#algorithms.append(ci_hcal_merger) algorithms.append(ci_hcal_merger)
ci_hcal_cl = IslandCluster("ci_hcal_cl", ci_hcal_cl = IslandCluster("ci_hcal_cl",
inputHitCollection=ci_hcal_merger.outputHitCollection, inputHitCollection=ci_hcal_merger.outputHitCollection,
...@@ -591,7 +591,7 @@ ci_hcal_cl = IslandCluster("ci_hcal_cl", ...@@ -591,7 +591,7 @@ ci_hcal_cl = IslandCluster("ci_hcal_cl",
splitCluster=False, splitCluster=False,
minClusterCenterEdep=30.*units.MeV, minClusterCenterEdep=30.*units.MeV,
localDistXY=[15.*units.cm, 15.*units.cm]) localDistXY=[15.*units.cm, 15.*units.cm])
#algorithms.append(ci_hcal_cl) algorithms.append(ci_hcal_cl)
ci_hcal_clreco = RecoCoG("ci_hcal_clreco", ci_hcal_clreco = RecoCoG("ci_hcal_clreco",
inputHitCollection=ci_hcal_cl.inputHitCollection, inputHitCollection=ci_hcal_cl.inputHitCollection,
...@@ -599,7 +599,7 @@ ci_hcal_clreco = RecoCoG("ci_hcal_clreco", ...@@ -599,7 +599,7 @@ ci_hcal_clreco = RecoCoG("ci_hcal_clreco",
outputClusterCollection="HcalEndcapPClusters", outputClusterCollection="HcalEndcapPClusters",
mcHits="HcalEndcapPHits", mcHits="HcalEndcapPHits",
logWeightBase=6.2) logWeightBase=6.2)
#algorithms.append(ci_hcal_clreco) algorithms.append(ci_hcal_clreco)
# Hcal Electron Endcap # Hcal Electron Endcap
ce_hcal_daq = calo_daq['hcal_neg_endcap'] ce_hcal_daq = calo_daq['hcal_neg_endcap']
...@@ -608,7 +608,7 @@ ce_hcal_digi = CalHitDigi("ce_hcal_digi", ...@@ -608,7 +608,7 @@ ce_hcal_digi = CalHitDigi("ce_hcal_digi",
inputHitCollection="HcalEndcapNHits", inputHitCollection="HcalEndcapNHits",
outputHitCollection="HcalEndcapNRawHits", outputHitCollection="HcalEndcapNRawHits",
**ce_hcal_daq) **ce_hcal_daq)
#algorithms.append(ce_hcal_digi) algorithms.append(ce_hcal_digi)
ce_hcal_reco = CalHitReco("ce_hcal_reco", ce_hcal_reco = CalHitReco("ce_hcal_reco",
inputHitCollection=ce_hcal_digi.outputHitCollection, inputHitCollection=ce_hcal_digi.outputHitCollection,
...@@ -616,7 +616,7 @@ ce_hcal_reco = CalHitReco("ce_hcal_reco", ...@@ -616,7 +616,7 @@ ce_hcal_reco = CalHitReco("ce_hcal_reco",
thresholdFactor=5.0, thresholdFactor=5.0,
samplingFraction=ce_hcal_sf, samplingFraction=ce_hcal_sf,
**ce_hcal_daq) **ce_hcal_daq)
#algorithms.append(ce_hcal_reco) algorithms.append(ce_hcal_reco)
ce_hcal_merger = CalHitsMerger("ce_hcal_merger", ce_hcal_merger = CalHitsMerger("ce_hcal_merger",
inputHitCollection=ce_hcal_reco.outputHitCollection, inputHitCollection=ce_hcal_reco.outputHitCollection,
...@@ -624,7 +624,7 @@ ce_hcal_merger = CalHitsMerger("ce_hcal_merger", ...@@ -624,7 +624,7 @@ ce_hcal_merger = CalHitsMerger("ce_hcal_merger",
readoutClass="HcalEndcapNHits", readoutClass="HcalEndcapNHits",
fields=["layer", "slice"], fields=["layer", "slice"],
fieldRefNumbers=[1, 0]) fieldRefNumbers=[1, 0])
#algorithms.append(ce_hcal_merger) algorithms.append(ce_hcal_merger)
ce_hcal_cl = IslandCluster("ce_hcal_cl", ce_hcal_cl = IslandCluster("ce_hcal_cl",
inputHitCollection=ce_hcal_merger.outputHitCollection, inputHitCollection=ce_hcal_merger.outputHitCollection,
...@@ -632,7 +632,7 @@ ce_hcal_cl = IslandCluster("ce_hcal_cl", ...@@ -632,7 +632,7 @@ ce_hcal_cl = IslandCluster("ce_hcal_cl",
splitCluster=False, splitCluster=False,
minClusterCenterEdep=30.*units.MeV, minClusterCenterEdep=30.*units.MeV,
localDistXY=[15.*units.cm, 15.*units.cm]) localDistXY=[15.*units.cm, 15.*units.cm])
#algorithms.append(ce_hcal_cl) algorithms.append(ce_hcal_cl)
ce_hcal_clreco = RecoCoG("ce_hcal_clreco", ce_hcal_clreco = RecoCoG("ce_hcal_clreco",
inputHitCollection=ce_hcal_cl.inputHitCollection, inputHitCollection=ce_hcal_cl.inputHitCollection,
...@@ -640,77 +640,77 @@ ce_hcal_clreco = RecoCoG("ce_hcal_clreco", ...@@ -640,77 +640,77 @@ ce_hcal_clreco = RecoCoG("ce_hcal_clreco",
outputClusterCollection="HcalEndcapNClusters", outputClusterCollection="HcalEndcapNClusters",
mcHits="HcalEndcapNHits", mcHits="HcalEndcapNHits",
logWeightBase=6.2) logWeightBase=6.2)
#algorithms.append(ce_hcal_clreco) algorithms.append(ce_hcal_clreco)
# Tracking # Tracking
trk_b_digi = TrackerDigi("trk_b_digi", trk_b_digi = TrackerDigi("trk_b_digi",
inputHitCollection="TrackerBarrelHits", inputHitCollection="TrackerBarrelHits",
outputHitCollection="TrackerBarrelRawHits", outputHitCollection="TrackerBarrelRawHits",
timeResolution=8) timeResolution=8)
#algorithms.append(trk_b_digi) algorithms.append(trk_b_digi)
trk_ec_digi = TrackerDigi("trk_ec_digi", trk_ec_digi = TrackerDigi("trk_ec_digi",
inputHitCollection="TrackerEndcapHits", inputHitCollection="TrackerEndcapHits",
outputHitCollection="TrackerEndcapRawHits", outputHitCollection="TrackerEndcapRawHits",
timeResolution=8) timeResolution=8)
#algorithms.append(trk_ec_digi) algorithms.append(trk_ec_digi)
vtx_b_digi = TrackerDigi("vtx_b_digi", vtx_b_digi = TrackerDigi("vtx_b_digi",
inputHitCollection="VertexBarrelHits", inputHitCollection="VertexBarrelHits",
outputHitCollection="VertexBarrelRawHits", outputHitCollection="VertexBarrelRawHits",
timeResolution=8) timeResolution=8)
#algorithms.append(vtx_b_digi) algorithms.append(vtx_b_digi)
if 'acadia' in detector_version: if 'acadia' in detector_version:
vtx_ec_digi = TrackerDigi("vtx_ec_digi", vtx_ec_digi = TrackerDigi("vtx_ec_digi",
inputHitCollection="VertexEndcapHits", inputHitCollection="VertexEndcapHits",
outputHitCollection="VertexEndcapRawHits", outputHitCollection="VertexEndcapRawHits",
timeResolution=8) timeResolution=8)
#algorithms.append( vtx_ec_digi ) algorithms.append( vtx_ec_digi )
else: else:
mm_b_digi = TrackerDigi("mm_b_digi", mm_b_digi = TrackerDigi("mm_b_digi",
inputHitCollection="MPGDTrackerBarrelHits", inputHitCollection="MPGDTrackerBarrelHits",
outputHitCollection="MPGDTrackerBarrelRawHits", outputHitCollection="MPGDTrackerBarrelRawHits",
timeResolution=8) timeResolution=8)
#algorithms.append( mm_b_digi ) algorithms.append( mm_b_digi )
gem_ec_digi = TrackerDigi("gem_ec_digi", gem_ec_digi = TrackerDigi("gem_ec_digi",
inputHitCollection="GEMTrackerEndcapHits", inputHitCollection="GEMTrackerEndcapHits",
outputHitCollection="GEMTrackerEndcapRawHits", outputHitCollection="GEMTrackerEndcapRawHits",
timeResolution=10) timeResolution=10)
#algorithms.append(gem_ec_digi) algorithms.append(gem_ec_digi)
# Tracker and vertex reconstruction # Tracker and vertex reconstruction
trk_b_reco = TrackerHitReconstruction("trk_b_reco", trk_b_reco = TrackerHitReconstruction("trk_b_reco",
inputHitCollection = trk_b_digi.outputHitCollection, inputHitCollection = trk_b_digi.outputHitCollection,
outputHitCollection="TrackerBarrelRecHits") outputHitCollection="TrackerBarrelRecHits")
#algorithms.append(trk_b_reco) algorithms.append(trk_b_reco)
trk_ec_reco = TrackerHitReconstruction("trk_ec_reco", trk_ec_reco = TrackerHitReconstruction("trk_ec_reco",
inputHitCollection = trk_ec_digi.outputHitCollection, inputHitCollection = trk_ec_digi.outputHitCollection,
outputHitCollection="TrackerEndcapRecHits") outputHitCollection="TrackerEndcapRecHits")
#algorithms.append(trk_ec_reco) algorithms.append(trk_ec_reco)
vtx_b_reco = TrackerHitReconstruction("vtx_b_reco", vtx_b_reco = TrackerHitReconstruction("vtx_b_reco",
inputHitCollection = vtx_b_digi.outputHitCollection, inputHitCollection = vtx_b_digi.outputHitCollection,
outputHitCollection="VertexBarrelRecHits") outputHitCollection="VertexBarrelRecHits")
#algorithms.append(vtx_b_reco) algorithms.append(vtx_b_reco)
if 'acadia' in detector_version: if 'acadia' in detector_version:
vtx_ec_reco = TrackerHitReconstruction("vtx_ec_reco", vtx_ec_reco = TrackerHitReconstruction("vtx_ec_reco",
inputHitCollection = vtx_ec_digi.outputHitCollection, inputHitCollection = vtx_ec_digi.outputHitCollection,
outputHitCollection="VertexEndcapRecHits") outputHitCollection="VertexEndcapRecHits")
#algorithms.append( vtx_ec_reco ) algorithms.append( vtx_ec_reco )
else: else:
mm_b_reco = TrackerHitReconstruction("mm_b_reco", mm_b_reco = TrackerHitReconstruction("mm_b_reco",
inputHitCollection = mm_b_digi.outputHitCollection, inputHitCollection = mm_b_digi.outputHitCollection,
outputHitCollection="MPGDTrackerBarrelRecHits") outputHitCollection="MPGDTrackerBarrelRecHits")
#algorithms.append( mm_b_reco ) 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 = [ input_tracking_hits = [
str(trk_b_reco.outputHitCollection), str(trk_b_reco.outputHitCollection),
...@@ -725,20 +725,20 @@ else: ...@@ -725,20 +725,20 @@ else:
trk_hit_col = TrackingHitsCollector("trk_hit_col", trk_hit_col = TrackingHitsCollector("trk_hit_col",
inputTrackingHits=input_tracking_hits, inputTrackingHits=input_tracking_hits,
trackingHits="trackingHits") trackingHits="trackingHits")
#algorithms.append( trk_hit_col ) algorithms.append( trk_hit_col )
# Hit Source linker # Hit Source linker
sourcelinker = TrackerSourceLinker("trk_srcslnkr", sourcelinker = TrackerSourceLinker("trk_srcslnkr",
inputHitCollection = trk_hit_col.trackingHits, inputHitCollection = trk_hit_col.trackingHits,
outputSourceLinks = "TrackSourceLinks", outputSourceLinks = "TrackSourceLinks",
outputMeasurements = "TrackMeasurements") outputMeasurements = "TrackMeasurements")
#algorithms.append(sourcelinker) algorithms.append(sourcelinker)
## Track param init ## Track param init
truth_trk_init = TrackParamTruthInit("truth_trk_init", truth_trk_init = TrackParamTruthInit("truth_trk_init",
inputMCParticles="MCParticles", inputMCParticles="MCParticles",
outputInitialTrackParameters="InitTrackParams") outputInitialTrackParameters="InitTrackParams")
#algorithms.append(truth_trk_init) algorithms.append(truth_trk_init)
# Tracking algorithms # Tracking algorithms
trk_find_alg = CKFTracking("trk_find_alg", trk_find_alg = CKFTracking("trk_find_alg",
...@@ -747,25 +747,25 @@ trk_find_alg = CKFTracking("trk_find_alg", ...@@ -747,25 +747,25 @@ trk_find_alg = CKFTracking("trk_find_alg",
inputInitialTrackParameters = truth_trk_init.outputInitialTrackParameters, inputInitialTrackParameters = truth_trk_init.outputInitialTrackParameters,
outputTrajectories = "trajectories", outputTrajectories = "trajectories",
chi2CutOff = [50.]) chi2CutOff = [50.])
#algorithms.append(trk_find_alg) algorithms.append(trk_find_alg)
parts_from_fit = ParticlesFromTrackFit("parts_from_fit", parts_from_fit = ParticlesFromTrackFit("parts_from_fit",
inputTrajectories = trk_find_alg.outputTrajectories, inputTrajectories = trk_find_alg.outputTrajectories,
outputParticles = "outputParticles", outputParticles = "outputParticles",
outputTrackParameters = "outputTrackParameters") outputTrackParameters = "outputTrackParameters")
#algorithms.append(parts_from_fit) algorithms.append(parts_from_fit)
# trajs_from_fit = TrajectoryFromTrackFit("trajs_from_fit", # trajs_from_fit = TrajectoryFromTrackFit("trajs_from_fit",
# inputTrajectories = trk_find_alg.outputTrajectories, # inputTrajectories = trk_find_alg.outputTrajectories,
# outputTrajectoryParameters = "outputTrajectoryParameters") # outputTrajectoryParameters = "outputTrajectoryParameters")
# #algorithms.append(trajs_from_fit) # algorithms.append(trajs_from_fit)
# Event building # Event building
parts_with_truth_pid = ParticlesWithTruthPID("parts_with_truth_pid", parts_with_truth_pid = ParticlesWithTruthPID("parts_with_truth_pid",
inputMCParticles = "MCParticles", inputMCParticles = "MCParticles",
inputTrackParameters = parts_from_fit.outputTrackParameters, inputTrackParameters = parts_from_fit.outputTrackParameters,
outputParticles = "ReconstructedChargedParticles") outputParticles = "ReconstructedChargedParticles")
#algorithms.append(parts_with_truth_pid) algorithms.append(parts_with_truth_pid)
match_clusters = MatchClusters("match_clusters", match_clusters = MatchClusters("match_clusters",
inputMCParticles = "MCParticles", inputMCParticles = "MCParticles",
...@@ -781,7 +781,7 @@ match_clusters = MatchClusters("match_clusters", ...@@ -781,7 +781,7 @@ match_clusters = MatchClusters("match_clusters",
str(ci_hcal_clreco.outputClusterCollection) str(ci_hcal_clreco.outputClusterCollection)
], ],
outputParticles = "ReconstructedParticles") outputParticles = "ReconstructedParticles")
#algorithms.append(match_clusters) algorithms.append(match_clusters)
## Far Forward for now stored separately ## Far Forward for now stored separately
fast_ff = FFSmearedParticles("fast_ff", fast_ff = FFSmearedParticles("fast_ff",
...@@ -793,19 +793,19 @@ fast_ff = FFSmearedParticles("fast_ff", ...@@ -793,19 +793,19 @@ fast_ff = FFSmearedParticles("fast_ff",
enableOMD = True, enableOMD = True,
ionBeamEnergy = ionBeamEnergy, ionBeamEnergy = ionBeamEnergy,
crossingAngle = -0.025) crossingAngle = -0.025)
#algorithms.append(fast_ff) algorithms.append(fast_ff)
# DRICH # DRICH
drich_digi = PhotoMultiplierDigi("drich_digi", drich_digi = PhotoMultiplierDigi("drich_digi",
inputHitCollection="DRICHHits", inputHitCollection="DRICHHits",
outputHitCollection="DRICHRawHits", outputHitCollection="DRICHRawHits",
quantumEfficiency=[(a*units.eV, b) for a, b in qe_data]) quantumEfficiency=[(a*units.eV, b) for a, b in qe_data])
#algorithms.append(drich_digi) algorithms.append(drich_digi)
drich_reco = PhotoMultiplierReco("drich_reco", drich_reco = PhotoMultiplierReco("drich_reco",
inputHitCollection=drich_digi.outputHitCollection, inputHitCollection=drich_digi.outputHitCollection,
outputHitCollection="DRICHRecHits") outputHitCollection="DRICHRecHits")
#algorithms.append(drich_reco) algorithms.append(drich_reco)
# FIXME # FIXME
#drich_cluster = PhotoRingClusters("drich_cluster", #drich_cluster = PhotoRingClusters("drich_cluster",
...@@ -819,11 +819,11 @@ if 'acadia' in detector_version: ...@@ -819,11 +819,11 @@ if 'acadia' in detector_version:
inputHitCollection="MRICHHits", inputHitCollection="MRICHHits",
outputHitCollection="MRICHRawHits", outputHitCollection="MRICHRawHits",
quantumEfficiency=[(a*units.eV, b) for a, b in qe_data]) quantumEfficiency=[(a*units.eV, b) for a, b in qe_data])
#algorithms.append(mrich_digi) algorithms.append(mrich_digi)
mrich_reco = PhotoMultiplierReco("mrich_reco", mrich_reco = PhotoMultiplierReco("mrich_reco",
inputHitCollection=mrich_digi.outputHitCollection, inputHitCollection=mrich_digi.outputHitCollection,
outputHitCollection="MRICHRecHits") outputHitCollection="MRICHRecHits")
#algorithms.append(mrich_reco) algorithms.append(mrich_reco)
# Inclusive kinematics # Inclusive kinematics
incl_kin_electron = InclusiveKinematicsElectron("incl_kin_electron", incl_kin_electron = InclusiveKinematicsElectron("incl_kin_electron",
...@@ -831,31 +831,31 @@ incl_kin_electron = InclusiveKinematicsElectron("incl_kin_electron", ...@@ -831,31 +831,31 @@ incl_kin_electron = InclusiveKinematicsElectron("incl_kin_electron",
inputParticles="ReconstructedParticles", inputParticles="ReconstructedParticles",
outputData="InclusiveKinematicsElectron" outputData="InclusiveKinematicsElectron"
) )
#algorithms.append(incl_kin_electron) algorithms.append(incl_kin_electron)
incl_kin_jb = InclusiveKinematicsJB("incl_kin_jb", incl_kin_jb = InclusiveKinematicsJB("incl_kin_jb",
inputMCParticles="MCParticles", inputMCParticles="MCParticles",
inputParticles="ReconstructedParticles", inputParticles="ReconstructedParticles",
outputData="InclusiveKinematicsJB" outputData="InclusiveKinematicsJB"
) )
#algorithms.append(incl_kin_jb) algorithms.append(incl_kin_jb)
incl_kin_da = InclusiveKinematicsDA("incl_kin_da", incl_kin_da = InclusiveKinematicsDA("incl_kin_da",
inputMCParticles="MCParticles", inputMCParticles="MCParticles",
inputParticles="ReconstructedParticles", inputParticles="ReconstructedParticles",
outputData="InclusiveKinematicsDA" outputData="InclusiveKinematicsDA"
) )
#algorithms.append(incl_kin_da) algorithms.append(incl_kin_da)
incl_kin_sigma = InclusiveKinematicsSigma("incl_kin_sigma", incl_kin_sigma = InclusiveKinematicsSigma("incl_kin_sigma",
inputMCParticles="MCParticles", inputMCParticles="MCParticles",
inputParticles="ReconstructedParticles", inputParticles="ReconstructedParticles",
outputData="InclusiveKinematicsSigma" outputData="InclusiveKinematicsSigma"
) )
#algorithms.append(incl_kin_sigma) algorithms.append(incl_kin_sigma)
incl_kin_esigma = InclusiveKinematicseSigma("incl_kin_esigma", incl_kin_esigma = InclusiveKinematicseSigma("incl_kin_esigma",
inputMCParticles="MCParticles", inputMCParticles="MCParticles",
inputParticles="ReconstructedParticles", inputParticles="ReconstructedParticles",
outputData="InclusiveKinematicseSigma" outputData="InclusiveKinematicseSigma"
) )
#algorithms.append(incl_kin_esigma) algorithms.append(incl_kin_esigma)
# Output # Output
podout = PodioOutput("out", filename=output_rec) podout = PodioOutput("out", filename=output_rec)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment